diff --git a/server/tests/conftest.py b/server/tests/conftest.py index 532ebff9..0acff79e 100644 --- a/server/tests/conftest.py +++ b/server/tests/conftest.py @@ -36,7 +36,13 @@ def dummy_processors(): mock_long_summary.return_value = "LLM LONG SUMMARY" mock_short_summary.return_value = {"short_summary": "LLM SHORT SUMMARY"} mock_translate.return_value = "Bonjour le monde" - yield mock_translate, mock_topic, mock_title, mock_long_summary, mock_short_summary # noqa + yield ( + mock_translate, + mock_topic, + mock_title, + mock_long_summary, + mock_short_summary, + ) # noqa @pytest.fixture @@ -159,6 +165,11 @@ def celery_config(): } +@pytest.fixture(scope="session") +def celery_includes(): + return ["reflector.pipelines.main_live_pipeline"] + + @pytest.fixture(scope="session") def fake_mp3_upload(): with patch( diff --git a/server/tests/test_transcripts_rtc_ws.py b/server/tests/test_transcripts_rtc_ws.py index 8502a0d9..c607fe06 100644 --- a/server/tests/test_transcripts_rtc_ws.py +++ b/server/tests/test_transcripts_rtc_ws.py @@ -32,7 +32,7 @@ class ThreadedUvicorn: @pytest.fixture -async def appserver(tmpdir, celery_session_app, celery_session_worker): +async def appserver(tmpdir, setup_database, celery_session_app, celery_session_worker): from reflector.settings import settings from reflector.app import app @@ -57,6 +57,7 @@ def celery_includes(): return ["reflector.pipelines.main_live_pipeline"] +@pytest.mark.usefixtures("setup_database") @pytest.mark.usefixtures("celery_session_app") @pytest.mark.usefixtures("celery_session_worker") @pytest.mark.asyncio @@ -213,6 +214,7 @@ async def test_transcript_rtc_and_websocket( assert audio_resp.headers["Content-Type"] == "audio/mpeg" +@pytest.mark.usefixtures("setup_database") @pytest.mark.usefixtures("celery_session_app") @pytest.mark.usefixtures("celery_session_worker") @pytest.mark.asyncio diff --git a/server/tests/test_transcripts_upload.py b/server/tests/test_transcripts_upload.py index 5414b498..3c9c240e 100644 --- a/server/tests/test_transcripts_upload.py +++ b/server/tests/test_transcripts_upload.py @@ -3,30 +3,9 @@ import asyncio from httpx import AsyncClient -@pytest.fixture(scope="session") -def celery_enable_logging(): - return True - - -@pytest.fixture(scope="session") -def celery_config(): - from tempfile import NamedTemporaryFile - - with NamedTemporaryFile() as f: - yield { - "broker_url": "memory://", - "result_backend": f"db+sqlite:///{f.name}", - } - - -@pytest.fixture(scope="session") -def celery_includes(): - return ["reflector.pipelines.main_live_pipeline"] - - @pytest.mark.usefixtures("setup_database") -@pytest.mark.usefixtures("celery_app") -@pytest.mark.usefixtures("celery_worker") +@pytest.mark.usefixtures("celery_session_app") +@pytest.mark.usefixtures("celery_session_worker") @pytest.mark.asyncio async def test_transcript_upload_file( tmpdir,