diff options
| author | Erik Johnston <erik@matrix.org> | 2014-09-01 18:21:29 +0100 |
|---|---|---|
| committer | Erik Johnston <erik@matrix.org> | 2014-09-01 18:21:29 +0100 |
| commit | 3faa2ae78c7ce2330ed7699ffe3fd08306055da9 (patch) | |
| tree | e97b09a70586b58cdc9f2533dae1d071167b4b71 /tests/utils.py | |
| parent | Add beginnings of ban support. (diff) | |
| parent | Follow API renaming. state -> presence. mtime_ago -> last_active_ago (diff) | |
| download | synapse-3faa2ae78c7ce2330ed7699ffe3fd08306055da9.tar.xz | |
Merge branch 'develop' of github.com:matrix-org/synapse into room_config
Diffstat (limited to 'tests/utils.py')
| -rw-r--r-- | tests/utils.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/utils.py b/tests/utils.py index 37b759febc..ea7d6893c6 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -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, |