group attributes join

getsentry · Jun 7, 2024 · 185cf5d · 185cf5d
1 parent 4dfe0eb
commit 185cf5d
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/tests/datasets/test_group_attributes_join.py b/tests/datasets/test_group_attributes_join.py
@@ -244,7 +244,7 @@ def query_events_inner_joined_group_attributes(self):
             ),
         )
 
-    @pytest.mark.clickhouse_db
+    @pytest.mark.clickhouse_db(storage_keys=["errors", "group_attributes"])
     @pytest.mark.redis_db
     def test_group_attributes_join(self) -> None:
         response = self.query_events_inner_joined_group_attributes()
@@ -283,7 +283,7 @@ def test_group_attributes_join(self) -> None:
             }.items()
         )
 
-    @pytest.mark.clickhouse_db
+    @pytest.mark.clickhouse_db(storage_keys=["errors", "group_attributes"])
     @pytest.mark.redis_db
     def test_group_attributes_inner_join(self) -> None:
         response = self.query_events_joined_group_attributes()
@@ -322,7 +322,7 @@ def test_group_attributes_inner_join(self) -> None:
             }.items()
         )
 
-    @pytest.mark.clickhouse_db
+    @pytest.mark.clickhouse_db(storage_keys=["errors", "group_attributes"])
     @pytest.mark.redis_db
     def test_group_attributes_join_after_delete(self) -> None:
         delete_row = self.initial_group_attributes.copy()
@@ -532,7 +532,9 @@ def assert_joined_final(
             ),
         )
 
-    @pytest.mark.clickhouse_db
+    @pytest.mark.clickhouse_db(
+        storage_keys=["errors", "group_attributes", "search_issues"]
+    )
     @pytest.mark.redis_db
     def test_group_attributes_join(self) -> None:
         response = self.query_search_issues_joined_group_attributes()
@@ -573,7 +575,7 @@ def test_group_attributes_join(self) -> None:
             }.items()
         )
 
-    @pytest.mark.clickhouse_db
+    @pytest.mark.clickhouse_db(storage_keys=["errors", "search_issues"])
     @pytest.mark.redis_db
     def test_group_attributes_join_after_delete(self) -> None:
         delete_row = self.initial_group_attributes.copy()