@@ -258,28 +258,38 @@ def test_notify_interested_services_ephemeral(self):
258
258
services = [interested_service ]
259
259
260
260
self .mock_store .get_app_services .return_value = services
261
- self .mock_store .get_type_stream_id_for_appservice .return_value = make_awaitable (579 )
261
+ self .mock_store .get_type_stream_id_for_appservice .return_value = make_awaitable (
262
+ 579
263
+ )
262
264
263
265
event = Mock (event_id = "event_1" )
264
- self .event_source .sources .receipt .get_new_events_as .return_value = make_awaitable (([event ], None ))
266
+ self .event_source .sources .receipt .get_new_events_as .return_value = (
267
+ make_awaitable (([event ], None ))
268
+ )
265
269
266
270
self .handler .notify_interested_services_ephemeral ("receipt_key" , 580 )
267
271
self .mock_scheduler .submit_ephemeral_events_for_as .assert_called_once_with (
268
272
interested_service , [event ]
269
273
)
270
274
self .mock_store .set_type_stream_id_for_appservice .assert_called_once_with (
271
- interested_service , "read_receipt" , 580 ,
275
+ interested_service ,
276
+ "read_receipt" ,
277
+ 580 ,
272
278
)
273
279
274
280
def test_notify_interested_services_ephemeral_out_of_order (self ):
275
281
interested_service = self ._mkservice (is_interested = True )
276
282
services = [interested_service ]
277
283
278
284
self .mock_store .get_app_services .return_value = services
279
- self .mock_store .get_type_stream_id_for_appservice .return_value = make_awaitable (580 )
285
+ self .mock_store .get_type_stream_id_for_appservice .return_value = make_awaitable (
286
+ 580
287
+ )
280
288
281
289
event = Mock (event_id = "event_1" )
282
- self .event_source .sources .receipt .get_new_events_as .return_value = make_awaitable (([event ], None ))
290
+ self .event_source .sources .receipt .get_new_events_as .return_value = (
291
+ make_awaitable (([event ], None ))
292
+ )
283
293
284
294
self .handler .notify_interested_services_ephemeral ("receipt_key" , 579 )
285
295
self .mock_scheduler .submit_ephemeral_events_for_as .assert_not_called ()
0 commit comments