a
    	¼3j/  ã                   @   s¶   d dl Z d dlZd dlZd dlmZmZ d dlmZ d dlm	Z	m
Z
 d dlmZ d dlmZ d dlmZmZ eG dd	„ d	ƒƒZG d
d„ deeƒZG dd„ deeƒZedkr²e ¡  dS )é    N)Úmessage_from_stringÚmessage_from_bytes)ÚEmailMessage)Ú	GeneratorÚBytesGenerator)ÚAddress©Úpolicy)ÚTestEmailBaseÚparameterizec                   @   sÄ   e Zd ZejZd$dd„Ze d¡e d¡e d¡dœZed ed	< e 	¡ Z
d
e
d< de
d	< dd„ e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 )%ÚTestGeneratorBaseNc                 C   s    |d u r| j n|}| j||dS )Nr   )r	   Úmsgfunc)ÚselfÚmsgr	   © r   úZ/www/server/python_manager/versions/3.9.10/lib/python3.9/test/test_email/test_generator.pyÚmsgmaker   s    zTestGeneratorBase.msgmakeraf              To: whom_it_may_concern@example.com
            From: nobody_you_want_to_know@example.com
            Subject: We the willing led by the unknowing are doing the
             impossible for the ungrateful. We have done so much for so long with so little
             we are now qualified to do anything with nothing.

            None
            a              To: whom_it_may_concern@example.com
            From:
             nobody_you_want_to_know@example.com
            Subject: We the willing led by the
             unknowing are doing the impossible for
             the ungrateful. We have done so much
             for so long with so little we are now
             qualified to do anything with nothing.

            None
            aè              To:
             whom_it_may_concern@example.com
            From:
             nobody_you_want_to_know@example.com
            Subject: We the
             willing led by the
             unknowing are doing
             the impossible for
             the ungrateful. We
             have done so much
             for so long with so
             little we are now
             qualified to do
             anything with
             nothing.

            None
            )r   é(   é   r   éd   a  To: whom_it_may_concern@example.com
From: nobody_you_want_to_know@example.com
Subject: We the willing led by the unknowing are doing the impossible for the ungrateful. We have done so much for so long with so little we are now qualified to do anything with nothing.

None
a  To: whom_it_may_concern@example.com
From: nobody_you_want_to_know@example.com
Subject: We the willing led by the unknowing are doing the impossible for the ungrateful. We have
 done so much for so long with so little we are now qualified to do anything with nothing.

None
c                 C   s   g | ]}|‘qS r   r   )Ú.0Únr   r   r   Ú
<listcomp>U   ó    zTestGeneratorBase.<listcomp>c                 C   sZ   |   |  | jd ¡¡}|  ¡ }| j||| jd}| |¡ |  | ¡ |  | j| ¡¡ d S )Nr   ©Zmaxheaderlenr	   )	r   ÚtypÚrefold_long_expectedÚioclassÚgenclassr	   ÚflattenÚassertEqualÚgetvalue©r   r   r   ÚsÚgr   r   r   Ú length_as_maxheaderlen_parameterW   s
    
