fix(n+1): fix environment-document n+1 for env_feature_version (#5332)

gagantrivedi · matthewelwell · commit 21b91022378d · 2026-03-09T17:06:17.000Z
diff --git a/api/environments/models.py b/api/environments/models.py
@@ -406,6 +406,7 @@ def _get_environment_document_from_db(
                         "feature",
                         "feature_state_value",
                         "identity",
+                        "environment_feature_version",
                         "identity__environment",
                     ).prefetch_related(
                         Prefetch(
diff --git a/api/tests/unit/environments/test_unit_environments_views_sdk_environment.py b/api/tests/unit/environments/test_unit_environments_views_sdk_environment.py
@@ -30,7 +30,7 @@
 
 
 @pytest.mark.parametrize(
-    "use_v2_feature_versioning, total_queries", [(True, 12), (False, 11)]
+    "use_v2_feature_versioning, total_queries", [(True, 16), (False, 15)]
 )
 def test_get_environment_document(
     organisation_one: "Organisation",

Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,7 @@`
`30`	`30`
`31`	`31`
`32`	`32`	`@pytest.mark.parametrize(`
`33`		`- "use_v2_feature_versioning, total_queries", [(True, 12), (False, 11)]`
	`33`	`+ "use_v2_feature_versioning, total_queries", [(True, 16), (False, 15)]`
`34`	`34`	`)`
`35`	`35`	`def test_get_environment_document(`
`36`	`36`	`organisation_one: "Organisation",`