Fix the presence tests

This commit is contained in:
Mark Haines 2015-05-20 13:22:18 +01:00
parent e01b825cc9
commit 8eca5bd50a
2 changed files with 6 additions and 10 deletions

View file

@ -295,6 +295,9 @@ class PresenceEventStreamTestCase(unittest.TestCase):
else:
return []
hs.handlers.room_member_handler.get_joined_rooms_for_user = get_rooms_for_user
hs.handlers.room_member_handler.get_room_members = (
lambda r: self.room_members if r == "a-room" else []
)
self.mock_datastore = hs.get_datastore()
self.mock_datastore.get_app_service_by_token = Mock(return_value=None)