Discovered that html5lib can't be made to support SoupStrainers, and changed the test suite appropriately.

1 files changed, 11 insertions, 2 deletions

diff --git a/tests/test_html5lib.py b/tests/test_html5lib.py
index 021c603..aa0bad2 100644
--- a/tests/test_html5lib.py
+++ b/tests/test_html5lib.py
@@ -1,5 +1,5 @@
-from beautifulsoup.builder.html5lib_builder import HTML5TreeBuilder
-from beautifulsoup.element import Comment
+from beautifulsoup.builder import HTML5TreeBuilder
+from beautifulsoup.element import Comment, SoupStrainer
 from test_lxml import (
     TestLXMLBuilder,
     TestLXMLBuilderInvalidMarkup,
@@ -13,6 +13,15 @@ class TestHTML5Builder(TestLXMLBuilder):
     def default_builder(self):
         return HTML5TreeBuilder()
 
+    def test_soupstrainer(self):
+        # The html5lib tree builder does not support SoupStrainers.
+        strainer = SoupStrainer("b")
+        markup = "<p>A <b>bold</b> statement.</p>"
+        soup = self.soup(markup,
+                         parseOnlyThese=strainer)
+        self.assertEquals(
+            soup.decode(), self.document_for(markup))
+
     def test_bare_string(self):
         # A bare string is turned into some kind of HTML document or
         # fragment recognizable as the original string.