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

Test cases for get_email

Browse Source
This commit is contained in:
Peter Eckersley
2015-11-16 16:57:48 -08:00
parent 371e57fc51
commit 2d07c017b2
1 changed files with 26 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
letsencrypt/tests/display/ops_test.py
+26 -6
View File
@@ -168,9 +168,9 @@ class GetEmailTest(unittest.TestCase):
zope.component.provideUtility(mock_display, interfaces.IDisplay) zope.component.provideUtility(mock_display, interfaces.IDisplay)
@classmethod @classmethod
def _call(cls): def _call(cls, **kwargs):
from letsencrypt.display.ops import get_email from letsencrypt.display.ops import get_email
return get_email() return get_email(**kwargs)
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")
@@ -178,18 +178,38 @@ class GetEmailTest(unittest.TestCase):
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")
with mock.patch("letsencrypt.display.ops.le_util" with mock.patch("letsencrypt.display.ops.le_util.safe_email") as mock_safe_email:
".safe_email") as mock_safe_email:
mock_safe_email.return_value = True mock_safe_email.return_value = True
self.assertTrue(self._call() is "foo@bar.baz") self.assertTrue(self._call() is "foo@bar.baz")
def test_ok_not_safe(self): def test_ok_not_safe(self):
self.input.return_value = (display_util.OK, "foo@bar.baz") self.input.return_value = (display_util.OK, "foo@bar.baz")
with mock.patch("letsencrypt.display.ops.le_util" with mock.patch("letsencrypt.display.ops.le_util.safe_email") as mock_safe_email:
".safe_email") as mock_safe_email:
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):
more_txt = "--register-unsafely-without-email"
invalid_txt = "There seem to be problems"
base_txt = "Enter email"
self.input.return_value = (display_util.OK, "foo@bar.baz")
with mock.patch("letsencrypt.display.ops.le_util.safe_email") as mock_safe_email:
mock_safe_email.return_value = True
self._call()
msg = self.input.call_args[0][0]
self.assertTrue(more_txt not in msg)
self.assertTrue(invalid_txt not in msg)
self.assertTrue(base_txt in msg)
self._call(more=True)
msg = self.input.call_args[0][0]
self.assertTrue(more_txt in msg)
self.assertTrue(invalid_txt not in msg)
self._call(more=True, invalid=True)
msg = self.input.call_args[0][0]
self.assertTrue(more_txt in msg)
self.assertTrue(invalid_txt in msg)
self.assertTrue(base_txt in msg)
class ChooseAccountTest(unittest.TestCase): class ChooseAccountTest(unittest.TestCase):
"""Tests for letsencrypt.display.ops.choose_account.""" """Tests for letsencrypt.display.ops.choose_account."""