Various minor fixes to unit-test structure around typing notifications
This commit is contained in:
parent
0268d40281
commit
04c7f3576e
|
@ -126,6 +126,13 @@ class TypingNotificationsTestCase(unittest.TestCase):
|
|||
return defer.succeed([])
|
||||
self.room_member_handler.get_room_members = get_room_members
|
||||
|
||||
def get_joined_rooms_for_user(user):
|
||||
if user in self.room_members:
|
||||
return defer.succeed([self.room_id])
|
||||
else:
|
||||
return defer.succeed([])
|
||||
self.room_member_handler.get_joined_rooms_for_user = get_joined_rooms_for_user
|
||||
|
||||
@defer.inlineCallbacks
|
||||
def fetch_room_distributions_into(room_id, localusers=None,
|
||||
remotedomains=None, ignore_user=None):
|
||||
|
|
|
@ -34,6 +34,8 @@ class RoomTypingTestCase(RestTestCase):
|
|||
""" Tests /rooms/$room_id/typing/$user_id REST API. """
|
||||
user_id = "@sid:red"
|
||||
|
||||
user = UserID.from_string(user_id)
|
||||
|
||||
@defer.inlineCallbacks
|
||||
def setUp(self):
|
||||
self.clock = MockClock()
|
||||
|
@ -75,7 +77,7 @@ class RoomTypingTestCase(RestTestCase):
|
|||
|
||||
def get_room_members(room_id):
|
||||
if room_id == self.room_id:
|
||||
return defer.succeed([UserID.from_string(self.user_id)])
|
||||
return defer.succeed([self.user])
|
||||
else:
|
||||
return defer.succeed([])
|
||||
|
||||
|
@ -115,7 +117,7 @@ class RoomTypingTestCase(RestTestCase):
|
|||
self.assertEquals(200, code)
|
||||
|
||||
self.assertEquals(self.event_source.get_current_key(), 1)
|
||||
events = yield self.event_source.get_new_events_for_user(self.user_id, 0, None)
|
||||
events = yield self.event_source.get_new_events_for_user(self.user, 0, None)
|
||||
self.assertEquals(
|
||||
events[0],
|
||||
[
|
||||
|
|
Loading…
Reference in New Issue