@@ -363,14 +363,16 @@ def test_accept_connection_multiple(self):
|
363 | 363 | sock.accept.return_value = (mock.Mock(), mock.Mock()) |
364 | 364 | backlog = 100 |
365 | 365 | # Mock the coroutine generation for a connection to prevent |
366 | | -# warnings related to un-awaited coroutines. |
| 366 | +# warnings related to un-awaited coroutines. _accept_connection2 |
| 367 | +# is an async function that is patched with AsyncMock. create_task |
| 368 | +# creates a task out of coroutine returned by AsyncMock, so use |
| 369 | +# asyncio.sleep(0) to ensure created tasks are complete to avoid |
| 370 | +# task pending warnings. |
367 | 371 | mock_obj = mock.patch.object |
368 | 372 | with mock_obj(self.loop, '_accept_connection2') as accept2_mock: |
369 | | -accept2_mock.return_value = None |
370 | | -with mock_obj(self.loop, 'create_task') as task_mock: |
371 | | -task_mock.return_value = None |
372 | | -self.loop._accept_connection( |
373 | | -mock.Mock(), sock, backlog=backlog) |
| 373 | +self.loop._accept_connection( |
| 374 | +mock.Mock(), sock, backlog=backlog) |
| 375 | +self.loop.run_until_complete(asyncio.sleep(0)) |
374 | 376 | self.assertEqual(sock.accept.call_count, backlog) |
375 | 377 | |
376 | 378 | |
|