Mocking Functions Using Python Mock

by

The general case would be to use patch from mock. Consider the following:

utils.py

def get_content():
    return 'stuff'

mymodule.py

from util import get_content


class MyClass(object):

    def func(self):
        return get_content()

test.py

import unittest

from mock import patch

from mymodule import MyClass

class Test(unittest.TestCase):

    @patch('mymodule.get_content')
    def test_func(self, get_content_mock):
        get_content_mock.return_value="mocked stuff"

        my_class = MyClass()
        self.assertEqual(my_class.func(), 'mocked stuff')
        self.assertEqual(get_content_mock.call_count, 1)
        get_content_mock.assert_called_once()

Note how get_content is mocked, it is not util.get_content, rather mymodule.get_content since we are using it in mymodule.

Above has been tested with mock v2.0.0, nosetests v1.3.7 and python v2.7.9.

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. Mocking a class: Mock() or patch()?
  8. Mock attributes in Python mock?
  9. How can I mock user input (from “input” in 3.x, or “raw_input” in 2.x) for a unit test?
  10. Mock Python’s built in print function
  11. How to test or mock “if __name__ == ‘__main__'” contents
  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 generate dynamic (parameterized) unit tests in Python?
  19. Trying to mock datetime.date.today(), but not working
  20. How to properly assert that an exception gets raised in pytest?
  21. How to mock an import
  22. How do I run all Python unit tests in a directory?
  23. How do you write tests for the argparse portion of a python module?
  24. Python unittest.TestCase execution order
  25. Where do the Python unit tests go? [closed]
  26. How to test a function with input call?
  27. Assert that a method was called in a Python unit test
  28. Unittest tests order
  29. python mock – patching a method without obstructing implementation
  30. ValueError: no such test method in : runTest

Leave a Comment