Fix ALL THE UNIT TESTS

This commit is contained in:
Kegan Dougal 2014-08-26 15:54:25 +01:00
parent 27979028b2
commit 8bd55cfdcb
5 changed files with 35 additions and 34 deletions

View file

@ -183,7 +183,7 @@ class MemoryDataStore(object):
def persist_event(self, event):
if event.type == RoomMemberEvent.TYPE:
room_id = event.room_id
user = event.target_user_id
user = event.state_key
membership = event.membership
self.members.setdefault(room_id, {})[user] = event
@ -196,7 +196,9 @@ class MemoryDataStore(object):
def get_current_state(self, room_id, event_type=None, state_key=""):
if event_type:
key = (room_id, event_type, state_key)
return self.current_state.get(key)
if self.current_state.get(key):
return [self.current_state.get(key)]
return None
else:
return [
e for e in self.current_state
@ -214,7 +216,7 @@ class MemoryDataStore(object):
def _format_call(args, kwargs):
return ", ".join(
["%r" % (a) for a in args] +
["%r" % (a) for a in args] +
["%s=%r" % (k, v) for k, v in kwargs.items()]
)