forked from MirrorHub/synapse
Fix unit tests
This commit is contained in:
parent
bed10f9880
commit
93b32d4515
|
@ -62,6 +62,7 @@ class TypingNotificationsTestCase(unittest.TestCase):
|
||||||
self.on_new_event = mock_notifier.on_new_event
|
self.on_new_event = mock_notifier.on_new_event
|
||||||
|
|
||||||
self.auth = Mock(spec=[])
|
self.auth = Mock(spec=[])
|
||||||
|
self.state_handler = Mock()
|
||||||
|
|
||||||
hs = yield setup_test_homeserver(
|
hs = yield setup_test_homeserver(
|
||||||
"test",
|
"test",
|
||||||
|
@ -75,6 +76,7 @@ class TypingNotificationsTestCase(unittest.TestCase):
|
||||||
"set_received_txn_response",
|
"set_received_txn_response",
|
||||||
"get_destination_retry_timings",
|
"get_destination_retry_timings",
|
||||||
]),
|
]),
|
||||||
|
state_handler=self.state_handler,
|
||||||
handlers=None,
|
handlers=None,
|
||||||
notifier=mock_notifier,
|
notifier=mock_notifier,
|
||||||
resource_for_client=Mock(),
|
resource_for_client=Mock(),
|
||||||
|
@ -113,6 +115,10 @@ class TypingNotificationsTestCase(unittest.TestCase):
|
||||||
return set(member.domain for member in self.room_members)
|
return set(member.domain for member in self.room_members)
|
||||||
self.datastore.get_joined_hosts_for_room = get_joined_hosts_for_room
|
self.datastore.get_joined_hosts_for_room = get_joined_hosts_for_room
|
||||||
|
|
||||||
|
def get_current_user_in_room(room_id):
|
||||||
|
return set(str(u) for u in self.room_members)
|
||||||
|
self.state_handler.get_current_user_in_room = get_current_user_in_room
|
||||||
|
|
||||||
self.auth.check_joined_room = check_joined_room
|
self.auth.check_joined_room = check_joined_room
|
||||||
|
|
||||||
# Some local users to test with
|
# Some local users to test with
|
||||||
|
|
|
@ -78,44 +78,3 @@ class RoomMemberStoreTestCase(unittest.TestCase):
|
||||||
)
|
)
|
||||||
)]
|
)]
|
||||||
)
|
)
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
|
||||||
def test_room_hosts(self):
|
|
||||||
yield self.inject_room_member(self.room, self.u_alice, Membership.JOIN)
|
|
||||||
|
|
||||||
self.assertEquals(
|
|
||||||
{"test"},
|
|
||||||
(yield self.store.get_joined_hosts_for_room(self.room.to_string()))
|
|
||||||
)
|
|
||||||
|
|
||||||
# Should still have just one host after second join from it
|
|
||||||
yield self.inject_room_member(self.room, self.u_bob, Membership.JOIN)
|
|
||||||
|
|
||||||
self.assertEquals(
|
|
||||||
{"test"},
|
|
||||||
(yield self.store.get_joined_hosts_for_room(self.room.to_string()))
|
|
||||||
)
|
|
||||||
|
|
||||||
# Should now have two hosts after join from other host
|
|
||||||
yield self.inject_room_member(self.room, self.u_charlie, Membership.JOIN)
|
|
||||||
|
|
||||||
self.assertEquals(
|
|
||||||
{"test", "elsewhere"},
|
|
||||||
(yield self.store.get_joined_hosts_for_room(self.room.to_string()))
|
|
||||||
)
|
|
||||||
|
|
||||||
# Should still have both hosts
|
|
||||||
yield self.inject_room_member(self.room, self.u_alice, Membership.LEAVE)
|
|
||||||
|
|
||||||
self.assertEquals(
|
|
||||||
{"test", "elsewhere"},
|
|
||||||
(yield self.store.get_joined_hosts_for_room(self.room.to_string()))
|
|
||||||
)
|
|
||||||
|
|
||||||
# Should have only one host after other leaves
|
|
||||||
yield self.inject_room_member(self.room, self.u_charlie, Membership.LEAVE)
|
|
||||||
|
|
||||||
self.assertEquals(
|
|
||||||
{"test"},
|
|
||||||
(yield self.store.get_joined_hosts_for_room(self.room.to_string()))
|
|
||||||
)
|
|
||||||
|
|
Loading…
Reference in a new issue