Fix test import and add UUID processor

getsentry · Jun 7, 2024 · 884fa18 · 884fa18
1 parent 72d160e
commit 884fa18
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/snuba/datasets/configuration/group_attributes/entities/group_attributes.yaml b/snuba/datasets/configuration/group_attributes/entities/group_attributes.yaml
@@ -38,6 +38,9 @@ storage_selector:
 
 query_processors:
   - processor: BasicFunctionsProcessor
+  - processor: UUIDColumnProcessor
+    args:
+      columns: [group_first_release_id]
 
 validate_data_model: error
 validators:

diff --git a/snuba/datasets/configuration/group_attributes/storages/group_attributes.yaml b/snuba/datasets/configuration/group_attributes/storages/group_attributes.yaml
@@ -59,6 +59,9 @@ allocation_policies:
 query_processors:
   - processor: TableRateLimit
   - processor: ConsistencyEnforcerProcessor
+  - processor: UUIDColumnProcessor
+    args:
+      columns: [group_first_release_id]
 
 mandatory_condition_checkers:
   - condition: ProjectIdEnforcer

diff --git a/tests/datasets/test_group_attributes_processor.py b/tests/datasets/test_group_attributes_processor.py
@@ -1,5 +1,6 @@
 from datetime import datetime
 from typing import Optional
+import uuid
 
 import pytest
 from sentry_kafka_schemas.schema_types.group_attributes_v1 import (