|
@@ -62,6 +62,7 @@ class EventPushActionsStoreTestCase(tests.unittest.TestCase):
|
|
|
{"notify_count": noitf_count, "highlight_count": highlight_count}
|
|
|
)
|
|
|
|
|
|
+ @defer.inlineCallbacks
|
|
|
def _inject_actions(stream, action):
|
|
|
event = Mock()
|
|
|
event.room_id = room_id
|
|
@@ -69,11 +70,12 @@ class EventPushActionsStoreTestCase(tests.unittest.TestCase):
|
|
|
event.internal_metadata.stream_ordering = stream
|
|
|
event.depth = stream
|
|
|
|
|
|
- tuples = [(user_id, action)]
|
|
|
-
|
|
|
- return self.store.runInteraction(
|
|
|
+ yield self.store.add_push_actions_to_staging(
|
|
|
+ event.event_id, user_id, action,
|
|
|
+ )
|
|
|
+ yield self.store.runInteraction(
|
|
|
"", self.store._set_push_actions_for_event_and_users_txn,
|
|
|
- event, tuples
|
|
|
+ event,
|
|
|
)
|
|
|
|
|
|
def _rotate(stream):
|