Your IP : 216.73.216.213
��Yf � � @ s� d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l Z d d l
Z
d d l Z d d l Z d d l
Z
d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z m Z y d d l Z Wn e k
r d Z Yn Xe j Z e j d k rQd d � Z e e j _ d d � Z Gd
d � d e j � Z! d d
� Z" d Z# d Z$ e j% e d � Gd d � d e j � � Z& Gd d � d e j � Z' e j% e d � Gd d � d e j � � Z( e j% e d � Gd d � d e j � � Z) d d d d d d i Z* d d f Z+ d Z, d! d d g d" d g i Z- Gd# d$ � d$ e. � Z/ Gd% d& � d& e j � Z0 Gd' d( � d( e j1 � Z2 e j% e d � Gd) d* � d* e j � � Z3 Gd+ d, � d, e2 � Z4 e j% e d � Gd- d. � d. e j � � Z5 e d/ d0 d1 �Z6 Gd2 d3 � d3 e0 � Z7 Gd4 d5 � d5 e2 � Z8 e j% e d � Gd6 d7 � d7 e j � � Z9 e j: d d8 d9 � � Z; e< d: k r�e; � d S);� N)� EmailMessage)� body_encode)�support� mock_socket�darwinc C s d S)N� )�selfr r �6/opt/alt/python35/lib64/python3.5/test/test_smtplib.py� handle_expt s r
c
C s� | j � | j � z� y | j � \ } } Wn t j k
rD Ynz Xd } xf | r� | d k r� t j g | g g � \ } } } | r� | j | � } | | d � } | d 8} qN W| j � Wd | j � | j � Xd S)Ni� r � )Zlisten�setZaccept�socket�timeout�select�send�close)
�evtZbuf�serv�conn�addr�n�r�w�eZsentr r r �server% s
!
r c @ s� e Z 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 S)� GeneralTestsc C s t t _ d | _ d S)N� )r �smtplibr
�port)r r r r �setUp= s zGeneralTests.setUpc C s
t t _ d S)N)r
r )r r r r �tearDownA s zGeneralTests.tearDownc C s) d } d } | j | t j | � � d S)Nzabc
.jkl
foo
..bluezabc
..jkl
foo
...blue)� assertEqualr Z quotedata)r Zteststr�expectedr r r �
testQuoteDataF s zGeneralTests.testQuoteDatac C s0 t j d � t j t | j � } | j � d S)Ns 220 Hola mundo)r �
reply_withr �SMTP�HOSTr r )r �smtpr r r �
testBasic1K s
zGeneralTests.testBasic1c C sI t j d � t j t | j d d �} | j | j d � | j � d S)Ns 220 Hola mundo�source_address� 127.0.0.1�M )r* r+ )r* r+ ) r r$ r r% r& r r! r) r )r r' r r r �testSourceAddressQ s
z GeneralTests.testSourceAddressc C s7 t j d � t j d t | j f � } | j � d S)Ns 220 Hola mundoz%s:%s)r r$ r r% r& r r )r r' r r r �
testBasic2Y s
zGeneralTests.testBasic2c C sI t j d � t j t | j d d �} | j | j d � | j � d S)Ns 220 Hola mundo�local_hostnameZtesthost) r r$ r r% r& r r! r. r )r r' r r r �testLocalHostName_ s
z GeneralTests.testLocalHostNamec C s� t j d � | j t j � � t j d � | j t j � d � z t j t | j � } Wd t j d � X| j | j
j � d � | j � d S)Ns 220 Hola mundo� )
r r$ � assertIsNone�getdefaulttimeout�setdefaulttimeoutr! r r% r& r �sock�
gettimeoutr )r r' r r r �testTimeoutDefaultf s
z GeneralTests.testTimeoutDefaultc C s� t j d � | j t j � � t j d � z t j t | j d d �} Wd t j d � X| j | j
j � � | j � d S)Ns 220 Hola mundor0 r )
r r$ r1 r
r2 r3 r r% r& r r4 r5 r )r r' r r r �testTimeoutNoner s
z GeneralTests.testTimeoutNonec C sO t j d � t j t | j d d �} | j | j j � d � | j � d S)Ns 220 Hola mundor r0 )
r r$ r r% r& r r! r4 r5 r )r r' r r r �testTimeoutValue} s
z GeneralTests.testTimeoutValuec
C s� t j d � t j � } | j d � t j � � } | j t | j � Wd QRX| j
� t j d t j
� } | j | j � | � d S)Ns 220 Hello worldr z ^connect:)r r$ r r% �set_debuglevelr �captured_stderr�connectr& r r �re�compile� MULTILINE� assertRegex�getvalue)r r' �stderrr" r r r �test_debuglevel� s
z GeneralTests.test_debuglevelc
C s� t j d � t j � } | j d � t j � � } | j t | j � Wd QRX| j
� t j d t j
� } | j | j � | � d S)Ns 220 Hello world� z#^\d{2}:\d{2}:\d{2}\.\d{6} connect: )r r$ r r% r9 r r: r; r& r r r< r= r> r? r@ )r r' rA r" r r r �test_debuglevel_2� s
z GeneralTests.test_debuglevel_2N)�__name__�
__module__� __qualname__r r r# r( r, r- r/ r6 r7 r8 rB rD r r r r r ; s
r c C s� | j � z� y{ t t d � r+ t j } n t j } d } xJ t j r� | d k r� | d t j � | j � ry | j � P| d 8} q= WWn t j
k
r� Yn XWd | j � s� t j d � | j � t j
� | j � Xd S)N�polli� r g{�G�z�?r g �?)r �hasattrr �asyncoreZpoll2rH Z
socket_mapZis_setr r
r �time�sleepZ close_all)r �serv_evt�
client_evtZpoll_funr r r r �debugging_server� s&
rO z&---------- MESSAGE FOLLOWS ----------
z&------------ END MESSAGE ------------
z!Threading required for this test.c @ s e Z d Z 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 d, d- � Z d S).�DebuggingServerTestsNc C s t j | _ t j t _ t j | _ t j � | _ | j t _ t
j � | _ t
j � | _
t j | _ t j � t _ t j t d f d d d �| _ | j j j � d | _ | j | j | j
f } t
j d t d | � | _ | j j � | j j � | j j � d S)
Nr �nowherer � decode_dataT�target�args���)rQ rU ) r
�getfqdn� real_getfqdnr �sys�stdout�
old_stdout�io�StringIO�output� threading�EventrM rN �smtpd� DEBUGSTREAM�old_DEBUGSTREAMZDebuggingServerr& r � getsocknamer �ThreadrO �thread�start�wait�clear)r � serv_argsr r r r � s"
zDebuggingServerTests.setUpc C s\ | j t _ | j j � | j j � | j j � | j t
_ t j
j � | j t _
d S)N)rW r
rV rN r rM rg re �joinrZ rX rY r` ra r rb )r r r r r � s
z DebuggingServerTests.tearDownc C s/ t j t | j d d d d �} | j � d S)Nr. � localhostr � )r r% r& r �quit)r r' r r r � testBasic� s !z DebuggingServerTests.testBasicc C s� t j � } yg t j t | j d d d d d d | f �} | j | j d | f � | j | j d � | j � WnK t
k
r� } z+ | j t j k r� | j
d | � � WYd d } ~ Xn Xd S)Nr. rk r rl r) z 127.0.0.1zcouldn't bind to port %d)r Zfind_unused_portr r% r& r r! r) r. rm �OSError�errnoZ
EADDRINUSEZskipTest)r r r' r r r r r, � s z&DebuggingServerTests.testSourceAddressc C sK t j t | j d d d d �} d } | j | j � | � | j � d S)Nr. rk r rl � � OK)rq rr )r r% r& r r! �nooprm )r r' r" r r r �testNOOP s !z DebuggingServerTests.testNOOPc C sK t j t | j d d d d �} d } | j | j � | � | j � d S)Nr. rk r rl rq � OK)rq ru )r r% r& r r! Zrsetrm )r r' r" r r r �testRSET s !z DebuggingServerTests.testRSETc C sK t j t | j d d d d �} d } | j | j � | � | j � d S)Nr. rk r rl rq �
SIZE 33554432
HELP)rq rw )r r% r& r r! �ehlorm )r r' r" r r r �testELHO
s !z DebuggingServerTests.testELHOc C sX t j t | j d d d d �} d } | j d � | j | j � | � | j � d S) Nr. rk r rl � � EXPN not implementedZEXPN)rz r{ )r r% r& r Zputcmdr! Zgetreplyrm )r r' r" r r r �testEXPNNotImplemented s
!
z+DebuggingServerTests.testEXPNNotImplementedc C sq t j t | j d d d d �} d d d f } | j | j d � | � | j | j d � | � | j � d S) Nr. rk r rl � s* Cannot VRFY user, but will accept message s and attempt deliveryznobody@nowhere.com)r r% r& r r! �vrfyZverifyrm )r r' r" r r r �testVRFY s !
z DebuggingServerTests.testVRFYc C sU t j t | j d d d d �} | j � d } | j | j � | � | j � d S)Nr. rk r rl � � Duplicate HELO/EHLO)r� r� )r r% r& r Zhelor! rm )r r' r" r r r �testSecondHELO$ s
!
z#DebuggingServerTests.testSecondHELOc C sI t j t | j d d d d �} | j | j � d d � | j � d S)Nr. rk r rl s# Supported commands: EHLO HELO MAIL s RCPT DATA RSET NOOP QUIT VRFY)r r% r& r r! �helprm )r r' r r r �testHELP- s !z DebuggingServerTests.testHELPc C s� d } t j t | j d d d d �} | j d d | � t j d � | j � | j j � | j
j � | j j
� d t | t f } | j | j j � | � d S)
NzA test messager. rk r rl �John�Sallyg{�G�z�?z%s%s
%s)r r% r& r �sendmailrK rL rm rN r rM rg r] �flush� MSG_BEGIN�MSG_ENDr! r@ )r �mr' �mexpectr r r �testSend3 s !
z DebuggingServerTests.testSendc C s� d } t j t | j d d d d �} | j d d | � t j d � | j � | j j � | j
j � | j j
� d t | j d
� t f } | j | j j � | � d S) Ns A test messager. rk r rl r� r� g{�G�z�?z%s%s
%s�ascii)r r% r& r r� rK rL rm rN r rM rg r] r� r� �decoder� r! r@ )r r� r' r� r r r �testSendBinaryD s !
z#DebuggingServerTests.testSendBinaryc C s� d } t j t | j d d d d �} | j d d | � t j d � | j � | j j � | j
j � | j j
� d t | t f } | j | j j � | � d S)
Nz.A test
.mes.sage.r. rk r rl r� r� g{�G�z�?z%s%s
%s)r r% r& r r� rK rL rm rN r rM rg r] r� r� r� r! r@ )r r� r' r� r r r �testSendNeedingDotQuoteR s !
z,DebuggingServerTests.testSendNeedingDotQuotec C s� d } t j t | j d d d d �} | j d d | � t j d � | j � | j j � | j
j � | j j
� d t | t f } | j | j j � | � t j j � } t j d
t j � } | j | | � d S) NzA test messager. rk r rl z<>r� g{�G�z�?z%s%s
%sz ^sender: <>$)r r% r& r r� rK rL rm rN r rM rg r] r� r� r� r! r@ r` ra r< r= r> r? )r r� r' r� �debugout�senderr r r �testSendNullSendera s !
z'DebuggingServerTests.testSendNullSenderc C s� t j j j d � } t j t | j d d d d �} | j | d d d d �t j
d
� | j � | j j
� | j j � | j j � t j d � | d <d t | j � t f } | j | j j � | � d S)
NzA test messager. rk r rl � from_addrr� �to_addrsr� g{�G�z�?zX-Peerz%s%s
%s)�email�mime�text�MIMETextr r% r& r � send_messagerK rL rm rN r rM rg r] r� r
�
gethostbynamer� � as_stringr� r! r@ )r r� r' r� r r r �testSendMessager s !
z$DebuggingServerTests.testSendMessagec C s� t j j j d � } d | d <d | d <d | d <d | d <t j t | j d
d d d
�} | j | � t j
d � | j � | j | d d � | j
j � | j j � | j j � t j d � | d <| d =d t | j � t f } | j | j j � | � t j j � } t j d t j � } | j | | � x< d D]4 } t j d j | � t j � } | j | | � qDWd S)NzA test messagez foo@bar.com�Fromr� �Toz Sally, Fred�CCz?John Root <root@localhost>, "Dinsdale" <warped@silly.walks.com>�Bccr. rk r rl g{�G�z�?zX-Peerz%s%s
%sz^sender: foo@bar.com$r� �Fred�root@localhost�warped@silly.walks.comz^recips: .*'{}'.*$)r� r� r� r� r� ) r� r� r� r� r r% r& r r� rK rL rm r! rN r rM rg r] r� r
r� r� r� r� r@ r` ra r< r= r> r? �format)r r� r' r� r� r� r �to_addrr r r � testSendMessageWithAddresses� s2
!
z1DebuggingServerTests.testSendMessageWithAddressesc C sQ t j j j d � } d | d <d | d <t j t | j d d d d �} | j | � t j
d
� | j � | j j
� | j j � | j j � t j d � | d <d t | j � t f } | j | j j � | � t j j � } t j d
t j � } | j | | � x< d D]4 } t j d j | � t j � } | j | | � qWd S)NzA test messagez foo@bar.comr� zJohn, Dinsdaler� r. rk r rl g{�G�z�?zX-Peerz%s%s
%sz^sender: foo@bar.com$r� �Dinsdalez^recips: .*'{}'.*$)r� r� ) r� r� r� r� r r% r& r r� rK rL rm rN r rM rg r] r� r
r� r� r� r� r! r@ r` ra r< r= r> r? r� )r r� r' r� r� r� r r� r r r � testSendMessageWithSomeAddresses� s(
!
z5DebuggingServerTests.testSendMessageWithSomeAddressesc C s� t j j j d � } d | d <d | d <t j t | j d d d d �} | j | d
d d d
�t j
d � | j � | j j
� | j j � | j j � t j d � | d <d t | j � t f } | j | j j � | � t j j � } t j d t j � } | j | | � x< d D]4 } t j d j | � t j � } | j | | � q!Wt j d t j � } | j | | � d S)NzA test messagez foo@bar.comr� zJohn, Dinsdaler� r. rk r rl r� zjoe@example.comr� zfoo@example.netg{�G�z�?zX-Peerz%s%s
%sz^sender: joe@example.com$r� r� z^recips: .*'{}'.*$z ^recips: .*'foo@example.net'.*$)r� r� )!r� r� r� r� r r% r& r r� rK rL rm rN r rM rg r] r� r
r� r� r� r� r! r@ r` ra r< r= r> r? r� ZassertNotRegex) r r� r' r� r� r� r r� Zrecipr r r �%testSendMessageWithSpecifiedAddresses� s,
!
z:DebuggingServerTests.testSendMessageWithSpecifiedAddressesc C s[ t j j j d � } d | d <d | d <d | d <t j t | j d d d
d �} | j | � t j
d � | j � | j j
� | j j � | j j � t j d � | d
<d t | j � t f } | j | j j � | � t j j � } t j d t j � } | j | | � x< d D]4 } t j d j | � t j � } | j | | � q Wd S)NzA test messagezBernard, Biancar� z#the_rescuers@Rescue-Aid-Society.comZSenderzJohn, Dinsdaler� r. rk r rl g{�G�z�?zX-Peerz%s%s
%sz-^sender: the_rescuers@Rescue-Aid-Society.com$r� r� z^recips: .*'{}'.*$)r� r� ) r� r� r� r� r r% r& r r� rK rL rm rN r rM rg r] r� r
r� r� r� r� r! r@ r` ra r< r= r> r? r� )r r� r' r� r� r� r r� r r r � testSendMessageWithMultipleFrom� s*
!
z4DebuggingServerTests.testSendMessageWithMultipleFromc C s� t j j j d � } d | d <d | d <d | d <d | d <d
| d <d | d
<d | d <d | d <t j t | j d d d d �} | j | � t j
d � | j � | j j
� | j j � | j j � | d =| d =t j d � | d <d t | j � t f } | j | j j � | � t j j � } t j d t j � } | j | | � x< d D]4 } t j d j | � t j � } | j | | � q_Wd S) NzA test messagez foo@bar.comr� r� r� z Sally, Fredr� z?John Root <root@localhost>, "Dinsdale" <warped@silly.walks.com>r� z Thu, 1 Jan 1970 17:42:00 +0000z Resent-Datezholy@grail.netz Resent-Fromz&Martha <my_mom@great.cooker.com>, Jeffz Resent-To�doe@losthope.netz
Resent-Bccr. rk r rl g{�G�z�?zX-Peerz%s%s
%sz^sender: holy@grail.net$�my_mom@great.cooker.com�Jeffz^recips: .*'{}'.*$)r� r� r� ) r� r� r� r� r r% r& r r� rK rL rm rN r rM rg r] r� r
r� r� r� r� r! r@ r` ra r< r= r> r? r� )r r� r' r� r� r� r r� r r r �testSendMessageResent� s8
!
z*DebuggingServerTests.testSendMessageResentc C s� t j j j d � } d | d <d | d <d | d <d | d <d
| d <d | d
<d | d <d | d <d | d <d | d <d | d
<t j t | j d d d d �} | j t � � | j
| � Wd QRX| j � d S)NzA test messagez foo@bar.comr� r� r� z Sally, Fredr� z?John Root <root@localhost>, "Dinsdale" <warped@silly.walks.com>r� z Thu, 1 Jan 1970 17:42:00 +0000z Resent-Datezholy@grail.netz Resent-Fromz&Martha <my_mom@great.cooker.com>, Jeffz Resent-Tozdoe@losthope.netz
Resent-Bccz Thu, 2 Jan 1970 17:42:00 +0000r. rk r rl ) r� r� r� r� r r% r& r � assertRaises�
ValueErrorr� r )r r� r' r r r �#testSendMessageMultipleResentRaises s
!z8DebuggingServerTests.testSendMessageMultipleResentRaises)rE rF rG �maxDiffr r rn r, rt rv ry r| r r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r rP � s.
! "rP c @ s( e Z d Z d d � Z d d � Z d S)�NonConnectingTestsc C s? t j � } | j t j | j � | j t j | j d � d S)Nztest msg)r r% r� �SMTPServerDisconnectedrx r )r r' r r r �testNotConnected* s z#NonConnectingTests.testNotConnectedc C s3 | j t t j d d � | j t t j d � d S)Nrk Zboguszlocalhost:bogus)r� ro r r% )r r r r �testNonnumericPort4 s
z%NonConnectingTests.testNonnumericPortN)rE rF rG r� r� r r r r r� ( s
r� c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�BadHELOServerTestsc C sJ t t _ t j d � t j | _ t j � | _ | j t _ d | _
d S)Ns 199 no hello for you!r ) r r r
r$ rX rY rZ r[ r\ r] r )r r r r r @ s
zBadHELOServerTests.setUpc C s t t _ | j t _ d S)N)r
r rZ rX rY )r r r r r H s zBadHELOServerTests.tearDownc C s) | j t j t j t | j d d � d S)Nrk rl )r� r ZSMTPConnectErrorr% r& r )r r r r �testFailingHELOL s z"BadHELOServerTests.testFailingHELON)rE rF rG r r r� r r r r r� = s r� c @ sM e Z d Z d d e j d d Z d d � Z d d � Z d d
� Z d S) �TooLongLineTestss 250 OK� .rC s
c C s� t j | _ t j � | _ | j t _ t j � | _ t j t j
t j � | _ | j j
d � t j | j � | _ | j | j | j f } t j d t d | � j � | j j � | j j � d S)N� rS rT )rX rY rZ r[ r\ r] r^ r_ r r
ZAF_INETZ SOCK_STREAMr4 Z
settimeoutr Z bind_portr �respdatard r rf rg rh )r Zservargsr r r r U s
zTooLongLineTests.setUpc C s | j j � | j t _ d S)N)r rg rZ rX rY )r r r r r c s
zTooLongLineTests.tearDownc C s) | j t j t j t | j d d � d S)Nrk rl )r� r �SMTPResponseExceptionr% r& r )r r r r �testLineTooLongg s z TooLongLineTests.testLineTooLongN) rE rF rG r Z_MAXLINEr� r r r� r r r r r� Q s r� zMr.A@somewhere.comzJohn AzMs.B@xn--fo-fka.comzSally BzMrs.C@somewhereesle.comzRuth CZ somepasswordz<PENCeUxFREJoU0NnbmhNWitOMjNGNndAZWx3b29kLmlubm9zb2Z0LmNvbT4=zlist-1zlist-2c @ s e Z d Z d S)�ResponseExceptionN)rE rF rG r r r r r� y s r� c sE e Z d Z d Z d Z d Z d Z d Z d Z d Z d Z
d Z � f d d � Z � f d d � Z
d d � Z d
d � Z d d
� Z d d d � Z d d d � Z d d d � Z d d � Z d d � Z d d � Z � f d d � Z � f d d � Z � f d d � Z � f d d! � Z � f d" d# � Z d$ d% � Z � S)&�SimSMTPChannelNr �c c s< d j d d � | D� � | _ t t | � j | | � d S)N� c S s g | ] } d j | � � q S)z 250-{0}
)r� )�.0�xr r r �
<listcomp>� s z+SimSMTPChannel.__init__.<locals>.<listcomp>)rj �_extrafeatures�superr� �__init__)r Zextra_featuresrT �kw)� __class__r r r� � s zSimSMTPChannel.__init__c s� | j | j k r� | j j | j � } t d t | � d t j �g | _ y | j | � WnR t
k
r� } z2 | j | _ | j d | j
| j f � d SWYd d } ~ Xn Xt � j � d S)NzData:�filez%s %s)�
smtp_state�AUTHZ _emptystringrj Zreceived_lines�print�reprr` ra � auth_objectr� �COMMAND�push� smtp_code�
smtp_errorr� �found_terminator)r �liner )r� r r r� � s z SimSMTPChannel.found_terminatorc C s1 | j s | j d � d S| j s3 d | j k rD | j d � d S| j d k rd | j d � d S| j � } t | � d
k r� | j d � d Sd | d j � j d
d � } y t | | � | _
Wn, t k
r� | j d j | � � d SYn X| j
| _ | j
t | � d k r&| d n d � d S)Nz503 Error: send EHLO firstr� z(500 Error: command "AUTH" not recognizedz3503 Bad sequence of commands: already authenticatedr rC z/501 Syntax: AUTH <mechanism> [initial-response]z_auth_%sr �-�_zQ504 Command parameter not implemented: unsupported authentication mechanism {!r})r rC )�
seen_greetingr� �
extended_smtpr� �authenticated_user�split�len�lower�replace�getattrr� �AttributeErrorr� r� r� )r �argrT Zauth_object_namer r r � smtp_AUTH� s.
zSimSMTPChannel.smtp_AUTHc C s< | r | | _ | j d � n
| j d � | j | _ d S)Nz 235 Authentication Succeededz&535 Authentication credentials invalid)r� r� r� r� )r �userZvalidr r r �_authenticated� s
z SimSMTPChannel._authenticatedc C s t j | j d � � j d � S)Nr� zutf-8)�base64Z decodebytes�encoder� )r �stringr r r �_decode_base64� s z SimSMTPChannel._decode_base64c C s� | d k r | j d � n� | j | � } y | j d � ^ } } } WnB t k
r� } z" | j d j | | � � d SWYd d } ~ Xn X| j | | t d k � d S)Nz334 � z=535 Splitting response {!r} into user and password failed: {}r )r� r� r� r� r� r� �sim_auth)r r� �logpassr� r� �passwordr r r r � _auth_plain� s
zSimSMTPChannel._auth_plainc C s� | d k r | j d � nc t | d � sM | j | � | _ | j d � n2 | j | � } | j | j | t d k � | ` d S)Nz334 VXNlcm5hbWU6�_auth_login_userz334 UGFzc3dvcmQ6r )r� rI r� r� r� r� )r r� r� r r r � _auth_login� s zSimSMTPChannel._auth_loginc C s� | d k r% | j d j t � � n� | j | � } y | j � \ } } WnB t k
r� } z" | j d j | | � � d SWYd d } ~ Xn Xt j t d j d � | j t � j d � d � j
� } | j | | | k � d S)Nz334 {}z<535 Splitting response {!r} into user and passwordfailed: {}Fr r� Zmd5) r� r� �sim_cram_md5_challenger� r� r� �hmacZHMACr� r� Z hexdigestr� )r r� r� r� Z hashed_passr Zvalid_hashed_passr r r �_auth_cram_md5� s
z SimSMTPChannel._auth_cram_md5c C s: d } | | j d } | j | � | | _ d | _ d S)NzH250-testhost
250-EXPN
250-SIZE 20000000
250-STARTTLS
250-DELIVERBY
z250 HELPT)r� r� r� r� )r r� �respr r r � smtp_EHLO� s
zSimSMTPChannel.smtp_EHLOc C sH | t k r3 | j d t | t j | � f � n | j d | � d S)Nz 250 %s %sz550 No such user: %s)� sim_usersr� r � quoteaddr)r r� r r r � smtp_VRFY� s 'zSimSMTPChannel.smtp_VRFYc C s� | j � } | t k r� t | } x� t | � D]j \ } } t j | � } | t | � d k r~ | j d t | | f � q/ | j d t | | f � q/ Wn
| j d � d S)Nr z 250-%s %sz 250 %s %sz550 No access for you!)r� � sim_lists� enumerater r� r� r� r� )r r� Z list_nameZ user_listr Z
user_emailZ quoted_addrr r r � smtp_EXPN� s
"zSimSMTPChannel.smtp_EXPNc sF | j d k r( t t | � j | � n | j | j � | j � d S)N)�
quit_responser� r� � smtp_QUITr� �close_when_done)r r� )r� r r r� s zSimSMTPChannel.smtp_QUITc sI | j d k r" t � j | � n# | j | j � | j rE | j � d S)N)�
mail_responser� � smtp_MAILr� �
disconnectr )r r� )r� r r r s
zSimSMTPChannel.smtp_MAILc sQ | j d k r# t � j | � d S| j d 7_ | j | j | j d � d S)Nr )�
rcpt_responser� � smtp_RCPT�
rcpt_countr� )r r� )r� r r r s
zSimSMTPChannel.smtp_RCPTc s# | j d 7_ t � j | � d S)Nr )�
rset_countr� � smtp_RSET)r r� )r� r r r s zSimSMTPChannel.smtp_RSETc s6 | j d k r" t � j | � n | j | j � d S)N)�
data_responser� � smtp_DATAr� )r r� )r� r r r
# s zSimSMTPChannel.smtp_DATAc C s � d S)Nr )r r r r � handle_error) s zSimSMTPChannel.handle_error) rE rF rG 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� z s4
r� c @ sR e Z d Z e Z d d � Z d d � Z d d � Z d d � Z d d
� Z d S) �
SimSMTPServerc O s# g | _ t j j | | | � d S)N)�_extra_featuresr` �
SMTPServerr� )r rT r� r r r r� 1 s zSimSMTPServer.__init__c C s+ | j | j | | | d | j �| _ d S)NrR )�
channel_classr
� _decode_data� _SMTPchannel)r r r r r r �handle_accepted5 s z SimSMTPServer.handle_acceptedc C s d S)Nr )r �peer�mailfrom�rcpttos�datar r r �process_message: s z SimSMTPServer.process_messagec C s | j j | � d S)N)r
�append)r Zfeaturer r r � add_feature= s zSimSMTPServer.add_featurec C s � d S)Nr )r r r r r @ s zSimSMTPServer.handle_errorN)
rE rF rG r� r r� r r r r r r r r r - s r c @ s e Z 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) S)*� SMTPSimTestsc C s� t j | _ t j t _ t j � | _ t j � | _ t t d f d d d �| _
| j
j j � d | _ | j
| j | j f } t j
d t d | � | _ | j j � | j j � | j j � d S)
Nr rQ r rR TrS rT rU )rQ rU )r
rV rW r r^ r_ rM rN r r& r rc r rd rO re rf rg rh )r ri r r r r I s
zSMTPSimTests.setUpc C s7 | j t _ | j j � | j j � | j j � d S)N) rW r
rV rN r rM rg re rj )r r r r r Z s
zSMTPSimTests.tearDownc C s/ t j t | j d d d d �} | j � d S)Nr. rk r r� )r r% r& r rm )r r' r r r rn b s !zSMTPSimTests.testBasicc
C s� t j t | j d d d d �} | j | j i � d d d d d d d
d d d i } | j � | j | j | � x$ | D] } | j | j | � � q| W| j | j d � � | j
� d S)
Nr. rk r r� �expnr� �sizeZ20000000ZstarttlsZ deliverbyr� zunsupported-feature) r r% r& r r! �esmtp_featuresrx �
assertTrue�has_extn� assertFalserm )r r' Zexpected_features�kr r r �testEHLOg s !
zSMTPSimTests.testEHLOc C s� t j t | j d d d d �} x[ t j � D]M \ } } d t d | t j | � f d � f } | j | j | � | � q. Wd } d d
| j
d � f } | j | j | � | � | j � d S) Nr. rk r r� rq z%s %sr� znobody@nowhere.comi& zNo such user: %s) r r% r& r r� �items�bytesr� r! r~ r� rm )r r' Z addr_spec�name�expected_known�u�expected_unknownr r r r | s ! zSMTPSimTests.testVRFYc C s� t j t | j d d d d �} x� t j � D]~ \ } } g } x2 | D]* } | j d t | t j | � f � qG Wd t d j
| � d � f } | j | j | � | � q. Wd } d } | j | j | � | � | j
� d S)
Nr. rk r r� z%s %srq �
r� zPSU-Members-List�& � No access for you!)r* r+ )r r% r& r r� r# r r� r� r$ rj r! r rm ) r r' ZlistnameZmembersZusersr� r& r' r( r r r �testEXPN� s !
( zSMTPSimTests.testEXPNc C si | j j d � t j t | j d d d d �} | j t d t d � } | j | d
� | j � d S) Nz
AUTH PLAINr. rk r r� r r �� � Authentication Succeeded)r- r. )
r r r r% r& r �loginr� r! r )r r' r� r r r �testAUTH_PLAIN� s
!zSMTPSimTests.testAUTH_PLAINc C si | j j d � t j t | j d d d d �} | j t d t d � } | j | d
� | j � d S) Nz
AUTH LOGINr. rk r r� r r r- � Authentication Succeeded)r- r1 )
r r r r% r& r r/ r� r! r )r r' r� r r r �testAUTH_LOGIN� s
!zSMTPSimTests.testAUTH_LOGINc C si | j j d � t j t | j d d d d �} | j t d t d � } | j | d
� | j � d S) Nz
AUTH CRAM-MD5r. rk r r� r r r- � Authentication Succeeded)r- r3 )
r r r r% r& r r/ r� r! r )r r' r� r r r �testAUTH_CRAM_MD5� s
!z SMTPSimTests.testAUTH_CRAM_MD5c C si | j j d � t j t | j d d d d �} | j t d t d � } | j | d
� | j � d S) Nz AUTH BOGUS PLAIN LOGIN CRAM-MD5r. rk r r� r r r- � Authentication Succeeded)r- r5 )
r r r r% r& r r/ r� r! r )r r' r� r r r �testAUTH_multiple� s
!z SMTPSimTests.testAUTH_multiplec C s d d d h } x' | D] } | j j d j | � � q Wx� | D]� } | j d | � �� t j t | j d d d d �} | j d
� t d t d | _
| _ d
| j � j
d d � } | j | t | | � � } | j | d � | j � Wd QRXq@ Wd S)NzCRAM-MD5ZPLAINZLOGINzAUTH {}� mechanismr. rk r r� Zfoor r Zauth_r� r� r- � Authentication Succeeded)r- r8 )r r r� ZsubTestr r% r& r rx r� r� r� r� r� �authr� r! r )r Z supportedr7 r' �methodr� r r r �test_auth_function� s
z SMTPSimTests.test_auth_functionc C s� t j t | j d d d d �} | j � \ } } | j | d � | j d | j � | j � | j d | j � | j
t | j � | j d | j � | j � | j d | j � | j � d S)Nr. rk r r� rq r ) r r% r& r rx r! �assertInr rm Z assertNotInr; Zehlo_or_helo_if_needed)r r' �code�messager r r �test_quit_resets_greeting� s
z&SMTPSimTests.test_quit_resets_greetingc C s� t j t | j � �) } | j � \ } } | j | d � Wd QRX| j t j | j d � t j t | j � � } | j � Wd QRX| j t j | j d � d S)Nrq s foo)
r r% r& r rs r! r� r� r r )r r' r= r> r r r �test_with_statement� s z SMTPSimTests.test_with_statementc C s� | j t j � �? } t j t | j � � } | j � d | j j _ Wd QRXWd QRX| j
| j j d � | j
| j j
d � d S)Nz421 QUIT FAILEDi� s QUIT FAILED)r� r r� r% r& r rs r r r� r! � exceptionr� r� )r �errorr' r r r � test_with_statement_QUIT_failure� s
z-SMTPSimTests.test_with_statement_QUIT_failurec C s� t j t | j d d d d �} | j � d | j j _ d | j j _ | j t j
� � | j d d d � Wd QRX| j | j
� d S)
Nr. rk r r� z 451 Requested action abortedTr� r� z test message)r r% r& r rs r r r r r� �SMTPSenderRefusedr� r1 r4 )r r' r r r �test__rest_from_mail_cmd� s !
z%SMTPSimTests.test__rest_from_mail_cmdc C s� t j t | j d d d d �} | j � d | j j _ | j t j � � | j
d d d � Wd QRX| j | j � | j
| j j j d � d S)
Nr. rk r r� z421 closing connectionr� r� z test messager )r r% r& r rs r r r r� rD r� r1 r4 r! r )r r' r r r �test_421_from_mail_cmd� s !
z#SMTPSimTests.test_421_from_mail_cmdc C s� t j t | j d d d d �} | j � d d g | j j _ | j t j � �# } | j
d d d d
g d � Wd QRX| j | j � | j
| j j j d � | j | j j d d d i � d S)Nr. rk r r� z 250 acceptedz 421 closingr� r� ZFrankZGeorgez test messager � � closing)rG rH )r r% r& r rs r r r r� ZSMTPRecipientsRefusedr� r1 r4 r! r ZassertDictEqualrA rT )r r' r r r r �test_421_from_rcpt_cmd� s !
#z#SMTPSimTests.test_421_from_rcpt_cmdc C s� Gd d � d t � } | | j _ t j t | j d d d d �} | j � | j t j � � | j
d d g d � Wd QRX| j | j � | j
| j j j d
� d S) Nc s" e Z d Z � f d d � Z � S)z=SMTPSimTests.test_421_from_data_cmd.<locals>.MySimSMTPChannelc s3 | j | j k r" | j d � n
t � j � d S)Nz 421 closing)r� ZDATAr� r� r� )r )r� r r r� s zNSMTPSimTests.test_421_from_data_cmd.<locals>.MySimSMTPChannel.found_terminator)rE rF rG r� r r )r� r �MySimSMTPChannel s rJ r. rk r r� z John@foo.orgz
Sally@foo.orgz test messager )r� r r r r% r& r rs r� Z
SMTPDataErrorr� r1 r4 r! r r )r rJ r' r r r �test_421_from_data_cmd s !
z#SMTPSimTests.test_421_from_data_cmdc C s� t j t | j d d d d �} | j | j � | j � | j | j � | j | j
d � � | j t j | j
d d d d d
d g �| j t j | j d d d
d g �d S)
Nr. rk r rl �smtputf8r� r� r� � mail_optionsz
BODY=8BITMIME�SMTPUTF8�options)r r% r& r �
addCleanupr rx r �
does_esmtpr r r� �SMTPNotSupportedErrorr� �mail)r r' r r r �4test_smtputf8_NotSupportedError_if_no_server_support s
zASMTPSimTests.test_smtputf8_NotSupportedError_if_no_server_supportc C sg t j t | j d d d d �} | j | j � | j t | j d d d � | j t | j d � d S) Nr. rk r rl ZAliceu Böbr� u Älice)
r r% r& r rP r r� �UnicodeEncodeErrorr� rS )r r' r r r �"test_send_unicode_without_SMTPUTF8$ s
z/SMTPSimTests.test_send_unicode_without_SMTPUTF8N)rE rF rG r r rn r" r r, r0 r2 r4 r6 r; r? r@ rC rE rF rI rK rT rV r r r r r F s(
r c @ s: e Z d Z d d � Z d d � Z d d d d � Z d S)�SimSMTPUTF8Serverc O s) d d g | _ t j j | | | � d S)NrN Z8BITMIME)r
r` r r� )r rT r� r r r r� . s zSimSMTPUTF8Server.__init__c C s4 | j | j | | | d | j d | j �| _ d S)NrR �enable_SMTPUTF8)r r
r rX r )r r r r r r r 5 s z!SimSMTPUTF8Server.handle_acceptedNc C s: | | _ | | _ | | _ | | _ | | _ | | _ d S)N)Z last_peer�
last_mailfrom� last_rcpttos� last_message�last_mail_options�last_rcpt_options)r r r r r rM Z rcpt_optionsr r r r < s z!SimSMTPUTF8Server.process_message)rE rF rG r� r r r r r r rW , s rW c @ sj e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
d S)�SMTPUTF8SimTestsNc C s� t j | _ t j t _ t j � | _ t j � | _ t t d f d d d d d �| _
| j
j j � d | _ | j
| j | j f } t j
d t d | � | _ | j j � | j j � | j j � d S) Nr rQ r rR FrX TrS rT rU )rQ rU )r
rV rW r r^ r_ rM rN rW r& r rc r rd rO re rf rg rh )r ri r r r r K s
zSMTPUTF8SimTests.setUpc C s7 | j t _ | j j � | j j � | j j � d S)N) rW r
rV rN r rM rg re rj )r r r r r ^ s
zSMTPUTF8SimTests.tearDownc C se t j t | j d d d d �} | j | j � | j � | j | j � | j | j d � � d S)Nr. rk r rl rL )
r r% r& r rP r rx r rQ r )r r' r r r �$test_test_server_supports_extensionsf s
z5SMTPUTF8SimTests.test_test_server_supports_extensionsc C s� d j d � } t j t | j d d d d �} | j | j � | j d d | d d
d g �| j | j j
d � | j | j j d g � | j | j j | � | j
d
| j j � | j
d | j j � | j | j j g � d S) Nu$ ¡a test message containing unicode!zutf-8r. rk r rl u Jőhnu SállyrM z
BODY=8BITMIMErN )r� r r% r& r rP r r� r! r rY rZ r[ r< r\ r] )r r� r' r r r �,test_send_unicode_with_SMTPUTF8_via_sendmailn s
z=SMTPUTF8SimTests.test_send_unicode_with_SMTPUTF8_via_sendmailc C s, d j d � } t j t | j d d d d �} | j | j � | j � | j | j d d d d
g �d � | j | j
d
� d � | j | j | � d � | j | j j
d � | j | j j d
g � | j | j j | � | j d | j j � | j d
| j j � | j | j j g � d S)Nu$ ¡a test message containing unicode!zutf-8r. rk r rl u JőrO z
BODY=8BITMIMErN rq � OKu János)rq ra )rq ra )rq ra )r� r r% r& r rP r rx r! rS Zrcptr r rY rZ r[ r< r\ r] )r r� r' r r r �1test_send_unicode_with_SMTPUTF8_via_low_level_API| s
zBSMTPUTF8SimTests.test_send_unicode_with_SMTPUTF8_via_low_level_APIc C s t � } d | d <d | d <d | d <| j d � t j d � } t j t | j d d
d d �} | j | j � | j
| j | � i � | j
| j j
d
� | j
| j j d g � | j
| j j j � | � | j d | j j � | j d | j j � | j
| j j g � d S)Nu Páolo <főo@bar.com>r� r� r� u Nudge nudge, wink, wink ὠ9�Subjectu1 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?
r. rk r rl u főo@bar.comz
BODY=8BITMIMErN )r Z set_content�textwrap�dedentr r% r& r rP r r! r� r rY rZ r[ r� r< r\ r] )r �msgr" r' r r r �2test_send_message_uses_smtputf8_if_addrs_non_ascii� s"
zCSMTPUTF8SimTests.test_send_message_uses_smtputf8_if_addrs_non_asciic C sx t � } d | d <d | d <d | d <t j t | j d d d d
�} | j | j � | j t j | j | � � d S) Nu Páolo <főo@bar.com>r� r� r� u Nudge nudge, wink, wink ὠ9rc r. rk r rl )
r r r% r& r rP r r� rR r� )r rf r' r r r �9test_send_message_error_on_non_ascii_addrs_if_no_smtputf8� s
zJSMTPUTF8SimTests.test_send_message_error_on_non_ascii_addrs_if_no_smtputf8) rE rF rG r� r r r_ r` rb rg rh r r r r r^ F s r^ s psu doesnotexistZeolr� c @ s e Z d Z d d � Z d S)�!SimSMTPAUTHInitialResponseChannelc C sf | j � } | d j � d k rU t | � d k rU | d t k rU | j d � d S| j d � d S)Nr �plainrC r z235 Okz571 Bad authentication)r� r� r� �EXPECTED_RESPONSEr� )r r� rT r r r r� � s
z+SimSMTPAUTHInitialResponseChannel.smtp_AUTHN)rE rF rG r� r r r r ri � s ri c @ s e Z d Z e Z d S)� SimSMTPAUTHInitialResponseServerN)rE rF rG ri r r r r r rl � s rl c @ s@ e Z d Z d d � Z d d � Z d d � Z d d � Z d S)
� SMTPAUTHInitialResponseSimTestsc C s� t j | _ t j t _ t j � | _ t j � | _ t t d f d d d �| _
| j
j j � d | _ | j
| j | j f } t j
d t d | � | _ | j j � | j j � | j j � d S)
Nr rQ r rR TrS rT rU )rQ rU )r
rV rW r r^ r_ rM rN rl r& r rc r rd rO re rf rg rh )r ri r r r r � s
z%SMTPAUTHInitialResponseSimTests.setUpc C s7 | j t _ | j j � | j j � | j j � d S)N) rW r
rV rN r rM rg re rj )r r r r r � s
z(SMTPAUTHInitialResponseSimTests.tearDownc C sO | j j d � t j t | j d d d d �} | j d d � | j � d S)Nz
AUTH PLAINr. rk r r� �psu� doesnotexist)r r r r% r& r r/ r )r r' r r r �%testAUTH_PLAIN_initial_response_login� s
zESMTPAUTHInitialResponseSimTests.testAUTH_PLAIN_initial_response_loginc C s| | j j d � t j t | j d d d d �} d | _ d | _ | j d | j � \ } } | j
� | j | d � d S)
Nz
AUTH PLAINr. rk r r� rn ro rj r- ) r r r r% r& r r� r� r9 Z
auth_plainr r! )r r' r= Zresponser r r �$testAUTH_PLAIN_initial_response_auth� s
zDSMTPAUTHInitialResponseSimTests.testAUTH_PLAIN_initial_response_authN)rE rF rG r r rp rq r r r r rm � s rm c C s# t j t t t t t t t � d S)N) r Z run_unittestr� rP r r� rm r r� )�verboser r r � test_main� s rs �__main__)=rJ r� Zemail.mime.textr� Z
email.messager Zemail.base64mimer Z
encode_base64Z email.utilsr� r
r` r r[ r< rX rK r rp rd ZunittestZtestr r r^ � ImportErrorr& �platformr
Z SMTPChannelr ZTestCaser rO r� r� Z
skipUnlessrP r� r� r� r� r� r� r� � Exceptionr� r� r r r rW r^ rk ri rl rm Z reap_threadsrs rE r r r r �<module> s~
_ � f � � r -