CV2-4508: include unconfirmed items when add new team task (#1872)

meedan · May 2, 2024 · e83862d · e83862d
1 parent c7dd0c3
commit e83862d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/app/models/team_task.rb b/app/models/team_task.rb
@@ -150,7 +150,7 @@ def add_to_sources
   end
 
   def add_to_project_medias
-    ProjectMedia.where({ team_id: self.team_id, archived: CheckArchivedFlags::FlagCodes::NONE })
+    ProjectMedia.where({ team_id: self.team_id, archived: [CheckArchivedFlags::FlagCodes::NONE, CheckArchivedFlags::FlagCodes::UNCONFIRMED] })
     .joins("LEFT JOIN annotations a ON a.annotation_type = 'task' AND a.annotated_type = 'ProjectMedia'
       AND a.annotated_id = project_medias.id
       AND task_team_task_id(a.annotation_type, a.data) = #{self.id}")

diff --git a/test/models/team_task_test.rb b/test/models/team_task_test.rb
@@ -119,7 +119,7 @@ def setup
     Team.stubs(:current).returns(t)
     Sidekiq::Testing.inline! do
       pm = create_project_media team: t
-      pm2 = create_project_media team: t
+      pm2 = create_project_media team: t, archived: CheckArchivedFlags::FlagCodes::UNCONFIRMED
       tt = create_team_task team_id: t.id, order: 2, description: 'Foo', options: [{ label: 'Foo' }]
       tt2 = create_team_task team_id: t.id, order: 1, description: 'Foo2', options: [{ label: 'Foo2' }]
       pm_tt = pm.annotations('task').select{|t| t.team_task_id == tt.id}.last