a
    CgH                     @   s   U d dl Z d dlZd dlmZmZ d dlmZmZmZm	Z	 d dl
mZ ddlmZmZ e	e ed< erd dlmZmZ eZeeed	k reZe jje d
dG dd deZdS )    N)BeautifulSoup	ResultSet)AnyListTupleType)Version   )SoupTestSOUP_SIEVE_PRESENT.SOUPSIEVE_EXCEPTION_ON_UNSUPPORTED_PSEUDOCLASS)__version__SelectorSyntaxErrorz2.6zSoup Sieve not installed)reasonc                   @   s2  e Zd ZdZdZdd Zeee eddddZ	e	Z
eeee f d	d
dZdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zd0d1 Zd2d3 Z d4d5 Z!d6d7 Z"d8d9 Z#d:d; Z$d<d= Z%d>d? Z&d@dA Z'dBdC Z(dDdE Z)dFdG Z*dHdI Z+dJdK Z,dLdM Z-dNdO Z.dPdQ Z/dRdS Z0dTdU Z1dVdW Z2dXdY Z3dZd[ Z4d\d] Z5d^d_ Z6d`da Z7dbdc Z8ddde Z9dfdg Z:dhdi Z;djdk Z<dldm Z=dndo Z>dpdq Z?drds Z@dtdu ZAdvdw ZBdxdy ZCdzd{ ZDd|d} ZEd~d ZFdd ZGdd ZHdS )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  
<!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 | jd| _d S )Nhtml.parser)r   HTML_soupself r   Q/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/bs4/tests/test_css.pysetup_method^   s    zTestCSSSelectors.setup_methodN)selectorexpected_idskwargsreturnc                 K   sj   | j j|fi |}t|ts"J dd |D }|  |  ||ksfJ d|d|d|f d S )Nc                 S   s   g | ]}|d  qS idr   ).0elr   r   r   
<listcomp>f       z7TestCSSSelectors.assert_css_selects.<locals>.<listcomp>z$Selector %s, expected [%s], got [%s]z, )r   select
isinstancer   sortjoin)r   r   r   r   resultsZel_idsr   r   r   assert_css_selectsa   s    z#TestCSSSelectors.assert_css_selects)testsc                 G   s   |D ]\}}|  || qd S )Nr(   )r   r)   r   r   r   r   r   assert_css_select_multipleq   s    z+TestCSSSelectors.assert_css_select_multiplec                 C   sb   | j jd}| j |}t|dks*J |D ]}|jdks.J q.| j |}d|d ks^J d S Ndiv   mainr   )r   csscompiler#   lenname
select_one)r   selelsr-   r    r   r   r   test_precompiledu   s    z!TestCSSSelectors.test_precompiledc                 C   sF   | j d}t|dksJ |d jdks.J |d jdgksBJ d S )Ntitler	   r   z	The title)r   r#   r2   r3   contentsr   r6   r   r   r   test_one_tag_one   s    z!TestCSSSelectors.test_one_tag_onec                 C   sT   | j d}t|dksJ |D ]}|jdks J q | j d}d|d ksPJ d S r,   )r   r#   r2   r3   r4   )r   r6   r-   r    r   r   r   test_one_tag_many   s    z"TestCSSSelectors.test_one_tag_manyc                 C   s   | j d}d |u sJ d S )NZnonexistenttag)r   r4   )r   matchr   r   r   (test_select_one_returns_none_if_no_match   s    z9TestCSSSelectors.test_select_one_returns_none_if_no_matchc                 C   s   |  dddg d S )Nzdiv divinnerdata1r*   r   r   r   r   test_tag_in_tag_one   s    z$TestCSSSelectors.test_tag_in_tag_onec                 C   s   dD ]}|  |g d qd S )N)html divhtml body divbody divr@   r/   r?   footerr*   r   r   r   r   r   test_tag_in_tag_many   s    z%TestCSSSelectors.test_tag_in_tag_manyc                 C   s>   | j ddgdd | j dddgdd | j dg d	d
d d S )NrB   r/   r	   )limitrC   r?      rD   rE   
   r*   r   r   r   r   
test_limit   s
    
