mirror of
https://github.com/certbot/certbot.git
synced 2026-03-23 10:53:11 -04:00
94 lines
2.9 KiB
Python
94 lines
2.9 KiB
Python
"""Tests for letsencrypt.reporter."""
|
|
import sys
|
|
import unittest
|
|
|
|
import six
|
|
|
|
|
|
class ReporterTest(unittest.TestCase):
|
|
"""Tests for letsencrypt.reporter.Reporter."""
|
|
|
|
def setUp(self):
|
|
from letsencrypt import reporter
|
|
self.reporter = reporter.Reporter()
|
|
|
|
self.old_stdout = sys.stdout
|
|
sys.stdout = six.StringIO()
|
|
|
|
def tearDown(self):
|
|
sys.stdout = self.old_stdout
|
|
|
|
def test_multiline_message(self):
|
|
self.reporter.add_message("Line 1\nLine 2", self.reporter.LOW_PRIORITY)
|
|
self.reporter.atexit_print_messages()
|
|
output = sys.stdout.getvalue()
|
|
self.assertTrue("Line 1\n" in output)
|
|
self.assertTrue("Line 2" in output)
|
|
|
|
def test_tty_print_empty(self):
|
|
sys.stdout.isatty = lambda: True
|
|
self.test_no_tty_print_empty()
|
|
|
|
def test_no_tty_print_empty(self):
|
|
self.reporter.print_messages()
|
|
self.assertEqual(sys.stdout.getvalue(), "")
|
|
try:
|
|
raise ValueError
|
|
except ValueError:
|
|
self.reporter.print_messages()
|
|
self.assertEqual(sys.stdout.getvalue(), "")
|
|
|
|
def test_atexit_print_messages(self):
|
|
self._add_messages()
|
|
self.reporter.atexit_print_messages()
|
|
output = sys.stdout.getvalue()
|
|
self.assertTrue("IMPORTANT NOTES:" in output)
|
|
self.assertTrue("High" in output)
|
|
self.assertTrue("Med" in output)
|
|
self.assertTrue("Low" in output)
|
|
|
|
def test_tty_successful_exit(self):
|
|
sys.stdout.isatty = lambda: True
|
|
self._successful_exit_common()
|
|
|
|
def test_no_tty_successful_exit(self):
|
|
self._successful_exit_common()
|
|
|
|
def test_tty_unsuccessful_exit(self):
|
|
sys.stdout.isatty = lambda: True
|
|
self._unsuccessful_exit_common()
|
|
|
|
def test_no_tty_unsuccessful_exit(self):
|
|
self._unsuccessful_exit_common()
|
|
|
|
def _successful_exit_common(self):
|
|
self._add_messages()
|
|
self.reporter.print_messages()
|
|
output = sys.stdout.getvalue()
|
|
self.assertTrue("IMPORTANT NOTES:" in output)
|
|
self.assertTrue("High" in output)
|
|
self.assertTrue("Med" in output)
|
|
self.assertTrue("Low" in output)
|
|
|
|
def _unsuccessful_exit_common(self):
|
|
self._add_messages()
|
|
try:
|
|
raise ValueError
|
|
except ValueError:
|
|
self.reporter.print_messages()
|
|
output = sys.stdout.getvalue()
|
|
self.assertTrue("IMPORTANT NOTES:" in output)
|
|
self.assertTrue("High" in output)
|
|
self.assertTrue("Med" not in output)
|
|
self.assertTrue("Low" not in output)
|
|
|
|
def _add_messages(self):
|
|
self.reporter.add_message("High", self.reporter.HIGH_PRIORITY)
|
|
self.reporter.add_message(
|
|
"Med", self.reporter.MEDIUM_PRIORITY, on_crash=False)
|
|
self.reporter.add_message(
|
|
"Low", self.reporter.LOW_PRIORITY, on_crash=False)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main() # pragma: no cover
|