o
    MK&hH                     @   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rGd dlmZmZ eZeeed	k rGe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                   @   s8  e Zd ZdZdZdd Zdedee dedd	fd
dZ	e	Z
deeee f f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/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 Z0dWdX Z1dYdZ Z2d[d\ Z3d]d^ Z4d_d` Z5dadb Z6dcdd Z7dedf Z8dgdh Z9didj Z:dkdl Z;dmdn Z<dodp Z=dqdr Z>dsdt Z?dudv Z@dwdx ZAdydz ZBd{d| ZCd}d~ ZDdd ZEdd 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   E/var/www/html/venv/lib/python3.10/site-packages/bs4/tests/test_css.pysetup_method^      zTestCSSSelectors.setup_methodselectorexpected_idskwargsreturnNc                 K   sj   | j j|fi |}t|tsJ dd |D }|  |  ||ks3J d|d|d|f d S )Nc                 S   s   g | ]}|d  qS idr   ).0elr   r   r   
<listcomp>f   s    z7TestCSSSelectors.assert_css_selects.<locals>.<listcomp>z$Selector %s, expected [%s], got [%s]z, )r   select
isinstancer   sortjoin)r   r   r   r   results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s!J d S )Ntitler	   r   z	The title)r   r#   r3   r4   contentsr   r7   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s(J d S r-   )r   r#   r3   r4   r5   )r   r7   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 )Nnonexistenttag)r   r5   )r   matchr   r   r   (test_select_one_returns_none_if_no_match   s   z9TestCSSSelectors.test_select_one_returns_none_if_no_matchc                 C      |  dddg d S )Nzdiv divinnerdata1r+   r   r   r   r   test_tag_in_tag_one      z$TestCSSSelectors.test_tag_in_tag_onec                 C   s   dD ]
}|  |g d qd S )N)html divhtml body divbody divrC   r0   rB   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 )NrF   r0   r	   )limitrG   rB      rH   rI   
   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   r3   r   r#   r   r   r   r   test_tag_no_match   s   z"TestCSSSelectors.test_tag_no_matchc                 C   <   t t | jd W d    d S 1 sw   Y  d S )Nztag%tpytestraisesr   r   r#   r   r   r   r   test_invalid_tag      "z!TestCSSSelectors.test_invalid_tagc                 C   rA   )Ncustom-dashed-tagdash1dash2r+   r   r   r   r   test_select_dashed_tag_ids   rE   z+TestCSSSelectors.test_select_dashed_tag_idsc                 C   s6   | j d}|d jdksJ |d d dksJ d S )Nzcustom-dashed-tag[id="dash2"]r   rZ   r   r\   )r   r#   r4   )r   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 )NrZ   )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 ]$}| j |}t|dksJ |d jdksJ |d d dgks&J qd S )N)z.onepzp.onepzhtml p.onepr	   r   pclassonep)r   r#   r3   r4   )r   r   r7   r   r   r   test_class_one   s   zTestCSSSelectors.test_class_onec                 C       | j d}t|dksJ d S )Nzdiv.onepr   r   r#   r3   r;   r   r   r   test_class_mismatched_tag      z*TestCSSSelectors.test_class_mismatched_tagc                 C      dD ]	}|  |dg qd S )N)z	div#innerz#innerzdiv div#innerrB   r+   rK   r   r   r   test_one_id      zTestCSSSelectors.test_one_idc                 C   ro   )Nz#doesnotexistr   rp   r;   r   r   r   test_bad_id   rr   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s%J |d dr.J d S )Nzdiv#inner p   rk   r	   rl   rm   r   )r   r#   r3   r4   has_attr)r   r7   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   rR   rK   r   r   r   test_a_bunch_of_emptys   s   z'TestCSSSelectors.test_a_bunch_of_emptysc                 C   rs   )N)z.class1zp.class1z.class2zp.class2z.class3zp.class3zhtml p.class2zdiv#inner .class2pmultir+   rK   r   r   r   test_multi_class_support   s   
z)TestCSSSelectors.test_multi_class_supportc                 C   rs   )N)z.class1.class3z.class3.class2z.class1.class2.class3r{   r+   rK   r   r   r   test_multi_class_selection   ru   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      |  ddg d S )Nz.s1 > a#s1a2 spanr   r+   r   r   r   r   test_child_selector_id   r   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"]ri   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"]r{   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"]ri   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"]r   zlink[rel^="style"]znotlink[rel^="notstyle"]z[rel^="notstyle"]zlink[rel^="notstyle"]zlink[href^="bla"]za[href^="http://"]r   mez[href^="http://"]z	[id^="p"]r{   r   z	[id^="m"]r0   zdiv[id^="m"]z
a[id^="m"]zdiv[data-tag^="dashed"]rC   ri   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"]r   zlink[href$=".css"]zlink[id$="1"]z	[id$="1"])rC   r   r   re   r~   s2a1r   r[   zdiv[id$="1"]rC   z[id$="noending"]ri   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"]r   zlink[rel*="style"]znotlink[rel*="notstyle"]z[rel*="notstyle"]zlink[rel*="notstyle"]zlink[href*="bla"]z[href*="http://"]r   r   z	[id*="p"]r{   r   zdiv[id*="m"]r0   z
a[id*="m"]z[href*=".css"]zlink[href*=".css"]zlink[id*="1"]z	[id*="1"])	rC   r   r   re   r~   r   r   r   r[   zdiv[id*="1"]rC   z[id*="noending"]z[href*="."])r   r   r   za[href*="."]zlink[href*="."]zdiv[id*="n"]rB   zdiv[id*="nn"]zdiv[data-tag*="edval"]ri   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"]ri   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])r   r   r   z	link[rel]r   za[rel]r   r   z[lang])r   r   r   r   zp[class]r   r{   z[blah]zp[blah]zdiv[data-tag]rC   ri   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soupchosenr   r   r   "test_quoted_space_in_selector_nameq  s   
z3TestCSSSelectors.test_quoted_space_in_selector_namec                 C   sr   t t | jd W d    n1 sw   Y  t t | jd W d    d S 1 s2w   Y  d S )Nza:no-such-pseudoclassza:nth-of-type(a))rV   rW   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s%J |d jdks.J | j d}t|dks<J | j d}t|dksJJ d S )	Nzdiv#inner p:nth-of-type(1)r	   r   	Some textzdiv#inner p:nth-of-type(3)Anotherzdiv#inner p:nth-of-type(4)zdiv p:nth-of-type(0)r   r#   r3   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   r   )Nz#inner > p:nth-of-type(2)r   r+   r   r   r   r   "test_id_child_selector_nth_of_type  r   z3TestCSSSelectors.test_id_child_selector_nth_of_typec                 C   s.   | j jddd}|d}| |ddg d S )Nr.   r0   r   rB   rC   )r   findr#   assert_selects_ids)r   rB   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 #innerrB   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 + h2rg   z#p1 + h2 + pr{   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s'J d S )	Nz#p1 ~ h2rg   rh   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   rT   )Nzh1 >rU   r   r   r   r   test_dangling_combinator  rY   z)TestCSSSelectors.test_dangling_combinatorc                 C      |  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  rE   zCTestCSSSelectors.test_sibling_combinator_wont_select_same_tag_twicec                 C   rA   )Nzx, yxidyidr+   r   r   r   r   test_multiple_select  rE   z%TestCSSSelectors.test_multiple_selectc                 C   rA   )Nzx,yr   r   r+   r   r   r   r   "test_multiple_select_with_no_space  rE   z3TestCSSSelectors.test_multiple_select_with_no_spacec                 C   rA   )Nzx,    yr   r   r+   r   r   r   r   $test_multiple_select_with_more_space  rE   z5TestCSSSelectors.test_multiple_select_with_more_spacec                 C   r   )Nzx, xr   r+   r   r   r   r   test_multiple_select_duplicated  r   z0TestCSSSelectors.test_multiple_select_duplicatedc                 C   rA   )Nzx, y ~ p[lang=fr]r   r   r+   r   r   r   r   test_multiple_select_sibling  rE   z-TestCSSSelectors.test_multiple_select_siblingc                 C   rA   )Nzx, y > zr   zidbr+   r   r   r   r   .test_multiple_select_tag_and_direct_descendant  rE   z?TestCSSSelectors.test_multiple_select_tag_and_direct_descendantc                 C   r   )Nzdiv > x, y, zr   r   zidar   zidabzidacr+   r   r   r   r   /test_multiple_select_direct_descendant_and_tags     z@TestCSSSelectors.test_multiple_select_direct_descendant_and_tagsc                 C   r   )Nzdiv x,y,  zr   r+   r   r   r   r   (test_multiple_select_indirect_descendant  r   z9TestCSSSelectors.test_multiple_select_indirect_descendantc                 C   sr   t t | jd W d    n1 sw   Y  t t | jd W d    d S 1 s2w   Y  d S )Nz,x, yzx,,yrU   r   r   r   r   test_invalid_multiple_select  s   "z-TestCSSSelectors.test_invalid_multiple_selectc                 C   rA   )Nzp[lang=en], p[lang=en-gb]r   r   r+   r   r   r   r   test_multiple_select_attrs  rE   z+TestCSSSelectors.test_multiple_select_attrsc                 C   r   )Nz*x, y > z[id=zida], z[id=zidab], z[id=zidb])r   r   r   r+   r   r   r   r   test_multiple_select_ids  r   z)TestCSSSelectors.test_multiple_select_idsc                 C   rA   )Nzbody > div > x, y > zr   r   r+   r   r   r   r   test_multiple_select_nested  rE   z,TestCSSSelectors.test_multiple_select_nestedc                 C   sN   d}t |d}|d}dt|ksJ |jddgdD ]}||v s$J qd S )Nz3<div class="c1"/><div class="c2"/><div class="c1"/>r   z.c1, .c2rw   c1c2)class_)r   r#   r3   rb   )r   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sJ d S )Nr.   rB   r   div[id=main]r0   )r   r   r1   closest)r   rB   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 sJ |jddu s$J d S )Nr.   rB   r   r0   r   FT)r   r   r1   r?   )r   rB   r0   r   r   r   
test_match  s   zTestCSSSelectors.test_matchc                 C   sJ   | j jd}t|tjsJ |\}}|d dksJ |d dks#J d S )Nrf   r   rg   rh   )r   r1   iselectr$   typesGeneratorType)r   genrg   rh   r   r   r   test_iselect  s
   zTestCSSSelectors.test_iselectc                 C   sh   | j jddd}|jd}t|jddksJ |jd}t|ts'J |\}|d dks2J d S )	Nr.   rB   r   rf   rN   zh2[id=header3]r   rh   )r   r   r1   filterr3   r$   r   )r   rB   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s"J d S )Nz.foo#barz
\.foo\#barz()[]{}z\(\)\[\]\{\}z.foo)r   r1   escape)r   mr   r   r   test_escape  s   
zTestCSSSelectors.test_escape)I__name__
__module____qualname____doc__r   r   strr   r   r)   assertSelectr   r,   r8   r<   r=   r@   rD   rL   rP   rS   rX   r]   r_   ra   rc   rj   rn   rq   rt   rv   ry   rz   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   r   r   r   r   r   $   s    1
	*	
r   )rV   r   bs4r   r   typingr   r   r   r   packaging.versionr    r
   r   	Exception__annotations__	soupsiever   r   r   NotImplementedErrormarkskipifr   r   r   r   r   <module>   s   
 