zTestCSSSelectors.test_limitc                 C   s   t | jddksJ d S )Ndelr   r2   r   r#   r   r   r   r   test_tag_no_match   s    z"TestCSSSelectors.test_tag_no_matchc                 C   s:   t t | jd W d    n1 s,0    Y  d S )Nztag%tpytestraisesr   r   r#   r   r   r   r   test_invalid_tag   s    z!TestCSSSelectors.test_invalid_tagc                 C   s   |  dddg 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 jdksJ |d d dks2J d S )Nzcustom-dashed-tag[id="dash2"]r   rT   r   rV   )r   r#   r3   )r   Zdashedr   r   r   test_select_dashed_by_id   s    z)TestCSSSelectors.test_select_dashed_by_idc                 C   s   | j dd jdksJ d S )Nzbody > custom-dashed-tagr   zHello there.)r   r#   textr   r   r   r   test_dashed_tag_text   s    z%TestCSSSelectors.test_dashed_tag_textc                 C   s    | j d| j dksJ d S )NrT   )r   r#   find_allr   r   r   r   #test_select_dashed_matches_find_all   s    z4TestCSSSelectors.test_select_dashed_matches_find_allc                 C   s   |  ddgfdddgf d S )Nh1header1h2header2header3r+   r   r   r   r   test_header_tags   s    
z!TestCSSSelectors.test_header_tagsc                 C   sR   dD ]H}| j |}t|dks$J |d jdks6J |d d dgksJ qd S )N)z.onepzp.onepzhtml p.onepr	   r   pclassonep)r   r#   r2   r3   )r   r   r6   r   r   r   test_class_one   s
    zTestCSSSelectors.test_class_onec                 C   s    | j d}t|dksJ d S )Nzdiv.onepr   r   r#   r2   r:   r   r   r   test_class_mismatched_tag   s    z*TestCSSSelectors.test_class_mismatched_tagc                 C   s   dD ]}|  |dg qd S )N)z	div#innerz#innerzdiv div#innerr?   r*   rG   r   r   r   test_one_id   s    zTestCSSSelectors.test_one_idc                 C   s    | j d}t|dksJ d S )Nz#doesnotexistr   rh   r:   r   r   r   test_bad_id   s    zTestCSSSelectors.test_bad_idc                 C   s`   | j d}t|dksJ |D ]}|jdks J q |d d dgksJJ |d dr\J d S )Nzdiv#inner p   rd   r	   re   rf   r   )r   r#   r2   r3   Zhas_attr)r   r6   r    r   r   r   test_items_in_id   s    z!TestCSSSelectors.test_items_in_idc                 C   s&   dD ]}t | j|dksJ qd S )N)zdiv#main delzdiv#main div.oopszdiv div#mainr   rN   rG   r   r   r   test_a_bunch_of_emptys   s    z'TestCSSSelectors.test_a_bunch_of_emptysc                 C   s   dD ]}|  |dg qd S )N)z.class1zp.class1z.class2zp.class2z.class3zp.class3zhtml p.class2zdiv#inner .class2pmultir*   rG   r   r   r   test_multi_class_support   s    
z)TestCSSSelectors.test_multi_class_supportc                 C   s   dD ]}|  |dg qd S )N)z.class1.class3z.class3.class2z.class1.class2.class3ro   r*   rG   r   r   r   test_multi_class_selection   s    z+TestCSSSelectors.test_multi_class_selectionc                 C   s"   |  dddg |  ddg 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   |  ddg d S )Nz.s1 > a#s1a2 spanrt   r*   r   r   r   r   test_child_selector_id   s    z'TestCSSSelectors.test_child_selector_idc                 C   st   |  ddgfddgfddgfddgfddgfddgfd	dgfd
g fddgfddgfddgfdg fdg fdg f d S )Nzp[class="onep"]p1z
p[id="p1"]z[class="onep"]z	[id="p1"]zlink[rel="stylesheet"]l1zlink[type="text/css"]zlink[href="blah.css"]zlink[href="no-blah.css"]z[rel="stylesheet"]z[type="text/css"]z[href="blah.css"]z[href="no-blah.css"]zp[href="no-blah.css"]rb   r   r   r   r   test_attribute_equals   s     z&TestCSSSelectors.test_attribute_equalsc                 C   s\   |  ddgfddgfddgfddgfddgfddgfdd	gfd
d	gfdd	gfdd	gf
 d S )Nzp[class~="class1"]ro   zp[class~="class2"]zp[class~="class3"]z[class~="class1"]z[class~="class2"]z[class~="class3"]za[rel~="friend"]bobza[rel~="met"]z[rel~="friend"]z[rel~="met"]rb   r   r   r   r   test_attribute_tilde  s    z%TestCSSSelectors.test_attribute_tildec                 C   sv   |  ddgfddgfdg fdg fdg fddgfdd	d
gfdd	d
gfdddgfdd
dgfddgfdd
gfddgf d S )Nz[rel^="style"]rx   zlink[rel^="style"]znotlink[rel^="notstyle"]z[rel^="notstyle"]zlink[rel^="notstyle"]zlink[href^="bla"]za[href^="http://"]rz   mez[href^="http://"]z	[id^="p"]ro   rw   z	[id^="m"]r/   zdiv[id^="m"]z
a[id^="m"]zdiv[data-tag^="dashed"]r@   rb   r   r   r   r   test_attribute_startswith  s    



