Skip to content
a version of pythons unittest.Mock with async methods
Branch: master
Clone or download
Latest commit 608be0c Sep 21, 2016
Type Name Latest commit message Commit time
Failed to load latest commit information.
aiomock Add __aenter__ and __aexit__ magics Sep 21, 2016
.gitignore Initial commit Sep 9, 2016
MANIFEST fix readme Sep 21, 2016


a version of pythons unittest.Mock with async methods

This mock works exactly like unitest.mock.Mock, except that it also includes the following:

  • mock.async_return_value - an awaitable version of return_value
  • mock.async_side_effect - an awaitable version of side_effect


pip install aiomock


imagine you have the following code

queue = asyncio.Queue()

async def wait_for_it():
    item = await queue.get()
    # do something with item
    return item

in order to test this, you cant use unittest.Mock.return_value because it is not awaitable. This means you have to write a function for every test

async def test_wait_for_it(monkeypatch):
    mock = Mock()
    async def my_get():
        return 1
    mock.get = my_get

    monkeypatch.setattr(asyncio.Queue, mock)

Its annoying that Mock has two excellant features, side_effect, and return, but you cant use them because we are async. This fixes that


from aiomock import AIOMock

async def test_wait_for_it(monkeypatch):
    mock = AIOMock()
    mock.get.async_return_value = 1

    monkeypatch.setattr(asyncio.Queue, mock)

or for side_effect, use

mock.func_name.async_side_effect = [1, 2]

You can even use it with lambdas, and it will convert it to an awaitable

mock.func_name.async_side_effect = lambda x: x+4
You can’t perform that action at this time.