Mock calls to asyncio.create_subprocess_shell

AsyncioCreateSubprocessShellCommunicateMock.py

# Used to mock calls to asyncio.create_subprocess_shell
# Patches the `communicate` object with a future as specified during construction
import asynctest
import mock


class AsyncioCreateSubprocessShellCommunicateMock:
    def __init__(self, future_result):
        process_mock = mock.Mock()
        future = asynctest.asyncio.Future()
        future.set_result(future_result)
        attrs = {"communicate.return_value": future}
        process_mock.configure_mock(**attrs)
        self.mock = process_mock