Your IP : 216.73.216.213
��Yf� � @ s� d d l Z d d l m Z m Z d d l Z d d l m Z d d l m Z d d l m
Z
m Z e Gd d � d e � � Z e Gd d � d e � � Z
e d
k r� e j � d S) � N)�
TestEmailBase� parameterize)�policy)� EmailMessage)�ContentManager�raw_data_managerc @ sK e Z d Z e j Z e Z d d8 d d9 d d: i Z d
d � Z d d
� Z d d � Z
Gd d � d e � Z e d d Z
Gd d � d e � Z e d d Z d d e f d d e f d d; d d< d d e f d d e
f d d= d" d> d$ d% e f d& d? d) d@ d dA i Z d. d/ � Z d0 d1 � Z d2 d3 � Z d4 d5 � Z d6 d7 � Z d- S)B�TestContentManagerZ full_type� �
text/plainZ
maintype_only� �textZnull_key� � c C sk d d d � } t � } | j | | � | j � } d | d <d | d <| j | j | d d �d � d S) Nc S s | d } | | f S)Nz X-Bar-Header� )�msg�foo�barr r �H/opt/alt/python35/lib64/python3.5/test/test_email/test_contentmanager.py�
foo_getter s
zATestContentManager.get_key_as_get_content_key.<locals>.foo_getterz
text/plainz Content-Typer z X-Bar-Headerr )r r )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� d d � } d d � } t � } | j | | � x9 | j j � D]( \ } } | | k rA | j | | � qA W| j � } d | d <d | d <d | d
<| j | j | � d � d S) Nc S s | d S)Nz X-Bar-Headerr )r r r r �
bar_getter! s zGTestContentManager.get_key_as_get_content_key_order.<locals>.bar_getterc S s | d S)Nz X-Foo-Headerr )r r r r r # s zGTestContentManager.get_key_as_get_content_key_order.<locals>.foo_getterz
text/plainz Content-Typer z X-Bar-Headerr z X-Foo-Header)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 sJ t � } | j � } d | d <| j t d � � | j | � Wd QRXd S)Nz
text/plainz Content-Type)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 Z d Z d S)z TestContentManager.BaseThingN)�__name__�
__module__� __qualname__r r r r � BaseThing7 s r* �.� TestContentManager.BaseThingc @ s e Z d Z d S)zTestContentManager.ThingN)r' r( r) r r r r �Thing: s r- �TestContentManager.Thing�typer � full_pathZqualname�nameZ base_type� Zbase_full_path� Z
base_qualname� Z base_name� Zstr_type� Z
str_full_path� � builtins.strZstr_name�
�str� Nc C s� d d d � } t � } | j | | � | j � } | j � } | j | | d d �| j | d d � | j | j � | � d S)Nc S s | | d <| j | � d S)Nz X-Foo-Header)� set_payload)r �objr r r r �
foo_setterN s
zATestContentManager.set_key_as_set_content_key.<locals>.foo_setterr r z X-Foo-Header)r �add_set_handlerr r- � set_contentr � get_payload)r r r r> r r �msg_objr r r �set_key_as_set_content_keyM s z-TestContentManager.set_key_as_set_content_keyc C s� d d � } d d � } t � } | j | | � x9 | j j � D]( \ } } | | k rA | j | | � qA W| j � } | j � } | j | | � | j | d d � | j | j � | � d S)Nc S s d | d <| j | � d S)Nr zX-FooBar-Header)r< )r r= r r r r> Z s
zGTestContentManager.set_key_as_set_content_key_order.<locals>.foo_setterc S s d | d <d S)Nr zX-FooBar-Headerr )r r= r r r �
bar_setter] s zGTestContentManager.set_key_as_set_content_key_order.<locals>.bar_setterzX-FooBar-Headerr ) r r? r r! r r- r@ r rA ) r r r r> rD r r" r rB r r r � set_key_as_set_content_key_orderY s z3TestContentManager.set_key_as_set_content_key_orderc
C sR t � } | j � } | j � } | j t | j � � | j | | � Wd QRXd S)N)r r r- r$ r% �testobject_full_pathr@ )r r r rB 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 sJ t � } | j � } d | d <| j t � � | j | d � Wd QRXd S)Nz
multipart/fooz Content-Type�test)r r � assertRaises� TypeErrorr@ )r r r r r r �.test_set_content_raises_if_called_on_multipartq s
zATestContentManager.test_set_content_raises_if_called_on_multipartc C s� | j � } d | d <d | d <d | d <| j d � t � } | j t d d � � | j d
d | �| j | d � | j | d � | j | d d � | j | j � � d S) Nr z Content-Fooz text/htmlz Content-TyperH �To�abcc _ s d S)Nr )�args�kwr r r �<lambda> s zITestContentManager.test_set_content_calls_clear_content.<locals>.<lambda>�xyz�content_manager) r r< r r? r: r@ Z assertIsNoner rA )r r r r r r �$test_set_content_calls_clear_contentx s
z7TestContentManager.test_set_content_calls_clear_content)r r
)r r )r
r )r r. )r
r- )r4 r, )r5 r* )r7 r8 )r9 r: )r; N)r' r( r) r �defaultr �messager r r# r& r: r* Zbaseobject_full_pathr- rF Zset_key_paramsrC rE rG rK rS r r r r r s: r c
@ s� e Z d Z e j j d d d e � Z e 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 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 d0 d1 � Z d2 d3 � Z d4 d5 � Z! d6 d7 � Z" d8 d9 � Z# d: d; � Z$ d< d= � Z% d> d? � Z& d@ dA � Z' dB dC � Z( dD dE � Z) dF dG � Z* dH dI � Z+ dJ dK � Z, dL dM � Z- dN dO � Z. dP dQ � Z/ dR dS � Z0 dT dU � Z1 dV dW � Z2 dX dY � Z3 dZ d[ � Z4 d\ d] � Z5 d^ d_ � Z6 d` da f f db dz dd d| dh d~ dk e � f f dl e � d f i Z7 dn do � Z8 dp dq � Z9 dr ds � Z: dt du � Z; dv dw � Z< dx S)��TestRawDataManager�max_line_length�< rR c C s5 | j t j d � � } | j t j | � d � d S)NzJ 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 s5 | j t j d � � } | j t j | � d � d S)NzP Content-Type: text/html
<p>Basic text.</p>
z<p>Basic text.</p>
)rY rZ r[ r r r )r r r r r �test_get_text_html� s z%TestRawDataManager.test_get_text_htmlc C s> | j t j d � j d � � } | j t j | � d � d S)Nu\ Content-Type: text/plain; charset=latin1
Basìc tëxt.
�latin1u Basìc tëxt.
)Z
_bytes_msgrZ r[ �encoder r r )r r r r r �test_get_text_plain_latin1� s z-TestRawDataManager.test_get_text_plain_latin1c C s5 | j t j d � � } | j t j | � d � d S)Nz� Content-Type: text/plain; charset="latin-1"
Content-Transfer-Encoding: quoted-printable
Bas=ECc t=EBxt.
u Basìc tëxt.
)rY rZ r[ r r r )r r r r r �+test_get_text_plain_latin1_quoted_printable� s z>TestRawDataManager.test_get_text_plain_latin1_quoted_printablec C s5 | j t j d � � } | j t j | � d � d S)Nz� Content-Type: text/plain; charset="utf8"
Content-Transfer-Encoding: base64
QmFzw6xjIHTDq3h0Lgo=
u Basìc tëxt.
)rY rZ r[ r r r )r r r r r � test_get_text_plain_utf8_base64� s z2TestRawDataManager.test_get_text_plain_utf8_base64c C s5 | j t j d � � } | j t j | � d � d S)Nz� Content-Type: text/plain; charset="utf8"
Content-Transfer-Encoding: quoted-printable
Bas=c3=acc t=c3=abxt=fd.
u Basìc tëxt�.
)rY rZ r[ r r r )r r r r r �-test_get_text_plain_bad_utf8_quoted_printable� s z@TestRawDataManager.test_get_text_plain_bad_utf8_quoted_printablec C s; | j t j d � � } | j t j | d d �d � d S)Nz� Content-Type: text/plain; charset="utf8"
Content-Transfer-Encoding: quoted-printable
Bas=c3=acc t=c3=abxt=fd.
�errors�ignoreu Basìc tëxt.
)rY rZ r[ r r r )r r 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; | j t j d � � } | j t j | d d �d � d S)Nu� Content-Type: text/plain; charset="utf8"
Content-Transfer-Encoding: base64
QmFzw6xjIHTDq3h0Lgoÿ=
rd re u Basìc tëxt.
)rY rZ r[ r r r )r r 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 sF | j t j d � � } | j t � � t j | d d �Wd QRXd S)NzJ Content-Type: text/plain
Basic text.
r re )rY rZ r[ rI rJ r r )r r r r r � test_get_text_invalid_keyword� s z0TestRawDataManager.test_get_text_invalid_keywordc C sy t j d � } xc d j � D]U } | j d | � �: | j | j | d � � } | j t j | � d � Wd QRXq Wd S)Nzu Content-Type: {}
Content-Transfer-Encoding: base64
Ym9ndXMgZGF0YQ==
z audio image video application�maintypez/foos
bogus data) rZ r[ �split�subTestrY �formatr r r )r �templateri r r r r �test_get_non_text� s z$TestRawDataManager.test_get_non_textc C sF | j t j d � � } | j t � � t j | d d �Wd QRXd S)Nz| Content-Type: image/jpg
Content-Transfer-Encoding: base64
Ym9ndXMgZGF0YQ==
rd re )rY rZ r[ rI rJ r r )r r r r r �!test_get_non_text_invalid_keyword� s z4TestRawDataManager.test_get_non_text_invalid_keywordc C s@ | j t j d � � } | j t � � t j | � Wd QRXd S)Nzm Content-Type: multipart/mixed; boundary="==="
--===
--===--
)rY rZ r[ rI r% r r )r r r r r � test_get_raises_on_multipart� s z/TestRawDataManager.test_get_raises_on_multipartc C s� t j d � } x� d j � D]� } | j d | � �� | j | j | � � } t j | � } | j | | j � | j
t j | � d � | j
| d d � | j
| d j d j d � Wd QRXq Wd S)
Nz� Content-Type: message/{}
To: foo@example.com
From: bar@example.com
Subject: example
an example message
zrfc822 external-body�subtypezan example message
�tozfoo@example.com�fromr r )
rZ r[ rj rk rY rl r r �assertIsInstancerU r Z addressesZusername)r rm rq r Zsub_msgr r r �)test_get_message_rfc822_and_external_body s z<TestRawDataManager.test_get_message_rfc822_and_external_bodyc C s? | j t j d � � } | j t j | � 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.
r9 s
To: foo@ex)rY rZ r[ r r r )r r 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� | j � } d } t j | | � | j t | � t j d � � | j | j d d � j d � | � | j | j � | � d S)NzSimple message.
z� Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: 7bit
Simple message.
�decodeTzutf-8)
r r r@ r r: rZ r[ rA rw r )r r �contentr r r �test_set_text_plain% s
%z&TestRawDataManager.test_set_text_plainc C s� | j � } d } t j | | d d �| j t | � t j d � � | j | j d d � j d � | � | j | j � | � d S)Nz<p>Simple message.</p>
rq �htmlz� Content-Type: text/html; charset="utf-8"
Content-Transfer-Encoding: 7bit
<p>Simple message.</p>
rw Tzutf-8)
r r r@ r r: rZ r[ rA rw r )r r rx r r r �test_set_text_html2 s
%z%TestRawDataManager.test_set_text_htmlc C s� | j � } d } t j | | d d �| j t | � t j d � � | j | j d d � j d � | � | j | j � | � d S)NzSimple message.
�charsetzlatin-1z� Content-Type: text/plain; charset="iso-8859-1"
Content-Transfer-Encoding: 7bit
Simple message.
rw Tzutf-8)
r r r@ r r: rZ r[ rA rw r )r r rx r r r � test_set_text_charset_latin_1? s
%z0TestRawDataManager.test_set_text_charset_latin_1c C s� | j � } d } t j | | � | j t | � t j d � j d � � | j | j d d � j d � | � | j | j
� | � 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.
zutf-8rw T) r r r@ r �bytesrZ r[ r_ rA rw r )r r rx r r r �5test_set_text_short_line_minimal_non_ascii_heuristicsL s %zHTestRawDataManager.test_set_text_short_line_minimal_non_ascii_heuristicsc C s� | j � } d } t j | | � | j t | � t j d � j d � � | j | j d d � j d � | � | j | j
� | � 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.
zutf-8rw T) r r r@ r r~ rZ r[ r_ rA rw r )r r rx r r r �4test_set_text_long_line_minimal_non_ascii_heuristicsY s %zGTestRawDataManager.test_set_text_long_line_minimal_non_ascii_heuristicsc C s� | j � } d d } t j | | � | j t | � t j d d
d � j d � � | j | j d d � j d � | � | j | j
� | � d S) N�
r9 us 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.
zz Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: quoted-printable
z�
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.
zutf-8rw Tz
z
) r r r@ r r~ rZ r[ r_ rA rw r )r r rx r r r �=test_set_text_11_lines_long_line_minimal_non_ascii_heuristicsj s %zPTestRawDataManager.test_set_text_11_lines_long_line_minimal_non_ascii_heuristicsc C s� | j � } d } t j | | � | j t | � t j d � j d � � | j | j d d � j d � | � | j | j
� | � d S)Nu áàäéèęöő.
u� Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: 8bit
áàäéèęöő.
zutf-8rw T) r r r@ r r~ rZ r[ r_ rA rw r )r r rx r r r �*test_set_text_maximal_non_ascii_heuristics| s %z=TestRawDataManager.test_set_text_maximal_non_ascii_heuristicsc C s� | j � } d
} t j | | � | j t | � t j d d d � j d � � | j | j d d � j d � | � | j | j
� | � d S) Nr� r9 u áàäéèęöő.
zn Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: 8bit
u+
áàäéèęöő.
zutf-8rw Tz
u
áàäéèęöő.
z
) r r r@ r r~ rZ r[ r_ rA rw r )r r rx r r r �3test_set_text_11_lines_maximal_non_ascii_heuristics� s %zFTestRawDataManager.test_set_text_11_lines_maximal_non_ascii_heuristicsc C s� | j � } d } t j | | � | j t | � t j d � j d � � | j | j d d � j d � | � | j | j
� | � d S)Nu� áàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöő.
a� Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: base64
w6HDoMOkw6nDqMSZw7bFkcOhw6DDpMOpw6jEmcO2xZHDocOgw6TDqcOoxJnD
tsWRw6HDoMOkw6nDqMSZw7bFkcOhw6DDpMOpw6jEmcO2xZHDocOgw6TDqcOo
xJnDtsWRw6HDoMOkw6nDqMSZw7bFkcOhw6DDpMOpw6jEmcO2xZHDocOgw6TD
qcOoxJnDtsWRw6HDoMOkw6nDqMSZw7bFkcOhw6DDpMOpw6jEmcO2xZHDocOg
w6TDqcOoxJnDtsWRLgo=
zutf-8rw T) r r r@ r r~ rZ r[ r_ rA rw r )r r rx r r r �4test_set_text_long_line_maximal_non_ascii_heuristics� s %zGTestRawDataManager.test_set_text_long_line_maximal_non_ascii_heuristicsc C s� | j � } d d } t j | d
d � | j t | � t j d d d � j d � � | j | j d d � j d � | � | j | j
� | � d S) Nr� r9 u� áàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöő.
zz Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: quoted-printable
a�
=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.
zutf-8rw Tz
z
z
) r r r@ r r~ rZ r[ r_ rA rw r )r r rx 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 s= | j � } | j t � � t j | d d d �Wd QRXd S)Nu áàäéèęöő.
�cte�7bit)r rI � UnicodeErrorr r@ )r r 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 s= | j � } | j t � � t j | d d d �Wd QRXd S)Nu áàäéèęöő.
r| �ascii)r rI r� r r@ )r r r r r �1test_set_text_non_ascii_with_charset_ascii_raises� s zDTestRawDataManager.test_set_text_non_ascii_with_charset_ascii_raisesc C sC | j � } | j t � �! t j | d d d d d �Wd QRXd S)Nu áàäéèęöő.
r� r� r| r� )r rI r� r r@ )r r 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� | j � } d | d <| j � } d | d <d | d <d | d <| j d � t j | | � | j t | � t j d � � | j d
� } | j | | j � | j t | � t | � � | j | j
� | j � | j t | j
� � t | � � d S) NzForwarded message�Subjectzpython@vivarium.orgrL zpolice@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 r@ r r r: rZ r[ rA rt rU r )r r rx Zpayloadr r r �test_set_message� s
z#TestRawDataManager.test_set_messagec C s� | j � } d | d <| j � } d | d <d | d <d | d <| j d � t j | | � | j t | � t j d � j d
� � | j | j d | j j
� t j d � � | j | j � | j
� | j t | j � � t | � � d S)
Nz
Escape reportr� zpolice@monty.orgrL 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.
zutf-8Z 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 r@ r r r~ rZ r[ r_ Z as_stringr rW rt r rU r: )r r rx 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 sc | j � } | j � } x� d j � D]� } x� d j � D]� } | j d | d | � �c | j t � � } | j | | d | �Wd QRXt | j � } | j | | � | j | | � Wd QRXq8 Wq% Wd } x� d j � D]� } | j d | d | � �c | j t � � } | j | | d | �Wd QRXt | j � } | j | | � | j | | � Wd QRXq� Wd S)Nzquoted-printable base64zrfc822 external-bodyr� rq z
external-bodyz 8bit binary) r rj rk rI �
ValueErrorr@ r: Z exceptionZassertIn)r r rx r� rq �ar�excr r r �#test_set_message_invalid_cte_raises$ s$ z6TestRawDataManager.test_set_message_invalid_cte_raisesc
C s� x� d t d � t d � f D]� } | j d | � �x | j � } t j | | d d � | j t | � t j d � � | j | j
d d � | � | j | j � | � Wd QRXq Wd S)Ns
bogus contentrx �image�jpegz� Content-Type: image/jpeg
Content-Transfer-Encoding: base64
Ym9ndXMgY29udGVudA==
rw T) � bytearray�
memoryviewrk r r r@ r r: rZ r[ rA r )r rx r r r r �test_set_image_jpg8 s
z%TestRawDataManager.test_set_image_jpgc C s� | j � } d d d } | j | d d d d �| j t | � t j d � j d � � | j | j d
d � | � | j | j � | � d S) Ns b�gus con
t
ent � z�d ZaudioZaifr� zquoted-printablea Content-Type: audio/aif
Content-Transfer-Encoding: quoted-printable
MIME-Version: 1.0
b=FFgus=09con=0At=0Dent=20zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz=
zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzlatin-1rw T) r r@ r r~ rZ r[ r_ rA r )r r rx r r r �,test_set_audio_aif_with_quoted_printable_cteH s z?TestRawDataManager.test_set_audio_aif_with_quoted_printable_ctec C s� | j � } d d d } | j | d d d d �| j t | � t j d � j d � d
d � | j | j d d
� | � | j | j � | � d S)Ns b�gus con
t
ent r� r� ZvideoZmpegr� Zbinaryz~ Content-Type: video/mpeg
Content-Transfer-Encoding: binary
MIME-Version: 1.0
r� s7 b�gus con
t
ent zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzs= zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzrw T) r r@ r r~ rZ r[ r_ rA r )r r rx r r r �#test_set_video_mpeg_with_binary_cte^ s z6TestRawDataManager.test_set_video_mpeg_with_binary_ctec C s� | j � } d d d d } | j | d d d d �| j t | � t j d � j d
� d d � | j | j d
d � | � | j | j � | � d S)Ns b�gus con
t
ent
r� rX s
� applicationz octet-streamr� Z8bitz� Content-Type: application/octet-stream
Content-Transfer-Encoding: 8bit
MIME-Version: 1.0
r� s b�gus con
t
ent
s= zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz
rw T) r r@ r r~ rZ r[ r_ rA r )r r rx r r r �/test_set_application_octet_stream_with_8bit_cteo s zBTestRawDataManager.test_set_application_octet_stream_with_8bit_ctec C s{ | j � } d } | j j } t j | | d | d d � | d d � | d d � f �| j t | � t j d � � d S) NzSimple message.
�headersrL zfoo@example.comr� r� zI'm talking to myself.z� 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 r@ r r: rZ r[ )r r rx r� r r r �$test_set_headers_from_header_objects� s z7TestRawDataManager.test_set_headers_from_header_objectsc C sK | j � } d } t j | | d d �| j t | � t j d � � d S)NzSimple message.
r� �X-Foo-Header: foo�X-Bar-Header: barz� Content-Type: text/plain; charset="utf-8"
X-Foo-Header: foo
X-Bar-Header: bar
Content-Transfer-Encoding: 7bit
Simple message.
)r� r� )r r r@ r r: rZ r[ )r r rx r r r � test_set_headers_from_strings� s z0TestRawDataManager.test_set_headers_from_stringsc C sF | j � } d } | j t d � � t j | | d d �Wd QRXd S)NzSimple message.
z Content-Typer� �Content-Type: foo/bar)r� )r r$ r� r r@ )r r rx r r r �<test_set_headers_with_invalid_duplicate_string_header_raises� s
zOTestRawDataManager.test_set_headers_with_invalid_duplicate_string_header_raisesc C s^ | j � } d } | j j } | j t d � �' t j | | d | d d � f �Wd QRXd S)NzSimple message.
z Content-Typer� z foo/bar)r r r� r$ r� r r@ )r r rx 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 sW | j � } d } | j t d � �, t j | | d d �t | d j � Wd QRXd S)NzSimple message.
za@fairly@@invalid@addressr� � To: a@fairly@@invalid@addressrL )r� )r r$ r� r r@ �print�defects)r r rx r r r �4test_set_headers_with_defective_string_header_raises� s zGTestRawDataManager.test_set_headers_with_defective_string_header_raisesc C so | j � } d } | j j } | j t d � �8 t j | | d | d d � f �t | d j � Wd QRXd S)NzSimple message.
za@fairly@@invalid@addressr� rL ) r r r� r$ r� r r@ r� r� )r r rx r� r r r �4test_set_headers_with_defective_header_header_raises� s zGTestRawDataManager.test_set_headers_with_defective_header_header_raisesc C s7 | j � } | j d d d �| j | d d � d S)Nr � disposition�inlinezContent-Disposition)r r@ r )r r r r r �test_set_disposition_inline� s z.TestRawDataManager.test_set_disposition_inlinec C s7 | j � } | j d d d �| j | d d � d S)Nr r� Z
attachmentzContent-Disposition)r r@ r )r r r r r � test_set_disposition_attachment� s z2TestRawDataManager.test_set_disposition_attachmentc C s7 | j � } | j d d d �| j | d d � d S)Nr r� zContent-Disposition)r r@ r )r r r r r �test_set_disposition_foo� s z+TestRawDataManager.test_set_disposition_fooc C s7 | j � } | j d d d �| j | d d � d S)Nr �filenamezbar.txtzContent-Dispositionz attachment; filename="bar.txt")r r@ r )r r r r r �test_set_filename� s
z$TestRawDataManager.test_set_filenamec C s= | j � } | j d d d d d �| j | d d � d S)Nr r� r� r� zbar.txtzContent-Dispositionzinline; filename="bar.txt")r r@ r )r r r r r �(test_set_filename_and_disposition_inline� s z;TestRawDataManager.test_set_filename_and_disposition_inlinec C sK | j � } | j d d d �| j t | � t j d � j d � � d S)Nr r� u ábárî.txta 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 r@ r r~ rZ r[ r_ )r r r r r �test_set_non_ascii_filename� s z.TestRawDataManager.test_set_non_ascii_filenameZ
text_plainrx Z text_htmlrz Zapplication_octet_stream� contentr� � octet_streamZ
image_jpegr� r� Zmessage_rfc822Zmessage_external_body�
external-bodyc C sN | j � } | j | | d d �| j | d d � | j | d d � d S) Nr� �To: foo@example.com�From: bar@simple.netrr zfoo@example.comrs zbar@simple.net)r� r� )r r@ r )r r= �mimetyper r r r �!content_object_as_header_receiver� s z4TestRawDataManager.content_object_as_header_receiverc C s: | j � } | j | | d d �| j | d d � d S)Nr� r� zContent-Disposition)r r@ r )r r= r� r r r r �-content_object_as_disposition_inline_receiver s z@TestRawDataManager.content_object_as_disposition_inline_receiverc C sq | j � } | j | | d d d d �| j | d d � | j | j � d � | j | d j d d � d S)Nr� r� r� u bár.txtzContent-Dispositionu inline; filename="bár.txt")r r@ r � get_filename�params)r r= r� r r r r �-content_object_as_non_ascii_filename_receiver s
z@TestRawDataManager.content_object_as_non_ascii_filename_receiverc C s: | j � } | j | | d d �| j | d d � d S)NZcidZsome_random_stuffz
Content-ID)r r@ r )r r= r� r r r r � content_object_as_cid_receiver
s z1TestRawDataManager.content_object_as_cid_receiverc C sh | j � } d d d d i } | j | | d | �t | t � rM d | d <| j | d j | � d S) Nr u bárrM rQ r� zutf-8r| z Content-Type)r r@ �
isinstancer: r r� )r r= r� r r� r r r �!content_object_as_params_receiver s
z4TestRawDataManager.content_object_as_params_receiverN�rz )rx r� �r� r� )r� r� �r� r� )r� r� )r� )=r' r( r) r rT Zcloner r rU r\ r] r` ra rb rc rf rg rh rn ro rp ru rv ry 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 rV � sz
0
rV �__main__)ZunittestZtest.test_emailr r rZ Zemailr Z
email.messager Zemail.contentmanagerr r r rV r' �mainr r r r �<module> s } � � �