96 lines
2.9 KiB
Python
Executable File
96 lines
2.9 KiB
Python
Executable File
#!/usr/bin/env python
|
|
from quixote.test.utest import UTest
|
|
from quixote.ptl.ptl_compile import compile_template
|
|
from io import StringIO
|
|
from quixote.html import TemplateIO, htmltext, _q_join, _q_format
|
|
|
|
try:
|
|
compile('f""\n', 'none', 'single')
|
|
except SyntaxError:
|
|
HAVE_FSTRINGS = False
|
|
else:
|
|
HAVE_FSTRINGS = True
|
|
|
|
|
|
def run_ptl(*source):
|
|
"""
|
|
Compile the given lines of source code using the ptl compiler
|
|
and run the resulting compiled code.
|
|
"""
|
|
# When the ptl compiler compiles a module, it places _q_TemplateIO
|
|
# and _q_htmltext into the globals of the module. Here, we don't
|
|
# have a module, but we provide these same globals for eval.
|
|
eval(compile_template(StringIO('\n'.join(source)), 'test'),
|
|
dict(_q_TemplateIO=TemplateIO, _q_htmltext=htmltext,
|
|
_q_join=_q_join, _q_format=_q_format))
|
|
|
|
class Test (UTest):
|
|
|
|
def check_html(self):
|
|
run_ptl(
|
|
'from quixote.html import htmltext',
|
|
'def f [html] (a):',
|
|
' "&"',
|
|
' a',
|
|
'assert type(f(1)) == htmltext',
|
|
'assert f("") == "&"',
|
|
'assert f("&") == "&&"',
|
|
'assert f(htmltext("&")) == "&&"')
|
|
|
|
def check_plain(self):
|
|
run_ptl(
|
|
'from quixote.html import htmltext',
|
|
'def f [plain] (a):',
|
|
' "&"',
|
|
' a',
|
|
'assert type(f(1)) == str',
|
|
'assert f("") == "&"',
|
|
'assert f("&") == "&&"',
|
|
'assert f(htmltext("&")) == "&&"',
|
|
'assert type(f(htmltext("&"))) == str')
|
|
|
|
def check_syntax(self):
|
|
run_ptl('def f(a):\n a')
|
|
try:
|
|
run_ptl('def f [] (a):\n a')
|
|
assert 0
|
|
except SyntaxError as e:
|
|
assert e.lineno == 1
|
|
try:
|
|
run_ptl('def f [HTML] (a):\n a')
|
|
assert 0
|
|
except SyntaxError as e:
|
|
assert e.lineno == 1
|
|
|
|
|
|
if HAVE_FSTRINGS:
|
|
def check_fstring(self):
|
|
run_ptl(
|
|
'from quixote.html import htmltext',
|
|
'def f [html] (a):',
|
|
' f"x{a}"',
|
|
'assert type(f(1)) == htmltext',
|
|
'assert f("&") == "x&"',
|
|
'assert f(htmltext("&")) == "x&"')
|
|
|
|
def check_q_join(self):
|
|
assert _q_join('x', '&') == 'x&'
|
|
assert _q_join('x', htmltext('&')) == 'x&'
|
|
|
|
def check_q_format(self):
|
|
assert _q_format('&') == '&'
|
|
assert _q_format(htmltext('&')) == '&'
|
|
assert _q_format('a', ord('r')) == "'a'"
|
|
assert _q_format(htmltext('a'), ord('r')) == "'a'"
|
|
assert _q_format(1, -1, '_>2') == '_1'
|
|
assert _q_format(1, -1, '_>2') == '_1'
|
|
assert _q_format(64, -1, 'c') == '@'
|
|
assert _q_format(38, -1, 'c') == '&'
|
|
|
|
|
|
def test_all():
|
|
Test()
|
|
|
|
if __name__ == "__main__":
|
|
test_all()
|