DEV: Fix test (#43)

* DEV: Fix test * lint
2023-04-18 15:43:10 -03:00 · 2023-04-18 15:43:10 -03:00 · 5a78b6a953
parent 4368ef29d8
commit 5a78b6a953
1 changed files with 2 additions and 1 deletions
--- a/spec/shared/chat_message_classificator_spec.rb
+++ b/spec/shared/chat_message_classificator_spec.rb
@ -15,7 +15,8 @@ describe DiscourseAi::ChatMessageClassificator do
    it "stores the model classification data" do
      classification.classify!(chat_message)

-      result = ClassificationResult.find_by(target: chat_message, classification_type: model.type)
+      result =
+        ClassificationResult.find_by(target_id: chat_message.id, classification_type: model.type)

      classification = result.classification.symbolize_keys