fix tests

elementary-data · Aug 29, 2023 · 6274d18 · 6274d18
1 parent 0cf08da
commit 6274d18
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/tests/mocks/fetchers/alerts_fetcher_mock.py b/tests/mocks/fetchers/alerts_fetcher_mock.py
@@ -52,6 +52,7 @@ def query_pending_test_alerts(self, *args, **kwargs):
                 suppression_status="pending",
                 sent_at=None,
                 test_created_at="2022-10-09 10:10:10",
+                elementary_unique_id="elementary.model_id_1.test_id_1.9cf2f5f6ad.None.generic",
             ),
             # Alert after suppression interval
             dict(
@@ -82,6 +83,7 @@ def query_pending_test_alerts(self, *args, **kwargs):
                 suppression_status="pending",
                 sent_at=None,
                 test_created_at="2022-10-09 10:10:10",
+                elementary_unique_id="elementary.model_id_1.test_id_4.9cf2f5f6ad.None.generic",
             ),
             # Alert without suppression interval
             dict(
@@ -112,6 +114,7 @@ def query_pending_test_alerts(self, *args, **kwargs):
                 suppression_status="pending",
                 sent_at=None,
                 test_created_at="2022-10-09 10:10:10",
+                elementary_unique_id="elementary.model_id_1.test_id_2.9cf2f5f6ad.None.generic",
             ),
             # First occurrence alert with suppression interval
             dict(
@@ -142,6 +145,7 @@ def query_pending_test_alerts(self, *args, **kwargs):
                 suppression_status="pending",
                 sent_at=None,
                 test_created_at="2022-10-09 10:10:10",
+                elementary_unique_id="elementary.model_id_1.test_id_3.9cf2f5f6ad.None.generic",
             ),
             # Duplicated alert that should be deduped
             dict(
@@ -174,6 +178,7 @@ def query_pending_test_alerts(self, *args, **kwargs):
                 suppression_status="pending",
                 sent_at=None,
                 test_created_at="2022-10-09 10:10:10",
+                elementary_unique_id="elementary.model_id_1.test_id_5.9cf2f5f6ad.None.generic",
             ),
         ]
 

diff --git a/tests/unit/monitor/alerts/group_alerts_by_table/mock_classes.py b/tests/unit/monitor/alerts/group_alerts_by_table/mock_classes.py
@@ -14,6 +14,7 @@ class MockAlert:
     owners: Optional[List[str]]
     subscribers: Optional[List[str]]
     tags: Optional[List[str]]
+    report_url: Optional[str]
     database_name: str = "elementary_test_db"
     schema_name: str = "master_elementary"
     concise_name: str = "Alert"