test response map

Author: azliu0

server_tests/test_cli.py

@@ -10,3 +10,8 @@ def test_seed_email(runner):
     result: Result = runner.invoke(args=["seed", "email"])
     print(result.stdout)
     assert result.exit_code == 0
+
+def test_seed_thread(runner):
+    result: Result = runner.invoke(args=["seed", "thread"])
+    print(result.stdout)
+    assert result.exit_code == 0

server_tests/test_email.py

@@ -10,12 +10,7 @@
 def test_thread_first_sender(app: APIFlask):
     """Test fetching the first thread."""
     with app.app_context():
-        thread = (
-            db.session.execute(
-                select(Thread)
-            )
-            .scalar_one()
-        )
+        thread = db.session.execute(select(Thread)).scalar_one()
         earliest_email = min(thread.emails, key=lambda x: x.date)
         assert earliest_email.sender == thread.first_sender
 
@@ -32,7 +27,7 @@ def test_get_threads(app: APIFlask, client: FlaskClient):
     thread = threads[0]
     thread_id = thread["id"]
     assert thread_id == 1
-    assert not thread["resolved"]  # resolved is false
+    assert not thread["resolved"]
     assert len(thread["emailList"]) == 5
 
     for email in thread["emailList"]:
@@ -44,3 +39,29 @@ def test_get_threads(app: APIFlask, client: FlaskClient):
         assert email["message_id"] is not None
         assert email["is_reply"] is not None
         assert email["thread_id"] == thread_id
+
+
+def test_get_response(app: APIFlask, client: FlaskClient):
+    """Test fetching a response."""
+    last_email_id = 5
+    response = client.post(
+        "/api/emails/get_response", data={"id": last_email_id}
+    )
+    assert_status(response, 200)
+
+    response = response.json
+    assert response is not None
+
+    # check that .map() is intact
+    assert response["confidence"] is not None
+    assert response["content"] is not None
+    assert response["questions"] is not None
+    assert response["documents"] is not None
+    assert response["id"] is not None
+    assert response["emailId"] is not None
+
+    for doc_list in response["documents"]:
+        for doc in doc_list:
+            assert doc["confidence"] is not None
+            assert doc["content"] is not None
+            assert doc["source"] is not None