Fix tests.

This commit is contained in:
Erik Johnston 2014-08-27 17:21:38 +01:00
parent 410a74b0f3
commit e111a06e0a
2 changed files with 3 additions and 4 deletions

View File

@ -74,8 +74,7 @@ class FederationTestCase(unittest.TestCase):
yield self.handlers.federation_handler.on_receive(event, False, False)
self.datastore.persist_event.assert_called_once_with(event, False)
self.notifier.on_new_room_event.assert_called_once_with(
event, store_id)
self.notifier.on_new_room_event.assert_called_once_with(event)
@defer.inlineCallbacks
def test_invite_join_target_this(self):

View File

@ -116,7 +116,7 @@ class RoomMemberHandlerTestCase(unittest.TestCase):
event
)
self.notifier.on_new_room_event.assert_called_once_with(
event, store_id)
event, extra_users=[self.hs.parse_userid(target_user_id)])
self.assertFalse(self.datastore.get_room.called)
self.assertFalse(self.datastore.store_room.called)
@ -175,7 +175,7 @@ class RoomMemberHandlerTestCase(unittest.TestCase):
event
)
self.notifier.on_new_room_event.assert_called_once_with(
event, store_id)
event, extra_users=[user])
join_signal_observer.assert_called_with(
user=user, room_id=room_id)