TIL enable_features

Author: emyller

api/tests/unit/features/test_unit_features_views.py

@@ -62,6 +62,7 @@
 from projects.tags.models import Tag
 from segments.models import Segment
 from tests.types import (
+    EnableFeaturesFixture,
     WithEnvironmentPermissionsCallable,
     WithProjectPermissionsCallable,
 )
@@ -3582,15 +3583,15 @@ def test_list_features_with_filter_by_search_value_boolean(  # noqa: FT003
 
 
 def test_FeatureViewSet_list__code_references_ui_stats_enabled__returns_counts(
+    enable_features: EnableFeaturesFixture,
     feature: Feature,
-    mocker: MockerFixture,
     project: Project,
     staff_client: APIClient,
     with_project_permissions: WithProjectPermissionsCallable,
 ) -> None:
     # Given
     with_project_permissions([VIEW_PROJECT])  # type: ignore[call-arg]
-    mocker.patch.object(views, "is__code_references_ui_stats__enabled", return_value=True)  # fmt: skip
+    enable_features("code_references_ui_stats")
     with freeze_time("2099-01-01T10:00:00-0300"):
         FeatureFlagCodeReferencesScan.objects.create(
             project=project,
@@ -3670,15 +3671,15 @@ def test_FeatureViewSet_list__code_references_ui_stats_enabled__returns_counts(
 
 @pytest.mark.usefixtures("feature")
 def test_FeatureViewSet_list__no_code_reference_scans__returns_empty_counts(
+    enable_features: EnableFeaturesFixture,
     environment: Environment,
-    mocker: MockerFixture,
     project: Project,
     staff_client: APIClient,
     with_project_permissions: WithProjectPermissionsCallable,
 ) -> None:
     # Given
     with_project_permissions([VIEW_PROJECT])  # type: ignore[call-arg]
-    mocker.patch.object(views, "is__code_references_ui_stats__enabled", return_value=True)  # fmt: skip
+    enable_features("code_references_ui_stats")
 
     # When
     response = staff_client.get(
@@ -3694,16 +3695,16 @@ def test_FeatureViewSet_list__no_code_reference_scans__returns_empty_counts(
 
 # TODO: Delete this after https://github.com/flagsmith/flagsmith/issues/6832 is resolved
 def test_FeatureViewSet_list__code_references_ui_stats_disabled__returns_empty_counts(
+    enable_features: EnableFeaturesFixture,
     environment: Environment,
     feature: Feature,
-    mocker: MockerFixture,
     project: Project,
     staff_client: APIClient,
     with_project_permissions: WithProjectPermissionsCallable,
 ) -> None:
     # Given
     with_project_permissions([VIEW_PROJECT])  # type: ignore[call-arg]
-    mocker.patch.object(views, "is__code_references_ui_stats__enabled", return_value=False)  # fmt: skip
+    enable_features()  # code_references_ui_stats not enabled
     FeatureFlagCodeReferencesScan.objects.create(
         project=project,
         repository_url="https://github.flagsmith.com/backend/",