chore: Write GSI index partition key for meta documents (#3146)

Flagsmith · Dec 12, 2023 · 76cc152 · 76cc152 · vercel · Dec 12, 2023
1 parent 352d6cc
commit 76cc152
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/api/tests/unit/util/mappers/test_unit_mappers_dynamodb.py b/api/tests/unit/util/mappers/test_unit_mappers_dynamodb.py
@@ -149,9 +149,9 @@ def test_map_environment_to_environment_v2_document__call_expected(
     assert result == {
         "document_key": ENVIRONMENTS_V2_ENVIRONMENT_META_DOCUMENT_KEY,
         "environment_id": str(environment.id),
+        "environment_api_key": expected_api_key,
         "allow_client_traits": True,
         "amplitude_config": None,
-        "api_key": expected_api_key,
         "dynatrace_config": None,
         "feature_states": [
             {

diff --git a/api/util/mappers/dynamodb.py b/api/util/mappers/dynamodb.py
@@ -55,9 +55,12 @@ def map_environment_to_environment_document(
 def map_environment_to_environment_v2_document(
     environment: "Environment",
 ) -> Document:
+    environment_document = map_environment_to_environment_document(environment)
+    environment_api_key = environment_document.pop("api_key")
     return {
-        **map_environment_to_environment_document(environment),
+        **environment_document,
         "document_key": ENVIRONMENTS_V2_ENVIRONMENT_META_DOCUMENT_KEY,
+        "environment_api_key": environment_api_key,
         "environment_id": str(environment.id),
     }