Fixing tests

meedan · Jan 24, 2024 · 4b5505d · 4b5505d
1 parent 2a4d6f9
commit 4b5505d
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/test/models/bot/alegre_v2_test.rb b/test/models/bot/alegre_v2_test.rb
@@ -652,20 +652,20 @@ def teardown
   end
 
   test "should not relate project media for audio if disabled on workspace" do
-    tbi = @team.team_bot_installations.select{ |x| x.user_id == @bot.id }
+    tbi = TeamBotInstallation.where(team: @team, user: @bot).last
     tbi.set_audio_similarity_enabled = false
     tbi.save!
     Bot::Alegre.stubs(:merge_suggested_and_confirmed).never
     pm = create_project_media team: @team, media: create_uploaded_audio
-    assert_equal {}, Bot::Alegre.get_similar_items_v2(pm)
+    assert_equal({}, Bot::Alegre.get_similar_items_v2(pm, nil))
   end
 
   test "should not relate project media for image if disabled on workspace" do
-    tbi = @team.team_bot_installations.select{ |x| x.user_id == @bot.id }
+    tbi = TeamBotInstallation.where(team: @team, user: @bot).last
     tbi.set_image_similarity_enabled = false
     tbi.save!
     Bot::Alegre.stubs(:merge_suggested_and_confirmed).never
     pm = create_project_media team: @team, media: create_uploaded_image
-    assert_equal {}, Bot::Alegre.get_similar_items_v2(pm)
+    assert_equal({}, Bot::Alegre.get_similar_items_v2(pm, nil))
   end
-end
+end