Mocking async call in python 3.5

Question

How do I mock async call from one native coroutine to other one using unittest.mock.patch?

I currently have quite an awkward solution:

class CoroutineMock(MagicMock):     def __await__(self, *args, **kwargs):         future = Future()         future.set_result(self)         result = yield from future         return result 

Then

class TestCoroutines(TestCase):     @patch('some.path', new_callable=CoroutineMock)     def test(self, mock):         some_action()         mock.assert_called_with(1,2,3) 

This works but looks ugly. Is there more pythonic way to do this?

Answer 1

Everyone's missing what's probably the simplest and clearest solution:

@patch('some.path') def test(self, mock):     f = asyncio.Future()     f.set_result('whatever result you want')     process_smtp_message.return_value = f     mock.assert_called_with(1, 2, 3) 

remember a coroutine can be thought of as just a function which is guaranteed to return a future which can, in turn be awaited.