You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
114 lines
3.1 KiB
Python
114 lines
3.1 KiB
Python
# -*- coding: utf8 -*-
|
|
|
|
"""Verify we can process html into a document to work off of."""
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import division, print_function, unicode_literals
|
|
|
|
from collections import defaultdict
|
|
from breadability._compat import (
|
|
to_unicode,
|
|
to_bytes,
|
|
unicode,
|
|
)
|
|
|
|
from breadability.document import (
|
|
convert_breaks_to_paragraphs,
|
|
decode_html,
|
|
OriginalDocument,
|
|
)
|
|
from .utils import load_snippet
|
|
|
|
|
|
def test_convert_br_tags_to_paragraphs():
|
|
returned = convert_breaks_to_paragraphs(
|
|
("<div>HI<br><br>How are you?<br><br> \t \n <br>"
|
|
"Fine\n I guess</div>"))
|
|
|
|
assert returned == "<div>HI</p><p>How are you?</p><p>Fine\n I guess</div>"
|
|
|
|
|
|
def test_convert_hr_tags_to_paragraphs():
|
|
returned = convert_breaks_to_paragraphs(
|
|
"<div>HI<br><br>How are you?<hr/> \t \n <br>Fine\n I guess</div>")
|
|
|
|
assert returned == "<div>HI</p><p>How are you?</p><p>Fine\n I guess</div>"
|
|
|
|
|
|
def test_readin_min_document():
|
|
"""Verify we can read in a min html document"""
|
|
doc = OriginalDocument(load_snippet('document_min.html'))
|
|
|
|
assert to_unicode(doc).startswith('<html>')
|
|
assert doc.title == 'Min Document Title'
|
|
|
|
|
|
def test_readin_with_base_url():
|
|
"""Passing a url should update links to be absolute links"""
|
|
doc = OriginalDocument(
|
|
load_snippet('document_absolute_url.html'),
|
|
url="http://blog.mitechie.com/test.html")
|
|
|
|
assert to_unicode(doc).startswith('<html>')
|
|
|
|
# find the links on the page and make sure each one starts with out
|
|
# base url we told it to use.
|
|
links = doc.links
|
|
assert len(links) == 3
|
|
# we should have two links that start with our blog url
|
|
# and one link that starts with amazon
|
|
link_counts = defaultdict(int)
|
|
for link in links:
|
|
if link.get('href').startswith('http://blog.mitechie.com'):
|
|
link_counts['blog'] += 1
|
|
else:
|
|
link_counts['other'] += 1
|
|
|
|
assert link_counts['blog'] == 2
|
|
assert link_counts['other'] == 1
|
|
|
|
|
|
def test_no_br_allowed():
|
|
"""We convert all <br/> tags to <p> tags"""
|
|
doc = OriginalDocument(load_snippet('document_min.html'))
|
|
|
|
assert doc.dom.find('.//br') is None
|
|
|
|
|
|
def test_empty_title():
|
|
"""We convert all <br/> tags to <p> tags"""
|
|
document = OriginalDocument(
|
|
"<html><head><title></title></head><body></body></html>")
|
|
|
|
assert document.title == ""
|
|
|
|
|
|
def test_title_only_with_tags():
|
|
"""We convert all <br/> tags to <p> tags"""
|
|
document = OriginalDocument(
|
|
"<html><head><title><em></em></title></head><body></body></html>")
|
|
|
|
assert document.title == ""
|
|
|
|
|
|
def test_no_title():
|
|
"""We convert all <br/> tags to <p> tags"""
|
|
document = OriginalDocument("<html><head></head><body></body></html>")
|
|
|
|
assert document.title == ""
|
|
|
|
|
|
def test_encoding():
|
|
text = "ľščťžýáíéäúňôůě".encode("iso-8859-2")
|
|
html = decode_html(text)
|
|
|
|
assert type(html) is unicode
|
|
|
|
|
|
def test_encoding_short():
|
|
text = to_bytes("ľščťžýáíé")
|
|
html = decode_html(text)
|
|
|
|
assert type(html) is unicode
|
|
assert html == "ľščťžýáíé"
|