Your IP : 216.73.216.52
U
i�fm� � @ s� d dl Z d dlmZmZ d dlZd dlmZ d dlmZ d dl m
Z
m Z eG dd� de��Z eG dd � d e��Z
ed
kr�e �� dS ) � N)�
TestEmailBase� parameterize)�policy)� EmailMessage)�ContentManager�raw_data_managerc
@ s� e Zd ZejZeZdddd�Zdd� Zdd� Z d d
� Z
G d d � d e �Z e d
d Z
G dd� de �Ze d
d Zdefdefddde fde
fddde fdd d d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)S )*�TestContentManager)� �
text/plain)� �text)� � )Z full_typeZ
maintype_only�null_keyc C sN d
dd�}t � }|�||� | �� }d|d< d|d< | �|j|dd�d � d S ) Nc S s | d }||fS �N� X-Bar-Header� )�msg�foo�barr r �H/opt/alt/python38/lib64/python3.8/test/test_email/test_contentmanager.py�
foo_getter s zATestContentManager.get_key_as_get_content_key.<locals>.foo_getterr
� Content-Typer r r �r )r r )N)r �add_get_handler�
_make_message� assertEqual� get_content)�self�order�keyr �cm�mr r r �get_key_as_get_content_key s
z-TestContentManager.get_key_as_get_content_keyc C s� dd� }dd� }t � }|�||� | j�� D ] \}}||kr,|�||� q,| �� }d|d< d|d< d |d
< | �|�|�d � d S ) Nc S s | d S r r �r r r r �
bar_getter! s zGTestContentManager.get_key_as_get_content_key_order.<locals>.bar_getterc S s | d S �N� X-Foo-Headerr r$ r r r r # s zGTestContentManager.get_key_as_get_content_key_order.<locals>.foo_getterr
r r r r r' )r r �get_key_params�valuesr r r )r r r r% r r! �
precedencer" r r r � get_key_as_get_content_key_order s z3TestContentManager.get_key_as_get_content_key_orderc C s< t � }| �� }d|d< | �td�� |�|� W 5 Q R X d S )Nr
r )r r �assertRaisesRegex�KeyErrorr �r r! r" r r r �:test_get_content_raises_if_unknown_mimetype_and_no_default0 s
zMTestContentManager.test_get_content_raises_if_unknown_mimetype_and_no_defaultc @ s e Zd ZdS )� TestContentManager.BaseThingN��__name__�
__module__� __qualname__r r r r � BaseThing7 s r5 �.r0 c @ s e Zd ZdS )�TestContentManager.ThingNr1 r r r r �Thing: s r8 r7 r r )r r7 )r
r8 � � )� r0 )� r5 � )� z builtins.str)�
�str)� N) �type� full_path�qualname�nameZ base_typeZbase_full_pathZ
base_qualnameZ base_nameZstr_typeZ
str_full_pathZstr_namer c C s` ddd�}t � }|�||� | �� }| �� }|j||dd� | �|d d� | �|�� |� d S )Nc S s || d< | � |� d S r& �� set_payload)r �objr r r r �
foo_setterN s zATestContentManager.set_key_as_set_content_key.<locals>.foo_setterr r r' )N)r �add_set_handlerr r8 � set_contentr � get_payload)r r r rI r! r" �msg_objr r r �set_key_as_set_content_keyM s
z-TestContentManager.set_key_as_set_content_keyc C s� dd� }dd� }t � }|�||� | j�� D ] \}}||kr,|�||� q,| �� }| �� }|�||� | �|d d� | �|�� |� d S )Nc S s d| d< | � |� d S )Nr �X-FooBar-HeaderrF �r rH r r r rI Z s zGTestContentManager.set_key_as_set_content_key_order.<locals>.foo_setterc S s d| d< d S )Nr rO r rP r r r �
bar_setter] s zGTestContentManager.set_key_as_set_content_key_order.<locals>.bar_setterrO r ) r rJ r( r) r r8 rK r rL ) r r r rI rQ r! r* r" rM r r r � set_key_as_set_content_key_orderY s z3TestContentManager.set_key_as_set_content_key_orderc C s@ t � }| �� }| �� }| �t| j�� |�||� W 5 Q R X d S �N)r r r8 r, r- �testobject_full_pathrK )r r! r" rM r r r �6test_set_content_raises_if_unknown_type_and_no_defaultj s
zITestContentManager.test_set_content_raises_if_unknown_type_and_no_defaultc C s< t � }| �� }d|d< | �t�� |�|d� W 5 Q R X d S )Nz
multipart/foor �test)r r � assertRaises� TypeErrorrK r. r r r �.test_set_content_raises_if_called_on_multipartq s
zATestContentManager.test_set_content_raises_if_called_on_multipartc C s� | � � }d|d< d|d< d|d< |�d� t� }|�tdd � � |jd
|d � | �|d � | �|d � | �|d d� | �|�� � d S ) Nr z Content-Fooz text/htmlr rV �To�abcc _ s d S rS r )�args�kwr r r �<lambda> � zITestContentManager.test_set_content_calls_clear_content.<locals>.<lambda>�xyz)�content_manager) r rG r rJ r@ rK Z assertIsNoner rL )r r" r! r r r �$test_set_content_calls_clear_contentx s
z7TestContentManager.test_set_content_calls_clear_contentN)r2 r3 r4 r �defaultr �messager( r# r+ r/ r@ r5 Zbaseobject_full_pathr8 rT Zset_key_paramsrN rR rU rY rb r r r r r s> � � r c @ s� e Zd Zejjded�ZeZdd� Z dd� Z
dd� Z d d
� Z d d � Z
d
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Z d+d,� Z d-d.� Z d/d0� Z d1d2� Z d3d4� Z!d5d6� Z"d7d8� Z#d9d:� Z$d;d<� Z%d=d>� Z&d?d@� Z'dAdB� Z(dCdD� Z)dEdF� Z*dGdH� Z+dIdJ� Z,dKdL� Z-dMdN� Z.dOdP� Z/dQdR� Z0dSdT� Z1dUdV� Z2dWdX� Z3dYdZ� Z4d[d\� Z5d]d^� Z6d_d`� Z7dadb� Z8dcdddedfe� dgfe� dhfdi�Z9djdk� Z:dldm� Z;dndo� Z<dpdq� Z=drds� Z>dtS )u�TestRawDataManager�< )�max_line_lengthra c C s& | � t�d��}| �t�|�d� d S )N�J Content-Type: text/plain
Basic text.
z Basic text.
��_str_msg�textwrap�dedentr r r �r r" r r r �test_get_text_plain� s z&TestRawDataManager.test_get_text_plainc C s& | � t�d��}| �t�|�d� d S )NzP Content-Type: text/html
<p>Basic text.</p>
z<p>Basic text.</p>
ri rm r r r �test_get_text_html� s �z%TestRawDataManager.test_get_text_htmlc C s, | � t�d��d��}| �t�|�d� d S )Nu\ Content-Type: text/plain; charset=latin1
Basìc tëxt.
�latin1� Basìc tëxt.
)Z
_bytes_msgrk rl �encoder r r rm r r r �test_get_text_plain_latin1� s �z-TestRawDataManager.test_get_text_plain_latin1c C s& | � t�d��}| �t�|�d� d S )Nz� Content-Type: text/plain; charset="latin-1"
Content-Transfer-Encoding: quoted-printable
Bas=ECc t=EBxt.
rq ri rm r r r �+test_get_text_plain_latin1_quoted_printable� s z>TestRawDataManager.test_get_text_plain_latin1_quoted_printablec C s& | � t�d��}| �t�|�d� d S )Nz� Content-Type: text/plain; charset="utf8"
Content-Transfer-Encoding: base64
QmFzw6xjIHTDq3h0Lgo=
rq ri rm r r r � test_get_text_plain_utf8_base64� s z2TestRawDataManager.test_get_text_plain_utf8_base64c C s& | � t�d��}| �t�|�d� d S )N�� Content-Type: text/plain; charset="utf8"
Content-Transfer-Encoding: quoted-printable
Bas=c3=acc t=c3=abxt=fd.
u Basìc tëxt�.
ri rm r r r �-test_get_text_plain_bad_utf8_quoted_printable� s z@TestRawDataManager.test_get_text_plain_bad_utf8_quoted_printablec C s* | � t�d��}| �tj|dd�d� d S )Nrv �ignore��errorsrq ri rm r r r �;test_get_text_plain_bad_utf8_quoted_printable_ignore_errors� s �zNTestRawDataManager.test_get_text_plain_bad_utf8_quoted_printable_ignore_errorsc C s* | � t�d��}| �tj|dd�d� d S )Nu� Content-Type: text/plain; charset="utf8"
Content-Transfer-Encoding: base64
QmFzw6xjIHTDq3h0Lgoÿ=
rx ry rq ri rm r r r �8test_get_text_plain_utf8_base64_recoverable_bad_CTE_data� s �zKTestRawDataManager.test_get_text_plain_utf8_base64_recoverable_bad_CTE_datac C s8 | � t�d��}| �t�� tj|dd� W 5 Q R X d S )Nrh rx r �rj rk rl rW rX r r rm r r r � test_get_text_invalid_keyword� s z0TestRawDataManager.test_get_text_invalid_keywordc
C sZ t �d�}d�� D ]B}| j|d��, | �|�|d ��}| �t�|�d� W 5 Q R X qd S )Nzu Content-Type: {}
Content-Transfer-Encoding: base64
Ym9ndXMgZGF0YQ==
z audio image video application)�maintypez/foos
bogus data) rk rl �split�subTestrj �formatr r r )r �templater r" r r r �test_get_non_text� s
z$TestRawDataManager.test_get_non_textc C s8 | � t�d��}| �t�� tj|dd� W 5 Q R X d S )Nz| Content-Type: image/jpg
Content-Transfer-Encoding: base64
Ym9ndXMgZGF0YQ==
rx ry r} rm r r r �!test_get_non_text_invalid_keyword� s z4TestRawDataManager.test_get_non_text_invalid_keywordc C s4 | � t�d��}| �t�� t�|� W 5 Q R X d S )Nzm Content-Type: multipart/mixed; boundary="==="
--===
--===--
)rj rk rl rW r- r r rm r r r � test_get_raises_on_multipart� s z/TestRawDataManager.test_get_raises_on_multipartc
C s� t �d�}d�� D ]~}| j|d��h | �|�|��}t�|�}| �|| j � | �
t�|�d� | �
|d d� | �
|d j d j d � W 5 Q R X qd S )
Nz� Content-Type: message/{}
To: foo@example.com
From: bar@example.com
Subject: example
an example message
�rfc822 external-body��subtypezan example message
�to�foo@example.com�fromr r )
rk rl r� r� rj r� r r �assertIsInstancerd r Z addressesZusername)r r� r� r" Zsub_msgr r r �)test_get_message_rfc822_and_external_body s
�z<TestRawDataManager.test_get_message_rfc822_and_external_bodyc C s. | � t�d��}| �t�|�d d� d� d S )Nz� Content-Type: message/partial
To: foo@example.com
From: bar@example.com
Subject: example
The real body is in another message.
r? s
To: foo@exri rm r r r �9test_get_message_non_rfc822_or_external_body_yields_bytes s zLTestRawDataManager.test_get_message_non_rfc822_or_external_body_yields_bytesc C s\ | � � }d}t�||� | �t|�t�d�� | �|jdd��d�|� | �|� � |� d S )N�Simple message.
z� Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: 7bit
Simple message.
T��decode�utf-8�
r r rK r r@ rk rl rL r� r �r r" �contentr r r �test_set_text_plain% s z&TestRawDataManager.test_set_text_plainc C s\ | � � }d}t�||� | �t|�t�d�� | �|jdd��d�d� | �|� � d� d S )Nr zp Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: 7bit
Tr� r� �
r� r� r r r �test_set_text_plain_null2 s z+TestRawDataManager.test_set_text_plain_nullc C s` | � � }d}tj||dd� | �t|�t�d�� | �|jdd��d�|� | �|� � |� d S )Nz<p>Simple message.</p>
�htmlr� z� Content-Type: text/html; charset="utf-8"
Content-Transfer-Encoding: 7bit
<p>Simple message.</p>
Tr� r� r� r� r r r �test_set_text_html? s z%TestRawDataManager.test_set_text_htmlc C s` | � � }d}tj||dd� | �t|�t�d�� | �|jdd��d�|� | �|� � |� d S )Nr� �latin-1��charsetz� Content-Type: text/plain; charset="iso-8859-1"
Content-Transfer-Encoding: 7bit
Simple message.
Tr� r� r� r� r r r � test_set_text_charset_latin_1L s z0TestRawDataManager.test_set_text_charset_latin_1c C s\ | � � }d}t�||� | �t|�t�d�� | �|jdd��d�|� | �|� � |� d S )NzTSimple but long message that is over 78 characters long to force transfer encoding.
z� Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: quoted-printable
Simple but long message that is over 78 characters long to =
force transfer encoding.
Tr� r� r� r� r r r �(test_set_text_plain_long_line_heuristicsY s z;TestRawDataManager.test_set_text_plain_long_line_heuristicsc C sb | � � }d}t�||� | �t|�t�d��d�� | �|jdd�� d�|� | �|�
� |� d S )Nu7 et là il est monté sur moi et il commence à m'éto.
u� Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: 8bit
et là il est monté sur moi et il commence à m'éto.
r� Tr� � r r rK r �bytesrk rl rr rL r� r r� r r r �5test_set_text_short_line_minimal_non_ascii_heuristicsh s �zHTestRawDataManager.test_set_text_short_line_minimal_non_ascii_heuristicsc C sb | � � }d}t�||� | �t|�t�d��d�� | �|jdd�� d�|� | �|�
� |� d S )Nus j'ai un problème de python. il est sorti de son vivarium. et là il est monté sur moi et il commence à m'éto.
a* Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: quoted-printable
j'ai un probl=C3=A8me de python. il est sorti de son vivari=
um. et l=C3=A0 il est mont=C3=A9 sur moi et il commence =
=C3=A0 m'=C3=A9to.
r� Tr� r� r� r r r �4test_set_text_long_line_minimal_non_ascii_heuristicsu s �zGTestRawDataManager.test_set_text_long_line_minimal_non_ascii_heuristicsc C sb | � � }d}t�||� | �t|�t�d��d�� | �|jdd�� d�|� | �|�
� |� d S )Nu}
j'ai un problème de python. il est sorti de son vivarium. et là il est monté sur moi et il commence à m'éto.
a@ Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: quoted-printable
j'ai un probl=C3=A8me de python. il est sorti de son vivari=
um. et l=C3=A0 il est mont=C3=A9 sur moi et il commence =
=C3=A0 m'=C3=A9to.
r� Tr� r� r� r r r �=test_set_text_11_lines_long_line_minimal_non_ascii_heuristics� s �zPTestRawDataManager.test_set_text_11_lines_long_line_minimal_non_ascii_heuristicsc C sb | � � }d}t�||� | �t|�t�d��d�� | �|jdd�� d�|� | �|�
� |� d S )N� áàäéèęöő.
u� Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: 8bit
áàäéèęöő.
r� Tr� r� r� r r r �*test_set_text_maximal_non_ascii_heuristics� s �z=TestRawDataManager.test_set_text_maximal_non_ascii_heuristicsc C sb | � � }d}t�||� | �t|�t�d��d�� | �|jdd�� d�|� | �|�
� |� d S )Nu
áàäéèęöő.
u� Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: 8bit
áàäéèęöő.
r� Tr� r� r� r r r �3test_set_text_11_lines_maximal_non_ascii_heuristics� s �zFTestRawDataManager.test_set_text_11_lines_maximal_non_ascii_heuristicsc C sb | � � }d}t�||� | �t|�t�d��d�� | �|jdd�� d�|� | �|�
� |� d S )Nu� áàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöő.
a� Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: base64
w6HDoMOkw6nDqMSZw7bFkcOhw6DDpMOpw6jEmcO2xZHDocOgw6TDqcOoxJnD
tsWRw6HDoMOkw6nDqMSZw7bFkcOhw6DDpMOpw6jEmcO2xZHDocOgw6TDqcOo
xJnDtsWRw6HDoMOkw6nDqMSZw7bFkcOhw6DDpMOpw6jEmcO2xZHDocOgw6TD
qcOoxJnDtsWRw6HDoMOkw6nDqMSZw7bFkcOhw6DDpMOpw6jEmcO2xZHDocOg
w6TDqcOoxJnDtsWRLgo=
r� Tr� r� r� r r r �4test_set_text_long_line_maximal_non_ascii_heuristics� s �
zGTestRawDataManager.test_set_text_long_line_maximal_non_ascii_heuristicsc C sb | � � }d}t�|d� | �t|�t�d��d�� | �|jdd�� d�|� | �|�
� |� d S )Nu�
áàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöő.
ak Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: quoted-printable
=C3=A1=C3=A0=C3=A4=C3=A9=C3=A8=C4=99=C3=B6=C5=91=C3=A1=C3=
=A0=C3=A4=C3=A9=C3=A8=C4=99=C3=B6=C5=91=C3=A1=C3=A0=C3=A4=
=C3=A9=C3=A8=C4=99=C3=B6=C5=91=C3=A1=C3=A0=C3=A4=C3=A9=C3=
=A8=C4=99=C3=B6=C5=91=C3=A1=C3=A0=C3=A4=C3=A9=C3=A8=C4=99=
=C3=B6=C5=91=C3=A1=C3=A0=C3=A4=C3=A9=C3=A8=C4=99=C3=B6=C5=
=91=C3=A1=C3=A0=C3=A4=C3=A9=C3=A8=C4=99=C3=B6=C5=91=C3=A1=
=C3=A0=C3=A4=C3=A9=C3=A8=C4=99=C3=B6=C5=91=C3=A1=C3=A0=C3=
=A4=C3=A9=C3=A8=C4=99=C3=B6=C5=91=C3=A1=C3=A0=C3=A4=C3=A9=
=C3=A8=C4=99=C3=B6=C5=91=C3=A1=C3=A0=C3=A4=C3=A9=C3=A8=C4=
=99=C3=B6=C5=91=C3=A1=C3=A0=C3=A4=C3=A9=C3=A8=C4=99=C3=B6=
=C5=91.
r� Tr� r� r� r r r �=test_set_text_11_lines_long_line_maximal_non_ascii_heuristics� s �zPTestRawDataManager.test_set_text_11_lines_long_line_maximal_non_ascii_heuristicsc C s2 | � � }| �t�� tj|ddd� W 5 Q R X d S )Nr� �7bit��cte�r rW � UnicodeErrorr rK rm r r r �,test_set_text_non_ascii_with_cte_7bit_raises� s z?TestRawDataManager.test_set_text_non_ascii_with_cte_7bit_raisesc C s2 | � � }| �t�� tj|ddd� W 5 Q R X d S )Nr� �asciir� r� rm r r r �1test_set_text_non_ascii_with_charset_ascii_raises� s zDTestRawDataManager.test_set_text_non_ascii_with_charset_ascii_raisesc C s4 | � � }| �t�� tj|dddd� W 5 Q R X d S )Nr� r� r� )r� r� r� rm r r r �>test_set_text_non_ascii_with_cte_7bit_and_charset_ascii_raises� s zQTestRawDataManager.test_set_text_non_ascii_with_cte_7bit_and_charset_ascii_raisesc C s� | � � }d|d< | � � }d|d< d|d< d|d< |�d� t�||� | �t|�t�d �� |�d
�}| �|| j � | �t|�t|�� | �|�
� | j � | �t|�
� �t|�� d S ) NzForwarded message�Subjectzpython@vivarium.orgrZ �police@monty.org�Fromzget back in your boxzOr face the comfy chair.a� Subject: Forwarded message
Content-Type: message/rfc822
Content-Transfer-Encoding: 8bit
To: python@vivarium.org
From: police@monty.org
Subject: get back in your box
Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: 7bit
MIME-Version: 1.0
Or face the comfy chair.
r ) r rK r r r@ rk rl rL r� rd r )r r" r� Zpayloadr r r �test_set_message� s
z#TestRawDataManager.test_set_messagec C s� | � � }d|d< | � � }d|d< d|d< d|d< |�d� t�||� | �t|�t�d ��d
�� | �|j| j j
d �t�d �� | � |� � | j
� | �t|� � �t|�� d S )
Nz
Escape reportr� r� rZ zvictim@monty.orgr� ZHelpu: j'ai un problème de python. il est sorti de son vivarium.u� Subject: Escape report
Content-Type: message/rfc822
Content-Transfer-Encoding: 8bit
To: police@monty.org
From: victim@monty.org
Subject: Help
Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: 8bit
MIME-Version: 1.0
j'ai un problème de python. il est sorti de son vivarium.
r� )Z maxheaderlena� Subject: Escape report
Content-Type: message/rfc822
Content-Transfer-Encoding: 8bit
To: police@monty.org
From: victim@monty.org
Subject: Help
Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: base64
MIME-Version: 1.0
aidhaSB1biBwcm9ibMOobWUgZGUgcHl0aG9uLiBpbCBlc3Qgc29ydGkgZGUgc29uIHZpdmFyaXVt
Lgo=
)r rK r r r� rk rl rr Z as_stringr rg r� r rd r@ r� r r r �4test_set_message_with_non_ascii_and_coercion_to_7bit s
��zGTestRawDataManager.test_set_message_with_non_ascii_and_coercion_to_7bitc C s | � � }| � � }d�� D ]t}d�� D ]f}| j||d��N | �t��}|j|||d� W 5 Q R X t|j�}| �||� | �||� W 5 Q R X q$qd}d�� D ]f}| j||d��N | �t��}|j|||d� W 5 Q R X t|j�}| �||� | �||� W 5 Q R X q�d S )Nzquoted-printable base64r� )r� r� r� �
external-bodyz 8bit binary) r r� r� rW �
ValueErrorrK r@ Z exceptionZassertIn)r r" r� r� r� Zar�excr r r �#test_set_message_invalid_cte_raises@ s$
z6TestRawDataManager.test_set_message_invalid_cte_raisesc
C s� dt d�td�fD ]n}| j|d��X | �� }t�||dd� | �t|�t� d�� | �|j
dd�|� | �|� � |� W 5 Q R X qd S )Ns
bogus content)r� �image�jpegz� Content-Type: image/jpeg
Content-Transfer-Encoding: base64
Ym9ndXMgY29udGVudA==
Tr� ) � bytearray�
memoryviewr� r r rK r r@ rk rl rL r )r r� r" r r r �test_set_image_jpgT s �z%TestRawDataManager.test_set_image_jpgc C sb | � � }d}|j|dddd� | �t|�t�d��d�� | �|jdd �|� | �|�� |� d S )
N�t b�gus con
t
ent zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzZaudioZaifzquoted-printabler� a Content-Type: audio/aif
Content-Transfer-Encoding: quoted-printable
MIME-Version: 1.0
b=FFgus=09con=0At=0Dent=20zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz=
zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzr� Tr� � r rK r r� rk rl rr rL r r� r r r �,test_set_audio_aif_with_quoted_printable_cted s �z?TestRawDataManager.test_set_audio_aif_with_quoted_printable_ctec C sj | � � }d}|j|dddd� | �t|�t�d��d�d d � | �|jd
d �|� | �|�� |� d S ) Nr� ZvideoZmpegZbinaryr� z~ Content-Type: video/mpeg
Content-Transfer-Encoding: binary
MIME-Version: 1.0
r� s7 b�gus con
t
ent zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzs= zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzTr� r� r� r r r �#test_set_video_mpeg_with_binary_ctez s �� �
z6TestRawDataManager.test_set_video_mpeg_with_binary_ctec C sj | � � }d}|j|dddd� | �t|�t�d��d�d d � | �|jd
d �|� | �|�� |� d S ) NsM b�gus con
t
ent
zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz
� applicationz octet-streamZ8bitr� z� Content-Type: application/octet-stream
Content-Transfer-Encoding: 8bit
MIME-Version: 1.0
r� s b�gus con
t
ent
s= zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz
Tr� r� r� r r r �/test_set_application_octet_stream_with_8bit_cte� s ���zBTestRawDataManager.test_set_application_octet_stream_with_8bit_ctec C sV | � � }d}| jj}tj|||dd�|dd�|dd�fd� | �t|�t�d�� d S ) Nr� rZ r� r� r� zI'm talking to myself.��headersz� Content-Type: text/plain; charset="utf-8"
To: foo@example.com
From: foo@example.com
Subject: I'm talking to myself.
Content-Transfer-Encoding: 7bit
Simple message.
) r r �header_factoryr rK r r@ rk rl �r r" r� r� r r r �$test_set_headers_from_header_objects� s �z7TestRawDataManager.test_set_headers_from_header_objectsc C s6 | � � }d}tj||dd� | �t|�t�d�� d S )Nr� )zX-Foo-Header: foozX-Bar-Header: barr� z� Content-Type: text/plain; charset="utf-8"
X-Foo-Header: foo
X-Bar-Header: bar
Content-Transfer-Encoding: 7bit
Simple message.
)r r rK r r@ rk rl r� r r r � test_set_headers_from_strings� s z0TestRawDataManager.test_set_headers_from_stringsc C s8 | � � }d}| �td�� tj||dd� W 5 Q R X d S )Nr� r )zContent-Type: foo/barr� )r r, r� r rK r� r r r �<test_set_headers_with_invalid_duplicate_string_header_raises� s zOTestRawDataManager.test_set_headers_with_invalid_duplicate_string_header_raisesc C sH | � � }d}| jj}| �td�� tj|||dd�fd� W 5 Q R X d S )Nr� r z foo/barr� )r r r� r, r� r rK r� r r r �<test_set_headers_with_invalid_duplicate_header_header_raises� s �zOTestRawDataManager.test_set_headers_with_invalid_duplicate_header_header_raisesc C sF | � � }d}| �td��$ tj||dd� t|d j� W 5 Q R X d S )Nr� �a@fairly@@invalid@address)z To: a@fairly@@invalid@addressr� rZ )r r, r� r rK �print�defectsr� r r r �4test_set_headers_with_defective_string_header_raises� s
zGTestRawDataManager.test_set_headers_with_defective_string_header_raisesc C sV | � � }d}| jj}| �td��, tj|||dd�fd� t|d j� W 5 Q R X d S )Nr� r� rZ r� ) r r r� r, r� r rK r� r� r� r r r �4test_set_headers_with_defective_header_header_raises� s �zGTestRawDataManager.test_set_headers_with_defective_header_header_raisesc C s* | � � }|jddd� | �|d d� d S )Nr �inline�� disposition�Content-Disposition�r rK r rm r r r �test_set_disposition_inline� s z.TestRawDataManager.test_set_disposition_inlinec C s* | � � }|jddd� | �|d d� d S )Nr Z
attachmentr� r� r� rm r r r � test_set_disposition_attachment� s z2TestRawDataManager.test_set_disposition_attachmentc C s* | � � }|jddd� | �|d d� d S )Nr r� r� r� rm r r r �test_set_disposition_foo� s z+TestRawDataManager.test_set_disposition_fooc C s* | � � }|jddd� | �|d d� d S )Nr �bar.txt��filenamer� z attachment; filename="bar.txt"r� rm r r r �test_set_filename� s
�z$TestRawDataManager.test_set_filenamec C s, | � � }|jdddd� | �|d d� d S )Nr r� r� �r� r� r� zinline; filename="bar.txt"r� rm r r r �(test_set_filename_and_disposition_inline� s z;TestRawDataManager.test_set_filename_and_disposition_inlinec C s6 | � � }|jddd� | �t|�t�d��d�� d S )Nr u ábárî.txtr� a Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: 7bit
Content-Disposition: attachment;
filename*=utf-8''%C3%A1b%C3%A1r%C3%AE.txt
MIME-Version: 1.0
foo
r� )r rK r r� rk rl rr rm r r r �test_set_non_ascii_filename� s
�z.TestRawDataManager.test_set_non_ascii_filename)r� r )r� )r� )� content)r� Z octet_stream)r� )r� r� r )r� )Z
text_plainZ text_htmlZapplication_octet_streamZ
image_jpegZmessage_rfc822Zmessage_external_bodyc C sB | � � }|j|f|�ddi� | �|d d� | �|d d� d S )Nr� )zTo: foo@example.comzFrom: bar@simple.netr� r� r� zbar@simple.netr� �r rH �mimetyper" r r r �!content_object_as_header_receiver s z4TestRawDataManager.content_object_as_header_receiverc C s2 | � � }|j|f|�ddi� | �|d d� d S )Nr� r� r� r� r� r r r �-content_object_as_disposition_inline_receiver s z@TestRawDataManager.content_object_as_disposition_inline_receiverc C sZ | � � }|j|f|�ddd�� | �|d d� | �|�� d� | �|d jd d� d S )Nr� u bár.txtr� r� u inline; filename="bár.txt"r� )r rK r � get_filename�paramsr� r r r �-content_object_as_non_ascii_filename_receiver" s
z@TestRawDataManager.content_object_as_non_ascii_filename_receiverc C s2 | � � }|j|f|�ddi� | �|d d� d S )NZcidZsome_random_stuffz
Content-IDr� r� r r r � content_object_as_cid_receiver) s z1TestRawDataManager.content_object_as_cid_receiverc C sP | � � }ddd�}|j|f|�d|i� t|t�r:d|d< | �|d j|� d S )Nu bárr` )r r[ r� r� r� r )r rK �
isinstancer@ r r� )r rH r� r" r� r r r �!content_object_as_params_receiver. s
z4TestRawDataManager.content_object_as_params_receiverN)?r2 r3 r4 r rc Zcloner r rd rn ro rs rt ru rw r{ r| r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� Zcontent_object_paramsr� r� r� r� r� r r r r re � s� �
0
�
re �__main__)ZunittestZtest.test_emailr r rk Zemailr Z
email.messager Zemail.contentmanagerr r r re r2 �mainr r r r �<module> s } 4