Your IP : 216.73.216.213
a
XC?hR7 � @ s� d dl Z d dlZd dlZd dlmZmZ d dlmZ d dlm Z m
Z
d dl m Z d dlm
Z
d dlZd dlmZmZ eG dd � d ��ZG d
d � d ee�ZG d d
� d
ee�Zedkr�e�� dS )� N)�message_from_string�message_from_bytes)� EmailMessage)� Generator�BytesGenerator)�Address��policy)�
TestEmailBase� parameterizec @ s� e Zd ZejZd(dd�Ze�d�e�d�e�d�d�Zed ed < e� � Z
d
e
d< d e
d < d d
� eD �Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� ZdS ))�TestGeneratorBaseNc C s |d u r| j n|}| j||d�S )Nr )r �msgfunc)�self�msgr � r �C/opt/alt/python39/lib64/python3.9/test/test_email/test_generator.py�msgmaker s zTestGeneratorBase.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 | ]}|�qS r r )�.0�nr r r �
<listcomp>V � z TestGeneratorBase.<listcomp>c C sZ | � | �| jd ��}| �� }| j||| jd�}|�|� | �|�� | �| 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_parameterX s
z2TestGeneratorBase.length_as_maxheaderlen_parameterc C s` | � | �| jd ��}| �� }| j|| jj|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 | � | �| jd ��}| �� }| j||| jjdd�d�}|�|� | �|� � | �| j| �� d S )Nr �
r' r r) r" r r r �,length_as_maxheaderlen_parm_overrides_policyf s �
z>TestGeneratorBase.length_as_maxheaderlen_parm_overrides_policyc C sb | � | �| jd ��}| �� }| j|| jjd|d�d�}|�|� | �|� � | �| jd �� d S )Nr Znone�Z
refold_sourcer( r r) r" r r r �8length_as_max_line_length_with_refold_none_does_not_foldn s �
zJTestGeneratorBase.length_as_max_line_length_with_refold_none_does_not_foldc C sb | � | �| jd ��}| �� }| j|| jjd|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_foldsv s �
zATestGeneratorBase.length_as_max_line_length_with_refold_all_foldsc C sT d}|}| � | �|��}| �� }| j|tjd�}|�|� | �|�� | �|�� 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|tjd�}|j|dd� | �|�� | �|�� d S )NzSubject: 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�}dtjdftjdftjjdd�dff}|D ]�\}}|rL|�dd�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_zFrom z>From )r r< r )�textwrap�dedentr �compat32�defaultr* �replace�subTestr r r r r r r! ) r r6 Zvariants�pZmangler7 r r# r$ r r r � test_set_mangle_from_via_policy� s
�
z1TestGeneratorBase.test_set_mangle_from_via_policyc C s` | � | �| jd ��}| �� }| j|tjjd d�d�}|�|� | � |�
� | �| jd �� 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|| jjdd�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|| jjdd�d�}|�|� | � |�
� | �|�� d S )NrF z� To: nobody
Content-Disposition:
attachment;
filename*0*=us-ascii''afilenamelongenoghtowraphere
None
r r' r rH rI r r r �;test_rfc2231_wrapping_switches_to_default_len_if_too_narrow� s
zMTestGeneratorBase.test_rfc2231_wrapping_switches_to_default_len_if_too_narrowc C sd | � | �t�d���}t�d�}| �� }| j|| jjdd�d�}|�|� | � |�
� | �|�� d S )N� To: nobody
Subject: Bad subject=?UTF-8?Q?=0A?=Bcc: injection@example.com
None
�P r' r rH rI r r r �test_keep_encoded_newlines� s
z,TestGeneratorBase.test_keep_encoded_newlinesc C sd | � | �t�d���}t�d�}| �� }| j|| jjdd�d�}|�|� | � |�
� | �|�� d S )NrL z� To: nobody
Subject: Bad subject
=?utf-8?q?=0A?=Bcc:
injection@example.com
None
� r' r rH rI r r r � test_keep_long_encoded_newlines� s
z1TestGeneratorBase.test_keep_long_encoded_newlines)N)�__name__�
__module__� __qualname__r r@ r r= r> r �copyr1 Z
length_paramsr% r+ r- r/ r2 r8 r; rD rE rJ rK rN rP r r r r r
s4
�* �
�
r c @ s* e Zd Zee�ZeZej Z
e Z dd� Z
dS )�
TestGeneratorc
C s� G dd� dt �}dD ]�}| j|d��f td| jd�}|d= ||�|d< | �tjj�� |�� W d � n1 sr0 Y W d � q1 s�0 Y qdS ) z=gh-121650: by default the generator prevents header injectionc @ s e Zd ZdZdd� ZdS )zBTestGenerator.test_verify_generated_headers.<locals>.LiteralHeader�Headerc [ s | S )Nr )r �kwargsr r r �fold s zGTestGenerator.test_verify_generated_headers.<locals>.LiteralHeader.foldN)rQ rR rS �namerX r r r r �
LiteralHeader s rZ )zValue
Bad Injection
Z NoNewLine)�textzHeader: Value
Bodyr rV N) �strrB r r Z assertRaises�email�errorsZHeaderWriteErrorZ as_string)r rZ r[ �messager r r � test_verify_generated_headers
s � z+TestGenerator.test_verify_generated_headersN)rQ rR rS � staticmethodr r
r r �io�StringIOr r\ r r` r r r r rU s
rU c @ sF e Zd Zee�ZeZej Z
dd� Z dd� Z dd� Z
dd� Zd d
� Zd S ) �TestBytesGeneratorc C s
|� d�S )N�ascii)�encode)r �xr r r �<lambda>' r zTestBytesGenerator.<lambda>c C sX d� d�}d� d�}t|�}t�� }t|| jjdd�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?=
re �7bit)�cte_typer )
rf r rb �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|| jjddd�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?
re rj r9 )rk r: r ) r= r> rf r rb rl r r r* r r r! )r r6 r r7 r# r$ r r r �&test_cte_type_7bit_transforms_8bit_cte4 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?
ri �
�
r )r � set_contentr= r> rf rA rb rl r r ZSMTPUTF8r r r! rI r r r �test_smtputf8_policyP s
� �
z'TestBytesGenerator.test_smtputf8_policyc C s� t � }tddd�|d< tddd�|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_namerp z bar@foo.comrq rr zNudge nudge, wink, winkrs 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?
rt ru r )r r rv r= r> rf rA rb rl r r r4 r r r! rI r r r �test_smtp_policye s
�
z#TestBytesGenerator.test_smtp_policyN)rQ rR rS ra r r
r r rb rl r r rm ro rw rx r r r r rd " s rd �__main__)rb r= Zunittestr] r r Z
email.messager Zemail.generatorr r Zemail.headerregistryr r Z email.errorsZtest.test_emailr
r r rU rd rQ �mainr r r r �<module> s v Y