From 57215c4dda1f381603479018deb6d3e122e68d1f Mon Sep 17 00:00:00 2001 From: Cammille Carinan Date: Tue, 5 Nov 2024 13:37:14 +0100 Subject: [PATCH] fix(subscriptions): return import on schema and update tests using async_lru --- api/src/damnit_api/graphql/schema.py | 2 +- api/tests/graphql/test_subscriptions.py | 12 ++---------- 2 files changed, 3 insertions(+), 11 deletions(-) diff --git a/api/src/damnit_api/graphql/schema.py b/api/src/damnit_api/graphql/schema.py index 4ad2b9b..0dc00cb 100644 --- a/api/src/damnit_api/graphql/schema.py +++ b/api/src/damnit_api/graphql/schema.py @@ -5,7 +5,7 @@ from strawberry.schema.config import StrawberryConfig from strawberry.type import has_object_definition -from . import models, mutations, queries +from . import models, mutations, queries, subscriptions class Schema(strawberry.Schema): diff --git a/api/tests/graphql/test_subscriptions.py b/api/tests/graphql/test_subscriptions.py index 4245332..b66d5b9 100644 --- a/api/tests/graphql/test_subscriptions.py +++ b/api/tests/graphql/test_subscriptions.py @@ -6,7 +6,7 @@ import pytest_asyncio from damnit_api.graphql.models import DamnitRun, serialize -from damnit_api.graphql.subscriptions import LATEST_DATA, POLLING_INTERVAL +from damnit_api.graphql.subscriptions import POLLING_INTERVAL from .const import ( EXAMPLE_VARIABLES, @@ -66,12 +66,6 @@ def mocked_new_count(mocker): ) -@pytest.fixture(autouse=True) -def clear_latest_data(): - """Automatically clears the latest data cache before every test.""" - LATEST_DATA.clear() - - @pytest.mark.asyncio async def test_latest_data( graphql_schema, @@ -221,9 +215,7 @@ async def test_latest_data_with_nonconcurrent_subscriptions( mocked_latest_rows.assert_called() break - await asyncio.sleep( - POLLING_INTERVAL * 3 - ) # give enough time to clear the cache + await asyncio.sleep(POLLING_INTERVAL * 3) # give enough time to clear the cache mocked_latest_rows.reset_mock() with patched_sleep: