Skip to content

Instantly share code, notes, and snippets.

@Sorseg
Created September 9, 2015 00:16
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 1 You must be signed in to fork a gist
  • Save Sorseg/14acb68f11e09ac4cb84 to your computer and use it in GitHub Desktop.
Save Sorseg/14acb68f11e09ac4cb84 to your computer and use it in GitHub Desktop.
Mocking await calls in python 3.5 is a problem
async def some_coroutine():
await another_coroutine()
async def another_coroutine():
pass
from unittest import TestCase, mock, main
import asyncio
from problem import some_coroutine
class Test(TestCase):
def setUp(self):
self.loop = asyncio.new_event_loop()
asyncio.set_event_loop(self.loop)
@mock.patch('problem.another_coroutine')
def test_some_coroutine(self, mock_another_coroutine):
async def run_test():
await some_coroutine()
mock_another_coroutine.assert_called_with()
self.loop.run_until_complete(run_test())
if __name__ == '__main__':
main()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment