python mock – patching a method without obstructing implementation

by

Similar solution with yours, but using wraps:

def test_something(self):
    spud = Potato()
    with patch.object(Potato, 'foo', wraps=spud.foo) as mock:
        forty_two = spud.foo(n=40)
        mock.assert_called_once_with(n=40)
    self.assertEqual(forty_two, 42)

According to the documentation:

wraps: Item for the mock object to wrap. If wraps is not None then
calling the Mock will pass the call through to the wrapped object
(returning the real result). Attribute access on the mock will return
a Mock object that wraps the corresponding attribute of the wrapped
object (so attempting to access an attribute that doesn’t exist will
raise an AttributeError).

class Potato(object):

    def spam(self, n):
        return self.foo(n=n)

    def foo(self, n):
        return self.bar(n)

    def bar(self, n):
        return n + 2


class PotatoTest(TestCase):

    def test_something(self):
        spud = Potato()
        with patch.object(Potato, 'foo', wraps=spud.foo) as mock:
            forty_two = spud.spam(n=40)
            mock.assert_called_once_with(n=40)
        self.assertEqual(forty_two, 42)

More Related Contents:

  1. How can I mock requests and the response?
  2. Trying to mock datetime.date.today(), but not working
  3. python mocking raw input in unittests
  4. How to mock an import
  5. How do I mock an open used in a with statement (using the Mock framework in Python)?
  6. Can I patch a Python decorator before it wraps a function?
  7. Python: Mocking a context manager
  8. Why python mock patch doesn’t work?
  9. Python mock multiple return values
  10. Python Mock object with method called multiple times
  11. How can one mock/stub python module like urllib
  12. Mocking a class: Mock() or patch()?
  13. Mock attributes in Python mock?
  14. How can I mock user input (from “input” in 3.x, or “raw_input” in 2.x) for a unit test?
  15. How to monkeypatch python’s datetime.datetime.now with py.test?
  16. Mocking boto3 S3 client method Python
  17. What is your favorite Python mocking library? [closed]
  18. Mock Python’s built in print function
  19. How to test or mock “if __name__ == ‘__main__'” contents
  20. Using mock patch to mock an instance method
  21. Mocking Functions Using Python Mock
  22. Python mock Patch os.environ and return value
  23. Customizing unittest.mock.mock_open for iteration
  24. Unit testing a python app that uses the requests library
  25. python: mock a module
  26. Django: How to create a model dynamically just for testing
  27. Can I “fake” a package (or at least a module) in python for testing purposes?
  28. How to mock a readonly property with mock?
  29. How to convert a string to integer entered in input functon? [duplicate]
  30. How can I rotate xticklabels in matplotlib so that the spacing between each xticklabel is equal?

Leave a Comment