Your IP : 216.73.216.196
o
0�i�7 � @ 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 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 �E/opt/alt/python310/lib64/python3.10/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 s 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_ �
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 �
�
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 r0 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 �
)�linesepr4 r6 r r r �%test_flatten_linesep_overrides_policy� s z7TestGeneratorBase.test_flatten_linesep_overrides_policyc C s� t �d�}dtjdftjdftjjdd�dff}|D ]H\}}|r&|�dd�n|}| j||d��+ | �| � |��}| �
� }| j ||d �}|� |� | �
|�� | � |�� W d � n1 s]w Y qd 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 r7 Zvariants�pZmangler8 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� r+ zGTestGeneratorBase.test_compat32_max_line_length_does_not_fold_when_nonec C �d | � | �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 r8 r" r# r r r �test_rfc2231_wrapping� s
z'TestGeneratorBase.test_rfc2231_wrappingc C rG )NrH z� To: nobody
Content-Disposition:
attachment;
filename*0*=us-ascii''afilenamelongenoghtowraphere
None
r r& r rJ rK 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 rJ rK r r r �test_keep_encoded_newlines� s
z,TestGeneratorBase.test_keep_encoded_newlinesc C rG )NrN z� To: nobody
Subject: Bad subject
=?utf-8?q?=0A?=Bcc:
injection@example.com
None
� r& r rJ rK r r r � test_keep_long_encoded_newlines� s
z1TestGeneratorBase.test_keep_long_encoded_newlines�N)�__name__�
__module__� __qualname__r rA r r> r? r �copyr2 Z
length_paramsr$ r* r- r/ r3 r9 r< rE rF rL rM rP rR r r r r r
s6
� *�
�
r c @ s* e Zd Zee�ZeZej Z
e Z dd� Z
dS )�
TestGeneratorc
C s� G dd� dt �}dD ]^}| j|d��N td| jd�}|d= ||�|d< | �tjj�� |�� W d � n1 s9w Y | �tjj�� |� � W d � n1 sTw Y W d � n1 scw Y q
d S )Nc @ s e Zd ZdZdd� ZdS )zBTestGenerator.test_verify_generated_headers.<locals>.LiteralHeader�Headerc [ s | S rS r )r �kwargsr r r �fold s zGTestGenerator.test_verify_generated_headers.<locals>.LiteralHeader.foldN)rT rU rV �namer[ r r r r �
LiteralHeader s r] )zValue
Bad Injection
Z NoNewLine)�textzHeader: Value
Bodyr rY )
�strrC r r Z assertRaises�email�errorsZHeaderWriteErrorZ as_stringZas_bytes)r r] r^ �messager r r � test_verify_generated_headers
s& �
�
�� ���z+TestGenerator.test_verify_generated_headersN)rT rU rV � staticmethodr r
r r �io�StringIOr r_ r rc r r r r rX s rX 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>) s
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?=
rh �7bit)�cte_typer )
ri r re �BytesIOr r r) r r r r6 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?
rh rm r: )rn r; r ) r> r? ri r re ro r r r) r r r )r r7 r r8 r" r# r r r �&test_cte_type_7bit_transforms_8bit_cte6 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?
rl �
�
r )r � set_contentr> r? ri rB re ro r r ZSMTPUTF8r r r rK r r r �test_smtputf8_policyR 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_namers z bar@foo.comrt ru zNudge nudge, wink, winkrv 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?
rw rx r )r r ry r> r? ri rB re ro r r r5 r r r rK r r r �test_smtp_policyg s
�
z#TestBytesGenerator.test_smtp_policyN)rT rU rV rd r r
r r re ro r r rp rr rz r{ r r r r rg $ s rg �__main__)re r> Zunittestr` r r Z
email.messager Zemail.generatorr r Zemail.headerregistryr r Z email.errorsZtest.test_emailr
r r rX rg rT �mainr r r r �<module> s$ v!Y �