Only check aggregate counts for our target room

matrix-org · Nov 28, 2022 · ed3809f · ed3809f
1 parent 4068079
commit ed3809f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/storage/test_event_push_actions.py b/tests/storage/test_event_push_actions.py
@@ -182,7 +182,7 @@ def _assert_counts(notif_count: int, highlight_count: int) -> None:
                     user_id,
                 )
             )
-            self.assertEqual(sum(aggregate_counts.values()), notif_count)
+            self.assertEqual(aggregate_counts[room_id], notif_count)
 
         def _create_event(highlight: bool = False) -> str:
             result = self.helper.send_event(
@@ -335,7 +335,7 @@ def _assert_counts(
                 )
             )
             self.assertEqual(
-                sum(aggregate_counts.values()), notif_count + thread_notif_count
+                aggregate_counts[room_id], notif_count + thread_notif_count
             )
 
         def _create_event(
@@ -517,7 +517,7 @@ def _assert_counts(
                 )
             )
             self.assertEqual(
-                sum(aggregate_counts.values()), notif_count + thread_notif_count
+                aggregate_counts[room_id], notif_count + thread_notif_count
             )
 
         def _create_event(