From ed3809f90caf932ba9133007a5329a6fa0d210d8 Mon Sep 17 00:00:00 2001 From: Nick Barrett Date: Mon, 28 Nov 2022 08:16:56 +0000 Subject: [PATCH] Only check aggregate counts for our target room --- tests/storage/test_event_push_actions.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/storage/test_event_push_actions.py b/tests/storage/test_event_push_actions.py index cc6d6c99d058..5fa8bd2d98ce 100644 --- 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(