going in circles but everything works now

Author: azliu0

server/fake_data.py

@@ -73,7 +73,7 @@ def _generate_response(email: Email, thread: Thread):
     increment_response_count(documents)
 
 
-def generate_fake_email(generate_response=True):
+def generate_fake_email():
     """Generate fake email."""
     thread = Thread()
     db.session.add(thread)
@@ -101,11 +101,10 @@ def generate_fake_email(generate_response=True):
     thread.resolved = False
     db.session.commit()
 
-    if generate_response:
-        _generate_response(email, thread)
+    _generate_response(email, thread)
 
 
-def generate_fake_thread(generate_response=True):
+def generate_fake_thread():
     """Generate fake unresolved thread."""
     thread = Thread()
     db.session.add(thread)
@@ -199,5 +198,4 @@ def generate_fake_thread(generate_response=True):
     thread.resolved = False
     db.session.commit()
 
-    if generate_response:
-        _generate_response(email_5, thread)
+    _generate_response(email_5, thread)

server_tests/conftest.py

@@ -77,7 +77,7 @@ def app(db_url: str, redis_host: str):
 
     with app.app_context():
         from server_tests.utils import seed_database
-        seed_database(app)
+        seed_database()
 
     yield app
 

server_tests/utils.py

@@ -2,7 +2,6 @@
 
 import logging
 
-from apiflask import APIFlask
 from werkzeug.test import TestResponse
 
 
@@ -17,11 +16,12 @@ def assert_status(response: TestResponse, status: int):
         )
         raise
 
-def seed_database(app: APIFlask):
+def seed_database():
     """Seeds the database with some fake data."""
 
-    with app.app_context():
-        from server.fake_data import generate_fake_thread, generate_test_documents
+    from server.fake_data import generate_fake_thread, generate_test_documents
+    from server.nlp.embeddings import embed_corpus
 
-        generate_test_documents()
-        generate_fake_thread(generate_response=False)
+    test_documents = generate_test_documents()
+    embed_corpus(test_documents)
+    generate_fake_thread()