Fix unit tests
This commit is contained in:
parent
6ff8c87484
commit
e440e28456
|
@ -230,10 +230,12 @@ class SlavedEventStoreTestCase(BaseSlavedStoreTestCase):
|
||||||
state_handler = self.hs.get_state_handler()
|
state_handler = self.hs.get_state_handler()
|
||||||
context = yield state_handler.compute_event_context(event)
|
context = yield state_handler.compute_event_context(event)
|
||||||
|
|
||||||
for user_id, actions in push_actions:
|
yield self.master_store.add_push_actions_to_staging(
|
||||||
yield self.master_store.add_push_actions_to_staging(
|
event.event_id, {
|
||||||
event.event_id, user_id, actions,
|
user_id: actions
|
||||||
)
|
for user_id, actions in push_actions
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
ordering = None
|
ordering = None
|
||||||
if backfill:
|
if backfill:
|
||||||
|
|
|
@ -71,7 +71,7 @@ class EventPushActionsStoreTestCase(tests.unittest.TestCase):
|
||||||
event.depth = stream
|
event.depth = stream
|
||||||
|
|
||||||
yield self.store.add_push_actions_to_staging(
|
yield self.store.add_push_actions_to_staging(
|
||||||
event.event_id, user_id, action,
|
event.event_id, {user_id: action},
|
||||||
)
|
)
|
||||||
yield self.store.runInteraction(
|
yield self.store.runInteraction(
|
||||||
"", self.store._set_push_actions_for_event_and_users_txn,
|
"", self.store._set_push_actions_for_event_and_users_txn,
|
||||||
|
|
Loading…
Reference in New Issue