def test_attrs_no_value(self):
doc, tag, text = SimpleDoc().tagtext()
with tag('paper-button', 'raised'):
text('I am a fancy button')
self.assertEqual(
doc.getvalue(),
"<paper-button raised>I am a fancy button</paper-button>")
def test_attrs_no_value(self):
doc, tag, text = SimpleDoc().tagtext()
with tag('paper-button', 'raised'):
text('I am a fancy button')
self.assertEqual(
doc.getvalue(),
"<paper-button raised>I am a fancy button</paper-button>"
)
def test_tag(self):
doc, tag, text = SimpleDoc().tagtext()
with tag('h1', id='main-title'):
with tag('a', href="/"):
text("Hello world!")
self.assertEqual(
doc.getvalue(),
'<h1 id="main-title"><a href="/">Hello world!</a></h1>')
def test_stag(self):
doc = SimpleDoc()
doc.stag('img', src='/salmon-plays-piano.jpg')
self.assertEqual(doc.getvalue(),
'<img src="/salmon-plays-piano.jpg" />')
doc = SimpleDoc(stag_end='>')
doc.stag('img', src='/salmon-plays-piano.jpg')
self.assertEqual(doc.getvalue(), '<img src="/salmon-plays-piano.jpg">')
def test_tag(self):
doc, tag, text = SimpleDoc().tagtext()
with tag('h1', id = 'main-title'):
with tag('a', href="/"):
text("Hello world!")
self.assertEqual(
doc.getvalue(),
'<h1 id="main-title"><a href="/">Hello world!</a></h1>'
)
def test_line(self):
doc, tag, text, line = SimpleDoc().ttl()
line('h1', 'Some interesting links')
line('a', "Python's strftime directives", href="http://strftime.org"),
line('a', "Example of good UX for a homepage", href="http://zombo.com")
self.assertEqual(
doc.getvalue(),
('<h1>Some interesting links</h1>'
'<a href="http://strftime.org">Python\'s strftime directives</a>'
'<a href="http://zombo.com">Example of good UX for a homepage</a>'
))
def test_line(self):
doc, tag, text, line = SimpleDoc().ttl()
line('h1', 'Some interesting links')
line('a', "Python's strftime directives", href="http://strftime.org"),
line('a', "Example of good UX for a homepage", href="http://zombo.com")
self.assertEqual(
doc.getvalue(),
(
'<h1>Some interesting links</h1>'
'<a href="http://strftime.org">Python\'s strftime directives</a>'
'<a href="http://zombo.com">Example of good UX for a homepage</a>'
)
)
def test_attributes_substitution(self):
doc = SimpleDoc()
doc.stag('rect', stroke_width=1)
add_svg_attributes()
doc.stag('circle', stroke_width=2)
reset_attr_substitutions()
doc.stag('line', stroke_width=3)
self.assertEqual(
doc.getvalue(),
'<rect stroke_width="1" /><circle stroke-width="2" /><line stroke_width="3" />'
)
def test_stag(self):
doc = SimpleDoc()
doc.stag('img', src = '/salmon-plays-piano.jpg')
self.assertEqual(
doc.getvalue(),
'<img src="/salmon-plays-piano.jpg" />'
)
doc = SimpleDoc(stag_end = '>')
doc.stag('img', src = '/salmon-plays-piano.jpg')
self.assertEqual(
doc.getvalue(),
'<img src="/salmon-plays-piano.jpg">'
)
def test_attrs(self):
doc, tag, text = SimpleDoc().tagtext()
with tag('div', id='article'):
if True:
doc.attr(klass='new')
else:
doc.attr(klass='old')
with tag('a', ('data-my-id', '89'), klass='alert'):
text('hi')
doc.stag('img', src='squirrel.jpg', klass='animal')
root = ET.fromstring(doc.getvalue())
self.assertEqual(root.attrib['class'], "new")
self.assertEqual(root[0].attrib['class'], "alert")
self.assertEqual(root[0].attrib['data-my-id'], '89')
self.assertEqual(root[1].attrib['src'], 'squirrel.jpg')
self.assertEqual(root[1].attrib['class'], 'animal')
self.assertRaises(KeyError, lambda: root[1].attrib['klass'])
def test_attrs(self):
doc, tag, text = SimpleDoc().tagtext()
with tag('div', id = 'article'):
if True:
doc.attr(klass = 'new')
else:
doc.attr(klass = 'old')
with tag('a', ('data-my-id', '89'), klass='alert'):
text('hi')
doc.stag('img', src='squirrel.jpg', klass='animal')
root = ET.fromstring(doc.getvalue())
self.assertEqual(root.attrib['class'], "new")
self.assertEqual(root[0].attrib['class'], "alert")
self.assertEqual(root[0].attrib['data-my-id'], '89')
self.assertEqual(root[1].attrib['src'], 'squirrel.jpg')
self.assertEqual(root[1].attrib['class'], 'animal')
self.assertRaises(
KeyError,
lambda: root[1].attrib['klass']
)
def test_cdata(self):
doc, tag, text = SimpleDoc().tagtext()
with tag('example'):
doc.cdata('6 > 8 & 54')
self.assertEqual(doc.getvalue(),
'<example><![CDATA[6 > 8 & 54]]></example>')
doc = SimpleDoc()
doc.cdata('Jean Michel', safe=True)
self.assertEqual(doc.getvalue(), '<![CDATA[Jean Michel]]>')
doc = SimpleDoc()
doc.cdata('A CDATA section should end with ]]>')
self.assertEqual(
doc.getvalue(),
'<![CDATA[A CDATA section should end with ]]]]><![CDATA[>]]>')
doc = SimpleDoc()
doc.cdata('Some data ]]><script src="malicious.js">')
self.assertEqual(
doc.getvalue(),
'<![CDATA[Some data ]]]]><![CDATA[><script src="malicious.js">]]>')
def link(doc: SimpleDoc, path: str):
"""A link tag"""
doc.stag("link", rel="stylesheet", type="text/css", href=path)
def script(doc: SimpleDoc, path: str):
"""A script tag"""
with doc.tag("script", src=path, type="text/javascript"):
pass
def test_cdata(self):
doc, tag, text = SimpleDoc().tagtext()
with tag('example'):
doc.cdata('6 > 8 & 54')
self.assertEqual(
doc.getvalue(),
'<example><![CDATA[6 > 8 & 54]]></example>'
)
doc = SimpleDoc()
doc.cdata('Jean Michel', safe = True)
self.assertEqual(doc.getvalue(), '<![CDATA[Jean Michel]]>')
doc = SimpleDoc()
doc.cdata('A CDATA section should end with ]]>')
self.assertEqual(
doc.getvalue(),
'<![CDATA[A CDATA section should end with ]]]]><![CDATA[>]]>'
)
doc = SimpleDoc()
doc.cdata('Some data ]]><script src="malicious.js">')
self.assertEqual(
doc.getvalue(),
'<![CDATA[Some data ]]]]><![CDATA[><script src="malicious.js">]]>'
)
def test_html_classes(self):
def class_elems(node):
return set(node.attrib['class'].split(' '))
doc, tag, text = SimpleDoc().tagtext()
with tag('p', klass = 'news'):
doc.add_class('highlight', 'today')
doc.discard_class('news')
doc.toggle_class('active', True)
with tag('a', href = '/', klass = 'small useless'):
doc.discard_class('useless')
with tag('a', href = '/', klass = 'important'):
doc.discard_class('important')
root = ET.fromstring(doc.getvalue())
self.assertEqual(
class_elems(root),
set(['highlight', 'today', 'active'])
)
self.assertEqual(
class_elems(root[0]),
set(['small'])
)
self.assertRaises(KeyError, lambda: class_elems(root[1]))
def test_html_classes(self):
def class_elems(node):
return set(node.attrib['class'].split(' '))
doc, tag, text = SimpleDoc().tagtext()
with tag('p', klass='news'):
doc.add_class('highlight', 'today')
doc.discard_class('news')
doc.toggle_class('active', True)
with tag('a', href='/', klass='small useless'):
doc.discard_class('useless')
with tag('a', href='/', klass='important'):
doc.discard_class('important')
root = ET.fromstring(doc.getvalue())
self.assertEqual(class_elems(root),
set(['highlight', 'today', 'active']))
self.assertEqual(class_elems(root[0]), set(['small']))
self.assertRaises(KeyError, lambda: class_elems(root[1]))
def test_text(self):
message = "Hello\nworld!"
doc = SimpleDoc()
doc.text(message)
self.assertEqual(doc.getvalue(), message)
doc = SimpleDoc(nl2br=True)
doc.text(message)
self.assertEqual(doc.getvalue(), "Hello<br />world!")
doc = SimpleDoc(nl2br=True, stag_end = '>')
doc.text(message)
self.assertEqual(doc.getvalue(), "Hello<br>world!")
doc = SimpleDoc(nl2br=True, stag_end = '>')
doc.text("\n\n\n")
self.assertEqual(doc.getvalue(), "<br><br><br>")
doc = SimpleDoc(nl2br=True, stag_end = '>')
doc.text("One\r\nTwo\r\nThree\r\n\r\n")
self.assertEqual(doc.getvalue(), "One<br>Two<br>Three<br><br>")
doc = SimpleDoc()
doc.text("1 < 2")
self.assertEqual(doc.getvalue(), "1 < 2")
doc = SimpleDoc()
doc.text("&")
self.assertEqual(doc.getvalue(), "&")
