Fix stream test.
This commit is contained in:
parent
e2cebe26e8
commit
3cd9c02f71
|
@ -48,7 +48,7 @@ class StreamStoreTestCase(unittest.TestCase):
|
||||||
self.depth = 1
|
self.depth = 1
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def inject_room_member(self, room, user, membership, prev_state=None):
|
def inject_room_member(self, room, user, membership, replaces_state=None):
|
||||||
self.depth += 1
|
self.depth += 1
|
||||||
|
|
||||||
event = self.event_factory.create_event(
|
event = self.event_factory.create_event(
|
||||||
|
@ -64,9 +64,12 @@ class StreamStoreTestCase(unittest.TestCase):
|
||||||
|
|
||||||
event.state_events = None
|
event.state_events = None
|
||||||
event.hashes = {}
|
event.hashes = {}
|
||||||
|
event.prev_state = []
|
||||||
|
event.auth_events = []
|
||||||
|
|
||||||
if prev_state:
|
if replaces_state:
|
||||||
event.prev_state = prev_state
|
event.prev_state = [(replaces_state, "hash")]
|
||||||
|
event.replaces_state = replaces_state
|
||||||
|
|
||||||
# Have to create a join event using the eventfactory
|
# Have to create a join event using the eventfactory
|
||||||
yield self.store.persist_event(
|
yield self.store.persist_event(
|
||||||
|
@ -90,6 +93,7 @@ class StreamStoreTestCase(unittest.TestCase):
|
||||||
|
|
||||||
event.state_events = None
|
event.state_events = None
|
||||||
event.hashes = {}
|
event.hashes = {}
|
||||||
|
event.auth_events = []
|
||||||
|
|
||||||
# Have to create a join event using the eventfactory
|
# Have to create a join event using the eventfactory
|
||||||
yield self.store.persist_event(
|
yield self.store.persist_event(
|
||||||
|
@ -216,7 +220,7 @@ class StreamStoreTestCase(unittest.TestCase):
|
||||||
|
|
||||||
event2 = yield self.inject_room_member(
|
event2 = yield self.inject_room_member(
|
||||||
self.room1, self.u_alice, Membership.JOIN,
|
self.room1, self.u_alice, Membership.JOIN,
|
||||||
prev_state=event1.event_id,
|
replaces_state=event1.event_id,
|
||||||
)
|
)
|
||||||
|
|
||||||
end = yield self.store.get_room_events_max_id()
|
end = yield self.store.get_room_events_max_id()
|
||||||
|
@ -233,4 +237,7 @@ class StreamStoreTestCase(unittest.TestCase):
|
||||||
|
|
||||||
event = results[0]
|
event = results[0]
|
||||||
|
|
||||||
self.assertTrue(hasattr(event, "prev_content"), msg="No prev_content key")
|
self.assertTrue(
|
||||||
|
hasattr(event, "prev_content"),
|
||||||
|
msg="No prev_content key"
|
||||||
|
)
|
||||||
|
|
Loading…
Reference in New Issue