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,9 +230,11 @@ 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, user_id, actions, event.event_id, {
user_id: actions
for user_id, actions in push_actions
},
) )
ordering = None ordering = None

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,