3 dC@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)Nz html.parser)rHTMLsoup)selfrTsz3TestCSSSelectors.assert_selects..z$Selector %s, expected [%s], got [%s]z, )r select isinstancerAssertionErrorsortjoin)rselector expected_idskwargsresultsZel_idsrrrassert_selectsQs zTestCSSSelectors.assert_selectscGs"x|D]\}}|j||qWdS)N)r)rtestsrrrrrassert_select_multiple]sz'TestCSSSelectors.assert_select_multiplecCsf|jjjd}|jj|}t|dks*tx|D]}|jdks0tq0W|jj|}d|dksbtdS)Ndivmainr)r csscompilerlenrname select_one)rselelsr"rrrrtest_precompiledas   z!TestCSSSelectors.test_precompiledcCsF|jjd}t|dkst|djdks.t|djdgksBtdS)Ntitlerrz The title)r rr'rr(contents)rr+rrrtest_one_tag_onels z!TestCSSSelectors.test_one_tag_onecCsX|jjd}t|dkstx|D]}|jdks"tq"W|jjd}d|dksTtdS)Nr"r#r$r)r rr'rr(r))rr+r"rrrrtest_one_tag_manyrs    z"TestCSSSelectors.test_one_tag_manycCs|jjd}d|kstdS)NZnonexistenttag)r r)r)rmatchrrr(test_select_one_returns_none_if_no_match{s z9TestCSSSelectors.test_select_one_returns_none_if_no_matchcCs |jjd}|jdddgdS)Nzdiv divinnerdata1)r rr)rr+rrrtest_tag_in_tag_ones z$TestCSSSelectors.test_tag_in_tag_onecCs&x dD]}|j|ddddgqWdS) Nhtml div html body divbody divr4r$r3footer)r6r7r8)r)rrrrrtest_tag_in_tag_manys z%TestCSSSelectors.test_tag_in_tag_manycCsB|jddgdd|jdddgdd|jdd ddd gd ddS) Nzhtml divr$r)limitz html body divr3zbody divr4r9 )r)rrrr test_limitszTestCSSSelectors.test_limitcCst|jjddkstdS)Ndelr)r'r rr)rrrrtest_tag_no_matchsz"TestCSSSelectors.test_tag_no_matchc Cs&tjt|jjdWdQRXdS)Nztag%t)pytestraisesr r r)rrrrtest_invalid_tags z!TestCSSSelectors.test_invalid_tagcCs|jdddgdS)Nzcustom-dashed-tagdash1dash2)r)rrrrtest_select_dashed_tag_idssz+TestCSSSelectors.test_select_dashed_tag_idscCs6|jjd}|djdkst|dddks2tdS)Nzcustom-dashed-tag[id="dash2"]rzcustom-dashed-tagrrE)r rr(r)rZdashedrrrtest_select_dashed_by_ids z)TestCSSSelectors.test_select_dashed_by_idcCs|jjddjdkstdS)Nzbody > custom-dashed-tagrz Hello there.)r rtextr)rrrrtest_dashed_tag_textsz%TestCSSSelectors.test_dashed_tag_textcCs |jjd|jjdkstdS)Nzcustom-dashed-tag)r rfind_allr)rrrr#test_select_dashed_matches_find_allsz4TestCSSSelectors.test_select_dashed_matches_find_allcCs|jddgfdddgfdS)Nh1header1h2header2header3)r!)rrrrtest_header_tagssz!TestCSSSelectors.test_header_tagscCsVxPd D]H}|jj|}t|dks&t|djdks8t|dddgkstqWdS) N.onepp.onep html p.oneprrpclassonep)rRrSrT)r rr'rr()rrr+rrrtest_class_ones   zTestCSSSelectors.test_class_onecCs |jjd}t|dkstdS)Nzdiv.onepr)r rr'r)rr+rrrtest_class_mismatched_tags z*TestCSSSelectors.test_class_mismatched_tagcCs xdD]}|j|dgqWdS)N div#inner#inner div div#innerr3)rZr[r\)r)rrrrr test_one_ids zTestCSSSelectors.test_one_idcCs |jjd}t|dkstdS)Nz #doesnotexistr)r rr'r)rr+rrr test_bad_ids zTestCSSSelectors.test_bad_idcCsf|jjd}t|dkstx|D]}|jdks"tq"W|dddgksNt|djd sbtdS)Nz div#inner prUrrVrWr)r rr'rr(Zhas_attr)rr+rrrrtest_items_in_ids   z!TestCSSSelectors.test_items_in_idcCs*x$dD]}t|jj|dkstqWdS)N div#main deldiv#main div.oops div div#mainr)rarbrc)r'r rr)rrrrrtest_a_bunch_of_emptyss z'TestCSSSelectors.test_a_bunch_of_emptyscCs xd D]}|j|d gqWdS) N.class1p.class1.class2p.class2.class3p.class3 html p.class2div#inner .class2pmulti)rerfrgrhrirjrkrl)r)rrrrrtest_multi_class_supportsz)TestCSSSelectors.test_multi_class_supportcCs xdD]}|j|dgqWdS)N.class1.class3.class3.class2.class1.class2.class3rm)rorprq)r)rrrrrtest_multi_class_selectionsz+TestCSSSelectors.test_multi_class_selectioncCs"|jdddg|jddgdS)Nz.s1 > as1a1s1a2z .s1 > a spans1a2s1)r)rrrrtest_child_selectorsz$TestCSSSelectors.test_child_selectorcCs|jddgdS)Nz.s1 > a#s1a2 spanru)r)rrrrtest_child_selector_idsz'TestCSSSelectors.test_child_selector_idcCst|jddgfddgfddgfddgfddgfddgfd 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_equalssz&TestCSSSelectors.test_attribute_equalsc Cs\|jddgfddgfddgfddgfddgfddgfdd gfd d gfd d gfd d gf dS) Nzp[class~="class1"]rmzp[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|jddgfddgfdgfdgfdgfddgfdd d gfd d d gfd d dgfdd dgfddgfdd gfddgf dS)Nz[rel^="style"]ryzlink[rel^="style"]znotlink[rel^="notstyle"]z[rel^="notstyle"]zlink[rel^="notstyle"]zlink[href^="bla"]za[href^="http://"]r{mez[href^="http://"]z [id^="p"]rmrxz [id^="m"]r$z div[id^="m"]z a[id^="m"]zdiv[data-tag^="dashed"]r4)r!)rrrrtest_attribute_startswiths    z*TestCSSSelectors.test_attribute_startswithc CsH|jddgfddgfddgfdddddd d d d gfd dgfdgfdS)Nz[href$=".css"]ryzlink[href$=".css"]z link[id$="1"]z [id$="1"]r4rxrMrss2a1rurDz div[id$="1"]z[id$="noending"])r!)rrrrtest_attribute_endswith sz(TestCSSSelectors.test_attribute_endswithcCs|jddgfddgfdgfdgfdgfddgfdd d gfd d d gfddgfdd gfddgfddgfddgfdddd ddddddg fddgfdgfdd d dgfdd d gfd dgfd!dd"gfd#d"gfd$dgfdS)%Nz[rel*="style"]ryzlink[rel*="style"]znotlink[rel*="notstyle"]z[rel*="notstyle"]zlink[rel*="notstyle"]zlink[href*="bla"]z[href*="http://"]r{r}z [id*="p"]rmrxz div[id*="m"]r$z a[id*="m"]z[href*=".css"]zlink[href*=".css"]z link[id*="1"]z [id*="1"]r4rMrsrtrrurDz div[id*="1"]z[id*="noending"]z [href*="."]z a[href*="."]zlink[href*="."]z div[id*="n"]r3z div[id*="nn"]zdiv[data-tag*="edval"])r!)rrrrtest_attribute_containss.     z(TestCSSSelectors.test_attribute_containscCs2|jddddgfddddgfddgfdgfdS) Nz p[lang|="en"]zlang-enz lang-en-gbz lang-en-usz [lang|="en"]z p[lang|="fr"]zlang-frz p[lang|="gb"])r!)rrrrtest_attribute_exact_or_hypen2s   z.TestCSSSelectors.test_attribute_exact_or_hypenc CsV|jddddgfddgfdddgfddd d d gfd d dgfdgfdgfddgfdS)Nz[rel]ryr{r}z link[rel]za[rel]z[lang]zlang-enz lang-en-gbz lang-en-uszlang-frzp[class]rxrmz[blah]zp[blah]z div[data-tag]r4)r!)rrrrtest_attribute_exists:s   z&TestCSSSelectors.test_attribute_existscCs,d}t|d}|jd\}d|jks(tdS)Nz]
nope
yes
z html.parserzdiv[style="display: right"]yes)rrstringr)rhtmlr Zchosenrrr"test_quoted_space_in_selector_nameFs  z3TestCSSSelectors.test_quoted_space_in_selector_namecCsHtjt|jjdWdQRXtjt|jjdWdQRXdS)Nza:no-such-pseudoclassza:nth-of-type(a))rArBNotImplementedErrorr rr )rrrrtest_unsupported_pseudoclassNs  z-TestCSSSelectors.test_unsupported_pseudoclasscCs|jjd}t|dkst|djdks.t|jjd}t|dksJt|djdks\t|jjd}t|dksxt|jjd}t|dkstdS) Nzdiv#inner p:nth-of-type(1)rrz Some textzdiv#inner p:nth-of-type(3)ZAnotherzdiv#inner p:nth-of-type(4)zdiv p:nth-of-type(0))r rr'rr)rr+rrrtest_nth_of_typeUs    z!TestCSSSelectors.test_nth_of_typecCs2|jjd}t|dkst|djdks.tdS)Nzdiv#inner > p:nth-of-type(1)rrz Some text)r rr'rr)rr+rrr"test_nth_of_type_direct_descendanths z3TestCSSSelectors.test_nth_of_type_direct_descendantcCs|jddgdS)Nz#inner > p:nth-of-type(2)rx)r)rrrr"test_id_child_selector_nth_of_typemsz3TestCSSSelectors.test_id_child_selector_nth_of_typecCs.|jjddd}|jd}|j|ddgdS)Nr"r$)rr3r4)r findrZassert_selects_ids)rr3selectedrrrtest_select_on_elementps z'TestCSSSelectors.test_select_on_elementcCs|jddg|jdgdS)Nz .fancy #innerr3z.normal #inner)r)rrrrtest_overspecified_child_idysz,TestCSSSelectors.test_overspecified_child_idcCsB|jddg|jddg|jddgg|jjdks>tdS)Nz#p1 + h2rOz #p1 + h2 + prmz#p1 + #header2 + .class1z#p1 + p)rr rr)rrrrtest_adjacent_sibling_selector}sz/TestCSSSelectors.test_adjacent_sibling_selectorcCsR|jdddg|jddg|jddg|jddgg|jjdksNtdS) Nz#p1 ~ h2rOrPz#p1 ~ #header2z #p1 ~ h2 + ar}z#p1 ~ h2 + [rel="me"]z #inner ~ h2)rr rr)rrrrtest_general_sibling_selectors z.TestCSSSelectors.test_general_sibling_selectorc Cs&tjt|jjdWdQRXdS)Nzh1 >)rArBr r r)rrrrtest_dangling_combinators z)TestCSSSelectors.test_dangling_combinatorcCs|jddddgdS)Nz p[lang] ~ pz lang-en-gbz lang-en-uszlang-fr)r)rrrr2test_sibling_combinator_wont_select_same_tag_twiceszCTestCSSSelectors.test_sibling_combinator_wont_select_same_tag_twicecCs|jdddgdS)Nzx, yxidyid)r)rrrrtest_multiple_selectsz%TestCSSSelectors.test_multiple_selectcCs|jdddgdS)Nzx,yrr)r)rrrr"test_multiple_select_with_no_spacesz3TestCSSSelectors.test_multiple_select_with_no_spacecCs|jdddgdS)Nzx, yrr)r)rrrr$test_multiple_select_with_more_spacesz5TestCSSSelectors.test_multiple_select_with_more_spacecCs|jddgdS)Nzx, xr)r)rrrrtest_multiple_select_duplicatedsz0TestCSSSelectors.test_multiple_select_duplicatedcCs|jdddgdS)Nzx, y ~ p[lang=fr]rzlang-fr)r)rrrrtest_multiple_select_siblingsz-TestCSSSelectors.test_multiple_select_siblingcCs|jdddgdS)Nzx, y > zrzidb)r)rrrr.test_multiple_select_tag_and_direct_descendantsz?TestCSSSelectors.test_multiple_select_tag_and_direct_descendantcCs|jdddddddgdS)Nz div > x, y, zrrzidarzidabzidac)r)rrrr/test_multiple_select_direct_descendant_and_tagssz@TestCSSSelectors.test_multiple_select_direct_descendant_and_tagscCs|jdddddddgdS)Nz div x,y, zrrrrrr)r)rrrr(test_multiple_select_indirect_descendantsz9TestCSSSelectors.test_multiple_select_indirect_descendantcCsHtjt|jjdWdQRXtjt|jjdWdQRXdS)Nz,x, yzx,,y)rArBr r r)rrrrtest_invalid_multiple_selects  z-TestCSSSelectors.test_invalid_multiple_selectcCs|jdddgdS)Nzp[lang=en], p[lang=en-gb]zlang-enz lang-en-gb)r)rrrrtest_multiple_select_attrssz+TestCSSSelectors.test_multiple_select_attrscCs|jddddgdS)Nz*x, y > z[id=zida], z[id=zidab], z[id=zidb]rrr)r)rrrrtest_multiple_select_idssz)TestCSSSelectors.test_multiple_select_idscCs|jdddgdS)Nzbody > div > x, y > zrr)r)rrrrtest_multiple_select_nestedsz,TestCSSSelectors.test_multiple_select_nestedcCsRd}t|d}|jd}dt|ks(tx$|jddgdD]}||ks:tq:WdS)Nz3
z html.parserz.c1, .c2r_c1c2)class_)rrr'rrJ)rZmarkupr relementrrrtest_select_duplicate_elementss   z/TestCSSSelectors.test_select_duplicate_elementscCs8|jjddd}|jjd}||jjdddks4tdS)Nr"r3)rz div[id=main]r$)r rr%closestr)rr3rrrr test_closests zTestCSSSelectors.test_closestcCsL|jjddd}|jjddd}|jjddks4t|jjddksHtdS)Nr"r3)rr$z div[id=main]FT)r rr%r1r)rr3r$rrr test_matchszTestCSSSelectors.test_matchcCsJ|jjjd}t|tjst|\}}|ddks6t|ddksFtdS)NrNrrOrP)r r%Ziselectrtypes GeneratorTyper)rgenrOrPrrr test_iselects zTestCSSSelectors.test_iselectcCsh|jjddd}|jjd}t|jjddks4t|jjd}t|tsNt|\}|ddksdtdS) Nr"r3)rrNr<zh2[id=header3]rrP)r rr%filterr'rrr)rr3rresultrrr test_filters  zTestCSSSelectors.test_filtercCsH|jjj}|ddkst|ddks*t|d|jjjdksDtdS)Nz.foo#barz \.foo\#barz()[]{}z \(\)\[\]\{\}z.foo)r r%escaper)rmrrr test_escapes zTestCSSSelectors.test_escapeN)E__name__ __module__ __qualname____doc__r rrZ assertSelectr!r,r/r0r2r5r:r>r@rCrFrGrIrKrQrXrYr]r^r`rdrnrrrvrwrzr|r~rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr s1       r )rArZ unittest.mockrZbs4rrrrrZ soupsiever markZskipifr rrrrs