dedupe helper

Signed-off-by: Jens Langhammer <jens@goauthentik.io>

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):

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()

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()

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()

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()

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()

authentik/enterprise/reports/tests/utils.py

@@ -1,6 +0,0 @@
-from unittest.mock import MagicMock, patch
-
-patch_license = patch(
-    "authentik.enterprise.models.LicenseUsageStatus.is_valid",
-    MagicMock(return_value=True),
-)