From 2c63a781f2fd7da33d7c9d9b63a17f563e850faf Mon Sep 17 00:00:00 2001 From: Andrew Liu <159852527+aliu39@users.noreply.github.com> Date: Mon, 9 Sep 2024 10:24:02 -0700 Subject: [PATCH] Fix snuba_perf_issue test --- .../endpoints/test_organization_group_index.py | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/tests/sentry/issues/endpoints/test_organization_group_index.py b/tests/sentry/issues/endpoints/test_organization_group_index.py index 8360ca13306528..1e2e3e2df3d744 100644 --- a/tests/sentry/issues/endpoints/test_organization_group_index.py +++ b/tests/sentry/issues/endpoints/test_organization_group_index.py @@ -17,6 +17,7 @@ PerformanceNPlusOneGroupType, PerformanceRenderBlockingAssetSpanGroupType, PerformanceSlowDBQueryGroupType, + ProfileFileIOGroupType, ) from sentry.models.activity import Activity from sentry.models.apitoken import ApiToken @@ -3148,10 +3149,11 @@ def test_snuba_perf_issue(self, mock_query: MagicMock) -> None: self.login_as(user=self.user) # give time for consumers to run and propogate changes to clickhouse sleep(1) - response = self.get_success_response( - sort="new", - query="user.email:myemail@example.com", - ) + with self.feature([ProfileFileIOGroupType.build_visible_feature_name()]): + response = self.get_success_response( + sort="new", + query="user.email:myemail@example.com", + ) assert len(response.data) == 2 assert {r["id"] for r in response.data} == { str(perf_group_id), @@ -3923,8 +3925,8 @@ def test_snuba_heavy_error_handled_boolean(self, _: MagicMock) -> None: def run_feedback_filtered_by_default_test(self, use_group_snuba_dataset: bool): with Feature( { - "organizations:feedback-ingest": True, - "organizations:feedback-visible": True, + FeedbackGroup.build_visible_feature_name(): True, + FeedbackGroup.build_ingest_feature_name(): True, "organizations:issue-search-snuba": use_group_snuba_dataset, } ): @@ -3957,8 +3959,8 @@ def test_feedback_filtered_by_default_use_snuba_search(self, _): def run_feedback_category_filter_test(self, use_group_snuba_dataset: bool): with Feature( { - "organizations:feedback-ingest": True, - "organizations:feedback-visible": True, + FeedbackGroup.build_visible_feature_name(): True, + FeedbackGroup.build_ingest_feature_name(): True, "organizations:issue-search-snuba": use_group_snuba_dataset, } ):