Implement Brad's more systematic solution for this

This commit is contained in:
Peter Eckersley
2016-06-22 15:50:21 -07:00
parent e231dae2c1
commit 8a28cb7352
2 changed files with 25 additions and 9 deletions
+1
View File
@@ -96,6 +96,7 @@ class NcursesDisplayTest(unittest.TestCase):
@mock.patch("certbot.display.util."
"dialog.Dialog.inputbox")
def test_input(self, mock_input):
mock_input.return_value = (mock.MagicMock(), mock.MagicMock())
self.displayer.input("message")
self.assertEqual(mock_input.call_count, 1)