From fdbe6f98764853ac719904b5a77c7c7c17e2196b Mon Sep 17 00:00:00 2001
From: Devin Gaffney <itsme@devingaffney.com>
Date: Wed, 24 Jan 2024 07:44:33 -0800
Subject: [PATCH] add tests for explicitly disabling similarity

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

diff --git a/test/models/bot/alegre_v2_test.rb b/test/models/bot/alegre_v2_test.rb
index 4d240caf94..ea3063a93d 100644
--- a/test/models/bot/alegre_v2_test.rb
+++ b/test/models/bot/alegre_v2_test.rb
@@ -650,4 +650,38 @@ def teardown
     assert_equal relationship.relationship_type, Relationship.suggested_type
     Bot::Alegre.unstub(:get_similar_items_v2)
   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.set_audio_similarity_enabled = false
+    tb.save!
+    pm1 = create_project_media team: @team, media: create_uploaded_audio
+    pm2 = create_project_media team: @team, media: create_uploaded_audio
+    Bot::Alegre.stubs(:get_similar_items_v2).returns({pm2.id=>{:score=>0.91, :context=>{"team_id"=>pm2.team_id, "has_custom_id"=>true, "project_media_id"=>pm2.id}, :model=>"audio", :source_field=>"audio", :target_field=>"audio", :relationship_type=>Relationship.suggested_type}})
+    relationship = nil
+    assert_no_difference 'Relationship.count' do
+      relationship = Bot::Alegre.relate_project_media(pm1)
+    end
+    assert_equal relationship, nil
+    Bot::Alegre.unstub(:get_similar_items_v2)
+    tbi.set_audio_similarity_enabled = true
+    tb.save!
+  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.set_image_similarity_enabled = false
+    tb.save!
+    pm1 = create_project_media team: @team, media: create_uploaded_image
+    pm2 = create_project_media team: @team, media: create_uploaded_image
+    Bot::Alegre.stubs(:get_similar_items_v2).returns({pm2.id=>{:score=>0.91, :context=>{"team_id"=>pm2.team_id, "has_custom_id"=>true, "project_media_id"=>pm2.id}, :model=>"audio", :source_field=>"audio", :target_field=>"audio", :relationship_type=>Relationship.suggested_type}})
+    relationship = nil
+    assert_no_difference 'Relationship.count' do
+      relationship = Bot::Alegre.relate_project_media(pm1)
+    end
+    assert_equal relationship, nil
+    Bot::Alegre.unstub(:get_similar_items_v2)
+    tbi.set_image_similarity_enabled = true
+    tb.save!
+  end
 end
\ No newline at end of file