A Python context manager for doing multiple mock.patches

multiple_mocks.py

@contextlib.contextmanager
def multiple_mocks(mock_specs):
    """
    `mock_specs` is a dict of mock target name => mock patch kwargs

    Example usage:

        with multiple_mocks(
                {'os.path.exists': {'side_effect': mock_path_exists},
                 'subprocess.Popen': {},
                 'os.symlink': {}}):
            do_stuff()
    """
    _mocks = []

    for target, mock_patch_kwargs in mock_specs.items():
        _mock = mock.patch(target, **mock_patch_kwargs)
        _mock.start()
        _mocks.append(_mock)

    yield

    for _mock in _mocks:
        _mock.stop()

We cannot use with multiple_mocks(...) as mocks with that, but nice idea.