mock plugin for pytest

pytest_mock.py

import pytest
import mock as mock_


class Mock(object):
    def __init__(self):
        self._patchers = []

    def _stop(self):
        for patcher in self._patchers:
            patcher.stop()

    def _patch(self, method, *args, **kwds):
        if method:
            patch_fn = getattr(mock_.patch, method)
        else:
            patch_fn = mock_.patch

        patcher = patch_fn(*args, **kwds)
        self._patchers.append(patcher)

        return patcher.start()

    def patch(self, *args, **kwds):
        return self._patch(None, *args, **kwds)

    def patch_object(self, *args, **kwds):
        return self._patch('object', *args, **kwds)

    def patch_dict(self, *args, **kwds):
        return self._patch('dict', *args, **kwds)

    def patch_multiple(self, *args, **kwds):
        return self._patch('multiple', *args, **kwds)


@pytest.fixture
def mock(request):
    mock = Mock()
    request.addfinalizer(mock._stop)
    return mock