diff --git a/src/backend/core/tests/conftest.py b/src/backend/core/tests/conftest.py index ea586dcb..00e830e1 100644 --- a/src/backend/core/tests/conftest.py +++ b/src/backend/core/tests/conftest.py @@ -2,6 +2,8 @@ from unittest import mock +from django.core.cache import cache + import pytest USER = "user" @@ -9,6 +11,12 @@ TEAM = "team" VIA = [USER, TEAM] +@pytest.fixture(autouse=True) +def clear_cache(): + """Fixture to clear the cache before each test.""" + cache.clear() + + @pytest.fixture def mock_user_teams(): """Mock for the "teams" property on the User model.""" diff --git a/src/backend/core/tests/documents/test_api_documents_ai_transform.py b/src/backend/core/tests/documents/test_api_documents_ai_transform.py index 18507b61..41e216ec 100644 --- a/src/backend/core/tests/documents/test_api_documents_ai_transform.py +++ b/src/backend/core/tests/documents/test_api_documents_ai_transform.py @@ -5,7 +5,6 @@ Test AI transform API endpoint for users in impress's core app. import random from unittest.mock import MagicMock, patch -from django.core.cache import cache from django.test import override_settings import pytest @@ -17,12 +16,6 @@ from core.tests.conftest import TEAM, USER, VIA pytestmark = pytest.mark.django_db -@pytest.fixture(autouse=True) -def clear_cache(): - """Fixture to clear the cache before each test.""" - cache.clear() - - @pytest.fixture def ai_settings(): """Fixture to set AI settings.""" diff --git a/src/backend/core/tests/documents/test_api_documents_ai_translate.py b/src/backend/core/tests/documents/test_api_documents_ai_translate.py index e077798e..a04a427d 100644 --- a/src/backend/core/tests/documents/test_api_documents_ai_translate.py +++ b/src/backend/core/tests/documents/test_api_documents_ai_translate.py @@ -5,7 +5,6 @@ Test AI translate API endpoint for users in impress's core app. import random from unittest.mock import MagicMock, patch -from django.core.cache import cache from django.test import override_settings import pytest @@ -17,12 +16,6 @@ from core.tests.conftest import TEAM, USER, VIA pytestmark = pytest.mark.django_db -@pytest.fixture(autouse=True) -def clear_cache(): - """Fixture to clear the cache before each test.""" - cache.clear() - - @pytest.fixture def ai_settings(): """Fixture to set AI settings.""" diff --git a/src/backend/core/tests/test_api_utils_ai_document_rate_throttles.py b/src/backend/core/tests/test_api_utils_ai_document_rate_throttles.py index dbb15467..7ab83bb9 100644 --- a/src/backend/core/tests/test_api_utils_ai_document_rate_throttles.py +++ b/src/backend/core/tests/test_api_utils_ai_document_rate_throttles.py @@ -4,10 +4,8 @@ Test throttling on documents for the AI endpoint. from unittest.mock import patch -from django.core.cache import cache from django.test import override_settings -import pytest from rest_framework.response import Response from rest_framework.test import APIRequestFactory from rest_framework.views import APIView @@ -25,12 +23,6 @@ class DocumentAPIView(APIView): return Response({"message": "Success"}) -@pytest.fixture(autouse=True) -def clear_cache(): - """Fixture to clear the cache before each test.""" - cache.clear() - - @override_settings(AI_DOCUMENT_RATE_THROTTLE_RATES={"minute": 3, "hour": 6, "day": 10}) @patch("time.time") def test_api_utils_ai_document_rate_throttle_minute_limit(mock_time): diff --git a/src/backend/core/tests/test_api_utils_ai_user_rate_throttles.py b/src/backend/core/tests/test_api_utils_ai_user_rate_throttles.py index 5fd52e27..01a25db4 100644 --- a/src/backend/core/tests/test_api_utils_ai_user_rate_throttles.py +++ b/src/backend/core/tests/test_api_utils_ai_user_rate_throttles.py @@ -5,7 +5,6 @@ Test throttling on users for the AI endpoint. from unittest.mock import patch from uuid import uuid4 -from django.core.cache import cache from django.test import override_settings import pytest @@ -29,12 +28,6 @@ class DocumentAPIView(APIView): return Response({"message": "Success"}) -@pytest.fixture(autouse=True) -def clear_cache(): - """Fixture to clear the cache before each test.""" - cache.clear() - - @override_settings(AI_USER_RATE_THROTTLE_RATES={"minute": 3, "hour": 6, "day": 10}) @patch("time.time") def test_api_utils_ai_user_rate_throttle_minute_limit(mock_time):