z*TestCSSSelectors.test_attribute_startswithc              	   C   s<   |  ddgfddgfddgfdg dfddgfd	g f d S )
Nz[href$=".css"]rx   zlink[href$=".css"]zlink[id$="1"]z	[id$="1"])r@   rx   rw   r^   rr   s2a1rt   rU   zdiv[id$="1"]r@   z[id$="noending"]rb   r   r   r   r   test_attribute_endswith&  s    z(TestCSSSelectors.test_attribute_endswithc                 C   s   |  ddgfddgfdg fdg fdg fddgfdd	d
gfdddgfddgfdd
gfddgfddgfddgfdg dfddgfdg fdg dfdd	d
gfddgfdddgfddgfd dgf d S )!Nz[rel*="style"]rx   zlink[rel*="style"]znotlink[rel*="notstyle"]z[rel*="notstyle"]zlink[rel*="notstyle"]zlink[href*="bla"]z[href*="http://"]rz   r|   z	[id*="p"]ro   rw   zdiv[id*="m"]r/   z
a[id*="m"]z[href*=".css"]zlink[href*=".css"]zlink[id*="1"]z	[id*="1"])	r@   rx   rw   r^   rr   rs   r~   rt   rU   zdiv[id*="1"]r@   z[id*="noending"]z[href*="."])rz   r|   rx   za[href*="."]zlink[href*="."]zdiv[id*="n"]r?   zdiv[id*="nn"]zdiv[data-tag*="edval"]rb   r   r   r   r   test_attribute_contains3  s4    




z(TestCSSSelectors.test_attribute_containsc                 C   s.   |  dg dfdg dfddgfdg 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"]rb   r   r   r   r   test_attribute_exact_or_hypen]  s    

z.TestCSSSelectors.test_attribute_exact_or_hypenc                 C   sP   |  dg dfddgfdddgfdg d	fd
ddgfdg fdg fddgf d S )Nz[rel])rx   rz   r|   z	link[rel]rx   za[rel]rz   r|   z[lang])r   r   r   r   zp[class]rw   ro   z[blah]zp[blah]zdiv[data-tag]r@   rb   r   r   r   r   test_attribute_existse  s    



z&TestCSSSelectors.test_attribute_existsc                 C   s,   d}t |d}|d\}d|jks(J d S )Nz]<div style="display: wrong">nope</div>
        <div style="display: right">yes</div>
        r   zdiv[style="display: right"]yes)r   r#   string)r   htmlsoupZchosenr   r   r   "test_quoted_space_in_selector_nameq  s    
