diff options
| -rw-r--r-- | test/test_xhtml.py | 35 |
1 files changed, 33 insertions, 2 deletions
diff --git a/test/test_xhtml.py b/test/test_xhtml.py index d81505da..4f0f2b4f 100644 --- a/test/test_xhtml.py +++ b/test/test_xhtml.py @@ -4,10 +4,19 @@ Test the functions in the `xhtml` module import pytest import xml - +import poezio.xhtml from poezio.xhtml import (poezio_colors_to_html, xhtml_to_poezio_colors, _parse_css as parse_css, clean_text) +class ConfigShim(object): + def __init__(self): + self.value = True + def get(self, *args, **kwargs): + return self.value + +config = ConfigShim() +poezio.xhtml.config = config + def test_clean_text(): example_string = '\x191}Toto \x192,-1}titi\x19b Tata' assert clean_text(example_string) == 'Toto titi Tata' @@ -47,10 +56,32 @@ def test_xhtml_to_poezio_colors(): with pytest.raises(xml.sax._exceptions.SAXParseException): xhtml_to_poezio_colors(b'<p>Invalid xml') +def test_xhtml_to_poezio_colors_disabled(): + config.value = False + start = b'<body xmlns="http://www.w3.org/1999/xhtml"><p>' + end = b'</p></body>' + xhtml = start + b'test' + end + assert xhtml_to_poezio_colors(xhtml) == 'test' + + xhtml = start + b'<a href="http://perdu.com">salut</a>' + end + assert xhtml_to_poezio_colors(xhtml) == '\x19usalut\x19o (http://perdu.com)' + + xhtml = start + b'<a href="http://perdu.com">http://perdu.com</a>' + end + assert xhtml_to_poezio_colors(xhtml) == '\x19uhttp://perdu.com\x19o' + + xhtml = b'<div style="font-weight:bold">Allo <div style="color:red">test <div style="color: blue">test2</div></div></div>' + assert xhtml_to_poezio_colors(xhtml, force=True) == 'Allo test test2' + + xhtml = (b'<div style="color:blue"><div style="color:yellow">' + b'<div style="color:blue">Allo <div style="color:red">' + b'test <div style="color: blue">test2</div></div></div></div></div>') + assert xhtml_to_poezio_colors(xhtml, force=True) == 'Allo test test2' + + config.value = True + def test_parse_css(): example_css = 'text-decoration: underline; color: red;' assert parse_css(example_css) == '\x19u\x19196}' example_css = 'text-decoration: underline coucou color: red;' assert parse_css(example_css) == '' - |