o tfC@svddlZddlZddlmZddlmZmZmZddlm Z m Z e r(ddl m Z ej je ddGd d d e ZdS) N) MagicMock)CSS BeautifulSoup ResultSet)SoupTestSOUP_SIEVE_PRESENT)SelectorSyntaxErrorzSoup Sieve not installed)reasonc@seZdZdZdZddZddZeZddZd d Z d d Z d dZ ddZ ddZ ddZddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7d8Z d9d:Z!d;d<Z"d=d>Z#d?d@Z$dAdBZ%dCdDZ&dEdFZ'dGdHZ(dIdJZ)dKdLZ*dMdNZ+dOdPZ,dQdRZ-dSdTZ.dUdVZ/dWdXZ0dYdZZ1d[d\Z2d]d^Z3d_d`Z4dadbZ5dcddZ6dedfZ7dgdhZ8didjZ9dkdlZ:dmdnZ;dodpZdudvZ?dwdxZ@dydzZAd{d|ZBd}d~ZCddZDdS)TestCSSSelectorszTest 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 The title Hello there.

An H1

Some text

Some more text

An H2

Another

Bob

Another H2

me span1a1 span1a2 test span2a1

English

English UK

English US

French

cCst|jd|_dS)N html.parser)rHTMLsoupselfr[/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/bs4/tests/test_css.py setup_methodNzTestCSSSelectors.setup_methodcKsj|jj|fi|}t|tsJdd|D}||||ks3Jd|d|d|fdS)NcSsg|]}|dqSidr).0elrrr Tsz3TestCSSSelectors.assert_selects..z$Selector %s, expected [%s], got [%s]z, )rselect isinstancersortjoin)rselector expected_idskwargsresultsZel_idsrrrassert_selectsQs  zTestCSSSelectors.assert_selectscGs|D] \}}|||qdS)Nr")rtestsrrrrrassert_select_multiple]s z'TestCSSSelectors.assert_select_multiplecCsb|jjd}|j|}t|dksJ|D] }|jdks Jq|j|}d|dks/JdSNdivmainr)rcsscompilerlenname select_one)rselelsr'rrrrtest_precompiledas  z!TestCSSSelectors.test_precompiledcCsF|jd}t|dksJ|djdksJ|djdgks!JdS)Ntitlerrz The title)rrr,r-contentsrr0rrrtest_one_tag_onels z!TestCSSSelectors.test_one_tag_onecCsT|jd}t|dksJ|D] }|jdksJq|jd}d|dks(JdSr&)rrr,r-r.)rr0r'rrrrtest_one_tag_manyrs  z"TestCSSSelectors.test_one_tag_manycCs|jd}d|ks JdS)NZnonexistenttag)rr.)rmatchrrr(test_select_one_returns_none_if_no_match{s z9TestCSSSelectors.test_select_one_returns_none_if_no_matchcCs |jd}|dddgdS)Nzdiv divinnerdata1)rrr"r4rrrtest_tag_in_tag_one z$TestCSSSelectors.test_tag_in_tag_onecCsdD] }||gdqdS)N)html div html body divbody divr:r)r9footerr#rrrrrtest_tag_in_tag_manysz%TestCSSSelectors.test_tag_in_tag_manycCs>|jddgdd|jdddgdd|jdgd d ddS) Nr=r)r)limitr>r9r?r@ r#rrrr test_limits   zTestCSSSelectors.test_limitcCst|jddks JdS)Ndelrr,rrrrrrtest_tag_no_matchsz"TestCSSSelectors.test_tag_no_matchcC<tt|jdWddS1swYdS)Nztag%tpytestraisesr rrrrrrtest_invalid_tag "z!TestCSSSelectors.test_invalid_tagcC|dddgdS)Ncustom-dashed-tagdash1dash2r#rrrrtest_select_dashed_tag_idsz+TestCSSSelectors.test_select_dashed_tag_idscCs6|jd}|djdksJ|dddksJdS)Nzcustom-dashed-tag[id="dash2"]rrRrrT)rrr-)rZdashedrrrtest_select_dashed_by_ids z)TestCSSSelectors.test_select_dashed_by_idcCs|jddjdks JdS)Nzbody > custom-dashed-tagrz Hello there.)rrtextrrrrtest_dashed_tag_textsz%TestCSSSelectors.test_dashed_tag_textcCs |jd|jdksJdS)NrR)rrfind_allrrrr#test_select_dashed_matches_find_alls z4TestCSSSelectors.test_select_dashed_matches_find_allcCs|ddgfdddgfdS)NZh1header1h2header2header3r%rrrrtest_header_tagss z!TestCSSSelectors.test_header_tagscCsRdD]$}|j|}t|dksJ|djdksJ|dddgks&JqdS)N)z.onepzp.onepz html p.oneprrpclassonep)rrr,r-)rrr0rrrtest_class_ones  zTestCSSSelectors.test_class_onecC |jd}t|dksJdS)Nzdiv.oneprrrr,r4rrrtest_class_mismatched_tagr<z*TestCSSSelectors.test_class_mismatched_tagcCdD] }||dgqdS)N)z div#innerz#innerz div div#innerr9r#rBrrr test_one_idszTestCSSSelectors.test_one_idcCrf)Nz #doesnotexistrrgr4rrr test_bad_idr<zTestCSSSelectors.test_bad_idcCs`|jd}t|dksJ|D] }|jdksJq|dddgks%J|ddr.JdS)Nz div#inner prbrrcrdr)rrr,r-Zhas_attr)rr0rrrrtest_items_in_ids z!TestCSSSelectors.test_items_in_idcCs&dD]}t|j|dksJqdS)N)z div#main delzdiv#main div.oopsz div div#mainrrIrBrrrtest_a_bunch_of_emptyssz'TestCSSSelectors.test_a_bunch_of_emptyscCri)N)z.class1zp.class1z.class2zp.class2z.class3zp.class3z html p.class2zdiv#inner .class2pmultir#rBrrrtest_multi_class_supportz)TestCSSSelectors.test_multi_class_supportcCri)N)z.class1.class3z.class3.class2z.class1.class2.class3ror#rBrrrtest_multi_class_selectionrqz+TestCSSSelectors.test_multi_class_selectioncCs"|dddg|ddgdS)Nz.s1 > as1a1s1a2z .s1 > a spans1a2s1r#rrrrtest_child_selectorsz$TestCSSSelectors.test_child_selectorcC|ddgdS)Nz.s1 > a#s1a2 spanrur#rrrrtest_child_selector_idrz'TestCSSSelectors.test_child_selector_idcCst|ddgfddgfddgfddgfddgfddgfd dgfd gfd dgfd dgfd dgfdgfdgfdgfdS)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`rrrrtest_attribute_equalss z&TestCSSSelectors.test_attribute_equalsc Cs\|ddgfddgfddgfddgfddgfddgfdd gfd d gfd d gfd d gf dS) Nzp[class~="class1"]rozp[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`rrrrtest_attribute_tildesz%TestCSSSelectors.test_attribute_tildecCsv|ddgfddgfdgfdgfdgfddgfdd d gfd d d gfd d dgfdd dgfddgfdd gfddgf dS)Nz[rel^="style"]rzzlink[rel^="style"]znotlink[rel^="notstyle"]z[rel^="notstyle"]zlink[rel^="notstyle"]zlink[href^="bla"]za[href^="http://"]r|mez[href^="http://"]z [id^="p"]roryz [id^="m"]r)z div[id^="m"]z a[id^="m"]zdiv[data-tag^="dashed"]r:r`rrrrtest_attribute_startswiths    z*TestCSSSelectors.test_attribute_startswithc Cs<|ddgfddgfddgfdgdfddgfd gfdS) Nz[href$=".css"]rzzlink[href$=".css"]z link[id$="1"]z [id$="1"])r:rzryr\rss2a1rurSz div[id$="1"]r:z[id$="noending"]r`rrrrtest_attribute_endswith s z(TestCSSSelectors.test_attribute_endswithcCs|ddgfddgfdgfdgfdgfddgfdd d gfd d d gfddgfdd gfddgfddgfddgfdgdfddgfdgfdgdfdd d gfddgfdddgfddgfd dgfdS)!Nz[rel*="style"]rzzlink[rel*="style"]znotlink[rel*="notstyle"]z[rel*="notstyle"]zlink[rel*="notstyle"]zlink[href*="bla"]z[href*="http://"]r|r~z [id*="p"]roryz div[id*="m"]r)z a[id*="m"]z[href*=".css"]zlink[href*=".css"]z link[id*="1"]z [id*="1"]) r:rzryr\rsrtrrurSz div[id*="1"]r:z[id*="noending"]z [href*="."])r|r~rzz a[href*="."]zlink[href*="."]z div[id*="n"]r9z div[id*="nn"]zdiv[data-tag*="edval"]r`rrrrtest_attribute_containss0      z(TestCSSSelectors.test_attribute_containscCs.|dgdfdgdfddgfdgfdS)Nz p[lang|="en"])lang-en lang-en-gb lang-en-usz [lang|="en"]z p[lang|="fr"]lang-frz p[lang|="gb"]r`rrrrtest_attribute_exact_or_hypen2s   z.TestCSSSelectors.test_attribute_exact_or_hypenc CsP|dgdfddgfdddgfdgd fd d d gfd gfdgfddgfdS)Nz[rel])rzr|r~z link[rel]rzza[rel]r|r~z[lang])rrrrzp[class]ryroz[blah]zp[blah]z div[data-tag]r:r`rrrrtest_attribute_exists:s    z&TestCSSSelectors.test_attribute_existscCs,d}t|d}|d\}d|jksJdS)Nz]
nope
yes
r zdiv[style="display: right"]yes)rrstring)rhtmlrZchosenrrr"test_quoted_space_in_selector_nameFs  z3TestCSSSelectors.test_quoted_space_in_selector_namecCsrtt|jdWdn1swYtt|jdWddS1s2wYdS)Nza:no-such-pseudoclassza:nth-of-type(a))rMrNNotImplementedErrorrrr rrrrtest_unsupported_pseudoclassNs  "z-TestCSSSelectors.test_unsupported_pseudoclasscCs|jd}t|dksJ|djdksJ|jd}t|dks%J|djdks.J|jd}t|dks p:nth-of-type(1)rrrrr4rrr"test_nth_of_type_direct_descendanths z3TestCSSSelectors.test_nth_of_type_direct_descendantcCrw)Nz#inner > p:nth-of-type(2)ryr#rrrr"test_id_child_selector_nth_of_typemrz3TestCSSSelectors.test_id_child_selector_nth_of_typecCs.|jjddd}|d}||ddgdS)Nr'r)rr9r:)rfindrZassert_selects_ids)rr9selectedrrrtest_select_on_elementps z'TestCSSSelectors.test_select_on_elementcCs|ddg|dgdS)Nz .fancy #innerr9z.normal #innerr#rrrrtest_overspecified_child_idysz,TestCSSSelectors.test_overspecified_child_idcCsB|ddg|ddg|ddgg|jdksJdS)Nz#p1 + h2r^z #p1 + h2 + proz#p1 + #header2 + .class1z#p1 + pr"rrrrrrtest_adjacent_sibling_selector}sz/TestCSSSelectors.test_adjacent_sibling_selectorcCsR|dddg|ddg|ddg|ddgg|jdks'JdS) Nz#p1 ~ h2r^r_z#p1 ~ #header2z #p1 ~ h2 + ar~z#p1 ~ h2 + [rel="me"]z #inner ~ h2rrrrrtest_general_sibling_selectors z.TestCSSSelectors.test_general_sibling_selectorcCrK)Nzh1 >rLrrrrtest_dangling_combinatorrPz)TestCSSSelectors.test_dangling_combinatorcC|dgddS)Nz p[lang] ~ p)rrrr#rrrr2test_sibling_combinator_wont_select_same_tag_twicerVzCTestCSSSelectors.test_sibling_combinator_wont_select_same_tag_twicecCrQ)Nzx, yxidyidr#rrrrtest_multiple_selectrVz%TestCSSSelectors.test_multiple_selectcCrQ)Nzx,yrrr#rrrr"test_multiple_select_with_no_spacerVz3TestCSSSelectors.test_multiple_select_with_no_spacecCrQ)Nzx, yrrr#rrrr$test_multiple_select_with_more_spacerVz5TestCSSSelectors.test_multiple_select_with_more_spacecCrw)Nzx, xrr#rrrrtest_multiple_select_duplicatedrz0TestCSSSelectors.test_multiple_select_duplicatedcCrQ)Nzx, y ~ p[lang=fr]rrr#rrrrtest_multiple_select_siblingrVz-TestCSSSelectors.test_multiple_select_siblingcCrQ)Nzx, y > zrzidbr#rrrr.test_multiple_select_tag_and_direct_descendantrVz?TestCSSSelectors.test_multiple_select_tag_and_direct_descendantcCr)Nz div > x, y, zrrZzidarzidabZzidacr#rrrr/test_multiple_select_direct_descendant_and_tagsrVz@TestCSSSelectors.test_multiple_select_direct_descendant_and_tagscCr)Nz div x,y, zrr#rrrr(test_multiple_select_indirect_descendantrVz9TestCSSSelectors.test_multiple_select_indirect_descendantcCsrtt|jdWdn1swYtt|jdWddS1s2wYdS)Nz,x, yzx,,yrLrrrrtest_invalid_multiple_selects  "z-TestCSSSelectors.test_invalid_multiple_selectcCrQ)Nzp[lang=en], p[lang=en-gb]rrr#rrrrtest_multiple_select_attrsrVz+TestCSSSelectors.test_multiple_select_attrscCr)Nz*x, y > z[id=zida], z[id=zidab], z[id=zidb])rrrr#rrrrtest_multiple_select_idsrVz)TestCSSSelectors.test_multiple_select_idscCrQ)Nzbody > div > x, y > zrrr#rrrrtest_multiple_select_nestedrVz,TestCSSSelectors.test_multiple_select_nestedcCsNd}t|d}|d}dt|ksJ|jddgdD]}||vs$JqdS)Nz3
r z.c1, .c2rlc1c2)class_)rrr,rZ)rmarkuprrelementrrrtest_select_duplicate_elementss  z/TestCSSSelectors.test_select_duplicate_elementscCs8|jjddd}|jd}||jjdddksJdS)Nr'r9r div[id=main]r))rrr*closest)rr9rrrr test_closests zTestCSSSelectors.test_closestcCsL|jjddd}|jjddd}|jddksJ|jddks$JdS)Nr'r9rr)rFT)rrr*r7)rr9r)rrr test_matchszTestCSSSelectors.test_matchcCsJ|jjd}t|tjsJ|\}}|ddksJ|ddks#JdS)Nr]rr^r_)rr*Ziselectrtypes GeneratorType)rgenr^r_rrr test_iselects zTestCSSSelectors.test_iselectcCsh|jjddd}|jd}t|jddksJ|jd}t|ts'J|\}|ddks2JdS) Nr'r9rr]rEzh2[id=header3]rr_)rrr*filterr,rr)rr9r!resultrrr test_filters  zTestCSSSelectors.test_filtercCsH|jjj}|ddks J|ddksJ|d|jjdks"JdS)Nz.foo#barz \.foo\#barz()[]{}z \(\)\[\]\{\}z.foo)rr*escape)rmrrr test_escapes zTestCSSSelectors.test_escapeN)E__name__ __module__ __qualname____doc__r rr"Z assertSelectr%r1r5r6r8r;rCrGrJrOrUrWrYr[rarerhrjrkrmrnrprrrvrxr{r}rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr s1       r )rMrZ unittest.mockrZbs4rrrrrZ soupsiever markZskipifr rrrrs