a
    #.c@=                     @   sl   d dl Z d dlZd dlZd dlmZ d dlmZmZmZ G dd deZ	G dd deZ
G dd	 d	eZdS )
    N)BeautifulSoup)EntitySubstitutionEncodingDetectorUnicodeDammitc                   @   sh   e Zd Z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S )TestUnicodeDammitz"Standalone tests of UnicodeDammit.c                 C   s   d}t |}|j|ksJ d S )Nu   I'm already Unicode! ☃r   unicode_markupselfmarkupdammit r   O/var/www/brookimports/venv/lib/python3.9/site-packages/bs4/tests/test_dammit.pytest_unicode_input   s    z$TestUnicodeDammit.test_unicode_inputc                 C   s   d}t |}|jdksJ d S )N   <foo></foo>u   <foo>‘’“”</foo>r   r	   r   r   r   test_smart_quotes_to_unicode   s    z.TestUnicodeDammit.test_smart_quotes_to_unicodec                 C   s"   d}t |dd}|jdksJ d S )Nr   xmlZsmart_quotes_toz+<foo>&#x2018;&#x2019;&#x201C;&#x201D;</foo>r   r	   r   r   r   !test_smart_quotes_to_xml_entities   s    z3TestUnicodeDammit.test_smart_quotes_to_xml_entitiesc                 C   s"   d}t |dd}|jdksJ d S )Nr   htmlr   z'<foo>&lsquo;&rsquo;&ldquo;&rdquo;</foo>r   r	   r   r   r   "test_smart_quotes_to_html_entities   s    z4TestUnicodeDammit.test_smart_quotes_to_html_entitiesc                 C   s"   d}t |dd}|jdksJ d S )Nr   asciir   z<foo>''""</foo>r   r	   r   r   r   test_smart_quotes_to_ascii#   s    z,TestUnicodeDammit.test_smart_quotes_to_asciic                 C   s0   d}t |}|j dksJ |jdks,J d S )Ns   Sacré bleu! ☃utf-8u   Sacré bleu! ☃r   original_encodinglowerr   )r
   utf8r   r   r   r   test_detect_utf8(   s    z"TestUnicodeDammit.test_detect_utf8c                 C   s4   d}t |dg}|j dks"J |jdks0J d S )N   
iso-8859-8u   םולשr   r
   hebrewr   r   r   r   test_convert_hebrew.   s    z%TestUnicodeDammit.test_convert_hebrewc                 C   s6   d}t |}|j dksJ |jd|ks2J d S )Ns   ケータイ Watchr   )r   r   r   r   encode)r
   utf_8r   r   r   r   /test_dont_see_smart_quotes_where_there_are_none4   s    zATestUnicodeDammit.test_dont_see_smart_quotes_where_there_are_nonec                 C   s,   d d}t|dg}|j dks(J d S )N   Räksmörgåsr   r    r$   r   r   r   r
   	utf8_datar   r   r   r    test_ignore_inappropriate_codecs:   s    
z2TestUnicodeDammit.test_ignore_inappropriate_codecsc                 C   s6   d d}dD ]"}t||g}|j dksJ qd S )Nr'   r   )z.utf8z...z
utF---16.!r(   )r
   r*   Zbad_encodingr   r   r   r   test_ignore_invalid_codecs?   s    
z,TestUnicodeDammit.test_ignore_invalid_codecsc                 C   sL   d d}t|dgd}|j dks*J t|ddgd}|jd ksHJ d S )Nr'   r   )Zexclude_encodingszwindows-1252r(   r)   r   r   r   test_exclude_encodingsE   s    
z(TestUnicodeDammit.test_exclude_encodingsN)__name__
__module____qualname____doc__r   r   r   r   r   r   r#   r&   r+   r,   r-   r   r   r   r   r      s   r   c                   @   sT   e Z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S )TestEncodingDetectorc                 C   s"   t d}t|j}d|v sJ d S )Ns'   <?xml version="1.0" encoding="UTF-" ?>u   utf-�)r   list	encodings)r
   Zdetectedr4   r   r   r   Ptest_encoding_detector_replaces_junk_in_encoding_name_with_replacement_characterU   s
    
