diff options
| author | Erik Johnston <erik@matrix.org> | 2021-05-05 17:27:05 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-05-05 17:27:05 +0100 |
| commit | 37623e33822d4e032ba6d1f523fb09b12fe27aab (patch) | |
| tree | da4c5662523edc77bae746321c2604a71ed983da /tests | |
| parent | Support stable MSC1772 spaces identifiers. (#9915) (diff) | |
| download | synapse-37623e33822d4e032ba6d1f523fb09b12fe27aab.tar.xz | |
Increase perf of handling presence when joining large rooms. (#9916)
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/handlers/test_presence.py | 14 |
1 files changed, 4 insertions, 10 deletions
diff --git a/tests/handlers/test_presence.py b/tests/handlers/test_presence.py index ce330e79cc..1ffab709fc 100644 --- a/tests/handlers/test_presence.py +++ b/tests/handlers/test_presence.py @@ -729,7 +729,7 @@ class PresenceJoinTestCase(unittest.HomeserverTestCase): ) self.assertEqual(expected_state.state, PresenceState.ONLINE) self.federation_sender.send_presence_to_destinations.assert_called_once_with( - destinations=["server2"], states={expected_state} + destinations={"server2"}, states=[expected_state] ) # @@ -740,7 +740,7 @@ class PresenceJoinTestCase(unittest.HomeserverTestCase): self._add_new_user(room_id, "@bob:server3") self.federation_sender.send_presence_to_destinations.assert_called_once_with( - destinations=["server3"], states={expected_state} + destinations={"server3"}, states=[expected_state] ) def test_remote_gets_presence_when_local_user_joins(self): @@ -788,14 +788,8 @@ class PresenceJoinTestCase(unittest.HomeserverTestCase): self.presence_handler.current_state_for_user("@test2:server") ) self.assertEqual(expected_state.state, PresenceState.ONLINE) - self.assertEqual( - self.federation_sender.send_presence_to_destinations.call_count, 2 - ) - self.federation_sender.send_presence_to_destinations.assert_any_call( - destinations=["server3"], states={expected_state} - ) - self.federation_sender.send_presence_to_destinations.assert_any_call( - destinations=["server2"], states={expected_state} + self.federation_sender.send_presence_to_destinations.assert_called_once_with( + destinations={"server2", "server3"}, states=[expected_state] ) def _add_new_user(self, room_id, user_id): |