dafddlZddlmZmZddlZddlmZddlmZddl m Z m Z eGddeZ eGdd eZ ed k(rejyy) N) TestEmailBase parameterize)policy) EmailMessage)ContentManagerraw_data_managerc eZdZejZeZddddZdZdZ dZ Gdd e Z ed zd zZ Gd d e Zed zdzZdefdefddde fde fddde fdddd ZdZdZdZdZd Zy!)"TestContentManager) text/plain)text)) full_type maintype_onlynull_keycd d}t}|j|||j}d|d<d|d<|j|j |ddy) Nc|d}||fSN X-Bar-Header)msgfoobars >/root/Python-3.12.4/Lib/test/test_email/test_contentmanager.py foo_getterzATestContentManager.get_key_as_get_content_key..foo_gettersn%C8Or Content-Typerrrr)rrN)radd_get_handler _make_message assertEqual get_content)selforderkeyrcmms rget_key_as_get_content_keyz-TestContentManager.get_key_as_get_content_keys`   3 +    (.!. u5~Frc<d}d}t}|j|||jjD]\}}||kDs |j|||j }d|d<d|d<d|d<|j |j |dy) Nc |dSrrrs r bar_getterzGTestContentManager.get_key_as_get_content_key_order..bar_getter! ~& &rc |dSN X-Foo-Headerrr.s rrzGTestContentManager.get_key_as_get_content_key_order..foo_getter#r0rr rrrrr3)rr"get_key_paramsvaluesr#r$r%)r&r'r(r/rr) precedencer*s r get_key_as_get_content_key_orderz3TestContentManager.get_key_as_get_content_key_order s ' '   3 +#2299; 4OJE!""3 3 4    (.!.!. *U4rct}|j}d|d<|jtd5|j |dddy#1swYyxYw)Nr r)rr#assertRaisesRegexKeyErrorr%r&r)r*s r:test_get_content_raises_if_unknown_mimetype_and_no_defaultzMTestContentManager.test_get_content_raises_if_unknown_mimetype_and_no_default0sS      (.  # #Hl ;  NN1    s AAc eZdZy)TestContentManager.BaseThingN__name__ __module__ __qualname__rrr BaseThingr>7 rrC.r>c eZdZy)TestContentManager.ThingNr?rrrThingrG:rDrrHrGrr )r rG)rrH)r>)rC) z builtins.str) str) N) type full_pathqualnamename base_typebase_full_path base_qualname base_namestr_type str_full_pathstr_namercdd}t}|j|||j}|j}|j ||d|j |dd|j |j |y)Nc0||d<|j|yr2 set_payload)robjrs r foo_setterzATestContentManager.set_key_as_set_content_key..foo_setterNs"%C  OOC rrr r3r!)radd_set_handlerr#rH set_contentr$ get_payload)r&r'r(rbr)r*msg_objs rset_key_as_set_content_keyz-TestContentManager.set_key_as_set_content_keyMsw !  3 +    **, q'u- >*E2 '2rcd}d}t}|j|||jjD]\}}||kDs |j|||j }|j }|j |||j|dd|j|j|y)Nc0d|d<|j|y)NrX-FooBar-Headerr_rras rrbzGTestContentManager.set_key_as_set_content_key_order..foo_setterZs%*C! " OOC rcd|d<y)Nrrjrrks r bar_setterzGTestContentManager.set_key_as_set_content_key_order..bar_setter]s %*C! "rrjr) rrcr4r5r#rHrdr$re) r&r'r(rbrmr)r6r*rfs r set_key_as_set_content_key_orderz3TestContentManager.set_key_as_set_content_key_orderYs ! +   3 +#2299; 4OJE!""3 3 4    **, q'" ,-u5 '2rct}|j}|j}|jt|j 5|j ||dddy#1swYyxYwr!)rr#rHr9r:testobject_full_pathrd)r&r)r*rfs r6test_set_content_raises_if_unknown_type_and_no_defaultzITestContentManager.test_set_content_raises_if_unknown_type_and_no_defaultjs\      **,  # #Hd.G.G H ' NN1g & ' ' 's A''A0ct}|j}d|d<|jt5|j |ddddy#1swYyxYw)Nz multipart/foortest)rr# assertRaises TypeErrorrdr;s r.test_set_content_raises_if_called_on_multipartzATestContentManager.test_set_content_raises_if_called_on_multipartqsS      +.   y ) & NN1f % & & &s AAc|j}d|d<d|d<d|d<|jdt}|jtd|j d | |j |d|j |d|j|dd|j |jy) Nrz Content-Fooz text/htmlrrsToabccyr!r)argskws rzITestContentManager.test_set_content_calls_clear_content..srxyz)content_manager) r#r`rrcrPrd assertIsNoner$re)r&r*r)s r$test_set_content_calls_clear_contentz7TestContentManager.test_set_content_calls_clear_contentxs     -'.$ e   3 89 eR 0 !M*+ !N+, 4&) !--/*rN)r@rArBrdefaultrmessager4r+r7r<rPrCbaseobject_full_pathrHrpset_key_paramsrgrnrqrvrrrrr r s^^FG/($ N G5  C #c>,JJ  #c>,FFL!57=*,!57A.J1('  N 33"'& +rr ceZdZejj deZeZdZ dZ dZ dZ dZ dZd Zd Zd Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%d Z&d!Z'd"Z(d#Z)d$Z*d%Z+d&Z,d'Z-d(Z.d)Z/d*Z0d+Z1d,Z2d-Z3d.Z4d/Z5d0Z6d1Z7d2Z8d3Z9d4d5d6d7ed8fed9fd:Z:d;Z;d<ZZ>d?Z?y@)ATestRawDataManager<)max_line_lengthrc|jtjd}|jt j |dy)NJ Content-Type: text/plain Basic text. z Basic text. _str_msgtextwrapdedentr$rr%r&r*s rtest_get_text_plainz&TestRawDataManager.test_get_text_plains> MM(//+  )55a8/Jrc|jtjd}|jt j |dy)NzP Content-Type: text/html

