Have MemoryDataStore's get_rooms_for_user_where_membership_is() return room membership event objects, as per interface contract

This commit is contained in:
Paul "LeoNerd" Evans 2014-09-01 16:14:07 +01:00
parent fd696f1243
commit 59d3955db1

View file

@ -188,8 +188,9 @@ class MemoryDataStore(object):
def get_rooms_for_user_where_membership_is(self, user_id, membership_list):
return [
r for r in self.members
if self.members[r].get(user_id).membership in membership_list
self.members[r].get(user_id) for r in self.members
if user_id in self.members[r] and
self.members[r][user_id].membership in membership_list
]
def get_room_events_stream(self, user_id=None, from_key=None, to_key=None,