Mock Python’s built in print function

by

I know that there is already an accepted answer but there is simpler solution for that problem – mocking the print in python 2.x. Answer is in the mock library tutorial: http://www.voidspace.org.uk/python/mock/patch.html and it is:

>>> from StringIO import StringIO
>>> def foo():
...     print 'Something'
...
>>> @patch('sys.stdout', new_callable=StringIO)
... def test(mock_stdout):
...     foo()
...     assert mock_stdout.getvalue() == 'Something\n'
...
>>> test()

Of course you can use also following assertion:

self.assertEqual("Something\n", mock_stdout.getvalue())

I have checked this solution in my unittests and it is working as expected. Hope this helps somebody. Cheers!

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. 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 does Python 2 compare string and int? Why do lists compare as greater than numbers, and tuples greater than lists?
  19. Alternative to dict comprehension prior to Python 2.7
  20. Unittest setUp/tearDown for several tests
  21. Strange behaviour with floats and string conversion
  22. How do you use subprocess.check_output() in Python?
  23. Python: Write unittest for console print
  24. Read Unicode characters from command-line arguments in Python 2.x on Windows
  25. Access self from decorator
  26. Run interactive Bash with popen and a dedicated TTY Python
  27. How to run django unit-tests on production database?
  28. python: mock a module
  29. How can I specify a database for Django Tests to use instead of having it build it every time?
  30. What arguments does Python sort() function have?

Leave a Comment