Fix unit tests

This commit is contained in:
Erik Johnston 2018-02-20 11:41:40 +00:00
parent 6ff8c87484
commit e440e28456
2 changed files with 7 additions and 5 deletions

View file

@ -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:

View file

@ -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,