diff options
Diffstat (limited to 'bs4/tests/test_tree.py')
| -rw-r--r-- | bs4/tests/test_tree.py | 20 |
1 files changed, 16 insertions, 4 deletions
diff --git a/bs4/tests/test_tree.py b/bs4/tests/test_tree.py index 3e75fae..a4e2a8a 100644 --- a/bs4/tests/test_tree.py +++ b/bs4/tests/test_tree.py @@ -1317,16 +1317,28 @@ class TestSubstitutions(SoupTest): </script> """ encoded = BeautifulSoup(doc).encode() - self.assertTrue("< < hey > >" in encoded) + self.assertTrue(b"< < hey > >" in encoded) + + def test_formatter_skips_style_tag_for_html_documents(self): + doc = """ + <style type="text/css"> + console.log("< < hey > > "); + </style> +""" + encoded = BeautifulSoup(doc).encode() + self.assertTrue(b"< < hey > >" in encoded) def test_formatter_processes_script_tag_for_xml_documents(self): doc = """ <script type="text/javascript"> - console.log("< < hey > > "); </script> """ - encoded = BeautifulSoup(doc).encode() - self.assertTrue("< < hey > >" in encoded) + soup = BeautifulSoup(doc, "xml") + # lxml would have stripped this while parsing, but we can add + # it later. + soup.script.string = 'console.log("< < hey > > ");' + encoded = soup.encode() + self.assertTrue(b"< < hey > >" in encoded) def test_prettify_accepts_formatter(self): soup = BeautifulSoup("<html><body>foo</body></html>") |