Assert that a method was called in a Python unit test

by

I use Mock (which is now unittest.mock on py3.3+) for this:

from mock import patch
from PyQt4 import Qt


@patch.object(Qt.QMessageBox, 'aboutQt')
def testShowAboutQt(self, mock):
    self.win.actionAboutQt.trigger()
    self.assertTrue(mock.called)

For your case, it could look like this:

import mock
from mock import patch


def testClearWasCalled(self):
   aw = aps.Request("nv1")
   with patch.object(aw, 'Clear') as mock:
       aw2 = aps.Request("nv2", aw)

   mock.assert_called_with(42) # or mock.assert_called_once_with(42)

Mock supports quite a few useful features, including ways to patch an object or module, as well as checking that the right thing was called, etc etc.

Caveat emptor! (Buyer beware!)

If you mistype assert_called_with (to assert_called_once or assert_called_wiht) your test may still run, as Mock will think this is a mocked function and happily go along, unless you use autospec=true. For more info read assert_called_once: Threat or Menace.

More Related Contents:

  1. How do you generate dynamic (parameterized) unit tests in Python?
  2. How do you test that a Python function throws an exception?
  3. How to properly assert that an exception gets raised in pytest?
  4. How do I run all Python unit tests in a directory?
  5. How unique is Python’s id()?
  6. How do you write tests for the argparse portion of a python module?
  7. Python unittest.TestCase execution order
  8. Where do the Python unit tests go? [closed]
  9. TransactionManagementError “You can’t execute queries until the end of the ‘atomic’ block” while using signals, but only during Unit Testing
  10. How to test a function with input call?
  11. Monkey patching a class in another module in Python
  12. Unittest setUp/tearDown for several tests
  13. Python Mocking a function from an imported module
  14. How to supply stdin, files and environment variable inputs to Python unit tests?
  15. Python mock multiple return values
  16. How can one mock/stub python module like urllib
  17. Mocking a class: Mock() or patch()?
  18. Python: Write unittest for console print
  19. Mock attributes in Python mock?
  20. Writing unit tests in Python: How do I start? [closed]
  21. Access self from decorator
  22. Python library ‘unittest’: Generate multiple tests programmatically [duplicate]
  23. How can I mock user input (from “input” in 3.x, or “raw_input” in 2.x) for a unit test?
  24. Python unittest – opposite of assertRaises?
  25. How to run django unit-tests on production database?
  26. how to share a variable across modules for all tests in py.test
  27. Is it possible for a unit test to assert that a method calls sys.exit()?
  28. How do I run unittest on a Tkinter app?
  29. ValueError: no such test method in : runTest
  30. Django: How to create a model dynamically just for testing

Leave a Comment