zeTestEncodingDetector.test_encoding_detector_replaces_junk_in_encoding_name_with_replacement_characterc                 C   s(   dD ]}t |dd}d|jksJ qd S )N)s&   <html><meta charset="euc-jp" /></html>s&   <html><meta charset='euc-jp' /></html>s$   <html><meta charset=euc-jp /></html>s#   <html><meta charset=euc-jp/></html>Tis_htmlzeuc-jp)r   r   r
   datar   r   r   r    test_detect_html5_style_meta_tag[   s    z5TestEncodingDetector.test_detect_html5_style_meta_tagc              	   C   s   d}t jj}ttj z`dd }|t j_t|}d|jks@J d|jv sNJ t	|d}|jsbJ W ttj
 |t j_nttj
 |t j_0 d S )NsT   ﻿<?xml version="1.0" encoding="UTF-8"?>
<html><b>بتر</b>
<i>ѐ</i></html>c                 S   s   d S Nr   )strr   r   r   noop{   s    zETestEncodingDetector.test_last_ditch_entity_replacement.<locals>.noopTu   �zhtml.parser)bs4r   Zchardet_dammitloggingdisableWARNINGr   Zcontains_replacement_charactersr   r   NOTSET)r
   docchardetr=   r   Zsoupr   r   r   "test_last_ditch_entity_replacemente   s    

z7TestEncodingDetector.test_last_ditch_entity_replacementc                 C   s,   d}t |}d|jksJ d|jks(J d S )N   < a >   < / a > u   <a>áé</a>utf-16le)r   r   r   r8   r   r   r   test_byte_order_mark_removed   s    z1TestEncodingDetector.test_byte_order_mark_removedc                 C   s   d}t |}t |dgd}d|jks(J t |dgd}d|jksDJ dgdd |jD ks^J d	}t |dgd
gd}d
|jksJ dd
gdd |jD ksJ d S )NrF   zutf-16)known_definite_encodingsr   )user_encodingsrG   c                 S   s   g | ]}|d  qS r   r   .0xr   r   r   
<listcomp>       zRTestEncodingDetector.test_known_definite_versus_user_encodings.<locals>.<listcomp>r   r    )rI   rJ   c                 S   s   g | ]}|d  qS rK   r   rL   r   r   r   rO      rP   r   r   tried_encodings)r
   r9   r   beforeafterr"   r   r   r   )test_known_definite_versus_user_encodings   s    z>TestEncodingDetector.test_known_definite_versus_user_encodingsc                 C   sH   d}t |dgdgdgd}d|jks(J g ddd |jD ksDJ d S )	Nr   	shift-jisr   r    )rI   Zoverride_encodingsrJ   )rV   r   r    c                 S   s   g | ]}|d  qS rK   r   rL   r   r   r   rO      rP   zKTestEncodingDetector.test_deprecated_override_encodings.<locals>.<listcomp>rQ   r!   r   r   r   "test_deprecated_override_encodings   s    z7TestEncodingDetector.test_deprecated_override_encodingsc                 C   st   d d}d d}|| | }tt |d W d    n1 sJ0    Y  t|}d|dkspJ d S )Nu	   ☃☃☃r   u   “Hi, I like Windows!”windows_1252u+   ☃☃☃“Hi, I like Windows!”☃☃☃)r$   pytestZraisesUnicodeDecodeErrordecoder   	detwingle)r
   r   rX   rC   fixedr   r   r   test_detwingle   s    
(
z#TestEncodingDetector.test_detwinglec                 C   s<   dD ]2}| d}|ds J t|}||ksJ qd S )N)u   œu   ₓu   ðr      )r$   endswithr   r\   )r
   Ztricky_unicode_charinputoutputr   r   r   +test_detwingle_ignores_multibyte_characters   s
    

z@TestEncodingDetector.test_detwingle_ignores_multibyte_charactersc                 C   s  d}| d}d}| d}tj}||ddd u s6J d||ddksJJ d||ddks^J d||ksnJ d||ks~J d	d
 }||| d u sJ ||| d u sJ ||| ddddksJ ||dddksJ |d	| dddksJ |d| ddd u sJ d S )Nz0<html><head><meta charset="utf-8"></head></html>r   z,<?xml version="1.0" encoding="ISO-8859-1" ?>Fr6   r   Tz
iso-8859-1    i  )r7   search_entire_document)re      a)r$   r   Zfind_declared_encoding)r
   Zhtml_unicodeZ
html_bytesZxml_unicodeZ	xml_bytesmZspacerr   r   r   test_find_declared_encoding   s(    

z0TestEncodingDetector.test_find_declared_encodingN)r.   r/   r0   r5   r:   rE   rH   rU   rW   r^   rc   rh   r   r   r   r   r2   S   s   
#r2   c                   @   s   e Zd Z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 S )!TestEntitySubstitutionz1Standalone tests of the EntitySubstitution class.c                 C   s
   t | _d S r;   )r   subr
   r   r   r   setup_method  s    z#TestEntitySubstitution.setup_methodc                 C   s   d}| j |dksJ d S )Nu   foo∀☃õbaru   foo&forall;☃&otilde;barrj   substitute_htmlr
   sr   r   r   test_simple_html_substitution  s    z4TestEntitySubstitution.test_simple_html_substitutionc                 C   s&   d}t |}| j|jdks"J d S )Ns   fooz&lsquo;&rsquo;foo&ldquo;&rdquo;)r   rj   rn   r   )r
   quotesr   r   r   r   test_smart_quote_substitution  s    z4TestEntitySubstitution.test_smart_quote_substitutionc                 C   s:   dD ]0\}}d}|| }|| }| j ||ksJ qd S )N)
)z&models;u   ⊧)z&Nfr;u   𝔑)z&ngeqq;u   ≧̸)z&not;   ¬)z&Not;u   ⫬z||)fjru   )z&gt;>)z&lt;<)z&amp;&z3 %s 4rm   )r
   entityutemplaterawZwith_entitiesr   r   r   test_html5_entity#  s
    z(TestEntitySubstitution.test_html5_entityc                 C   s<   d}d}| j ||ksJ d}d}| j ||ks8J d S )Nu   fjords ⊔ penguinszfjords &sqcup; penguinsu   fjords ⊔︀ penguinszfjords &sqcups; penguinsrm   )r
   r9   r   r   r   r   )test_html5_entity_with_variation_selectorD  s    z@TestEntitySubstitution.test_html5_entity_with_variation_selectorc                 C   s   d}| j |d|ksJ d S )NWelcome to "my bar"Frj   Zsubstitute_xmlro   r   r   r   Itest_xml_converstion_includes_no_quotes_if_make_quoted_attribute_is_falseP  s    z`TestEntitySubstitution.test_xml_converstion_includes_no_quotes_if_make_quoted_attribute_is_falsec                 C   s0   | j dddksJ | j dddks,J d S )NZWelcomeTz	"Welcome"z	Bob's Barz"Bob's Bar"r   rk   r   r   r   6test_xml_attribute_quoting_normally_uses_double_quotesT  s    zMTestEntitySubstitution.test_xml_attribute_quoting_normally_uses_double_quotesc                 C   s   d}| j |ddksJ d S )Nr   Tz'Welcome to "my bar"'r   ro   r   r   r   Otest_xml_attribute_quoting_uses_single_quotes_when_value_contains_double_quotesX  s    zfTestEntitySubstitution.test_xml_attribute_quoting_uses_single_quotes_when_value_contains_double_quotesc                 C   s   d}| j |ddksJ d S )NWelcome to "Bob's Bar"Tz""Welcome to &quot;Bob's Bar&quot;"r   ro   r   r   r   btest_xml_attribute_quoting_escapes_single_quotes_when_value_contains_both_single_and_double_quotes\  s    zyTestEntitySubstitution.test_xml_attribute_quoting_escapes_single_quotes_when_value_contains_both_single_and_double_quotesc                 C   s   d}| j ||ksJ d S )Nr   r   )r
   quotedr   r   r   <test_xml_quotes_arent_escaped_when_value_is_not_being_quoted`  s    zSTestEntitySubstitution.test_xml_quotes_arent_escaped_when_value_is_not_being_quotedc                 C   s   | j ddksJ d S )Nzfoo<bar>zfoo&lt;bar&gt;r   rk   r   r   r   'test_xml_quoting_handles_angle_bracketsd  s    z>TestEntitySubstitution.test_xml_quoting_handles_angle_bracketsc                 C   s   | j ddksJ d S )NzAT&TzAT&amp;Tr   rk   r   r   r   #test_xml_quoting_handles_ampersandsg  s    z:TestEntitySubstitution.test_xml_quoting_handles_ampersandsc                 C   s   | j ddksJ d S )N&Aacute;T&Tz&amp;Aacute;T&amp;Tr   rk   r   r   r   Etest_xml_quoting_including_ampersands_when_they_are_part_of_an_entityj  s    z\TestEntitySubstitution.test_xml_quoting_including_ampersands_when_they_are_part_of_an_entityc                 C   s   | j ddksJ d S )Nr   z&Aacute;T&amp;T)rj   Z"substitute_xml_containing_entitiesrk   r   r   r   Dtest_xml_quoting_ignoring_ampersands_when_they_are_part_of_an_entitym  s    z[TestEntitySubstitution.test_xml_quoting_ignoring_ampersands_when_they_are_part_of_an_entityc                 C   s   d}| j ||ksJ dS )z:There's no need to do this except inside attribute values.zBob's "bar"Nrm   )r
   textr   r   r    test_quotes_not_html_substitutedp  s    z7TestEntitySubstitution.test_quotes_not_html_substitutedN)r.   r/   r0   r1   rl   rq   rs   r}   r~   r   r   r   r   r   r   r   r   r   r   r   r   r   r   ri     s    !ri   )rY   r?   r>   r   Z
bs4.dammitr   r   r   objectr   r2   ri   r   r   r   r   <module>   s   G ?