z3TestCSSSelectors.test_quoted_space_in_selector_namec                 C   sp   t t | jd W d    n1 s,0    Y  t t | jd W d    n1 sb0    Y  d S )Nza:no-such-pseudoclassza:nth-of-type(a))rQ   rR   r   r   r#   r   r   r   r   r   test_unsupported_pseudoclassy  s    *z-TestCSSSelectors.test_unsupported_pseudoclassc                 C   s   | j d}t|dksJ |d jdks.J | j d}t|dksJJ |d jdks\J | j d}t|dksxJ | j d}t|dksJ d S )	Nzdiv#inner p:nth-of-type(1)r	   r   	Some textzdiv#inner p:nth-of-type(3)ZAnotherzdiv#inner p:nth-of-type(4)zdiv p:nth-of-type(0)r   r#   r2   r   r:   r   r   r   test_nth_of_type  s    z!TestCSSSelectors.test_nth_of_typec                 C   s2   | j d}t|dksJ |d jdks.J d S )Nzdiv#inner > p:nth-of-type(1)r	   r   r   r   r:   r   r   r   "test_nth_of_type_direct_descendant  s    z3TestCSSSelectors.test_nth_of_type_direct_descendantc                 C   s   |  ddg d S )Nz#inner > p:nth-of-type(2)rw   r*   r   r   r   r   "test_id_child_selector_nth_of_type  s    z3TestCSSSelectors.test_id_child_selector_nth_of_typec                 C   s.   | j jddd}|d}| |ddg d S )Nr-   r/   r   r?   r@   )r   findr#   Zassert_selects_ids)r   r?   selectedr   r   r   test_select_on_element  s    
z'TestCSSSelectors.test_select_on_elementc                 C   s   |  ddg |  dg  d S )Nz.fancy #innerr?   z.normal #innerr*   r   r   r   r   test_overspecified_child_id  s    z,TestCSSSelectors.test_overspecified_child_idc                 C   sB   |  ddg |  ddg |  ddg g | jdks>J d S )Nz#p1 + h2r`   z#p1 + h2 + pro   z#p1 + #header2 + .class1z#p1 + pr(   r   r#   r   r   r   r   test_adjacent_sibling_selector  s    z/TestCSSSelectors.test_adjacent_sibling_selectorc                 C   sR   |  dddg |  ddg |  ddg |  ddg g | jdksNJ d S )	Nz#p1 ~ h2r`   ra   z#p1 ~ #header2z#p1 ~ h2 + ar|   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 d    n1 s,0    Y  d S )Nzh1 >rP   r   r   r   r   test_dangling_combinator  s    z)TestCSSSelectors.test_dangling_combinatorc                 C   s   |  dg d d S )Nzp[lang] ~ p)r   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   |  dddg d S )Nzx, yxidyidr*   r   r   r   r   test_multiple_select  s    z%TestCSSSelectors.test_multiple_selectc                 C   s   |  dddg d S )Nzx,yr   r   r*   r   r   r   r   "test_multiple_select_with_no_space  s    z3TestCSSSelectors.test_multiple_select_with_no_spacec                 C   s   |  dddg d S )Nzx,    yr   r   r*   r   r   r   r   $test_multiple_select_with_more_space  s    z5TestCSSSelectors.test_multiple_select_with_more_spacec                 C   s   |  ddg d S )Nzx, xr   r*   r   r   r   r   test_multiple_select_duplicated  s    z0TestCSSSelectors.test_multiple_select_duplicatedc                 C   s   |  dddg d S )Nzx, y ~ p[lang=fr]r   r   r*   r   r   r   r   test_multiple_select_sibling  s    z-TestCSSSelectors.test_multiple_select_siblingc                 C   s   |  dddg d S )Nzx, 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   |  dg d d S )Nzdiv > x, y, zr   r   Zzidar   zidabZzidacr*   r   r   r   r   /test_multiple_select_direct_descendant_and_tags  s    z@TestCSSSelectors.test_multiple_select_direct_descendant_and_tagsc                 C   s   |  dg d d S )Nzdiv x,y,  zr   r*   r   r   r   r   (test_multiple_select_indirect_descendant  s    z9TestCSSSelectors.test_multiple_select_indirect_descendantc                 C   sp   t t | jd W d    n1 s,0    Y  t t | jd W d    n1 sb0    Y  d S )Nz,x, yzx,,yrP   r   r   r   r   test_invalid_multiple_select  s    *z-TestCSSSelectors.test_invalid_multiple_selectc                 C   s   |  dddg d S )Nzp[lang=en], p[lang=en-gb]r   r   r*   r   r   r   r   test_multiple_select_attrs  s    z+TestCSSSelectors.test_multiple_select_attrsc                 C   s   |  dg d 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   |  dddg d S )Nzbody > 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}dt|ks(J |jddgdD ]}||v s8J q8d S )Nz3<div class="c1"/><div class="c2"/><div class="c1"/>r   z.c1, .c2rl   Zc1c2)class_)r   r#   r2   r[   )r   Zmarkupr   r   elementr   r   r   test_select_duplicate_elements  s    

z/TestCSSSelectors.test_select_duplicate_elementsc                 C   s8   | j jddd}|jd}|| j jdddks4J d S )Nr-   r?   r   div[id=main]r/   )r   r   r0   closest)r   r?   r   r   r   r   test_closest  s    zTestCSSSelectors.test_closestc                 C   sL   | j jddd}| j jddd}|jddu s4J |jddu sHJ d S )Nr-   r?   r   r/   r   FT)r   r   r0   r=   )r   r?   r/   r   r   r   
test_match  s    zTestCSSSelectors.test_matchc                 C   sJ   | j jd}t|tjsJ |\}}|d dks6J |d dksFJ d S )Nr_   r   r`   ra   )r   r0   Ziselectr$   typesGeneratorType)r   genr`   ra   r   r   r   test_iselect  s
    zTestCSSSelectors.test_iselectc                 C   sh   | j jddd}|jd}t|jddks4J |jd}t|tsNJ |\}|d dksdJ d S )	Nr-   r?   r   r_   rJ   zh2[id=header3]r   ra   )r   r   r0   filterr2   r$   r   )r   r?   r'   resultr   r   r   test_filter
  s    zTestCSSSelectors.test_filterc                 C   sH   | j jj}|ddksJ |ddks*J |d| j jdksDJ d S )Nz.foo#barz
\.foo\#barz()[]{}z\(\)\[\]\{\}z.foo)r   r0   escape)r   mr   r   r   test_escape  s    
zTestCSSSelectors.test_escape)I__name__
__module____qualname____doc__r   r   strr   r   r(   ZassertSelectr   r+   r7   r;   r<   r>   rA   rH   rL   rO   rS   rW   rX   rZ   r\   rc   rg   ri   rj   rk   rm   rn   rp   rq   ru   rv   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   r   r   r   r   $   s   1
	*	
r   )rQ   r   Zbs4r   r   typingr   r   r   r   Zpackaging.versionr    r
   r   	Exception__annotations__Z	soupsiever   r   r   NotImplementedErrormarkZskipifr   r   r   r   r   <module>   s   
