Added methods spy and stub to mocker fixture

Author: fogo

pytest_mock.py

@@ -31,6 +31,31 @@ def stopall(self):
             p.stop()
         self._patches[:] = []
 
+    def spy(self, obj, method_name):
+        """
+        Creates a spy of method. It will run method normally, but it is now possible to use `mock`
+        call features with it, like call count.
+
+        :param object obj:
+            An object.
+
+        :param unicode method_name:
+            A method in object.
+
+        :return: mock.MagicMock
+            Spy object.
+        """
+        return self.patch.object(obj, method_name, side_effect=getattr(obj, method_name))
+
+    def stub(self):
+        """
+        Creates a stub method. It accepts any arguments. Ideal to register to callbacks in tests.
+
+        :return: mock.MagicMock
+            Stub object.
+        """
+        return mock_module.MagicMock(spec=lambda *args, **kwargs: None)
+
     class _Patcher(object):
         """
         Object to provide the same interface as mock.patch, mock.patch.object,

test_pytest_mock.py

@@ -136,3 +136,26 @@ def test_mocker_has_mock_class_as_attribute_for_instantiation():
 
     mocker = MockFixture()
     assert isinstance(mocker.Mock(), mock_module.Mock)
+
+
+def test_mocker_spy(mocker):
+    class Foo(object):
+
+        def bar(self, arg):
+            return arg * 2
+
+    foo = Foo()
+    spy = mocker.spy(foo, 'Bar')
+
+    assert foo.Bar(arg=10) == 20
+    spy.assert_called_once_with(arg=10)
+
+
+def test_mocker_stub(mocker):
+    def Foo(on_something):
+        on_something('foo', 'bar')
+
+    stub = mocker.stub()
+
+    Foo(stub)
+    stub.assert_called_once_with('foo', 'bar')