Python mock multiple return values

by

You can assign an iterable to side_effect, and the mock will return the next value in the sequence each time it is called:

>>> from unittest.mock import Mock
>>> m = Mock()
>>> m.side_effect = ['foo', 'bar', 'baz']
>>> m()
'foo'
>>> m()
'bar'
>>> m()
'baz'

Quoting the Mock() documentation:

If side_effect is an iterable then each call to the mock will return the next value from the iterable.

More Related Contents:

  1. Python Mock object with method called multiple times
  2. Mock attributes in Python mock?
  3. Customizing unittest.mock.mock_open for iteration
  4. python mocking raw input in unittests
  5. Can I patch a Python decorator before it wraps a function?
  6. Python Mocking a function from an imported module
  7. Why python mock patch doesn’t work?
  8. How can one mock/stub python module like urllib
  9. Mocking a class: Mock() or patch()?
  10. How can I mock user input (from “input” in 3.x, or “raw_input” in 2.x) for a unit test?
  11. Mock Python’s built in print function
  12. How to test or mock “if __name__ == ‘__main__'” contents
  13. Mocking Functions Using Python Mock
  14. Python mock Patch os.environ and return value
  15. Unit testing a python app that uses the requests library
  16. Django: How to create a model dynamically just for testing
  17. Can I “fake” a package (or at least a module) in python for testing purposes?
  18. How to mock a readonly property with mock?
  19. PATH issue with pytest ‘ImportError: No module named YadaYadaYada’
  20. How do I mock an open used in a with statement (using the Mock framework in Python)?
  21. Getting Python’s unittest results in a tearDown() method
  22. How to test Python 3.4 asyncio code?
  23. Continuing in Python’s unittest when an assertion fails
  24. Mocking boto3 S3 client method Python
  25. What is your favorite Python mocking library? [closed]
  26. Can Python’s unittest test in parallel, like nose can?
  27. Using mock patch to mock an instance method
  28. What is the difference between setUp() and setUpClass() in Python unittest?
  29. Is it possible for a unit test to assert that a method calls sys.exit()?
  30. How do I run unittest on a Tkinter app?

Leave a Comment