From 31807262c0750a4f71a793579092e9e4a3b51abc Mon Sep 17 00:00:00 2001
From: Bianca Yang <21014901+xrdt@users.noreply.github.com>
Date: Wed, 31 Jan 2024 19:32:36 -0800
Subject: [PATCH] fix: fix structure of group properties param in
 feature_enabled call (#20079)

fix structure of group properties param
---
 ee/billing/quota_limiting.py           | 4 +---
 ee/billing/test/test_quota_limiting.py | 2 +-
 2 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/ee/billing/quota_limiting.py b/ee/billing/quota_limiting.py
index 0476ce5d8bb2a..e492d3338cf7d 100644
--- a/ee/billing/quota_limiting.py
+++ b/ee/billing/quota_limiting.py
@@ -96,9 +96,7 @@ def org_quota_limited_until(organization: Organization, resource: QuotaResource)
         QUOTA_LIMIT_DATA_RETENTION_FLAG,
         organization.id,
         groups={"organization": str(organization.id)},
-        group_properties={
-            "organization": str(organization.id),
-        },
+        group_properties={"organization": {"id": str(organization.id)}},
     ):
         # Don't drop data for this org but record that they __would have__ been
         # limited.
diff --git a/ee/billing/test/test_quota_limiting.py b/ee/billing/test/test_quota_limiting.py
index bffed0f9f3578..ce71caa5a5697 100644
--- a/ee/billing/test/test_quota_limiting.py
+++ b/ee/billing/test/test_quota_limiting.py
@@ -60,7 +60,7 @@ def test_dont_quota_limit_feature_flag_enabled(self, patch_feature_enabled, patc
             QUOTA_LIMIT_DATA_RETENTION_FLAG,
             self.organization.id,
             groups={"organization": str(self.organization.id)},
-            group_properties={"organization": str(self.organization.id)},
+            group_properties={"organization": {"id": str(self.organization.id)}},
         )
         patch_capture.assert_called_once_with(
             str(self.organization.id),