✅ 🤡 add get interaction test

agn-7 · Nov 18, 2023 · b369853 · b369853
1 parent e50b07b
commit b369853
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/tests/test_crud.py b/tests/test_crud.py
@@ -28,3 +28,16 @@ def test_get_interactions(db):
     assert interactions[1].settings["model_name"] == "model2"
 
 
+def test_get_interaction(db):
+    timestamp = utils.convert_timezone(datetime.now())
+    interaction = models.Interaction(
+        settings=dict(model_name="model", role="role", prompt="prompt"),
+        created_at=timestamp,
+        updated_at=timestamp,
+    )
+    db.add(interaction)
+    db.commit()
+
+    retrieved_interaction = crud.get_interaction(db, interaction.id)
+    assert retrieved_interaction.id == interaction.id
+    assert retrieved_interaction.settings["model_name"] == "model"