Trying to solve some flakiness on integration tests

.github/workflows/integration_tests.yaml

@@ -57,12 +57,17 @@ jobs:
         uses: supercharge/[email protected]
         with:
           mongodb-version: 7.0.8
-      - name: Install dependencies and check code
+      - name: Install dependencies
         run: |
           poetry env use '3.10'
           source .venv/bin/activate
           poetry install --with docs,test
-          coverage run -m pytest florist/tests/integration && coverage xml && coverage report -m
+      - name: Run integration tests
+        run: |
+          source .venv/bin/activate
+          coverage run -m pytest -s -vv florist/tests/integration
+          coverage xml
+          coverage report -m
       - name: Upload python coverage to Codecov
         uses: Wandalen/[email protected]
         with:

florist/tests/integration/api/test_train.py

@@ -17,9 +17,21 @@
 
 async def test_train():
     # Define services
-    server_config = uvicorn.Config("florist.api.server:app", host="localhost", port=8000, log_level="debug")
+    server_config = uvicorn.Config(
+        "florist.api.server:app",
+        host="localhost",
+        port=8000,
+        log_level="debug",
+        timeout_graceful_shutdown=120,
+    )
     server_service = TestUvicornServer(config=server_config)
-    client_config = uvicorn.Config("florist.api.client:app", host="localhost", port=8001, log_level="debug")
+    client_config = uvicorn.Config(
+        "florist.api.client:app",
+        host="localhost",
+        port=8001,
+        log_level="debug",
+        timeout_graceful_shutdown=120,
+    )
     client_service = TestUvicornServer(config=client_config)
 
     # TODO figure out how to run fastapi with the test DB so we can use the fixture here