diff --git a/authentik/enterprise/lifecycle/tests/test_api.py b/authentik/enterprise/lifecycle/tests/test_api.py index 67838e7b76..885b8f7266 100644 --- a/authentik/enterprise/lifecycle/tests/test_api.py +++ b/authentik/enterprise/lifecycle/tests/test_api.py @@ -5,11 +5,11 @@ from rest_framework.test import APITestCase from authentik.core.models import Application, Group from authentik.core.tests.utils import create_test_admin_user, create_test_user from authentik.enterprise.lifecycle.models import LifecycleIteration, LifecycleRule, ReviewState -from authentik.enterprise.reports.tests.utils import patch_license +from authentik.enterprise.tests import enterprise_test from authentik.lib.generators import generate_id -@patch_license +@enterprise_test() class TestLifecycleRuleAPI(APITestCase): def setUp(self): @@ -179,7 +179,7 @@ class TestLifecycleRuleAPI(APITestCase): self.assertFalse(LifecycleRule.objects.filter(pk=rule.pk).exists()) -@patch_license +@enterprise_test() class TestIterationAPI(APITestCase): def setUp(self): @@ -268,7 +268,7 @@ class TestIterationAPI(APITestCase): self.assertIn("user_can_review", response.data["results"][0]) -@patch_license +@enterprise_test() class TestReviewAPI(APITestCase): def setUp(self): diff --git a/authentik/enterprise/reports/tests/test_api.py b/authentik/enterprise/reports/tests/test_api.py index 9c99515262..1443f4573b 100644 --- a/authentik/enterprise/reports/tests/test_api.py +++ b/authentik/enterprise/reports/tests/test_api.py @@ -4,11 +4,11 @@ from rest_framework.test import APITestCase from authentik.core.models import User from authentik.core.tests.utils import create_test_admin_user -from authentik.enterprise.reports.tests.utils import patch_license +from authentik.enterprise.tests import enterprise_test from authentik.events.models import Event -@patch_license +@enterprise_test() class TestExportAPI(APITestCase): def setUp(self) -> None: self.user = create_test_admin_user() diff --git a/authentik/enterprise/reports/tests/test_event_export.py b/authentik/enterprise/reports/tests/test_event_export.py index 95506ef198..95c787ca6b 100644 --- a/authentik/enterprise/reports/tests/test_event_export.py +++ b/authentik/enterprise/reports/tests/test_event_export.py @@ -3,11 +3,11 @@ from django.test.testcases import TestCase from authentik.core.tests.utils import create_test_user from authentik.enterprise.reports.models import DataExport -from authentik.enterprise.reports.tests.utils import patch_license +from authentik.enterprise.tests import enterprise_test from authentik.events.models import Event, EventAction -@patch_license +@enterprise_test() class TestEventExport(TestCase): def setUp(self) -> None: self.user = create_test_user() diff --git a/authentik/enterprise/reports/tests/test_permissions.py b/authentik/enterprise/reports/tests/test_permissions.py index 22bf93048d..020970e3ca 100644 --- a/authentik/enterprise/reports/tests/test_permissions.py +++ b/authentik/enterprise/reports/tests/test_permissions.py @@ -2,10 +2,10 @@ from django.urls import reverse from rest_framework.test import APITestCase from authentik.core.tests.utils import create_test_user -from authentik.enterprise.reports.tests.utils import patch_license +from authentik.enterprise.tests import enterprise_test -@patch_license +@enterprise_test() class TestExportPermissions(APITestCase): def setUp(self) -> None: self.user = create_test_user() diff --git a/authentik/enterprise/reports/tests/test_schema.py b/authentik/enterprise/reports/tests/test_schema.py index 03e7990b56..4240114bf1 100644 --- a/authentik/enterprise/reports/tests/test_schema.py +++ b/authentik/enterprise/reports/tests/test_schema.py @@ -1,10 +1,10 @@ from django.test.testcases import TestCase from drf_spectacular.generators import SchemaGenerator -from authentik.enterprise.reports.tests.utils import patch_license +from authentik.enterprise.tests import enterprise_test -@patch_license +@enterprise_test() class TestSchemaMatch(TestCase): def setUp(self) -> None: generator = SchemaGenerator() diff --git a/authentik/enterprise/reports/tests/test_user_export.py b/authentik/enterprise/reports/tests/test_user_export.py index 18b4cf665a..164b7572d9 100644 --- a/authentik/enterprise/reports/tests/test_user_export.py +++ b/authentik/enterprise/reports/tests/test_user_export.py @@ -7,10 +7,10 @@ from authentik.admin.files.tests.utils import FileTestFileBackendMixin from authentik.core.models import User from authentik.core.tests.utils import create_test_user from authentik.enterprise.reports.models import DataExport -from authentik.enterprise.reports.tests.utils import patch_license +from authentik.enterprise.tests import enterprise_test -@patch_license +@enterprise_test() class TestUserExport(FileTestFileBackendMixin, TestCase): def setUp(self) -> None: super().setUp() diff --git a/authentik/enterprise/reports/tests/utils.py b/authentik/enterprise/reports/tests/utils.py deleted file mode 100644 index 5d5edc306c..0000000000 --- a/authentik/enterprise/reports/tests/utils.py +++ /dev/null @@ -1,6 +0,0 @@ -from unittest.mock import MagicMock, patch - -patch_license = patch( - "authentik.enterprise.models.LicenseUsageStatus.is_valid", - MagicMock(return_value=True), -)