| import sys |
| |
| from mako import exceptions |
| from mako.lookup import TemplateLookup |
| from mako.template import Template |
| from mako.testing.exclusions import requires_no_pygments_exceptions |
| from mako.testing.exclusions import requires_pygments_14 |
| from mako.testing.fixtures import TemplateTest |
| from mako.testing.helpers import result_lines |
| |
| |
| class ExceptionsTest(TemplateTest): |
| def test_html_error_template(self): |
| """test the html_error_template""" |
| code = """ |
| % i = 0 |
| """ |
| try: |
| template = Template(code) |
| template.render_unicode() |
| assert False |
| except exceptions.CompileException: |
| html_error = exceptions.html_error_template().render_unicode() |
| assert ( |
| "CompileException: Fragment 'i = 0' is not " |
| "a partial control statement at line: 2 char: 1" |
| ) in html_error |
| assert "<style>" in html_error |
| html_error_stripped = html_error.strip() |
| assert html_error_stripped.startswith("<html>") |
| assert html_error_stripped.endswith("</html>") |
| |
| not_full = exceptions.html_error_template().render_unicode( |
| full=False |
| ) |
| assert "<html>" not in not_full |
| assert "<style>" in not_full |
| |
| no_css = exceptions.html_error_template().render_unicode(css=False) |
| assert "<style>" not in no_css |
| else: |
| assert False, ( |
| "This function should trigger a CompileException, " |
| "but didn't" |
| ) |
| |
| def test_text_error_template(self): |
| code = """ |
| % i = 0 |
| """ |
| try: |
| template = Template(code) |
| template.render_unicode() |
| assert False |
| except exceptions.CompileException: |
| text_error = exceptions.text_error_template().render_unicode() |
| assert "Traceback (most recent call last):" in text_error |
| assert ( |
| "CompileException: Fragment 'i = 0' is not a partial " |
| "control statement" |
| ) in text_error |
| |
| @requires_pygments_14 |
| def test_utf8_html_error_template_pygments(self): |
| """test the html_error_template with a Template containing UTF-8 |
| chars""" |
| |
| code = """# -*- coding: utf-8 -*- |
| % if 2 == 2: /an error |
| ${'привет'} |
| % endif |
| """ |
| try: |
| template = Template(code) |
| template.render_unicode() |
| except exceptions.CompileException: |
| html_error = exceptions.html_error_template().render() |
| assert ( |
| "CompileException: Fragment 'if 2 == 2: /an " |
| "error' is not a partial control statement " |
| "at line: 2 char: 1" |
| ).encode( |
| sys.getdefaultencoding(), "htmlentityreplace" |
| ) in html_error |
| |
| assert ( |
| "".encode(sys.getdefaultencoding(), "htmlentityreplace") |
| in html_error |
| ) |
| else: |
| assert False, ( |
| "This function should trigger a CompileException, " |
| "but didn't" |
| ) |
| |
| @requires_no_pygments_exceptions |
| def test_utf8_html_error_template_no_pygments(self): |
| """test the html_error_template with a Template containing UTF-8 |
| chars""" |
| |
| code = """# -*- coding: utf-8 -*- |
| % if 2 == 2: /an error |
| ${'привет'} |
| % endif |
| """ |
| try: |
| template = Template(code) |
| template.render_unicode() |
| except exceptions.CompileException: |
| html_error = exceptions.html_error_template().render() |
| assert ( |
| "CompileException: Fragment 'if 2 == 2: /an " |
| "error' is not a partial control statement " |
| "at line: 2 char: 1" |
| ).encode( |
| sys.getdefaultencoding(), "htmlentityreplace" |
| ) in html_error |
| assert ( |
| "${'привет'}".encode( |
| sys.getdefaultencoding(), "htmlentityreplace" |
| ) |
| in html_error |
| ) |
| else: |
| assert False, ( |
| "This function should trigger a CompileException, " |
| "but didn't" |
| ) |
| |
| def test_format_closures(self): |
| try: |
| exec("def foo():" " raise RuntimeError('test')", locals()) |
| foo() # noqa |
| except: |
| html_error = exceptions.html_error_template().render() |
| assert "RuntimeError: test" in str(html_error) |
| |
| def test_py_utf8_html_error_template(self): |
| try: |
| foo = "日本" # noqa |
| raise RuntimeError("test") |
| except: |
| html_error = exceptions.html_error_template().render() |
| assert "RuntimeError: test" in html_error.decode("utf-8") |
| assert "foo = "日本"" in html_error.decode( |
| "utf-8" |
| ) or "foo = "日本"" in html_error.decode("utf-8") |
| |
| def test_py_unicode_error_html_error_template(self): |
| try: |
| raise RuntimeError("日本") |
| except: |
| html_error = exceptions.html_error_template().render() |
| assert "RuntimeError: 日本".encode("ascii", "ignore") in html_error |
| |
| @requires_pygments_14 |
| def test_format_exceptions_pygments(self): |
| l = TemplateLookup(format_exceptions=True) |
| |
| l.put_string( |
| "foo.html", |
| """ |
| <%inherit file="base.html"/> |
| ${foobar} |
| """, |
| ) |
| |
| l.put_string( |
| "base.html", |
| """ |
| ${self.body()} |
| """, |
| ) |
| |
| assert ( |
| '<table class="syntax-highlightedtable">' |
| in l.get_template("foo.html").render_unicode() |
| ) |
| |
| @requires_no_pygments_exceptions |
| def test_format_exceptions_no_pygments(self): |
| l = TemplateLookup(format_exceptions=True) |
| |
| l.put_string( |
| "foo.html", |
| """ |
| <%inherit file="base.html"/> |
| ${foobar} |
| """, |
| ) |
| |
| l.put_string( |
| "base.html", |
| """ |
| ${self.body()} |
| """, |
| ) |
| |
| assert '<div class="sourceline">${foobar}</div>' in result_lines( |
| l.get_template("foo.html").render_unicode() |
| ) |
| |
| @requires_pygments_14 |
| def test_utf8_format_exceptions_pygments(self): |
| """test that htmlentityreplace formatting is applied to |
| exceptions reported with format_exceptions=True""" |
| |
| l = TemplateLookup(format_exceptions=True) |
| l.put_string( |
| "foo.html", """# -*- coding: utf-8 -*-\n${'привет' + foobar}""" |
| ) |
| |
| assert "'привет'</span>" in l.get_template( |
| "foo.html" |
| ).render().decode("utf-8") |
| |
| @requires_no_pygments_exceptions |
| def test_utf8_format_exceptions_no_pygments(self): |
| """test that htmlentityreplace formatting is applied to |
| exceptions reported with format_exceptions=True""" |
| |
| l = TemplateLookup(format_exceptions=True) |
| l.put_string( |
| "foo.html", """# -*- coding: utf-8 -*-\n${'привет' + foobar}""" |
| ) |
| |
| assert ( |
| '<div class="sourceline">${'привет' + foobar}</div>' |
| in result_lines( |
| l.get_template("foo.html").render().decode("utf-8") |
| ) |
| ) |
| |
| def test_mod_no_encoding(self): |
| mod = __import__("test.foo.mod_no_encoding").foo.mod_no_encoding |
| try: |
| mod.run() |
| except: |
| t, v, tback = sys.exc_info() |
| exceptions.html_error_template().render_unicode( |
| error=v, traceback=tback |
| ) |
| |
| def test_custom_tback(self): |
| try: |
| raise RuntimeError("error 1") |
| foo("bar") # noqa |
| except: |
| t, v, tback = sys.exc_info() |
| |
| try: |
| raise RuntimeError("error 2") |
| except: |
| html_error = exceptions.html_error_template().render_unicode( |
| error=v, traceback=tback |
| ) |
| |
| # obfuscate the text so that this text |
| # isn't in the 'wrong' exception |
| assert ( |
| "".join(reversed(");touq&rab;touq&(oof")) in html_error |
| or "".join(reversed(");43#&rab;43#&(oof")) in html_error |
| ) |
| |
| def test_tback_no_trace_from_py_file(self): |
| try: |
| t = self._file_template("runtimeerr.html") |
| t.render() |
| except: |
| t, v, tback = sys.exc_info() |
| |
| # and don't even send what we have. |
| html_error = exceptions.html_error_template().render_unicode( |
| error=v, traceback=None |
| ) |
| |
| assert self.indicates_unbound_local_error(html_error, "y") |
| |
| def test_tback_trace_from_py_file(self): |
| t = self._file_template("runtimeerr.html") |
| try: |
| t.render() |
| assert False |
| except: |
| html_error = exceptions.html_error_template().render_unicode() |
| |
| assert self.indicates_unbound_local_error(html_error, "y") |
| |
| def test_code_block_line_number(self): |
| l = TemplateLookup() |
| l.put_string( |
| "foo.html", |
| """ |
| <% |
| msg = "Something went wrong." |
| raise RuntimeError(msg) # This is the line. |
| %> |
| """, |
| ) |
| t = l.get_template("foo.html") |
| try: |
| t.render() |
| except: |
| text_error = exceptions.text_error_template().render_unicode() |
| assert 'File "foo.html", line 4, in render_body' in text_error |
| assert "raise RuntimeError(msg) # This is the line." in text_error |
| else: |
| assert False |
| |
| def test_module_block_line_number(self): |
| l = TemplateLookup() |
| l.put_string( |
| "foo.html", |
| """ |
| <%! |
| def foo(): |
| msg = "Something went wrong." |
| raise RuntimeError(msg) # This is the line. |
| %> |
| ${foo()} |
| """, |
| ) |
| t = l.get_template("foo.html") |
| try: |
| t.render() |
| except: |
| text_error = exceptions.text_error_template().render_unicode() |
| assert 'File "foo.html", line 7, in render_body' in text_error |
| assert 'File "foo.html", line 5, in foo' in text_error |
| assert "raise RuntimeError(msg) # This is the line." in text_error |
| else: |
| assert False |
| |
| def test_alternating_file_names(self): |
| l = TemplateLookup() |
| l.put_string( |
| "base.html", |
| """ |
| <%! |
| def broken(): |
| raise RuntimeError("Something went wrong.") |
| %> body starts here |
| <%block name="foo"> |
| ${broken()} |
| </%block> |
| """, |
| ) |
| l.put_string( |
| "foo.html", |
| """ |
| <%inherit file="base.html"/> |
| <%block name="foo"> |
| ${parent.foo()} |
| </%block> |
| """, |
| ) |
| t = l.get_template("foo.html") |
| try: |
| t.render() |
| except: |
| text_error = exceptions.text_error_template().render_unicode() |
| assert ( |
| """ |
| File "base.html", line 5, in render_body |
| %> body starts here |
| File "foo.html", line 4, in render_foo |
| ${parent.foo()} |
| File "base.html", line 7, in render_foo |
| ${broken()} |
| File "base.html", line 4, in broken |
| raise RuntimeError("Something went wrong.") |
| """ |
| in text_error |
| ) |
| else: |
| assert False |
