|
import pytest |
|
from unittest.mock import patch |
|
from bs4.builder import DetectsXMLParsedAsHTML |
|
|
|
class TestDetectsXMLParsedAsHTML(object): |
|
|
|
@pytest.mark.parametrize( |
|
"markup,looks_like_xml", |
|
[("No xml declaration", False), |
|
("<html>obviously HTML</html", False), |
|
("<?xml ><html>Actually XHTML</html>", False), |
|
("<?xml> < html>Tricky XHTML</html>", False), |
|
("<?xml ><no-html-tag>", True), |
|
] |
|
) |
|
def test_warn_if_markup_looks_like_xml(self, markup, looks_like_xml): |
|
|
|
|
|
with patch('bs4.builder.DetectsXMLParsedAsHTML._warn') as mock: |
|
for data in markup, markup.encode('utf8'): |
|
result = DetectsXMLParsedAsHTML.warn_if_markup_looks_like_xml( |
|
data |
|
) |
|
assert result == looks_like_xml |
|
if looks_like_xml: |
|
assert mock.called |
|
else: |
|
assert not mock.called |
|
mock.reset_mock() |
|
|
