U lu fC@svddlZddlZddlmZddlmZmZmZddlm Z m Z e rPddl 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)rHTMLsoupselfr6/tmp/pip-unpacked-wheel-pg4vfqbd/bs4/tests/test_css.py setup_methodNszTestCSSSelectors.setup_methodcKsf|jj|f|}t|tstdd|D}||||ksbtd|d|d|fdS)NcSsg|] }|dqSidr).0elrrr Tsz3TestCSSSelectors.assert_selects..z$Selector %s, expected [%s], got [%s]z, )rselect isinstancerAssertionErrorsortjoin)rselector expected_idskwargsresultsZel_idsrrrassert_selectsQs zTestCSSSelectors.assert_selectscGs|D]\}}|||qdS)Nr")rtestsrrrrrassert_select_multiple]s z'TestCSSSelectors.assert_select_multiplecCsb|jjd}|j|}t|dks*t|D]}|jdks.tq.|j|}d|dks^tdSNdivmainr)rcsscompilerlenrname select_one)rselelsr'rrrrtest_precompiledas  z!TestCSSSelectors.test_precompiledcCsF|jd}t|dkst|djdks.t|djdgksBtdS)Ntitlerrz The title)rrr,rr-contentsrr0rrrtest_one_tag_onels z!TestCSSSelectors.test_one_tag_onecCsT|jd}t|dkst|D]}|jdks tq |jd}d|dksPtdSr&)rrr,rr-r.)rr0r'rrrrtest_one_tag_manyrs   z"TestCSSSelectors.test_one_tag_manycCs|jd}d|kstdS)NZnonexistenttag)rr.r)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_ones z$TestCSSSelectors.test_tag_in_tag_onecCs"dD]}||ddddgqdS)N)html div html body divbody divr:r)r9footerr#rrrrrtest_tag_in_tag_manysz%TestCSSSelectors.test_tag_in_tag_manycCsB|jddgdd|jdddgdd|jdd ddd gd ddS) Nr<r)r)limitr=r9r>r:r? r#rrrr test_limits zTestCSSSelectors.test_limitcCst|jddkstdS)Ndelrr,rrrrrrrtest_tag_no_matchsz"TestCSSSelectors.test_tag_no_matchc Cs&tt|jdW5QRXdS)Nztag%tpytestraisesr rrrrrrtest_invalid_tags z!TestCSSSelectors.test_invalid_tagcCs|dddgdS)Ncustom-dashed-tagdash1dash2r#rrrrtest_select_dashed_tag_idssz+TestCSSSelectors.test_select_dashed_tag_idscCs6|jd}|djdkst|dddks2tdS)Nzcustom-dashed-tag[id="dash2"]rrMrrO)rrr-r)rZdashedrrrtest_select_dashed_by_ids z)TestCSSSelectors.test_select_dashed_by_idcCs|jddjdkstdS)Nzbody > custom-dashed-tagrz Hello there.)rrtextrrrrrtest_dashed_tag_textsz%TestCSSSelectors.test_dashed_tag_textcCs |jd|jdkstdS)NrM)rrfind_allrrrrr#test_select_dashed_matches_find_allsz4TestCSSSelectors.test_select_dashed_matches_find_allcCs|ddgfdddgfdS)Nh1header1h2header2header3r%rrrrtest_header_tagss z!TestCSSSelectors.test_header_tagscCsRdD]H}|j|}t|dks$t|djdks6t|dddgkstqdS)N)z.onepzp.onepz html p.oneprrpclassonep)rrr,rr-)rrr0rrrtest_class_ones  zTestCSSSelectors.test_class_onecCs |jd}t|dkstdS)Nzdiv.oneprrrr,rr4rrrtest_class_mismatched_tags z*TestCSSSelectors.test_class_mismatched_tagcCsdD]}||dgqdS)N)z div#innerz#innerz div div#innerr9r#r@rrr test_one_idszTestCSSSelectors.test_one_idcCs |jd}t|dkstdS)Nz #doesnotexistrrar4rrr test_bad_ids zTestCSSSelectors.test_bad_idcCs`|jd}t|dkst|D]}|jdks tq |dddgksJt|ddr\tdS)Nz div#inner pr]rr^r_r)rrr,rr-Zhas_attr)rr0rrrrtest_items_in_ids  z!TestCSSSelectors.test_items_in_idcCs&dD]}t|j|dkstqdS)N)z div#main delzdiv#main div.oopsz div div#mainrrGr@rrrtest_a_bunch_of_emptyssz'TestCSSSelectors.test_a_bunch_of_emptyscCsdD]}||dgqdS)N)z.class1zp.class1z.class2zp.class2z.class3zp.class3z html p.class2zdiv#inner .class2pmultir#r@rrrtest_multi_class_supportsz)TestCSSSelectors.test_multi_class_supportcCsdD]}||dgqdS)N)z.class1.class3z.class3.class2z.class1.class2.class3rhr#r@rrrtest_multi_class_selectionsz+TestCSSSelectors.test_multi_class_selectioncCs"|dddg|ddgdS)Nz.s1 > as1a1s1a2z .s1 > a spans1a2s1r#rrrrtest_child_selectorsz$TestCSSSelectors.test_child_selectorcCs|ddgdS)Nz.s1 > a#s1a2 spanrmr#rrrrtest_child_selector_idsz'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"]rhzp[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"]rqzlink[rel^="style"]znotlink[rel^="notstyle"]z[rel^="notstyle"]zlink[rel^="notstyle"]zlink[href^="bla"]za[href^="http://"]rsmez[href^="http://"]z [id^="p"]rhrpz [id^="m"]r)z div[id^="m"]z a[id^="m"]zdiv[data-tag^="dashed"]r:r[rrrrtest_attribute_startswiths    z*TestCSSSelectors.test_attribute_startswithcCsH|ddgfddgfddgfdddddd d d d gfd dgfdgfdS)Nz[href$=".css"]rqzlink[href$=".css"]z link[id$="1"]z [id$="1"]r:rprWrks2a1rmrNz div[id$="1"]z[id$="noending"]r[rrrrtest_attribute_endswith sz(TestCSSSelectors.test_attribute_endswithcCs|ddgfddgfdgfdgfdgfddgfdd d gfd d d gfddgfdd gfddgfddgfddgfdddd ddddddg fddgfdgfdd d dgfdd d gfd dgfd!dd"gfd#d"gfd$dgfdS)%Nz[rel*="style"]rqzlink[rel*="style"]znotlink[rel*="notstyle"]z[rel*="notstyle"]zlink[rel*="notstyle"]zlink[href*="bla"]z[href*="http://"]rsruz [id*="p"]rhrpz div[id*="m"]r)z a[id*="m"]z[href*=".css"]zlink[href*=".css"]z link[id*="1"]z [id*="1"]r:rWrkrlrwrmrNz div[id*="1"]z[id*="noending"]z [href*="."]z a[href*="."]zlink[href*="."]z div[id*="n"]r9z div[id*="nn"]zdiv[data-tag*="edval"]r[rrrrtest_attribute_containss0     z(TestCSSSelectors.test_attribute_containscCs2|ddddgfddddgfddgfdgfdS) 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 CsV|ddddgfddgfdddgfddd d d gfd d dgfdgfdgfddgfdS)Nz[rel]rqrsruz link[rel]za[rel]z[lang]rzr{r|r}zp[class]rprhz[blah]zp[blah]z div[data-tag]r:r[rrrrtest_attribute_exists:s   z&TestCSSSelectors.test_attribute_existscCs,d}t|d}|d\}d|jks(tdS)Nz]
nope
yes
r zdiv[style="display: right"]yes)rrstringr)rhtmlrZchosenrrr"test_quoted_space_in_selector_nameFs  z3TestCSSSelectors.test_quoted_space_in_selector_namec CsHtt|jdW5QRXtt|jdW5QRXdS)Nza:no-such-pseudoclassza:nth-of-type(a))rJrKNotImplementedErrorrrr rrrrtest_unsupported_pseudoclassNs  z-TestCSSSelectors.test_unsupported_pseudoclasscCs|jd}t|dkst|djdks.t|jd}t|dksJt|djdks\t|jd}t|dksxt|jd}t|dkstdS) Nzdiv#inner p:nth-of-type(1)rr Some textzdiv#inner p:nth-of-type(3)ZAnotherzdiv#inner p:nth-of-type(4)zdiv p:nth-of-type(0)rrr,rrr4rrrtest_nth_of_typeUs    z!TestCSSSelectors.test_nth_of_typecCs2|jd}t|dkst|djdks.tdS)Nzdiv#inner > p:nth-of-type(1)rrrrr4rrr"test_nth_of_type_direct_descendanths z3TestCSSSelectors.test_nth_of_type_direct_descendantcCs|ddgdS)Nz#inner > p:nth-of-type(2)rpr#rrrr"test_id_child_selector_nth_of_typemsz3TestCSSSelectors.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|jdks>tdS)Nz#p1 + h2rYz #p1 + h2 + prhz#p1 + #header2 + .class1z#p1 + pr"rrrrrrrtest_adjacent_sibling_selector}sz/TestCSSSelectors.test_adjacent_sibling_selectorcCsR|dddg|ddg|ddg|ddgg|jdksNtdS) Nz#p1 ~ h2rYrZz#p1 ~ #header2z #p1 ~ h2 + aruz#p1 ~ h2 + [rel="me"]z #inner ~ h2rrrrrtest_general_sibling_selectors z.TestCSSSelectors.test_general_sibling_selectorc Cs&tt|jdW5QRXdS)Nzh1 >rIrrrrtest_dangling_combinators z)TestCSSSelectors.test_dangling_combinatorcCs|ddddgdS)Nz p[lang] ~ pr{r|r}r#rrrr2test_sibling_combinator_wont_select_same_tag_twiceszCTestCSSSelectors.test_sibling_combinator_wont_select_same_tag_twicecCs|dddgdS)Nzx, yxidyidr#rrrrtest_multiple_selectsz%TestCSSSelectors.test_multiple_selectcCs|dddgdS)Nzx,yrrr#rrrr"test_multiple_select_with_no_spacesz3TestCSSSelectors.test_multiple_select_with_no_spacecCs|dddgdS)Nzx, yrrr#rrrr$test_multiple_select_with_more_spacesz5TestCSSSelectors.test_multiple_select_with_more_spacecCs|ddgdS)Nzx, xrr#rrrrtest_multiple_select_duplicatedsz0TestCSSSelectors.test_multiple_select_duplicatedcCs|dddgdS)Nzx, y ~ p[lang=fr]rr}r#rrrrtest_multiple_select_siblingsz-TestCSSSelectors.test_multiple_select_siblingcCs|dddgdS)Nzx, y > zrzidbr#rrrr.test_multiple_select_tag_and_direct_descendantsz?TestCSSSelectors.test_multiple_select_tag_and_direct_descendantc Cs|dddddddgdS)Nz div > x, y, zrrzidarzidabzidacr#rrrr/test_multiple_select_direct_descendant_and_tagssz@TestCSSSelectors.test_multiple_select_direct_descendant_and_tagsc Cs|dddddddgdS)Nz div x,y, zrrrrrrr#rrrr(test_multiple_select_indirect_descendantsz9TestCSSSelectors.test_multiple_select_indirect_descendantc CsHtt|jdW5QRXtt|jdW5QRXdS)Nz,x, yzx,,yrIrrrrtest_invalid_multiple_selects  z-TestCSSSelectors.test_invalid_multiple_selectcCs|dddgdS)Nzp[lang=en], p[lang=en-gb]rzr{r#rrrrtest_multiple_select_attrssz+TestCSSSelectors.test_multiple_select_attrscCs|ddddgdS)Nz*x, y > z[id=zida], z[id=zidab], z[id=zidb]rrrr#rrrrtest_multiple_select_idssz)TestCSSSelectors.test_multiple_select_idscCs|dddgdS)Nzbody > div > x, y > zrrr#rrrrtest_multiple_select_nestedsz,TestCSSSelectors.test_multiple_select_nestedcCsNd}t|d}|d}dt|ks(t|jddgdD]}||ks8tq8dS)Nz3
r z.c1, .c2reZc1c2)class_)rrr,rrT)rZmarkuprrelementrrrtest_select_duplicate_elementss   z/TestCSSSelectors.test_select_duplicate_elementscCs8|jjddd}|jd}||jjdddks4tdS)Nr'r9r div[id=main]r))rrr*closestr)rr9rrrr test_closests zTestCSSSelectors.test_closestcCsL|jjddd}|jjddd}|jddks4t|jddksHtdS)Nr'r9rr)rFT)rrr*r7r)rr9r)rrr test_matchszTestCSSSelectors.test_matchcCsJ|jjd}t|tjst|\}}|ddks6t|ddksFtdS)NrXrrYrZ)rr*Ziselectrtypes GeneratorTyper)rgenrYrZrrr test_iselects zTestCSSSelectors.test_iselectcCsh|jjddd}|jd}t|jddks4t|jd}t|tsNt|\}|ddksdtdS) Nr'r9rrXrCzh2[id=header3]rrZ)rrr*filterr,rrr)rr9r!resultrrr test_filters  zTestCSSSelectors.test_filtercCsH|jjj}|ddkst|ddks*t|d|jjdksDtdS)Nz.foo#barz \.foo\#barz()[]{}z \(\)\[\]\{\}z.foo)rr*escaper)rmrrr test_escapes zTestCSSSelectors.test_escapeN)E__name__ __module__ __qualname____doc__r rr"Z assertSelectr%r1r5r6r8r;rArErHrLrPrQrSrUr\r`rbrcrdrfrgrirjrnrorrrtrvrxryr~rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr s1       r )rJrZ unittest.mockrZbs4rrrrrZ soupsiever markZskipifr rrrrs