1from test.support import temp_dir
2from test.support.script_helper import assert_python_failure
3import unittest
4import sys
5import cgitb
6
7class TestCgitb(unittest.TestCase):
8
9    def test_fonts(self):
10        text = "Hello Robbie!"
11        self.assertEqual(cgitb.small(text), "<small>{}</small>".format(text))
12        self.assertEqual(cgitb.strong(text), "<strong>{}</strong>".format(text))
13        self.assertEqual(cgitb.grey(text),
14                         '<font color="#909090">{}</font>'.format(text))
15
16    def test_blanks(self):
17        self.assertEqual(cgitb.small(""), "")
18        self.assertEqual(cgitb.strong(""), "")
19        self.assertEqual(cgitb.grey(""), "")
20
21    def test_html(self):
22        try:
23            raise ValueError("Hello World")
24        except ValueError as err:
25            # If the html was templated we could do a bit more here.
26            # At least check that we get details on what we just raised.
27            html = cgitb.html(sys.exc_info())
28            self.assertIn("ValueError", html)
29            self.assertIn(str(err), html)
30
31    def test_text(self):
32        try:
33            raise ValueError("Hello World")
34        except ValueError:
35            text = cgitb.text(sys.exc_info())
36            self.assertIn("ValueError", text)
37            self.assertIn("Hello World", text)
38
39    def test_syshook_no_logdir_default_format(self):
40        with temp_dir() as tracedir:
41            rc, out, err = assert_python_failure(
42                  '-c',
43                  ('import cgitb; cgitb.enable(logdir=%s); '
44                   'raise ValueError("Hello World")') % repr(tracedir),
45                  PYTHONIOENCODING='utf-8')
46        out = out.decode()
47        self.assertIn("ValueError", out)
48        self.assertIn("Hello World", out)
49        self.assertIn("<strong>&lt;module&gt;</strong>", out)
50        # By default we emit HTML markup.
51        self.assertIn('<p>', out)
52        self.assertIn('</p>', out)
53
54    def test_syshook_no_logdir_text_format(self):
55        # Issue 12890: we were emitting the <p> tag in text mode.
56        with temp_dir() as tracedir:
57            rc, out, err = assert_python_failure(
58                  '-c',
59                  ('import cgitb; cgitb.enable(format="text", logdir=%s); '
60                   'raise ValueError("Hello World")') % repr(tracedir),
61                  PYTHONIOENCODING='utf-8')
62        out = out.decode()
63        self.assertIn("ValueError", out)
64        self.assertIn("Hello World", out)
65        self.assertNotIn('<p>', out)
66        self.assertNotIn('</p>', out)
67
68
69if __name__ == "__main__":
70    unittest.main()
71