U
|
¡ý°dC ã @ sv d dl Z d dlZd dlmZ d dlmZmZmZ ddlm Z m
|
Z
|
e
|
rPd dl m Z e j je
|
ddG d d
|
d
|
e ZdS ) é N)Ú MagicMock)ÚCSSÚ BeautifulSoupÚ ResultSeté )ÚSoupTestÚSOUP_SIEVE_PRESENT)ÚSelectorSyntaxErrorzSoup Sieve not installed)Úreasonc @ s e Zd ZdZdZdd Zdd ZeZdd Zd d
|
Z d d Z
|
d d Z dd Z dd Z dd Zdd Zdd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Z d1d2 Z d3d4 Z d5d6 Z d7d8 Z d9d: Z!d;d< Z"d=d> Z#d?d@ Z$dAdB Z%dCdD Z&dEdF Z'dGdH Z(dIdJ Z)dKdL Z*dMdN Z+dOdP Z,dQdR Z-dSdT Z.dUdV Z/dWdX Z0dYdZ Z1d[d\ Z2d]d^ Z3d_d` Z4dadb Z5dcdd Z6dedf Z7dgdh Z8didj Z9dkdl Z:dmdn Z;dodp Z<dqdr Z=dsdt Z>dudv Z?dwdx Z@dydz ZAd{d| ZBd}d~ ZCdd ZDdS )ÚTestCSSSelectorszÞTest basic CSS selector functionality.
|
|
This functionality is implemented in soupsieve, which has a much
|
more comprehensive test suite, so this is basically an extra check
|
that soupsieve works as expected.
|
aþ
|
<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01//EN"
|
"http://www.w3.org/TR/html4/strict.dtd">
|
<html>
|
<head>
|
<title>The title</title>
|
<link rel="stylesheet" href="blah.css" type="text/css" id="l1">
|
</head>
|
<body>
|
<custom-dashed-tag class="dashed" id="dash1">Hello there.</custom-dashed-tag>
|
<div id="main" class="fancy">
|
<div id="inner">
|
<h1 id="header1">An H1</h1>
|
<p>Some text</p>
|
<p class="onep" id="p1">Some more text</p>
|
<h2 id="header2">An H2</h2>
|
<p class="class1 class2 class3" id="pmulti">Another</p>
|
<a href="http://bob.example.org/" rel="friend met" id="bob">Bob</a>
|
<h2 id="header3">Another H2</h2>
|
<a id="me" href="http://simonwillison.net/" rel="me">me</a>
|
<span class="s1">
|
<a href="#" id="s1a1">span1a1</a>
|
<a href="#" id="s1a2">span1a2 <span id="s1a2s1">test</span></a>
|
<span class="span2">
|
<a href="#" id="s2a1">span2a1</a>
|
</span>
|
<span class="span3"></span>
|
<custom-dashed-tag class="dashed" id="dash2"/>
|
<div data-tag="dashedvalue" id="data1"/>
|
</span>
|
</div>
|
<x id="xid">
|
<z id="zida"/>
|
<z id="zidab"/>
|
<z id="zidac"/>
|
</x>
|
<y id="yid">
|
<z id="zidb"/>
|
</y>
|
<p lang="en" id="lang-en">English</p>
|
<p lang="en-gb" id="lang-en-gb">English UK</p>
|
<p lang="en-us" id="lang-en-us">English US</p>
|
<p lang="fr" id="lang-fr">French</p>
|
</div>
|
|
<div id="footer">
|
</div>
|
c C s t | jd| _d S )Nú html.parser)r ÚHTMLÚsoup©Úself© r úId:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\bs4/tests/test_css.pyÚ setup_methodN s z TestCSSSelectors.setup_methodc K sf | j j|f|}t|ts tdd |D }| ¡ | ¡ ||ksbtd|d |¡d |¡f d S )Nc S s g | ] }|d qS ©Úidr )Ú.0Úelr r r Ú
|
<listcomp>T s z3TestCSSSelectors.assert_selects.<locals>.<listcomp>z$Selector %s, expected [%s], got [%s]z, )r ÚselectÚ
|
isinstancer ÚAssertionErrorÚsortÚjoin)r ÚselectorÚ expected_idsÚkwargsÚresultsZel_idsr r r Úassert_selectsQ s ÿz TestCSSSelectors.assert_selectsc G s |D ]\}}| ||¡ qd S )N©r" )r Útestsr r r r r Úassert_select_multiple] s z'TestCSSSelectors.assert_select_multiplec C sb | j j d¡}| j |¡}t|dks*t|D ]}|jdks.tq.| j |¡}d|d ks^td S ©NÚdivé Úmainr )r ÚcssÚcompiler Úlenr ÚnameÚ
|
select_one)r ÚselÚelsr' r r r r Útest_precompileda s z!TestCSSSelectors.test_precompiledc C sF | j d¡}t|dks t|d jdks.t|d jdgksBtd S )NÚtitler r z The title)r r r, r r- Úcontents©r r0 r r r Útest_one_tag_onel s z!TestCSSSelectors.test_one_tag_onec C sT | j d¡}t|dks t|D ]}|jdks tq | j d¡}d|d ksPtd S r& )r r r, r r- r. )r r0 r' r r r r Útest_one_tag_manyr s z"TestCSSSelectors.test_one_tag_manyc C s | j d¡}d |kstd S )NZnonexistenttag)r r. r )r Úmatchr r r Ú(test_select_one_returns_none_if_no_match{ s z9TestCSSSelectors.test_select_one_returns_none_if_no_matchc C s | j d¡}| dddg¡ d S )Nzdiv divÚinnerÚdata1)r r r" r4 r r r Útest_tag_in_tag_one s z$TestCSSSelectors.test_tag_in_tag_onec C s" dD ]}| |ddddg¡ qd S )N)úhtml divú html body divúbody divr: r) r9 Úfooterr# ©r r r r r Útest_tag_in_tag_many s z%TestCSSSelectors.test_tag_in_tag_manyc C sB | j ddgdd | j dddgdd | j dd ddd
|
gd d d S ) Nr< r) r )Úlimitr= r9 é r> r: r? é
|
r# r r r r Ú
|
test_limit s
|
ÿzTestCSSSelectors.test_limitc C s t | j d¡dkstd S )NÚdelr ©r, r r r r r r r Útest_tag_no_match s z"TestCSSSelectors.test_tag_no_matchc C s& t t¡ | j d¡ W 5 Q R X d S )Nztag%t©ÚpytestÚraisesr r r r r r r Útest_invalid_tag s z!TestCSSSelectors.test_invalid_tagc C s | dddg¡ d S )Núcustom-dashed-tagÚdash1Údash2r# r r r r Útest_select_dashed_tag_ids s z+TestCSSSelectors.test_select_dashed_tag_idsc C s6 | j d¡}|d jdks t|d d dks2td S )Nz custom-dashed-tag[id="dash2"]r rM r rO )r r r- r )r Zdashedr r r Útest_select_dashed_by_id s z)TestCSSSelectors.test_select_dashed_by_idc C s | j d¡d jdkstd S )Nzbody > custom-dashed-tagr z Hello there.)r r Útextr r r r r Útest_dashed_tag_text s z%TestCSSSelectors.test_dashed_tag_textc C s | j d¡| j d¡ks td S )NrM )r r Úfind_allr r r r r Ú#test_select_dashed_matches_find_all¡ s z4TestCSSSelectors.test_select_dashed_matches_find_allc C s | ddgfdddgf¡ d S )NÚh1Úheader1Úh2Úheader2Úheader3©r% r r r r Útest_header_tags¤ s
|
þz!TestCSSSelectors.test_header_tagsc C sR dD ]H}| j |¡}t|dks$t|d jdks6t|d d dgkstqd S )N)z.onepzp.onepz html p.onepr r ÚpÚclassÚonep)r r r, r r- )r r r0 r r r Útest_class_oneª s
|
z TestCSSSelectors.test_class_onec C s | j d¡}t|dks td S )Nzdiv.onepr ©r r r, r r4 r r r Útest_class_mismatched_tag± s z*TestCSSSelectors.test_class_mismatched_tagc C s dD ]}| |dg¡ qd S )N)z div#innerz#innerz div div#innerr9 r# r@ r r r Ú test_one_idµ s z TestCSSSelectors.test_one_idc C s | j d¡}t|dks td S )Nz #doesnotexistr ra r4 r r r Ú test_bad_id¹ s z TestCSSSelectors.test_bad_idc C s` | j d¡}t|dks t|D ]}|jdks tq |d d dgksJt|d d¡r\td S )Nz div#inner pé r] r r^ r_ r )r r r, r r- Zhas_attr)r r0 r r r r Útest_items_in_id½ s z!TestCSSSelectors.test_items_in_idc C s& dD ] }t | j |¡dkstqd S )N)z div#main delzdiv#main div.oopsz div div#mainr rG r@ r r r Útest_a_bunch_of_emptysÅ s z'TestCSSSelectors.test_a_bunch_of_emptysc C s dD ]}| |dg¡ qd S )N)z.class1zp.class1z.class2zp.class2z.class3zp.class3z html p.class2zdiv#inner .class2Úpmultir# r@ r r r Útest_multi_class_supportÉ s z)TestCSSSelectors.test_multi_class_supportc C s dD ]}| |dg¡ qd S )N)z.class1.class3z.class3.class2z.class1.class2.class3rh r# r@ r r r Útest_multi_class_selectionÎ s z+TestCSSSelectors.test_multi_class_selectionc C s" | dddg¡ | ddg¡ d S )Nz.s1 > aÚs1a1Ús1a2z .s1 > a spanÚs1a2s1r# r r r r Útest_child_selectorÓ s z$TestCSSSelectors.test_child_selectorc C s | ddg¡ d S )Nz.s1 > a#s1a2 spanrm r# r r r r Útest_child_selector_id× s z'TestCSSSelectors.test_child_selector_idc C st | ddgfddgfddgfddgfddgfddgfd dgfd
|
g fd dgfd dgfd dgfdg fdg fdg f¡ d S )Nzp[class="onep"]Úp1z
|
p[id="p1"]z[class="onep"]z [id="p1"]zlink[rel="stylesheet"]Úl1zlink[type="text/css"]zlink[href="blah.css"]zlink[href="no-blah.css"]z[rel="stylesheet"]z[type="text/css"]z[href="blah.css"]z[href="no-blah.css"]zp[href="no-blah.css"]r[ r r r r Útest_attribute_equalsÚ s òz&TestCSSSelectors.test_attribute_equalsc C s\ | ddgfddgfddgfddgfddgfddgfdd gfd
|
d gfd d gfd d gf¡
|
d S ) Nzp[class~="class1"]rh zp[class~="class2"]zp[class~="class3"]z[class~="class1"]z[class~="class2"]z[class~="class3"]za[rel~="friend"]Úbobz a[rel~="met"]z[rel~="friend"]z [rel~="met"]r[ r r r r Útest_attribute_tildeì s öz%TestCSSSelectors.test_attribute_tildec C sv | ddgfddgfdg fdg fdg fddgfdd d
|
gfd d d
|
gfd d dgfdd
|
dgfddgfdd
|
gfddgf¡ d S )Nz[rel^="style"]rq zlink[rel^="style"]znotlink[rel^="notstyle"]z[rel^="notstyle"]zlink[rel^="notstyle"]zlink[href^="bla"]za[href^="http://"]rs Úmez[href^="http://"]z [id^="p"]rh rp z [id^="m"]r) z div[id^="m"]z
|
a[id^="m"]zdiv[data-tag^="dashed"]r: r[ r r r r Útest_attribute_startswithú s
|
|
|
|
óz*TestCSSSelectors.test_attribute_startswithc C sH | ddgfddgfddgfdddddd d
|
d d gfd dgfdg f¡ d S )Nz[href$=".css"]rq zlink[href$=".css"]z link[id$="1"]z [id$="1"]r: rp rW rk Ús2a1rm rN z div[id$="1"]z[id$="noending"]r[ r r r r Útest_attribute_endswith s úz(TestCSSSelectors.test_attribute_endswithc C sÐ | ddgfddgfdg fdg fdg fddgfdd d
|
gfd d d gfddgfdd
|
gfddgfddgfddgfdddd ddddddg fd dgfd g fd d d
|
dgfd d d
|
gfd dgfd!dd"gfd#d"gfd$dgf¡ d S )%Nz[rel*="style"]rq zlink[rel*="style"]znotlink[rel*="notstyle"]z[rel*="notstyle"]zlink[rel*="notstyle"]zlink[href*="bla"]z[href*="http://"]rs ru z [id*="p"]rh rp z div[id*="m"]r) z
|
a[id*="m"]z[href*=".css"]zlink[href*=".css"]z link[id*="1"]z [id*="1"]r: rW rk rl rw rm rN z div[id*="1"]z[id*="noending"]z [href*="."]z a[href*="."]zlink[href*="."]z div[id*="n"]r9 z div[id*="nn"]zdiv[data-tag*="edval"]r[ r r r r Útest_attribute_contains s0
|
|
|
|
çz(TestCSSSelectors.test_attribute_containsc C s2 | ddddgfddddgfddgfdg f¡ d S ) Nz p[lang|="en"]úlang-enú
|
lang-en-gbú
|
lang-en-usz [lang|="en"]z p[lang|="fr"]úlang-frz p[lang|="gb"]r[ r r r r Ú test_attribute_exact_or_hypen2 s üz.TestCSSSelectors.test_attribute_exact_or_hypenc C sV | ddddgfddgfdddgfddd d
|
d gfd d dgfdg fdg fddgf¡ d S )Nz[rel]rq rs ru z link[rel]za[rel]z[lang]rz r{ r| r} zp[class]rp rh z[blah]zp[blah]z div[data-tag]r: r[ r r r r Útest_attribute_exists: s
|
|
øz&TestCSSSelectors.test_attribute_existsc C s, d}t |d}| d¡\}d|jks(td S )Nz]<div style="display: wrong">nope</div>
|
<div style="display: right">yes</div>
|
r zdiv[style="display: right"]Úyes)r r Ústringr )r Úhtmlr Zchosenr r r Ú"test_quoted_space_in_selector_nameF s
|
z3TestCSSSelectors.test_quoted_space_in_selector_namec C sH t t¡ | j d¡ W 5 Q R X t t¡ | j d¡ W 5 Q R X d S )Nza:no-such-pseudoclassza:nth-of-type(a))rJ rK ÚNotImplementedErrorr r r r r r r Ú test_unsupported_pseudoclassN s z-TestCSSSelectors.test_unsupported_pseudoclassc C s | j d¡}t|dks t|d jdks.t| j d¡}t|dksJt|d jdks\t| j d¡}t|dksxt| j d¡}t|dkstd S ) Nzdiv#inner p:nth-of-type(1)r r ú Some textzdiv#inner p:nth-of-type(3)ZAnotherzdiv#inner p:nth-of-type(4)zdiv p:nth-of-type(0)©r r r, r r r4 r r r Útest_nth_of_typeU s z!TestCSSSelectors.test_nth_of_typec C s2 | j d¡}t|dks t|d jdks.td S )Nz div#inner > p:nth-of-type(1)r r r r r4 r r r Ú"test_nth_of_type_direct_descendanth s z3TestCSSSelectors.test_nth_of_type_direct_descendantc C s | ddg¡ d S )Nz#inner > p:nth-of-type(2)rp r# r r r r Ú"test_id_child_selector_nth_of_typem s z3TestCSSSelectors.test_id_child_selector_nth_of_typec C s. | j jddd}| d¡}| |ddg¡ d S )Nr' r) r r9 r: )r Úfindr Zassert_selects_ids)r r9 Úselectedr r r Útest_select_on_elementp s
|
z'TestCSSSelectors.test_select_on_elementc C s | ddg¡ | dg ¡ d S )Nz .fancy #innerr9 z.normal #innerr# r r r r Útest_overspecified_child_idy s z,TestCSSSelectors.test_overspecified_child_idc C sB | ddg¡ | ddg¡ | ddg¡ g | j d¡ks>td S )Nz#p1 + h2rY z #p1 + h2 + prh z#p1 + #header2 + .class1z#p1 + p©r" r r r r r r r Ú test_adjacent_sibling_selector} s z/TestCSSSelectors.test_adjacent_sibling_selectorc C sR | dddg¡ | ddg¡ | ddg¡ | ddg¡ g | j d¡ksNtd S ) Nz#p1 ~ h2rY rZ z#p1 ~ #header2z #p1 ~ h2 + aru z#p1 ~ h2 + [rel="me"]z #inner ~ h2r r r r r Ú test_general_sibling_selector s
|
z.TestCSSSelectors.test_general_sibling_selectorc C s& t t¡ | j d¡ W 5 Q R X d S )Nzh1 >rI r r r r Útest_dangling_combinator s z)TestCSSSelectors.test_dangling_combinatorc C s | ddddg¡ d S )Nz p[lang] ~ pr{ r| r} r# r r r r Ú2test_sibling_combinator_wont_select_same_tag_twice s zCTestCSSSelectors.test_sibling_combinator_wont_select_same_tag_twicec C s | dddg¡ d S )Nzx, yÚxidÚyidr# r r r r Útest_multiple_select s z%TestCSSSelectors.test_multiple_selectc C s | dddg¡ d S )Nzx,yr r r# r r r r Ú"test_multiple_select_with_no_space s z3TestCSSSelectors.test_multiple_select_with_no_spacec C s | dddg¡ d S )Nzx, yr r r# r r r r Ú$test_multiple_select_with_more_space s z5TestCSSSelectors.test_multiple_select_with_more_spacec C s | ddg¡ d S )Nzx, xr r# r r r r Ú test_multiple_select_duplicated s z0TestCSSSelectors.test_multiple_select_duplicatedc C s | dddg¡ d S )Nzx, y ~ p[lang=fr]r r} r# r r r r Ú test_multiple_select_sibling s z-TestCSSSelectors.test_multiple_select_siblingc C s | dddg¡ d S )Nzx, y > zr Úzidbr# r r r r Ú.test_multiple_select_tag_and_direct_descendant¡ s z?TestCSSSelectors.test_multiple_select_tag_and_direct_descendantc C s | dddddddg¡ d S )Nz div > x, y, zr r Úzidar ÚzidabÚzidacr# r r r r Ú/test_multiple_select_direct_descendant_and_tags¤ s z@TestCSSSelectors.test_multiple_select_direct_descendant_and_tagsc C s | dddddddg¡ d S )Nz div x,y, zr r r r r r r# r r r r Ú(test_multiple_select_indirect_descendant§ s z9TestCSSSelectors.test_multiple_select_indirect_descendantc C sH t t¡ | j d¡ W 5 Q R X t t¡ | j d¡ W 5 Q R X d S )Nz,x, yzx,,yrI r r r r Ú test_invalid_multiple_selectª s z-TestCSSSelectors.test_invalid_multiple_selectc C s | dddg¡ d S )Nzp[lang=en], p[lang=en-gb]rz r{ r# r r r r Útest_multiple_select_attrs° s z+TestCSSSelectors.test_multiple_select_attrsc C s | ddddg¡ d S )Nz*x, y > z[id=zida], z[id=zidab], z[id=zidb]r r r r# r r r r Útest_multiple_select_ids³ s z)TestCSSSelectors.test_multiple_select_idsc C s | dddg¡ d S )Nzbody > div > x, y > zr r r# r r r r Útest_multiple_select_nested¶ s z,TestCSSSelectors.test_multiple_select_nestedc C sN d}t |d}| d¡}dt|ks(t|jddgdD ]}||ks8tq8d S )Nz3<div class="c1"/><div class="c2"/><div class="c1"/>r z.c1, .c2re Zc1Úc2)Úclass_)r r r, r rT )r Zmarkupr r Úelementr r r Ú test_select_duplicate_elements¹ s
|
|
z/TestCSSSelectors.test_select_duplicate_elementsc C s8 | j jddd}|j d¡}|| j jdddks4td S )Nr' r9 r ú div[id=main]r) )r r r* Úclosestr )r r9 r« r r r Ú test_closestÇ s z TestCSSSelectors.test_closestc C sL | j jddd}| j jddd}|j d¡dks4t|j d¡dksHtd S )Nr' r9 r r) rª FT)r r r* r7 r )r r9 r) r r r Ú
|
test_matchÌ s zTestCSSSelectors.test_matchc C sJ | j j d¡}t|tjs t|\}}|d dks6t|d dksFtd S )NrX r rY rZ )r r* Ziselectr ÚtypesÚ GeneratorTyper )r ÚgenrY rZ r r r Ú test_iselectÒ s
|
z TestCSSSelectors.test_iselectc C sh | j jddd}|j d¡}t|j d¡dks4t|j d¡}t|tsNt|\}|d dksdtd S ) Nr' r9 r rX rC zh2[id=header3]r rZ )r r r* Úfilterr, r r r )r r9 r! Úresultr r r Ú test_filterÙ s z TestCSSSelectors.test_filterc C sH | j jj}|ddkst|ddks*t|d| j j d¡ksDtd S )Nz.foo#barz
|
\.foo\#barz()[]{}z \(\)\[\]\{\}z.foo)r r* Úescaper )r Úmr r r Ú test_escapeã s
|
z TestCSSSelectors.test_escapeN)EÚ__name__Ú
|
__module__Ú __qualname__Ú__doc__r r r" Z assertSelectr% r1 r5 r6 r8 r; rA rE rH rL rP rQ rS rU r\ r` rb rc rd rf rg ri rj rn ro rr rt rv rx ry r~ r r r
r r r r r r r r r r r r r r r r r¡ r¢ r£ r¤ r¥ r© r¬ r r± r´ r· r r r r r s 1
|
|
|
r )rJ r® Z unittest.mockr Zbs4r r r Ú r r Z soupsiever ÚmarkZskipifr r r r r Ú<module> s
|
