From c4d908a727937255828025548e9c804959e57126 Mon Sep 17 00:00:00 2001 From: Marcelo Lotif Date: Wed, 27 Nov 2024 14:35:52 -0500 Subject: [PATCH] cleanup --- .github/workflows/integration_tests.yaml | 2 +- florist/tests/integration/api/test_train.py | 16 ++++++++++++++-- 2 files changed, 15 insertions(+), 3 deletions(-) diff --git a/.github/workflows/integration_tests.yaml b/.github/workflows/integration_tests.yaml index 9d460c4..7c3f457 100644 --- a/.github/workflows/integration_tests.yaml +++ b/.github/workflows/integration_tests.yaml @@ -62,7 +62,7 @@ jobs: poetry env use '3.10' source .venv/bin/activate poetry install --with docs,test - coverage run -m pytest -s -vv florist/tests/integration && coverage xml && coverage report -m + coverage run -m pytest florist/tests/integration && coverage xml && coverage report -m - name: Upload python coverage to Codecov uses: Wandalen/wretry.action@v3.7.3 with: diff --git a/florist/tests/integration/api/test_train.py b/florist/tests/integration/api/test_train.py index 13de57d..1ea8366 100644 --- a/florist/tests/integration/api/test_train.py +++ b/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", timeout_graceful_shutdown=60) + server_config = uvicorn.Config( + "florist.api.server:app", + host="localhost", + port=8000, + log_level="debug", + timeout_graceful_shutdown=60, + ) server_service = TestUvicornServer(config=server_config) - client_config = uvicorn.Config("florist.api.client:app", host="localhost", port=8001, log_level="debug", timeout_graceful_shutdown=60) + client_config = uvicorn.Config( + "florist.api.client:app", + host="localhost", + port=8001, + log_level="debug", + timeout_graceful_shutdown=60, + ) client_service = TestUvicornServer(config=client_config) # TODO figure out how to run fastapi with the test DB so we can use the fixture here