begin setting up more tests

Author: azliu0

server/controllers/emails.py

@@ -442,6 +442,7 @@ def get_response():
     response = db.session.execute(
         select(Response).where(Response.email_id == data["id"])
     ).scalar()
+
     if not response:
         return {"message": "Response not found"}, 400
     return response.map()
@@ -524,23 +525,26 @@ def get_threads():
 
     Get a list of all threads.
     """
-    thread_list = db.session.execute(select(Thread)).all()
-    print("thread list", thread_list)
-    # thread_list = db.session.execute(
-    #     select(Thread).order_by(Thread.resolved, Thread.last_email.desc())
-    # ).all()
-    # email_list = [
-    #     {
-    #         "id": thread.id,
-    #         "resolved": thread.resolved,
-    #         "emailList": [
-    #             thread_email.map()
-    #             for thread_email in db.session.execute(
-    #                 select(Email).where(Email.thread_id == thread.id)
-    #             ).all()
-    #         ],
-    #     }
-    #     for thread in thread_list
-    # ]
-    email_list = []
+    thread_list = (
+        db.session.execute(
+            select(Thread).order_by(Thread.resolved, Thread.last_email.desc())
+        )
+        .scalars()
+        .all()
+    )
+    email_list = [
+        {
+            "id": thread.id,
+            "resolved": thread.resolved,
+            "emailList": [
+                thread_email.map()
+                for thread_email in db.session.execute(
+                    select(Email).where(Email.thread_id == thread.id)
+                )
+                .scalars()
+                .all()
+            ],
+        }
+        for thread in thread_list
+    ]
     return email_list

server/models/response.py

@@ -54,12 +54,25 @@ class Response(db.Model):
     )
 
     def map(self):
-        """Map the response to a dictionary."""
+        """Map the response to a dictionary.
+
+        Groups documents and document_confidences into a list of lists clustered by
+        question.
+        """
+        doc_confs = []
+        docs = []
+        cur_idx = 0
+        for num_docs in self.docs_per_question:
+            doc_confs.append(self.document_confidences[cur_idx : cur_idx + num_docs])
+            docs.append(self.documents[cur_idx : cur_idx + num_docs])
+            cur_idx += num_docs
+        docs = [[doc.map() for doc in doc_list] for doc_list in docs]
         return {
             "id": self.id,
             "content": self.response,
             "questions": self.questions,
-            "document_confidences": self.document_confidences,
+            "documents": docs,
+            "document_confidences": doc_confs,
             "confidence": self.confidence,
             "emailId": self.email_id,
         }

server_tests/test_email.py

@@ -0,0 +1,10 @@
+from apiflask import APIFlask
+from flask.testing import FlaskClient
+
+from server_tests.utils import assert_status
+
+
+def test_get_threads(app: APIFlask, client: FlaskClient):
+    """Test fetching threads."""
+    response = client.get("/api/emails/get_threads")
+    assert_status(response, 200)

server_tests/utils.py

@@ -0,0 +1,17 @@
+"""Utils for testing."""
+
+import logging
+
+from werkzeug.test import TestResponse
+
+
+def assert_status(response: TestResponse, status: int):
+    """Asserts a response's status code, logging the response if it fails."""
+    try:
+        assert response.status_code == status
+    except AssertionError:
+        logging.error(
+            f"Assertion failed: expected {status}, got {response.status_code}. "
+            f"Response body: {response.data.decode()}"
+        )
+        raise