diff options
author | Patrick Cloke <clokep@users.noreply.github.com> | 2021-12-21 08:25:34 -0500 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-12-21 13:25:34 +0000 |
commit | b6102230a7391d1acaa50cc6c389813f7e0fab84 (patch) | |
tree | 605757fe7627d00ea873fb7a69f63128fbe53432 /tests/storage | |
parent | Various opentracing enhancements (#11619) (diff) | |
download | synapse-b6102230a7391d1acaa50cc6c389813f7e0fab84.tar.xz |
Add type hints to event_push_actions. (#11594)
Diffstat (limited to 'tests/storage')
-rw-r--r-- | tests/storage/test_event_push_actions.py | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/tests/storage/test_event_push_actions.py b/tests/storage/test_event_push_actions.py index bb5939ba4a..738f3ad1dc 100644 --- a/tests/storage/test_event_push_actions.py +++ b/tests/storage/test_event_push_actions.py @@ -14,6 +14,8 @@ from unittest.mock import Mock +from synapse.storage.databases.main.event_push_actions import NotifCounts + from tests.unittest import HomeserverTestCase USER_ID = "@user:example.com" @@ -57,11 +59,11 @@ class EventPushActionsStoreTestCase(HomeserverTestCase): ) self.assertEquals( counts, - { - "notify_count": noitf_count, - "unread_count": 0, # Unread counts are tested in the sync tests. - "highlight_count": highlight_count, - }, + NotifCounts( + notify_count=noitf_count, + unread_count=0, # Unread counts are tested in the sync tests. + highlight_count=highlight_count, + ), ) def _inject_actions(stream, action): |