U
}g @ s` d Z ddlZddlZddlmZmZmZ ddlmZ ddl m
Z
ddl m Z G dd d e Z
dS )
zGTests to ensure that the html.parser tree builder generates good
trees. N)_DuplicateAttributeHandlerBeautifulSoupHTMLParserHTMLParserTreeBuilder)ParserRejectedMarkup)Any )HTMLTreeBuilderSmokeTestc @ s` e Zd ZeZdd Zdd Zdd Zdd Zd d
Z d d Z
d
d Z dd Z dd Z
dd ZdS )TestHTMLParserTreeBuilderc
C s8 dddg}|D ]$}t t | | W 5 Q R X qd S )Ns
)pytestZraisesr soup)selfZ
bad_markupmarkup r =/tmp/pip-unpacked-wheel-kgiupv3k/bs4/tests/test_htmlparser.pytest_rejected_input s
z-TestHTMLParserTreeBuilder.test_rejected_inputc C s d S Nr r r r r test_namespaced_system_doctype( s z8TestHTMLParserTreeBuilder.test_namespaced_system_doctypec C s d S r r r r r r test_namespaced_public_doctype, s z8TestHTMLParserTreeBuilder.test_namespaced_public_doctypec C s: | d}t|d}t|}t|jt|js6tdS )zfUnlike most tree builders, HTMLParserTreeBuilder and will
be restored after pickling.
z
foo N)r pickledumpsloads
isinstanceZbuildertypeAssertionError)r treeZdumpedZloadedr r r test_builder_is_pickled0 s
z1TestHTMLParserTreeBuilder.test_builder_is_pickledc C s | dd | dd d S )Nz
z
z Z assert_soupr r r r )test_redundant_empty_element_closing_tags9 s zCTestHTMLParserTreeBuilder.test_redundant_empty_element_closing_tagsc C s | dd d S )Nz
foo &# barzfoo &# barr r r r r test_empty_element= s z,TestHTMLParserTreeBuilder.test_empty_elementc C sv d}| |}d|jjks td|jjks.td|jdjksDt| j |dd}d |jjksbtd |jjksrtd S )Nz=
text
r
sourcelineF)Zstore_line_numbers)r pr# r Z sourceposfindnamer r
r r r r test_tracking_line_numbersB s
z4TestHTMLParserTreeBuilder.test_tracking_line_numbersc s d }d|jd ks tdg|jd ks4td|jd ksFtttd d fdd }|d d |tjd |tjd
d d }||d
d
dg d S )Nz;Zurl3hrefclsclassid)on_duplicate_attributeexpectedreturnc sX j | d}|jd k s t||jd ks.tdg|jd ksBtd|jd ksTtd S )N)r- r) r* r+ r, )r ar )r- r. r r
r r r assert_attribute^ s
zOTestHTMLParserTreeBuilder.test_on_duplicate_attribute..assert_attributeurl1c S s. t | | ts | | g| |< | | | d S r )r listappend)attrskeyvaluer r r
accumulatep s zITestHTMLParserTreeBuilder.test_on_duplicate_attribute..accumulateurl2)r r0 r r r r ZREPLACEZIGNORE)r r r2 r9 r r1 r test_on_duplicate_attributeP s
z5TestHTMLParserTreeBuilder.test_on_duplicate_attributec C sj dD ]`\}}}d| }| |j}| }d|d }||ksDt|jdd}d| }||kstqd S )N))z⇄u ⇄s ⇄)z⊧u ⊧s ⊧)z𝔑u 𝔑s 𝔑)z≧̸u ≧̸s ≧̸)z¬ ¬s ¬)z⫬u ⫬s ⫬)z"" ")z∴ ∴ ∴)z ∴r? r@ )z ∴r? r@ )zfjfjs fj)z⊔u ⊔s ⊔)z⊔︀u ⊔︀s ⊔︀)z'' ')z|| |z
%s
s
%s
utf8html) formatter)r divencoder ) r Z
input_elementZoutput_unicodeZoutput_elementr
rI Zwithout_elementexpectZ with_elementr r r test_html5_attributesw s z/TestHTMLParserTreeBuilder.test_html5_attributesc C sD d}| |}d|j ks td}| |}d|j ks@td S )Nza &nosuchentity b
z a &nosuchentity b
za &nosuchentity; b
)r r$ decoder r' r r r test_invalid_html_entity s
z2TestHTMLParserTreeBuilder.test_invalid_html_entityN)__name__
__module__ __qualname__r Zdefault_builderr r r r r r! r( r; rL rN r r r r r s ' r )__doc__r r
Zbs4.builder._htmlparserr r r Zbs4.exceptionsr typingr r r r r r r r s