chore(billing): Add profile chunk to org stats v2 endpoint

src/sentry/api/endpoints/organization_stats_v2.py

@@ -99,7 +99,16 @@ class OrgStatsQueryParamsSerializer(serializers.Serializer):
     )
 
     category = serializers.ChoiceField(
-        ("error", "transaction", "attachment", "replay", "profile", "profile_duration", "monitor"),
+        (
+            "error",
+            "transaction",
+            "attachment",
+            "replay",
+            "profile",
+            "profile_duration",
+            "profile_chunk",
+            "monitor",
+        ),
         required=False,
         help_text=(
             "Filter by data category. Each category represents a different type of data:\n\n"
@@ -109,6 +118,7 @@ class OrgStatsQueryParamsSerializer(serializers.Serializer):
             "- `replay`: Session replay events\n"
             "- `profile`: Performance profiles\n"
             "- `profile_duration`: Profile duration data (note: cannot be combined with other categories since quantity represents milliseconds)\n"
+            "- `profile_chunk`: Profile chunk\n"
             "- `monitor`: Cron monitor events"
         ),
     )

tests/apidocs/endpoints/organizations/test_org_stats_v2.py

@@ -71,3 +71,18 @@ def test_profile_duration_category(self):
         request = RequestFactory().get(self.url)
 
         self.validate_schema(request, response)
+
+    def test_profile_chunk_category(self):
+        """
+        Test that the organization stats endpoint correctly handles profile chunk category.
+        """
+        query = {
+            "interval": "1d",
+            "field": "sum(quantity)",
+            "groupBy": "category",
+            "category": "profile_chunk",
+        }
+        response = self.client.get(self.url, query, format="json")
+        request = RequestFactory().get(self.url)
+
+        self.validate_schema(request, response)

tests/snuba/api/endpoints/test_organization_stats_v2.py

@@ -98,6 +98,17 @@ def setUp(self):
             3,
         )
 
+        # Add profile_chunk outcome data
+        self.store_outcomes(
+            {
+                "org_id": self.org.id,
+                "timestamp": self.now - timedelta(hours=1),
+                "project_id": self.project.id,
+                "category": DataCategory.PROFILE_CHUNK,
+                "quantity": 10,
+            },
+        )
+
     def do_request(self, query, user=None, org=None, status_code=200):
         self.login_as(user=user or self.user)
         org_slug = (org or self.organization).slug
@@ -474,6 +485,11 @@ def test_org_simple(self):
                     "series": {"sum(quantity)": [0, 0, 3000]},
                     "totals": {"sum(quantity)": 3000},
                 },
+                {
+                    "by": {"category": "profile_chunk", "outcome": "accepted", "reason": "none"},
+                    "series": {"sum(quantity)": [0, 0, 10]},
+                    "totals": {"sum(quantity)": 10},
+                },
                 {
                     "by": {
                         "category": "transaction",
@@ -930,6 +946,61 @@ def test_profile_duration_groupby(self):
             ],
         }
 
+    @freeze_time("2021-03-14T12:27:28.303Z")
+    def test_profile_chunk_filter(self):
+        """Test that profile_chunk data is correctly filtered and returned"""
+        response = self.do_request(
+            {
+                "project": [-1],
+                "statsPeriod": "1d",
+                "interval": "1d",
+                "field": ["sum(quantity)"],
+                "category": ["profile_chunk"],
+            },
+            status_code=200,
+        )
+
+        assert result_sorted(response.data) == {
+            "start": "2021-03-13T00:00:00Z",
+            "end": "2021-03-15T00:00:00Z",
+            "intervals": ["2021-03-13T00:00:00Z", "2021-03-14T00:00:00Z"],
+            "groups": [
+                {
+                    "by": {},
+                    "series": {"sum(quantity)": [0, 10]},
+                    "totals": {"sum(quantity)": 10},
+                }
+            ],
+        }
+
+    @freeze_time("2021-03-14T12:27:28.303Z")
+    def test_profile_chunk_groupby(self):
+        """Test that profile_chunk data is correctly grouped"""
+        response = self.do_request(
+            {
+                "project": [-1],
+                "statsPeriod": "1d",
+                "interval": "1d",
+                "field": ["sum(quantity)"],
+                "groupBy": ["category"],
+                "category": ["profile_chunk"],
+            },
+            status_code=200,
+        )
+
+        assert result_sorted(response.data) == {
+            "start": "2021-03-13T00:00:00Z",
+            "end": "2021-03-15T00:00:00Z",
+            "intervals": ["2021-03-13T00:00:00Z", "2021-03-14T00:00:00Z"],
+            "groups": [
+                {
+                    "by": {"category": "profile_chunk"},
+                    "series": {"sum(quantity)": [0, 10]},
+                    "totals": {"sum(quantity)": 10},
+                }
+            ],
+        }
+
 
 def result_sorted(result):
     """sort the groups of the results array by the `by` object, ensuring a stable order"""