Mock patch multiple targets in Python

mock_patch_multiple_targets.py

@contextlib.contextmanager
def multiple_targets(mock_patches):
    """
    `mock_patches` is a list (or iterable) of mock.patch objects

    Example usage:

        with mock.patch.multiple_targets([
                mock.patch('os.path.exists', side_effect=mock_path_exists),
                mock.patch('subprocess.Popen'),
                mock.patch('os.mkdir'),
                mock.patch('os.symlink')]):
            do_stuff()

    Or if you need to do stuff with the mock objects:

        with mock.patch.multiple_targets([
                mock.patch('os.path.exists', side_effect=mock_path_exists),
                mock.patch('subprocess.Popen'),
                mock.patch('os.mkdir'),
                mock.patch('os.symlink')]) as (mpe, msp, mom, mos):
    """

    mocks = []

    for mock_patch in mock_patches:
        _mock = mock_patch.start()
        mocks.append(_mock)

    yield mocks

    for mock_patch in mock_patches:
        mock_patch.stop()


mock.patch.multiple_targets = multiple_targets

Thanks!

Thank you.

thanks!