diff --git a/console_conf/controllers/tests/test_chooser.py b/console_conf/controllers/tests/test_chooser.py index 1349b483..6c5a6d82 100644 --- a/console_conf/controllers/tests/test_chooser.py +++ b/console_conf/controllers/tests/test_chooser.py @@ -148,9 +148,10 @@ class TestChooserController(unittest.TestCase): # go back now c.back() c.ui.set_body.assert_called_with("current") + c.ui.set_body.reset_mock() # nothing c.back() - c.ui.set_body.not_called() + c.ui.set_body.assert_not_called() @mock.patch( "console_conf.controllers.chooser.ChooserCurrentSystemView", @@ -169,7 +170,7 @@ class TestChooserController(unittest.TestCase): self.assertEqual(v, "current") # going back does nothing c.back() - c.ui.set_body.not_called() + c.ui.set_body.assert_not_called() @mock.patch( "console_conf.controllers.chooser.ChooserCurrentSystemView",