From b75235b3dc80bd79a06a9800d956403a4517e823 Mon Sep 17 00:00:00 2001 From: Peter Eckersley Date: Thu, 21 Jan 2016 15:27:23 -0800 Subject: [PATCH] Close test coverage gaps (And fix a bug in one test...) --- letsencrypt/tests/display/util_test.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/letsencrypt/tests/display/util_test.py b/letsencrypt/tests/display/util_test.py index 8759a7faa..a16eb544e 100644 --- a/letsencrypt/tests/display/util_test.py +++ b/letsencrypt/tests/display/util_test.py @@ -291,6 +291,12 @@ class NoninteractiveDisplayTest(unittest.TestCase): self.mock_stdout = mock.MagicMock() self.displayer = display_util.NoninteractiveDisplay(self.mock_stdout) + def test_notification_no_pause(self): + self.displayer.notification("message", 10) + string = self.mock_stdout.write.call_args[0][0] + + self.assertTrue("message" in string) + def test_input(self): d = "an incomputable value" ret = self.displayer.input("message", default=d) @@ -310,7 +316,7 @@ class NoninteractiveDisplayTest(unittest.TestCase): def test_checklist(self): d = [1, 3] - ret = self.displayer.menu("message", TAGS, default=d) + ret = self.displayer.checklist("message", TAGS, default=d) self.assertEqual(ret, (display_util.OK, d)) self.assertRaises(errors.MissingCommandlineFlag, self.displayer.checklist, "message", TAGS)