Basic text.

z

Basic text.

rrs rtest_get_text_htmlz%TestRawDataManager.test_get_text_htmlsA MM(//+  )55a8/ 1rc|jtjdjd}|j t j |dy)Nu\ Content-Type: text/plain; charset=latin1 Basìc tëxt. latin1Basìc tëxt. ) _bytes_msgrrencoder$rr%rs rtest_get_text_plain_latin1z-TestRawDataManager.test_get_text_plain_latin1sK OOHOO-!  # )55a8:KLrc|jtjd}|jt j |dy)Nz Content-Type: text/plain; charset="latin-1" Content-Transfer-Encoding: quoted-printable Bas=ECc t=EBxt. rrrs r+test_get_text_plain_latin1_quoted_printablez>TestRawDataManager.test_get_text_plain_latin1_quoted_printable? MM(//+  )55a8:KLrc|jtjd}|jt j |dy)Nz Content-Type: text/plain; charset="utf8" Content-Transfer-Encoding: base64 QmFzw6xjIHTDq3h0Lgo= rrrs rtest_get_text_plain_utf8_base64z2TestRawDataManager.test_get_text_plain_utf8_base64rrc|jtjd}|jt j |dy)N Content-Type: text/plain; charset="utf8" Content-Transfer-Encoding: quoted-printable Bas=c3=acc t=c3=abxt=fd. uBasìc tëxt�. rrs r-test_get_text_plain_bad_utf8_quoted_printablez@TestRawDataManager.test_get_text_plain_bad_utf8_quoted_printables? MM(//+  )55a8:NOrc|jtjd}|jt j |ddy)Nrignoreerrorsrrrs r;test_get_text_plain_bad_utf8_quoted_printable_ignore_errorszNTestRawDataManager.test_get_text_plain_bad_utf8_quoted_printable_ignore_errorsC MM(//+  )55aI* ,rc|jtjd}|jt j |ddy)Nu Content-Type: text/plain; charset="utf8" Content-Transfer-Encoding: base64 QmFzw6xjIHTDq3h0Lgoÿ= rrrrrs r8test_get_text_plain_utf8_base64_recoverable_bad_CTE_datazKTestRawDataManager.test_get_text_plain_utf8_base64_recoverable_bad_CTE_datarrc|jtjd}|jt5t j |ddddy#1swYyxYw)Nrrr rrrrtrurr%rs rtest_get_text_invalid_keywordz0TestRawDataManager.test_get_text_invalid_keywordsU MM(//+    y ) :  ( ( 9 : : : AA$c6tjd}djD]e}|j|5|j |j |dz}|j tj|ddddgy#1swYrxYw)Nzu Content-Type: {} Content-Transfer-Encoding: base64 Ym9ndXMgZGF0YQ== zaudio image video application)maintypez/foos bogus data) rrsplitsubTestrformatr$rr%)r&templaterr*s rtest_get_non_textz$TestRawDataManager.test_get_non_texts??$ 8==? QHx0 QMM(//(6/"BC  !1!=!=a!@-P Q Q Q Q Qs A BB c|jtjd}|jt5t j |ddddy#1swYyxYw)Nz| Content-Type: image/jpg Content-Transfer-Encoding: base64 Ym9ndXMgZGF0YQ== rrrrs r!test_get_non_text_invalid_keywordz4TestRawDataManager.test_get_non_text_invalid_keywordsU MM(//+    y ) =  ( (8 < = = =rc|jtjd}|jt5t j |dddy#1swYyxYw)Nzm Content-Type: multipart/mixed; boundary="===" --=== --===-- )rrrrtr:rr%rs rtest_get_raises_on_multipartz/TestRawDataManager.test_get_raises_on_multipartsS MM(//+    x ( ,  ( ( + , , ,s AA"ctjd}djD]}|j|5|j |j |}t j|}|j||j|jt j|d|j|dd|j|djdjd dddy#1swYxYw) Nz Content-Type: message/{} To: foo@example.com From: bar@example.com Subject: example an example message rfc822 external-bodysubtypezan example message tofoo@example.comfromrr) rrrrrrrr%assertIsInstancerr$ addressesusername)r&rrr*sub_msgs r)test_get_message_rfc822_and_external_bodyz>D c|jtjd}|jt j |dddy)Nz Content-Type: message/partial To: foo@example.com From: bar@example.com Subject: example The real body is in another message. rOs To: foo@exrrs r9test_get_message_non_rfc822_or_external_body_yields_byteszLTestRawDataManager.test_get_message_non_rfc822_or_external_body_yields_bytessE MM(//+  )55a8"=}MrcR|j}d}tj|||jt |t j d|j|jdjd||j|j|y)NSimple message. z Content-Type: text/plain; charset="utf-8" Content-Transfer-Encoding: 7bit Simple message. Tdecodeutf-8 r#rrdr$rPrrrerr%r&r*contents rtest_set_text_plainz&TestRawDataManager.test_set_text_plain%s    %$$Q0 Q2"  d3::7CWM '2rcR|j}d}tj|||jt |t j d|j|jdjdd|j|jdy)Nrzp Content-Type: text/plain; charset="utf-8" Content-Transfer-Encoding: 7bit Trr rrs rtest_set_text_plain_nullz+TestRawDataManager.test_set_text_plain_null2s    $$Q0 Q2"  d3::7CTJ $/rcV|j}d}tj||d|jt |t j d|j|jdjd||j|j|y)Nz

Simple message.

htmlrz Content-Type: text/html; charset="utf-8" Content-Transfer-Encoding: 7bit

Simple message.

Trrrrs rtest_set_text_htmlz%TestRawDataManager.test_set_text_html?s    ,$$Q@ Q2"  d3::7CWM '2rcV|j}d}tj||d|jt |t j d|j|jdjd||j|j|y)Nrlatin-1charsetz Content-Type: text/plain; charset="iso-8859-1" Content-Transfer-Encoding: 7bit Simple message. Trrrrs rtest_set_text_charset_latin_1z0TestRawDataManager.test_set_text_charset_latin_1Ls    %$$QC Q2"  d3::7CWM '2rcR|j}d}tj|||jt |t j d|j|jdjd||j|j|y)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. Trrrrs r(test_set_text_plain_long_line_heuristicsz;TestRawDataManager.test_set_text_plain_long_line_heuristicsYs    9$$Q0 Q2"  d3::7CWM '2rcp|j}d}tj|||jt |t j djd|j|jdjd||j|j|y)Nu7et 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. rTr r#rrdr$bytesrrrrerr%rs r5test_set_text_short_line_minimal_non_ascii_heuristicszHTestRawDataManager.test_set_text_short_line_minimal_non_ascii_heuristicshs    L$$Q0 q8??4$   " d3::7CWM '2rcp|j}d}tj|||jt |t j djd|j|jdjd||j|j|y)Nusj'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. rTrrrs r4test_set_text_long_line_minimal_non_ascii_heuristicszGTestRawDataManager.test_set_text_long_line_minimal_non_ascii_heuristicsus    # $$Q0 q8??4$ " d3::7CWM '2rcp|j}d}tj|||jt |t j djd|j|jdjd||j|j|y)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. rTrrrs r=test_set_text_11_lines_long_line_minimal_non_ascii_heuristicszPTestRawDataManager.test_set_text_11_lines_long_line_minimal_non_ascii_heuristicss    # $$Q0 q8??4$ " d3::7CWM '2rcp|j}d}tj|||jt |t j djd|j|jdjd||j|j|y)Náàäéèęöő. u Content-Type: text/plain; charset="utf-8" Content-Transfer-Encoding: 8bit áàäéèęöő. rTrrrs r*test_set_text_maximal_non_ascii_heuristicsz=TestRawDataManager.test_set_text_maximal_non_ascii_heuristicss    '$$Q0 q8??4$   " d3::7CWM '2rcp|j}d}tj|||jt |t j djd|j|jdjd||j|j|y)Nu áàäéèęöő. u Content-Type: text/plain; charset="utf-8" Content-Transfer-Encoding: 8bit áàäéèęöő. rTrrrs r3test_set_text_11_lines_maximal_non_ascii_heuristicszFTestRawDataManager.test_set_text_11_lines_maximal_non_ascii_heuristicss    1$$Q0 q8??4$   " d3::7CWM '2rcp|j}d}tj|||jt |t j djd|j|jdjd||j|j|y)Nuáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöőáàäéèęöő. a Content-Type: text/plain; charset="utf-8" Content-Transfer-Encoding: base64 w6HDoMOkw6nDqMSZw7bFkcOhw6DDpMOpw6jEmcO2xZHDocOgw6TDqcOoxJnD tsWRw6HDoMOkw6nDqMSZw7bFkcOhw6DDpMOpw6jEmcO2xZHDocOgw6TDqcOo xJnDtsWRw6HDoMOkw6nDqMSZw7bFkcOhw6DDpMOpw6jEmcO2xZHDocOgw6TD qcOoxJnDtsWRw6HDoMOkw6nDqMSZw7bFkcOhw6DDpMOpw6jEmcO2xZHDocOg w6TDqcOoxJnDtsWRLgo= rTrrrs r4test_set_text_long_line_maximal_non_ascii_heuristicszGTestRawDataManager.test_set_text_long_line_maximal_non_ascii_heuristicss    Y $$Q0 q8?? 4 $ " d3::7CWM '2rcp|j}d}tj|d|jt |t j djd|j|jdjd||j|j|y)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. rTrrrs r=test_set_text_11_lines_long_line_maximal_non_ascii_heuristicszPTestRawDataManager.test_set_text_11_lines_long_line_maximal_non_ascii_heuristicss    Y $$Q)n o q8??4$ " d3::7CWM '2rc|j}|jt5tj|dddddy#1swYyxYw)Nr7bitcter#rt UnicodeErrorrrdrs r,test_set_text_non_ascii_with_cte_7bit_raisesz?TestRawDataManager.test_set_text_non_ascii_with_cte_7bit_raisessJ       | , N  ( (+@f M N N N AAc|j}|jt5tj|dddddy#1swYyxYw)Nrasciirrrs r1test_set_text_non_ascii_with_charset_ascii_raiseszDTestRawDataManager.test_set_text_non_ascii_with_charset_ascii_raisessJ       | , S  ( (+@' R S S Src|j}|jt5tj|ddddddy#1swYyxYw)Nrrr)rrrrs r>test_set_text_non_ascii_with_cte_7bit_and_charset_ascii_raiseszQTestRawDataManager.test_set_text_non_ascii_with_cte_7bit_and_charset_ascii_raisessM       | , _  ( (+@fV] ^ _ _ _s A  Acp|j}d|d<|j}d|d<d|d<d|d<|jdtj|||jt |t j d |jd }|j||j|jt |t ||j|j|j|jt |jt |y) NzForwarded messageSubjectzpython@vivarium.orgrxpolice@monty.orgFromzget 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#rdrr$rPrrrerrr%)r&r*rpayloads rtest_set_messagez#TestRawDataManager.test_set_messages    *) $$&- ,3 67$$Q0 Q 2 " --" gt||4 Ws7|4 ammot||< Q]]_-s7| g}n& I q8??4$  & $' N$N O d3W= '2rc |j}d}|jj}tj|||dd|dd|ddf|j t |tjdy) NrrxrrrzI'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#rheader_factoryrrdr$rPrrr&r*rr's r$test_set_headers_from_header_objectsz7TestRawDataManager.test_set_headers_from_header_objectss    %33$$Q 4!2 3 6#4 5 9&> ?:A B Q2" rc|j}d}tj||d|jt |t j dy)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#rrdr$rPrrrs rtest_set_headers_from_stringsz0TestRawDataManager.test_set_headers_from_stringssR    %$$Q:" # Q2" rc|j}d}|jtd5tj||ddddy#1swYyxYw)Nrr)zContent-Type: foo/barr%)r#r9rrrdrs r*    s A  Ac |j}d}|jj}|jtd5t j |||ddfdddy#1swYyxYw)Nrrz foo/barr%)r#rr'r9rrrdr(s r=    s !A))A2c|j}d}|jtd5tj||dt |dj dddy#1swYyxYw)Nra@fairly@@invalid@address)zTo: a@fairly@@invalid@addressr%rx)r#r9rrrdprintdefectsrs r4test_set_headers_with_defective_string_header_raiseszGTestRawDataManager.test_set_headers_with_defective_string_header_raisessd    %  # #J0K L #  ( (G>2  !D'// "  # # #s 1A##A,c |j}d}|jj}|jtd5t j |||ddft|djdddy#1swYyxYw)Nrr1rxr%) r#rr'r9rrrdr2r3r(s r4test_set_headers_with_defective_header_header_raiseszGTestRawDataManager.test_set_headers_with_defective_header_header_raisess    %33  # #J0K L #  ( (Gt%@A>D  !D'// "  # # #s 9BB ct|j}|jdd|j|ddy)Nrinline dispositionContent-Dispositionr#rdr$rs rtest_set_disposition_inlinez.TestRawDataManager.test_set_disposition_inlines6     e 2 018     c;H;(; 018     c>H>*=> <*=>rc|j}ddd}|j|g|d|it|trd|d<|j |dj |y)Nubárr~)rryr_rrr)r#rd isinstancerPr$r_)r&rarYr*r_s r!content_object_as_params_receiverz4TestRawDataManager.content_object_as_params_receiver:sa    . c4H4V4 c3  'F9  >*116:rN)@r@rArBrrclonerrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr rrrr#r)r+r-r/r4r6r=r@rBrGrJrLrNcontent_object_paramsrZr\r`rdrgrrrrrsr^^ ! !"2B"DFGK1MMMP,,: Q=,O( N 3 0 3 3 3 33"3$ 3 33&3>N S _ =:.=`,(; 3,3"3"$ ##= A :; Q " &+%F5"9b/"))-?!@ 6= R? ;rr__main__)unittesttest.test_emailrrremailr email.messageremail.contentmanagerrrr rr@mainrrrrqso7&Az+z+z+zx ;x ;x ;v zHMMOr