Your IP : 216.73.216.213
���] � � @ 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 �1/opt/alt/python35/lib64/python3.5/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 � )�listen�set�accept�socket�timeout�select�send�close)
�evt�buf�serv�conn�addr�n�r�w�e�sentr 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% r2 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 mundor4 r )
r r( r5 r r6 r7 r! r) r* r" r8 r9 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 r4 )
r r( r! r) r* r" r% r8 r9 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) r= r r> r? r* r" r r@ rA rB rC rD )r r+ rE r& r r r �test_debuglevel_2� s
z GeneralTests.test_debuglevel_2N)�__name__�
__module__� __qualname__r# r$ r' r, r0 r1 r3 r: r; r< rF rH 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 �asyncoreZpoll2rL Z
socket_map�is_setr r r �time�sleepZ close_all)r �serv_evt�
client_evtZpoll_funr r r r �debugging_server� s&
rT 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���)rV rZ ) r �getfqdn� real_getfqdnr �sys�stdout�
old_stdout�io�StringIO�output� threading�EventrR rS �smtpd� DEBUGSTREAM�old_DEBUGSTREAMZDebuggingServerr* r � getsocknamer" �ThreadrT �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)r\ r r[ rS r
rR rl rj �joinr_ r] r^ re rf r rg )r r r r r$ � s
z DebuggingServerTests.tearDownc C s/ t j t | j d d d d �} | j � d S)Nr2 � 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)Nr2 rp r rq r- z 127.0.0.1zcouldn't bind to port %d)r Zfind_unused_portr! r) r* r" r% r- r2 rr �OSError�errnoZ
EADDRINUSEZskipTest)r r" r+ r r r r r0 � s z&DebuggingServerTests.testSourceAddressc C sK t j t | j d d d d �} d } | j | j � | � | j � d S)Nr2 rp r rq � � OK)rv rw )r! r) r* r" r% �nooprr )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)Nr2 rp r rq rv � OK)rv rz )r! r) r* r" r% Zrsetrr )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)Nr2 rp r rq rv �
SIZE 33554432
HELP)rv r| )r! r) r* r" r% �ehlorr )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) Nr2 rp r rq � � EXPN not implementedZEXPN)r r� )r! r) r* r" Zputcmdr% Zgetreplyrr )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) Nr2 rp r rq � s* Cannot VRFY user, but will accept message s and attempt deliveryznobody@nowhere.com)r! r) r* r" r% �vrfyZverifyrr )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)Nr2 rp r rq � � Duplicate HELO/EHLO)r� r� )r! r) r* r" Zhelor% rr )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)Nr2 rp r rq s# Supported commands: EHLO HELO MAIL s RCPT DATA RSET NOOP QUIT VRFY)r! r) r* r" r% �helprr )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 messager2 rp r rq �John�Sallyg{�G�z�?z%s%s
%s)r! r) r* r" �sendmailrP rQ rr rS r
rR rl rb �flush� MSG_BEGIN�MSG_ENDr% rD )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 messager2 rp r rq r� r� g{�G�z�?z%s%s
%s�ascii)r! r) r* r" r� rP rQ rr rS r
rR rl rb r� r� �decoder� r% rD )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.r2 rp r rq r� r� g{�G�z�?z%s%s
%s)r! r) r* r" r� rP rQ rr rS r
rR rl rb r� r� r� r% rD )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 messager2 rp r rq z<>r� g{�G�z�?z%s%s
%sz ^sender: <>$)r! r) r* r" r� rP rQ rr rS r
rR rl rb r� r� r� r% rD re rf r@ rA rB rC )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 messager2 rp r rq � from_addrr� �to_addrsr� g{�G�z�?zX-Peerz%s%s
%s)�email�mime�text�MIMETextr! r) r* r" � send_messagerP rQ rr rS r
rR rl rb r� r �
gethostbynamer� � as_stringr� r% rD )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>�Bccr2 rp r rq 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� rP rQ rr r% rS r
rR rl rb r� r r� r� r� r� rD re rf r@ rA rB rC �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� r2 rp r rq 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� rP rQ rr rS r
rR rl rb r� r r� r� r� r� r% rD re rf r@ rA rB rC 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� r2 rp r rq 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� rP rQ rr rS r
rR rl rb r� r r� r� r� r� r% rD re rf r@ rA rB rC 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� r2 rp r rq 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� rP rQ rr rS r
rR rl rb r� r r� r� r� r� r% rD re rf r@ rA rB rC 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-Bccr2 rp r rq 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� rP rQ rr rS r
rR rl rb r� r r� r� r� r� r% rD re rf r@ rA rB rC 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 +0000r2 rp r rq ) r� r� r� r� r! r) r* r" � assertRaises�
ValueErrorr� r )r r� r+ r r r �#testSendMessageMultipleResentRaises s
!z8DebuggingServerTests.testSendMessageMultipleResentRaises)rI rJ rK �maxDiffr# r$ rs r0 ry r{ r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r rU � s.
! "rU 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� �SMTPServerDisconnectedr} 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)Nrp Zboguszlocalhost:bogus)r� rt r! r) )r r r r �testNonnumericPort4 s
z%NonConnectingTests.testNonnumericPortN)rI rJ rK 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( r] r^ r_ r` ra rb r" )r r r r r# @ s
zBadHELOServerTests.setUpc C s t t _ | j t _ d S)N)r r! r_ r] r^ )r r r r r$ H s zBadHELOServerTests.tearDownc C s) | j t j t j t | j d d � d S)Nrp rq )r� r! ZSMTPConnectErrorr) r* r" )r r r r �testFailingHELOL s z"BadHELOServerTests.testFailingHELON)rI rJ rK 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� .rG 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� rX rY )r] r^ r_ r` ra rb rc rd r r �AF_INET� SOCK_STREAMr8 �
settimeoutr Z bind_portr" �respdatari r rk rl rm )r Zservargsr r r r# U s
zTooLongLineTests.setUpc C s | j j � | j t _ d S)N)r rl r_ r] r^ )r r r r r$ c s
zTooLongLineTests.tearDownc C s) | j t j t j t | j d d � d S)Nrp rq )r� r! �SMTPResponseExceptionr) r* r" )r r r r �testLineTooLongg s z TooLongLineTests.testLineTooLongN) rI rJ rK 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)rI rJ rK 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>)ro �_extrafeatures�superr� �__init__)r Zextra_featuresrY �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 _emptystringro Zreceived_lines�print�reprre rf � 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 rG z/501 Syntax: AUTH <mechanism> [initial-response]z_auth_%sr �-�_zQ504 Command parameter not implemented: unsupported authentication mechanism {!r})r rG )�
seen_greetingr� �
extended_smtpr� �authenticated_user�split�len�lower�replace�getattrr� �AttributeErrorr� r� r� )r �argrY 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� rM 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� �md5) r� r� �sim_cram_md5_challenger� r� r� �hmacZHMACr� r� � 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) rI rJ rK 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_featuresre �
SMTPServerr� )r rY r� r r r r� 1 s zSimSMTPServer.__init__c C s+ | j | j | | | d | j �| _ d S)NrW )�
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)
rI rJ rK 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 rV r rW TrX rY rZ )rV rZ )r r[ r\ r rc rd rR rS r r* r rh r" ri rT rj rk rl rm )r rn r r r r# I s
zSMTPSimTests.setUpc C s7 | j t _ | j j � | j j � | j j � d S)N) r\ r r[ rS r
rR rl rj ro )r r r r r$ Z s
zSMTPSimTests.tearDownc C s/ t j t | j d d d d �} | j � d S)Nr2 rp r r� )r! r) r* r" rr )r r+ r r r rs 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)
Nr2 rp r r� �expnr� �sizeZ20000000ZstarttlsZ deliverbyr� zunsupported-feature) r! r) r* r" r% �esmtp_featuresr} �
assertTrue�has_extn� assertFalserr )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) Nr2 rp r r� rv z%s %sr� znobody@nowhere.comi& zNo such user: %s) r! r) r* r" r �items�bytesr r% r� r� rr )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)
Nr2 rp r r� z%s %srv �
r� zPSU-Members-List�& � No access for you!)r4 r5 )r! r) r* r" r r- r" r r r. ro r% r% rr ) r r+ Zlistname�membersZusersr� r0 r1 r2 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 PLAINr2 rp r r� r r �� � Authentication Succeeded)r8 r9 )
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 LOGINr2 rp r r� r r r8 � Authentication Succeeded)r8 r< )
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-MD5r2 rp r r� r r r8 � Authentication Succeeded)r8 r> )
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-MD5r2 rp r r� r r r8 � Authentication Succeeded)r8 r@ )
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 {}� mechanismr2 rp r r� Zfoor r Zauth_r� r� r8 � Authentication Succeeded)r8 rC )r r# r� ZsubTestr! r) r* r" r} r� r� r� r� r� �authr� r% r )r Z supportedrB 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)Nr2 rp r r� rv r& ) r! r) r* r" r} r% �assertInr' rr 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)Nrv s foo)
r! r) r* r" rx r% r� r� r r )r r+ rH rI 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" rx 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)
Nr2 rp r r� z 451 Requested action abortedTr� r� z test message)r! r) r* r" rx r r r r
r� �SMTPSenderRefusedr� r5 r8 )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)
Nr2 rp r r� z421 closing connectionr� r� z test messager )r! r) r* r" rx r r r r� rO r� r5 r8 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)Nr2 rp r r� z 250 acceptedz 421 closingr� r� ZFrankZGeorgez test messager � � closing)rR rS )r! r) r* r" rx r r r r� ZSMTPRecipientsRefusedr� r5 r8 r% r ZassertDictEqualrL rY )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)rI rJ rK r� r r )r� r �MySimSMTPChannel s rU r2 rp r r� z John@foo.orgz
Sally@foo.orgz test messager )r� r r r! r) r* r" rx r� Z
SMTPDataErrorr� r5 r8 r% r r )r rU 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)
Nr2 rp r rq �smtputf8r� r� r� � mail_optionsz
BODY=8BITMIME�SMTPUTF8�options)r! r) r* r" �
addCleanupr r} 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) Nr2 rp r rq ZAliceu Böbr� u Älice)
r! r) r* r" r[ r r� �UnicodeEncodeErrorr� r^ )r r+ r r r �"test_send_unicode_without_SMTPUTF8$ s
z/SMTPSimTests.test_send_unicode_without_SMTPUTF8N)rI rJ rK r# r$ rs r, r� r7 r; r= r? rA rF rJ rK rN rP rQ rT rV r_ ra 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)NrY Z8BITMIME)r re r r� )r rY r� r r r r� . s zSimSMTPUTF8Server.__init__c C s4 | j | j | | | d | j d | j �| _ d S)NrW �enable_SMTPUTF8)r r r rc 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 rX Z rcpt_optionsr r r r! < s z!SimSMTPUTF8Server.process_message)rI rJ rK r� r r! r r r r rb , s rb 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 rV r rW Frc TrX rY rZ )rV rZ )r r[ r\ r rc rd rR rS rb r* r rh r" ri rT rj rk rl rm )r rn r r r r# K s
zSMTPUTF8SimTests.setUpc C s7 | j t _ | j j � | j j � | j j � d S)N) r\ r r[ rS r
rR rl rj ro )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)Nr2 rp r rq rW )
r! r) r* r" r[ r r} r( r\ 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-8r2 rp r rq u Jőhnu SállyrX z
BODY=8BITMIMErY )r� r! r) r* r" r[ r r� r% r rd re rf rG rg rh )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-8r2 rp r rq u JőrZ z
BODY=8BITMIMErY rv � OKu János)rv rl )rv rl )rv rl )r� r! r) r* r" r[ r r} r% r^ Zrcptr r rd re rf rG rg rh )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?
r2 rp r rq u főo@bar.comz
BODY=8BITMIMErY )r Z set_content�textwrap�dedentr! r) r* r" r[ r r% r� r rd re rf r� rG rg rh )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 ὠ9rn r2 rp r rq )
r r! r) r* r" r[ r r� r] r� )r rq 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) rI rJ rK r� r# r$ rj rk rm rr rs r r r r ri F s ri 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 �plainrG r z235 Okz571 Bad authentication)r� r� r� �EXPECTED_RESPONSEr� )r r� rY r r r r� � s
z+SimSMTPAUTHInitialResponseChannel.smtp_AUTHN)rI rJ rK r� r r r r rt � s rt c @ s e Z d Z e Z d S)� SimSMTPAUTHInitialResponseServerN)rI rJ rK rt r r r r r rw � s rw 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 rV r rW TrX rY rZ )rV rZ )r r[ r\ r rc rd rR rS rw r* r rh r" ri rT rj rk rl rm )r rn r r r r# � s
z%SMTPAUTHInitialResponseSimTests.setUpc C s7 | j t _ | j j � | j j � | j j � d S)N) r\ r r[ rS r
rR rl rj ro )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 PLAINr2 rp 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 PLAINr2 rp r r� ry rz ru r8 ) r r# r! r) r* r" r� r� rD Z
auth_plainr r% )r r+ rH �responser r r �$testAUTH_PLAIN_initial_response_auth� s
zDSMTPAUTHInitialResponseSimTests.testAUTH_PLAIN_initial_response_authN)rI rJ rK r# r$ r{ r} r r r r rx � s rx c C s# t j t t t t t t t � d S)N) r Z run_unittestr� rU r r� rx r$ r� )�verboser r r � test_main� s r �__main__)=rN r� Zemail.mime.textr� Z
email.messager Zemail.base64mimer Z
encode_base64Z email.utilsr� r re r! r` r@ r] rP r ru ro ZunittestZtestr r rc � ImportErrorr* �platformr
Z SMTPChannelr ZTestCaser rT r� r� Z
skipUnlessrU r� r� r� r r� r� r � Exceptionr� r� r r r$ rb ri rv rt rw rx Z reap_threadsr rI r r r r �<module> s~
_ � f � � r -