Mocking a class: Mock() or patch()?

by

mock.patch is a very very different critter than mock.Mock. patch replaces the class with a mock object and lets you work with the mock instance. Take a look at this snippet:

>>> class MyClass(object):
...   def __init__(self):
...     print 'Created MyClass@{0}'.format(id(self))
... 
>>> def create_instance():
...   return MyClass()
... 
>>> x = create_instance()
Created MyClass@4299548304
>>> 
>>> @mock.patch('__main__.MyClass')
... def create_instance2(MyClass):
...   MyClass.return_value="foo"
...   return create_instance()
... 
>>> i = create_instance2()
>>> i
'foo'
>>> def create_instance():
...   print MyClass
...   return MyClass()
...
>>> create_instance2()
<mock.Mock object at 0x100505d90>
'foo'
>>> create_instance()
<class '__main__.MyClass'>
Created MyClass@4300234128
<__main__.MyClass object at 0x100505d90>

patch replaces MyClass in a way that allows you to control the usage of the class in functions that you call. Once you patch a class, references to the class are completely replaced by the mock instance.

mock.patch is usually used when you are testing something that creates a new instance of a class inside of the test. mock.Mock instances are clearer and are preferred. If your self.sut.something method created an instance of MyClass instead of receiving an instance as a parameter, then mock.patch would be appropriate here.

More Related Contents:

  1. python mocking raw input in unittests
  2. Can I patch a Python decorator before it wraps a function?
  3. Why python mock patch doesn’t work?
  4. Python mock multiple return values
  5. Python Mock object with method called multiple times
  6. How can one mock/stub python module like urllib
  7. Mock attributes in Python mock?
  8. How can I mock user input (from “input” in 3.x, or “raw_input” in 2.x) for a unit test?
  9. Mock Python’s built in print function
  10. How to test or mock “if __name__ == ‘__main__'” contents
  11. Mocking Functions Using Python Mock
  12. Python mock Patch os.environ and return value
  13. Customizing unittest.mock.mock_open for iteration
  14. Unit testing a python app that uses the requests library
  15. Django: How to create a model dynamically just for testing
  16. Can I “fake” a package (or at least a module) in python for testing purposes?
  17. How to mock a readonly property with mock?
  18. How do you test that a Python function throws an exception?
  19. How unique is Python’s id()?
  20. Unittest setUp/tearDown for several tests
  21. Python Mocking a function from an imported module
  22. How to supply stdin, files and environment variable inputs to Python unit tests?
  23. Writing unit tests in Python: How do I start? [closed]
  24. Python library ‘unittest’: Generate multiple tests programmatically [duplicate]
  25. how to test if a variable is pd.NaT?
  26. How to monkeypatch python’s datetime.datetime.now with py.test?
  27. How to run django unit-tests on production database?
  28. how to share a variable across modules for all tests in py.test
  29. python: mock a module
  30. How can I specify a database for Django Tests to use instead of having it build it every time?

Leave a Comment