Created
September 9, 2015 00:16
-
-
Save Sorseg/14acb68f11e09ac4cb84 to your computer and use it in GitHub Desktop.
Mocking await calls in python 3.5 is a problem
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
async def some_coroutine(): | |
await another_coroutine() | |
async def another_coroutine(): | |
pass |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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