starred/certbot-certbot
1
0
Fork 0
mirror of https://github.com/certbot/certbot.git synced 2026-08-01 19:02:52 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix up display/ops.py coverage

Browse Source
This commit is contained in:
Brad Warren
2016-05-25 22:12:06 -07:00
parent 653935b937
commit 8d802ef6fd
1 changed files with 13 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
certbot/tests/display/ops_test.py
+13 -11
View File
@@ -39,6 +39,7 @@ class GetEmailTest(unittest.TestCase):
def test_cancel_none(self): def test_cancel_none(self):
self.input.return_value = (display_util.CANCEL, "foo@bar.baz") self.input.return_value = (display_util.CANCEL, "foo@bar.baz")
self.assertRaises(errors.Error, self._call) self.assertRaises(errors.Error, self._call)
self.assertRaises(errors.Error, self._call, optional=False)
def test_ok_safe(self): def test_ok_safe(self):
self.input.return_value = (display_util.OK, "foo@bar.baz") self.input.return_value = (display_util.OK, "foo@bar.baz")
@@ -52,23 +53,24 @@ class GetEmailTest(unittest.TestCase):
mock_safe_email.side_effect = [False, True] mock_safe_email.side_effect = [False, True]
self.assertTrue(self._call() is "foo@bar.baz") self.assertTrue(self._call() is "foo@bar.baz")
def test_more_and_invalid_flags(self): def test_invalid_flag(self):
optional_txt = "--register-unsafely-without-email"
invalid_txt = "There seem to be problems" invalid_txt = "There seem to be problems"
base_txt = "Enter email"
self.input.return_value = (display_util.OK, "foo@bar.baz") self.input.return_value = (display_util.OK, "foo@bar.baz")
with mock.patch("certbot.display.ops.le_util.safe_email") as mock_safe_email: with mock.patch("certbot.display.ops.le_util.safe_email") as mock_safe_email:
mock_safe_email.return_value = True mock_safe_email.return_value = True
self._call() self._call()
msg = self.input.call_args[0][0] self.assertTrue(invalid_txt not in self.input.call_args[0][0])
self.assertTrue(optional_txt not in msg)
self.assertTrue(invalid_txt not in msg)
self.assertTrue(base_txt in msg)
self._call(invalid=True) self._call(invalid=True)
msg = self.input.call_args[0][0] self.assertTrue(invalid_txt in self.input.call_args[0][0])
self.assertTrue(optional_txt in msg)
self.assertTrue(invalid_txt in msg) def test_optional_flag(self):
self.assertTrue(base_txt in msg) self.input.return_value = (display_util.OK, "foo@bar.baz")
with mock.patch("certbot.display.ops.le_util.safe_email") as mock_safe_email:
mock_safe_email.side_effect = [False, True]
self._call(optional=False)
for call in self.input.call_args_list:
self.assertTrue(
"--register-unsafely-without-email" not in call[0][0])
class ChooseAccountTest(unittest.TestCase): class ChooseAccountTest(unittest.TestCase):