z2TestGeneratorBase.length_as_maxheaderlen_parameterc                 C   s`   |   |  | jd ¡¡}|  ¡ }| j|| jj|dd}| |¡ |  | 	¡ |  | j| ¡¡ d S ©Nr   ©Úmax_line_lengthr   ©
r   r   r   r   r   r	   Úcloner   r    r!   r"   r   r   r   Ú length_as_max_line_length_policy^   s
    
z2TestGeneratorBase.length_as_max_line_length_policyc                 C   sb   |   |  | jd ¡¡}|  ¡ }| j||| jjddd}| |¡ |  | 	¡ |  | j| ¡¡ d S )Nr   é
   r'   r   r)   r"   r   r   r   Ú,length_as_maxheaderlen_parm_overrides_policye   s    ÿ
z>TestGeneratorBase.length_as_maxheaderlen_parm_overrides_policyc                 C   sb   |   |  | jd ¡¡}|  ¡ }| j|| jjd|dd}| |¡ |  | 	¡ |  | jd ¡¡ d S )Nr   Znone©Zrefold_sourcer(   r   r)   r"   r   r   r   Ú8length_as_max_line_length_with_refold_none_does_not_foldm   s    ÿ

zJTestGeneratorBase.length_as_max_line_length_with_refold_none_does_not_foldc                 C   sb   |   |  | jd ¡¡}|  ¡ }| j|| jjd|dd}| |¡ |  | 	¡ |  | j
| ¡¡ d S )Nr   Úallr.   r   )r   r   r   r   r   r	   r*   r   r    r!   Úrefold_all_expectedr"   r   r   r   Ú/length_as_max_line_length_with_refold_all_foldsu   s    ÿ

zATestGeneratorBase.length_as_max_line_length_with_refold_all_foldsc                 C   sT   d}|}|   |  |¡¡}|  ¡ }| j|tjd}| |¡ |  | ¡ |  |¡¡ d S )NzSubject: test

test body
r   ©	r   r   r   r   r	   ÚSMTPr   r    r!   ©r   ÚsourceÚexpectedr   r#   r$   r   r   r   Útest_crlf_control_via_policy}   s    
z.TestGeneratorBase.test_crlf_control_via_policyc                 C   sX   d}|}|   |  |¡¡}|  ¡ }| j|tjd}|j|dd |  | ¡ |  |¡¡ d S )NzSubject: test

test body
r   Ú
)Úlinesepr3   r5   r   r   r   Ú%test_flatten_linesep_overrides_policy†   s    z7TestGeneratorBase.test_flatten_linesep_overrides_policyc           	   	   C   sÊ   t  d¡}dtjdftjdftjjdddff}|D ]\}}|rL| dd¡n|}| j||dV |  |  	|¡¡}|  
¡ }| j||d	}| |¡ |  | ¡ |  	|¡¡ W d   ƒ q41 sº0    Y  q4d S )
Nz…            Subject: test that
             from is mangled in the body!

            From time to time I write a rhyme.
            )NTTF)Úmangle_from_zFrom z>From )r	   r<   r   )ÚtextwrapÚdedentr	   Úcompat32Údefaultr*   ÚreplaceZsubTestr   r   r   r   r   r    r!   )	r   r6   ZvariantsÚpZmangler7   r   r#   r$   r   r   r   Útest_set_mangle_from_via_policy   s    
ü
z1TestGeneratorBase.test_set_mangle_from_via_policyc                 C   s`   |   |  | jd ¡¡}|  ¡ }| j|tjjd dd}| |¡ |  	| 
¡ |  | jd ¡¡ d S r&   )r   r   r   r   r   r	   r?   r*   r   r    r!   )r   r   r#   r$   r   r   r   Ú5test_compat32_max_line_length_does_not_fold_when_none¥   s
    
zGTestGeneratorBase.test_compat32_max_line_length_does_not_fold_when_nonec                 C   sd   |   |  t d¡¡¡}t d¡}|  ¡ }| j|| jjddd}| |¡ |  	| 
¡ |  |¡¡ d S )Nú—            To: nobody
            Content-Disposition: attachment;
             filename="afilenamelongenoghtowraphere"

            None
            z½            To: nobody
            Content-Disposition: attachment;
             filename*0*=us-ascii''afilename;
             filename*1*=longenoghtowraphere

            None
            é!   r'   r   ©r   r   r=   r>   r   r   r	   r*   r   r    r!   ©r   r   r7   r#   r$   r   r   r   Útest_rfc2231_wrapping¬   s    

z'TestGeneratorBase.test_rfc2231_wrappingc                 C   sd   |   |  t d¡¡¡}t d¡}|  ¡ }| j|| jjddd}| |¡ |  	| 
¡ |  |¡¡ d S )NrE   z¯            To: nobody
            Content-Disposition:
             attachment;
             filename*0*=us-ascii''afilenamelongenoghtowraphere

            None
            r   r'   r   rG   rH   r   r   r   Ú;test_rfc2231_wrapping_switches_to_default_len_if_too_narrowÃ   s    

zMTestGeneratorBase.test_rfc2231_wrapping_switches_to_default_len_if_too_narrow)N)Ú__name__Ú
__module__Ú__qualname__r	   r@   r   r=   r>   r   Úcopyr1   Zlength_paramsr%   r+   r-   r/   r2   r8   r;   rC   rD   rI   rJ   r   r   r   r   r      s0   
	ê*ÿ
ÿ
		r   c                   @   s"   e Zd ZeeƒZeZej	Z
eZdS )ÚTestGeneratorN)rK   rL   rM   Ústaticmethodr   r   r   r   ÚioÚStringIOr   Ústrr   r   r   r   r   rO   Ü   s   rO   c                   @   sF   e Zd ZeeƒZeZej	Z
dd„ Zdd„ Zdd„ Zdd„ Zd	d
„ ZdS )ÚTestBytesGeneratorc                 C   s
   |  d¡S )NÚascii)Úencode)r   Úxr   r   r   Ú<lambda>é   r   zTestBytesGenerator.<lambda>c                 C   sX   d  d¡}d  d¡}t|ƒ}t ¡ }t|| jjddd}| |¡ |  | 	¡ |¡ d S )Nu5   Subject: Maintenant je vous prÃ©sente mon collÃ¨gue

