Try side_effect def my_side_effect(*args, **kwargs): if args[0] == 42: return “Called with 42” elif args[0] == 43: return “Called with 43” elif kwargs[‘foo’] == 7: return “Foo is seven” mockobj.mockmethod.side_effect = my_side_effect
You can assign an iterable to side_effect, and the mock will return the next value in the sequence each time it is called: >>> from unittest.mock import Mock >>> m = Mock() >>> m.side_effect = [‘foo’, ‘bar’, ‘baz’] >>> m() ‘foo’ >>> m() ‘bar’ >>> m() ‘baz’ Quoting the Mock() documentation: If side_effect is an … Read more
The solution was actually quite simple: I just needed to convert __call__ method of mock into coroutine: class AsyncMock(MagicMock): async def __call__(self, *args, **kwargs): return super(AsyncMock, self).__call__(*args, **kwargs) This works perfectly, when mock is called, code receives native coroutine Example usage: @mock.patch(‘my.path.asyncio.sleep’, new_callable=AsyncMock) def test_stuff(sleep): # code
When you are using the patch decorator from the unittest.mock package you are patching it in the namespace that is under test (in this case app.mocking.get_user_name), not the namespace the function is imported from (in this case app.my_module.get_user_name). To do what you describe with @patch try something like the below: from mock import patch from … Read more