From 4b5505d7fe8efdca4d54cc25014bf3900f547852 Mon Sep 17 00:00:00 2001
From: Caio <117518+caiosba@users.noreply.github.com>
Date: Wed, 24 Jan 2024 13:47:55 -0300
Subject: [PATCH] Fixing tests

---
 test/models/bot/alegre_v2_test.rb | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/test/models/bot/alegre_v2_test.rb b/test/models/bot/alegre_v2_test.rb
index 44475f52ad..89c8d57bc3 100644
--- 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
\ No newline at end of file
+end