a
    CgE                     @   st   d dl Z 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                   @   sb   e Zd ZdZdd Zejdg d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   T/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/bs4/tests/test_dammit.pytest_unicode_input   s    z$TestUnicodeDammit.test_unicode_inputz smart_quotes_to,expect_converted))Nu   ‘’“”)xmlz &#x2018;&#x2019;&#x201C;&#x201D;)htmlz&lsquo;&rsquo;&ldquo;&rdquo;)asciiz''""c                 C   s,   d}t |dg|dj}|d|ks(J dS )zbVerify the functionality of the smart_quotes_to argument
        to the UnicodeDammit constructor.s   <foo></foo>windows-1252)known_definite_encodingssmart_quotes_toz<foo>{}</foo>N)r   r   format)r   r   Zexpect_convertedr	   Z	convertedr   r   r   test_smart_quotes_to   s    z&TestUnicodeDammit.test_smart_quotes_toc                 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_hebrew0   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_none6   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_codecsA   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 u sHJ d S )Nr#   r   )Zexclude_encodingsr   r$   r%   r   r   r   test_exclude_encodingsG   s
    
z(TestUnicodeDammit.test_exclude_encodingsN)__name__
__module____qualname____doc__r   pytestmarkparametrizer   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detectedr3   r   r   r   Ptest_encoding_detector_replaces_junk_in_encoding_name_with_replacement_characterV   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u 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_replacementg   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   r7   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 )NrE   zutf-16)r   r   )user_encodingsrF   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   )r   rH   c                 S   s   g | ]}|d  qS rI   r   rJ   r   r   r   rM      rN   )r   r   tried_encodings)r   r8   r
   beforeafterr   r   r   r   )test_known_definite_versus_user_encodings   s    
z>TestEncodingDetector.test_known_definite_versus_user_encodingsc                 C   s   d}t jdd&}t|dgdgdgd}W d    n1 s<0    Y  |\}|j}t|ts`J |jtksnJ d|jks|J g dd	d
 |j	D ksJ d S )Nr   T)record	shift-jisr   r   )r   Zoverride_encodingsrH   )rT   r   r   c                 S   s   g | ]}|d  qS rI   r   rJ   r   r   r   rM      rN   zKTestEncodingDetector.test_deprecated_override_encodings.<locals>.<listcomp>)
warningscatch_warningsr   message
isinstanceDeprecationWarningfilename__file__r   rO   )r   r   wr
   warningrW   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    r.   ZraisesUnicodeDecodeErrordecoder   	detwingle)r   r   r_   rB   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   rb   )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" ?>Fr5   r   Tz
iso-8859-1    i  )r6   search_entire_document)rk      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+   r,   r4   r9   rD   rG   rR   r^   rd   ri   rn   r   r   r   r   r1   U   s   
% r1   c                   @   s   e Zd ZdZdd Zejdddg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ejd!g d"d#d$ Zejd%g d&d'd( Zejd)d*g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_methodzoriginal,substituted)u   foo∀☃õbaru   foo&forall;☃&otilde;bar)u   ‘’foo“”z&lsquo;&rsquo;foo&ldquo;&rdquo;c                 C   s   | j ||ksJ d S r:   rp   substitute_html)r   originalZsubstitutedr   r   r   test_substitute_html!  s    z+TestEntitySubstitution.test_substitute_htmlc                 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||)fjrx   )z&gt;>)z&lt;<z3 %s 4rs   )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; penguinsrs   )r   r8   r	   r   r   r   )test_html5_entity_with_variation_selectorI  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rp   Zsubstitute_xmlr   sr   r   r   Itest_xml_converstion_includes_no_quotes_if_make_quoted_attribute_is_falseU  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   rq   r   r   r   6test_xml_attribute_quoting_normally_uses_double_quotesY  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   r   r   r   r   Otest_xml_attribute_quoting_uses_single_quotes_when_value_contains_double_quotes]  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   r   r   r   r   btest_xml_attribute_quoting_escapes_single_quotes_when_value_contains_both_single_and_double_quotesc  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_quotedi  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   rq   r   r   r   'test_xml_quoting_handles_angle_bracketsm  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   rq   r   r   r   #test_xml_quoting_handles_ampersandsp  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   rq   r   r   r   Etest_xml_quoting_including_ampersands_when_they_are_part_of_an_entitys  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)rp   Z"substitute_xml_containing_entitiesrq   r   r   r   Dtest_xml_quoting_ignoring_ampersands_when_they_are_part_of_an_entityv  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"Nrs   )r   textr   r   r    test_quotes_not_html_substituted|  s    z7TestEntitySubstitution.test_quotes_not_html_substitutedzmarkup, old))z	foo & barzfoo &amp; bar)zfoo&zfoo&amp;)z
foo&&& barzfoo&amp;&amp;&amp; bar)zx=1&y=2zx=1&amp;y=2)z&123z&amp;123)z&abcz&amp;abc)z
foo &0 barzfoo &amp;0 bar)zfoo &lolwat barzfoo &amp;lolwat barc                 C   s,   | j ||ksJ | j ||ks(J d S r:   rp   rt   substitute_html5_raw)r   r	   oldr   r   r   'test_unambiguous_ampersands_not_escaped  s    z>TestEntitySubstitution.test_unambiguous_ampersands_not_escapedzmarkup,html,html5,html5raw))&divide;&amp;divide;r   r   )z
&nonesuch;&amp;nonesuch;r   r   )z&#247;
&amp;#247;r   r   )z&#xa1;
&amp;#xa1;r   r   c                 C   s@   | j ||ksJ | j ||ks(J | j ||ks<J d S r:   )rp   rt   Zsubstitute_html5r   )r   r	   r   html5Zhtml5rawr   r   r   'test_when_entity_ampersands_are_escaped  s    z>TestEntitySubstitution.test_when_entity_ampersands_are_escapedzmarkup,expect)z&nosuchentity;z&amp;nosuchentity;c                 C   s,   | j ||ksJ | j ||ks(J d S r:   r   )r   r	   expectr   r   r   !test_ambiguous_ampersands_escaped  s    z8TestEntitySubstitution.test_ambiguous_ampersands_escapedN)r*   r+   r,   r-   rr   r.   r/   r0   rv   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   ro     sF   

	
ro   )r.   r>   rU   r=   r   Z
bs4.dammitr   r   r   objectr   r1   ro   r   r   r   r   <module>   s   G G