Address review issues

This commit is contained in:
Peter Eckersley
2016-06-25 12:28:47 -07:00
parent 8a28cb7352
commit 23f0ccbc8e
2 changed files with 18 additions and 5 deletions
+2
View File
@@ -113,6 +113,7 @@ class NcursesDisplayTest(unittest.TestCase):
@mock.patch("certbot.display.util."
"dialog.Dialog.checklist")
def test_checklist(self, mock_checklist):
mock_checklist.return_value = (mock.MagicMock(), mock.MagicMock())
self.displayer.checklist("message", TAGS)
choices = [
@@ -126,6 +127,7 @@ class NcursesDisplayTest(unittest.TestCase):
@mock.patch("certbot.display.util.dialog.Dialog.dselect")
def test_directory_select(self, mock_dselect):
mock_dselect.return_value = (mock.MagicMock(), mock.MagicMock())
self.displayer.directory_select("message")
self.assertEqual(mock_dselect.call_count, 1)