Tornado Asynchronous Mock

async_mock_example.py

from mock import CallableMixin, NonCallableMock

class AsyncCallableMixin(CallableMixin):
    """Change the __call__ method such that it does not return but call the
    `callback` kwarg with the return value.
    """

    def __call__(_mock_self, *args, **kwargs):
        cb = kwargs.get('callback', None)

        _mock_self._mock_check_sig(*args, **kwargs)

        if cb:
            del kwargs['callback']
            result = _mock_self._mock_call(*args, **kwargs)
            cb(result)
        else:
            return _mock_self._mock_call(*args, **kwargs)


class AsyncMock(AsyncCallableMixin, NonCallableMock):
    pass