mirror of
https://github.com/certbot/certbot.git
synced 2026-06-09 00:32:12 -04:00
525c427c60
* Cleanup some useless type ignore directives * Cleanup one more type ignore directive Co-authored-by: Adrien Ferrand <aferrand@ecomundo.eu>
89 lines
2.7 KiB
Python
89 lines
2.7 KiB
Python
"""Tests for certbot._internal.reporter."""
|
|
import io
|
|
import sys
|
|
import unittest
|
|
|
|
|
|
try:
|
|
import mock
|
|
except ImportError: # pragma: no cover
|
|
from unittest import mock
|
|
|
|
|
|
class ReporterTest(unittest.TestCase):
|
|
"""Tests for certbot._internal.reporter.Reporter."""
|
|
def setUp(self):
|
|
from certbot._internal import reporter
|
|
self.reporter = reporter.Reporter(mock.MagicMock(quiet=False))
|
|
|
|
self.old_stdout = sys.stdout
|
|
sys.stdout = io.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.print_messages()
|
|
output = sys.stdout.getvalue()
|
|
self.assertIn("Line 1\n", output)
|
|
self.assertIn("Line 2", 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_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.assertIn("IMPORTANT NOTES:", output)
|
|
self.assertIn("High", output)
|
|
self.assertIn("Med", output)
|
|
self.assertIn("Low", output)
|
|
|
|
def _unsuccessful_exit_common(self):
|
|
self._add_messages()
|
|
try:
|
|
raise ValueError
|
|
except ValueError:
|
|
self.reporter.print_messages()
|
|
output = sys.stdout.getvalue()
|
|
self.assertIn("IMPORTANT NOTES:", output)
|
|
self.assertIn("High", output)
|
|
self.assertNotIn("Med", output)
|
|
self.assertNotIn("Low", 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
|