Fix #132: add test for #115, single character fragments (#264)

gsnedders · gsnedders · commit 5197f557afe1 · 2016-06-06T22:40:46.000+01:00
r=nobody!
diff --git a/html5lib/tests/test_treewalkers.py b/html5lib/tests/test_treewalkers.py
@@ -1,5 +1,7 @@
 from __future__ import absolute_import, division, unicode_literals
 
+import itertools
+
 import pytest
 
 try:
@@ -100,6 +102,24 @@ def test_treewalker_six_mix():
             yield runTreewalkerEditTest, intext, expected, attrs, tree
 
 
+@pytest.mark.parametrize("tree,char", itertools.product(sorted(treeTypes.items()), ["x", "\u1234"]))
+def test_fragment_single_char(tree, char):
+    expected = [
+        {'data': char, 'type': 'Characters'}
+    ]
+
+    treeName, treeClass = tree
+    if treeClass is None:
+        pytest.skip("Treebuilder not loaded")
+
+    parser = html5parser.HTMLParser(tree=treeClass["builder"])
+    document = parser.parseFragment(char)
+    document = treeClass.get("adapter", lambda x: x)(document)
+    output = Lint(treeClass["walker"](document))
+
+    assert list(output) == expected
+
+
 @pytest.mark.skipif(treeTypes["lxml"] is None, reason="lxml not importable")
 def test_lxml_xml():
     expected = [
