diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_html5lib.py | 13 | ||||
| -rw-r--r-- | tests/test_lxml.py | 11 | ||||
| -rw-r--r-- | tests/test_tree.py | 9 |
3 files changed, 24 insertions, 9 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. diff --git a/tests/test_lxml.py b/tests/test_lxml.py index 88c866d..de2ce7b 100644 --- a/tests/test_lxml.py +++ b/tests/test_lxml.py @@ -3,8 +3,8 @@ import re from beautifulsoup import BeautifulSoup -from beautifulsoup.builder.lxml_builder import LXMLTreeBuilder -from beautifulsoup.element import Comment, Doctype +from beautifulsoup.builder import LXMLTreeBuilder, LXMLTreeBuilderForXML +from beautifulsoup.element import Comment, Doctype, SoupStrainer from beautifulsoup.testing import SoupTest @@ -320,6 +320,12 @@ class TestLXMLBuilder(SoupTest): self.assertFalse(soup.p.is_empty_element) self.assertEquals(str(soup.p), "<p></p>") + def test_soupstrainer(self): + strainer = SoupStrainer("b") + soup = self.soup("A <b>bold</b> <meta /> <i>statement</i>", + parseOnlyThese=strainer) + self.assertEquals(soup.decode(), "<b>bold</b>") + class TestLXMLBuilderInvalidMarkup(SoupTest): """Tests of invalid markup for the LXML tree builder. @@ -505,7 +511,6 @@ class TestLXMLBuilderEncodingConversion(SoupTest): self.HEBREW_DOCUMENT.decode("iso-8859-8").encode("utf-8")) -from beautifulsoup.builder.lxml_builder import LXMLTreeBuilderForXML class TestLXMLXMLBuilder(SoupTest): """Test XML-specific parsing behavior. diff --git a/tests/test_tree.py b/tests/test_tree.py index 8cbd309..384d518 100644 --- a/tests/test_tree.py +++ b/tests/test_tree.py @@ -524,14 +524,15 @@ class TestTreeModification(SoupTest): def test_new_tag_creation(self): builder = BeautifulSoup.default_builder() - soup = self.soup("", builder=builder) + soup = self.soup("<body></body>", builder=builder) a = Tag(soup, builder, 'a') ol = Tag(soup, builder, 'ol') a['href'] = 'http://foo.com/' - soup.insert(0, a) - soup.insert(1, ol) + soup.body.insert(0, a) + soup.body.insert(1, ol) self.assertEqual( - soup.decode(), '<a href="http://foo.com/"></a><ol></ol>') + soup.body.encode(), + '<body><a href="http://foo.com/"></a><ol></ol></body>') def test_append_to_contents_moves_tag(self): doc = """<p id="1">Don't leave me <b>here</b>.</p> |