úutf-8zPSubject: Maintenant je vous =?unknown-8bit?q?pr=C3=A9sente_mon_coll=C3=A8gue?=

rU   Ú7bit)Úcte_typer   )
rV   r   rQ   ÚBytesIOr   r	   r*   r   r    r!   r5   r   r   r   Ú'test_cte_type_7bit_handles_unknown_8bitë   s    ÿÿ
z:TestBytesGenerator.test_cte_type_7bit_handles_unknown_8bitc                 C   sf   t  d¡ d¡}t|ƒ}t  d¡ d¡}t ¡ }t|| jjdddd}| 	|¡ |  
| ¡ |¡ d S )	Nu/              From: foo@bar.com
            To: Dinsdale
            Subject: Nudge nudge, wink, wink
            Mime-Version: 1.0
            Content-Type: text/plain; charset="latin-1"
            Content-Transfer-Encoding: 8bit

            oh lÃ  lÃ , know what I mean, know what I mean?
            Úlatin1a@              From: foo@bar.com
            To: Dinsdale
            Subject: Nudge nudge, wink, wink
            Mime-Version: 1.0
            Content-Type: text/plain; charset="iso-8859-1"
            Content-Transfer-Encoding: quoted-printable

            oh l=E0 l=E0, know what I mean, know what I mean?
            rU   rZ   r9   )r[   r:   r   )r=   r>   rV   r   rQ   r\   r   r	   r*   r   r    r!   )r   r6   r   r7   r#   r$   r   r   r   Ú&test_cte_type_7bit_transforms_8bit_cteö   s    
	÷

	÷
ÿ

z9TestBytesGenerator.test_cte_type_7bit_transforms_8bit_ctec                 C   st   t ƒ }d|d< d|d< d|d< | d¡ t d¡ d	¡ d
d¡}t ¡ }t|t	j
d}| |¡ |  | ¡ |¡ d S )Nu   PÃ¡olo <fÅ‘o@bar.com>ÚFromÚDinsdaleÚTou   Nudge nudge, wink, wink á½ 9ÚSubjectu/   oh lÃ  lÃ , know what I mean, know what I mean?u<              From: PÃ¡olo <fÅ‘o@bar.com>
            To: Dinsdale
            Subject: Nudge nudge, wink, wink á½ 9
            Content-Type: text/plain; charset="utf-8"
            Content-Transfer-Encoding: 8bit
            MIME-Version: 1.0

            oh lÃ  lÃ , know what I mean, know what I mean?
            rY   ó   
ó   
r   )r   Úset_contentr=   r>   rV   rA   rQ   r\   r   r	   ZSMTPUTF8r   r    r!   rH   r   r   r   Útest_smtputf8_policy  s    

	÷	÷

z'TestBytesGenerator.test_smtputf8_policyc                 C   s‚   t ƒ }tddd|d< tddd|d< d|d	< | d
¡ t d¡ ¡  dd¡}t ¡ }t	|t
jd}| |¡ |  | ¡ |¡ d S )Nzfoo@bar.comu   PÃ¡olo)Z	addr_specZdisplay_namer`   zbar@foo.comra   rb   zNudge nudge, wink, winkrc   z+oh boy, know what I mean, know what I mean?aP              From: =?utf-8?q?P=C3=A1olo?= <foo@bar.com>
            To: Dinsdale <bar@foo.com>
            Subject: Nudge nudge, wink, wink
            Content-Type: text/plain; charset="utf-8"
            Content-Transfer-Encoding: 7bit
            MIME-Version: 1.0

            oh boy, know what I mean, know what I mean?
            rd   re   r   )r   r   rf   r=   r>   rV   rA   rQ   r\   r   r	   r4   r   r    r!   rH   r   r   r   Útest_smtp_policy'  s    
	÷

z#TestBytesGenerator.test_smtp_policyN)rK   rL   rM   rP   r   r   r   r   rQ   r\   r   r   r]   r_   rg   rh   r   r   r   r   rT   ä   s   rT   Ú__main__)rQ   r=   ZunittestZemailr   r   Zemail.messager   Zemail.generatorr   r   Zemail.headerregistryr   r	   Ztest.test_emailr
   r   r   rO   rT   rK   Úmainr   r   r   r   Ú<module>   s    PY