Your IP : 216.73.216.213
�
g f� �
@ s� 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 rd d � Z e e j _ n 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! i d d 6d d 6d d 6Z" d d f Z# d Z$ i d d 6d! d" 6d# d$ 6Z% d% Z& i d d g d& 6d g d' 6Z' Gd( d) � d) e j � Z( Gd* d+ � d+ e j) � Z* e
j e d � Gd, d- � d- e
j � � Z+ e j, d d. d/ � � Z- e. d0 k r�e- � n d S)1� N)�support� mock_socket�darwinc C s d S)N� )�selfr r �6/opt/alt/python34/lib64/python3.4/test/test_smtplib.py� handle_expt s r c
C s� | j d � | j � z� y | j � \ } } Wn t j k
rG Yn} Xd } xi | r� | d k r� t j g | g g � \ } } } | r� | j | � } | | d � } n | d 8} qQ W| j � Wd | j � | j � Xd S)N� i� 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 S)� GeneralTestsc C s t t _ d | _ d S)N� )r �smtplibr �port)r r r r �setUp8 s zGeneralTests.setUpc C s
t t _ d S)N)r r )r r r r �tearDown< 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 �
testQuoteDataA 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 �
testBasic1F 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 �testSourceAddressL 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 �
testBasic2T 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 �testLocalHostNameZ 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 �testTimeoutDefaulta 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 mundor/ r
)
r r# r0 r r1 r2 r r$ r% r r3 r4 r )r r& r r r �testTimeoutNonem 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
r/ )
r r# r r$ r% r r r3 r4 r )r r& r r r �testTimeoutValuex s
z GeneralTests.testTimeoutValueN)
�__name__�
__module__� __qualname__r r r" r' r+ r, r. r5 r6 r7 r r r r r 6 s r c C s� | j � z� y~ t t d � r+ t j } n t j } d } xM t j r� | d k r� | d t j � | j � r| | j � Pn | d 8} q= WWn t j
k
r� Yn XWd | j � s� t j d � | j � n t j
� | j � Xd S)N�polli� r g{�G�z�?r
g �?)r �hasattrr �asyncoreZpoll2r; Z
socket_mapZis_setr r r
�time�sleepZ close_all)r �serv_evt�
client_evtZpoll_funr r r r �debugging_server� s&
rB 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 � | _ | j j j � d | _ | j | j | j
f } t
j d t d | � | _ | j j � | j j � | j j � d S)Nr �nowherer
�target�args���)rD rG ) r �getfqdn� real_getfqdnr �sys�stdout�
old_stdout�io�StringIO�output� threading�Eventr@ rA �smtpd� DEBUGSTREAM�old_DEBUGSTREAMZDebuggingServerr% r � getsocknamer �ThreadrB �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)rI r rH rA r r@ rY rW �joinrL rJ rK rR rS r rT )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 � WnN t
k
r� } z. | j t j k r� | j
d | � n � WYd d } ~ Xn Xd S)Nr- r] r
r^ r( z 127.0.0.1zcouldn't bind to port %d)r Zfind_unused_portr r$ r% r r r( r- r_ �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- r] r
r^ � � OK)rc rd )r r$ r% r r �noopr_ )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- r] r
r^ rc � OK)rc rg )r r$ r% r r Zrsetr_ )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- r] r
r^ rc �
SIZE 33554432
HELP)rc ri )r r$ r% r r �ehlor_ )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- r] r
r^ � � EXPN not implementedZEXPN)rl rm )r r$ r% r Zputcmdr Zgetreplyr_ )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- r] r
r^ � s* Cannot VRFY user, but will accept message s and attempt deliveryznobody@nowhere.com)r r$ r% r r �vrfyZverifyr_ )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- r] r
r^ � � Duplicate HELO/EHLO)rr rs )r r$ r% r Zhelor r_ )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- r] r
r^ s# Supported commands: EHLO HELO MAIL s RCPT DATA RSET NOOP QUIT VRFY)r r$ r% r r �helpr_ )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- r] r
r^ �John�Sallyg{�G�z�?z%s%s
%s)r r$ r% r �sendmailr> r? r_ rA r r@ rY rO �flush� MSG_BEGIN�MSG_ENDr �getvalue)r �mr& �mexpectr r r �testSend 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- r] r
r^ rw rx g{�G�z�?z%s%s
%s�ascii)r r$ r% r ry r> r? r_ rA r r@ rY rO rz r{ �decoder| r r} )r r~ r& r r r r �testSendBinary) 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- r] r
r^ rw rx g{�G�z�?z%s%s
%s)r r$ r% r ry r> r? r_ rA r r@ rY rO rz r{ r| r r} )r r~ r& r r r r �testSendNeedingDotQuote7 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- r] r
r^ z<>rx g{�G�z�?z%s%s
%sz ^sender: <>$)r r$ r% r ry r> r? r_ rA r r@ rY rO rz r{ r| r r} rR rS �re�compile� MULTILINE� assertRegex)r r~ r& r �debugout�senderr r r �testSendNullSenderF 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- r] r
r^ � from_addrrw �to_addrsrx g{�G�z�?zX-Peerz%s%s
%s)�email�mime�text�MIMETextr r$ r% r � send_messager> r? r_ rA r r@ rY rO rz r �
gethostbynamer{ � as_stringr| r r} )r r~ r& r r r r �testSendMessageW 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�Fromrw �Toz Sally, Fred�CCz?John Root <root@localhost>, "Dinsdale" <warped@silly.walks.com>�Bccr- r] r
r^ g{�G�z�?zX-Peerz%s%s
%sz^sender: foo@bar.com$rx �Fred�root@localhost�warped@silly.walks.comz^recips: .*'{}'.*$)zJohnzSallyr� r� r� ) r� r� r� r� r r$ r% r r� r> r? r_ r rA r r@ rY rO rz r r� r{ r� r| r} rR rS r� r� r� r� �format)r r~ r& r r� r� r �to_addrr r r � testSendMessageWithAddressesg 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- r] r
r^ g{�G�z�?zX-Peerz%s%s
%sz^sender: foo@bar.com$rw �Dinsdalez^recips: .*'{}'.*$)zJohnr� ) r� r� r� r� r r$ r% r r� r> r? r_ rA r r@ rY rO rz r r� r{ r� r| r r} rR rS 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- r] r
r^ r� zjoe@example.comr� zfoo@example.netg{�G�z�?zX-Peerz%s%s
%sz^sender: joe@example.com$rw r� z^recips: .*'{}'.*$z ^recips: .*'foo@example.net'.*$)zJohnzDinsdale)!r� r� r� r� r r$ r% r r� r> r? r_ rA r r@ rY rO rz r r� r{ r� r| r r} rR rS 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- r] r
r^ g{�G�z�?zX-Peerz%s%s
%sz-^sender: the_rescuers@Rescue-Aid-Society.com$rw r� z^recips: .*'{}'.*$)zJohnzDinsdale) r� r� r� r� r r$ r% r r� r> r? r_ rA r r@ rY rO rz r r� r{ r� r| r r} rR rS 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� rw 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- r] r
r^ 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� r> r? r_ rA r r@ rY rO rz r r� r{ r� r| r r} rR rS 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 QX| j � d S)NzA test messagez foo@bar.comr� rw 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- r] r
r^ ) r� r� r� r� r r$ r% r � assertRaises�
ValueErrorr� r )r r~ r& r r r �#testSendMessageMultipleResentRaises� s
!z8DebuggingServerTests.testSendMessageMultipleResentRaises)r8 r9 r: ZmaxDiffr r r` r+ rf rh rk rn rq rt rv r� r� r� r� r� r� r� r� r� r� r� r r r r rC � s.
! "rC 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� �SMTPServerDisconnectedrj 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)Nr] Zboguszlocalhost:bogus)r� ra r r$ )r r r r �testNonnumericPort s
z%NonConnectingTests.testNonnumericPortN)r8 r9 r: 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# rJ rK rL rM rN rO r )r r r r r % s
zBadHELOServerTests.setUpc C s t t _ | j t _ d S)N)r r rL rJ rK )r r r r r - s zBadHELOServerTests.tearDownc C s) | j t j t j t | j d d � d S)Nr] r^ )r� r ZSMTPConnectErrorr$ r% r )r r r r �testFailingHELO1 s z"BadHELOServerTests.testFailingHELON)r8 r9 r: 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� .� 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� rE rF )rJ rK rL rM rN rO rP rQ r r ZAF_INETZ SOCK_STREAMr3 Z
settimeoutr Z bind_portr �respdatarV r rX rY rZ )r Zservargsr r r r : s
zTooLongLineTests.setUpc C s | j j � | j t _ d S)N)r rY rL rJ rK )r r r r r H s
zTooLongLineTests.tearDownc C s) | j t j t j t | j d d � d S)Nr] r^ )r� r �SMTPResponseExceptionr$ r% r )r r r r �testLineTooLongL s z TooLongLineTests.testLineTooLongN) r8 r9 r: r Z_MAXLINEr� r r r� r r r r r� 6 s r� zJohn AzMr.A@somewhere.comzSally BzMs.B@xn--fo-fka.comzRuth CzMrs.C@somewhereesle.comZ somepasswordz<PENCeUxFREJoU0NnbmhNWitOMjNGNndAZWx3b29kLmlubm9zb2Z0LmNvbT4=ZTXIuQUBzb21ld2hlcmUuY29t�loginz,AE1yLkFAc29tZXdoZXJlLmNvbQBzb21lcGFzc3dvcmQ=�plainZDTXIUQUBZB21LD2HLCMUUY29TIDG4OWQ0MJKWZGQ4ODNMNDA4NTGXMDRLZWMYZJDMODG1zcram-md5ZC29TZXBHC3N3B3JKzlist-1zlist-2c s� e Z d Z d Z d Z d Z d Z d Z d Z d Z � f d d � Z
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 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>r s z+SimSMTPChannel.__init__.<locals>.<listcomp>)r\ �_extrafeatures�superr� �__init__)r Zextra_featuresrF �kw)� __class__r r r� p s zSimSMTPChannel.__init__c C s: d } | | j d } | j | � | | _ d | _ d S)NzH250-testhost
250-EXPN
250-SIZE 20000000
250-STARTTLS
250-DELIVERBY
z250 HELPT)r� �pushZ
seen_greetingZ
extended_smtp)r �argZrespr r r � smtp_EHLOu 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!)�lower� sim_lists� enumerater r� �lenr� r� )r r� Z list_nameZ user_listr Z
user_emailZ quoted_addrr r r � smtp_EXPN� s
"zSimSMTPChannel.smtp_EXPNc C s� | j � j � d k r2 | j d j t � � d S| j � \ } } | j � } | t k rm | j d � d S| d k r� | t d k r� | j d � n9 | d k r� | t d k r� | j d � n
| j d � d S) Nzcram-md5z334 {}z504 auth type unimplementedr� z235 plain auth okr� z
334 Password:z550 No access for you!)�stripr� r� r� �sim_cram_md5_challenge�split�sim_auth_credentials)r r� ZmechZauthr r r � smtp_AUTH� s
zSimSMTPChannel.smtp_AUTHc 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 sL | j d k r" t � j | � n& | j | j � | j rH | j � n 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)r8 r9 r: r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r )r� r r� f s$
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_featuresrR �
SMTPServerr� )r rF r� r r r r� � s zSimSMTPServer.__init__c C s" | j | j | | | � | _ d S)N)�
channel_classr� � _SMTPchannel)r r r r r r �handle_accepted� s z SimSMTPServer.handle_acceptedc C s d S)Nr )r ZpeerZmailfromZrcpttos�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)
r8 r9 r: 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# S)$� SMTPSimTestsc C s� t j | _ t j t _ t j � | _ t j � | _ t t d f d � | _
| j
j j � d | _ | j
| j | j f } t j
d t d | � | _ | j j � | j j � | j j � d S)Nr rD r
rE rF rG )znowhererG )r rH rI r rP rQ r@ rA r� r% r rU r rV rB rW rX rY rZ )r r[ r r r r � s
zSMTPSimTests.setUpc C s7 | j t _ | j j � | j j � | j j � d S)N) rI r rH rA r r@ rY rW r\ )r r r r r � s
zSMTPSimTests.tearDownc C s/ t j t | j d d d d �} | j � d S)Nr- r] r
r� )r r$ r% r r_ )r r& r r r r` � s !zSMTPSimTests.testBasicc C s� t j t | j d d d d �} | j | j i � i d d 6d d 6d d 6d d
6d d 6} | j � | j | j | � x$ | D] } | j | j | � � q� W| j | j d � � | j
� d S)
Nr- r] r
r� r� �expnZ20000000�sizeZstarttlsZ deliverbyru zunsupported-feature) r r$ r% r r �esmtp_featuresrj Z
assertTrueZhas_extnZ assertFalser_ )r r& Zexpected_features�kr r r �testEHLO 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- r] r
r� rc z%s %sr� znobody@nowhere.comi& zNo such user: %s) r r$ r% r r� �items�bytesr� r rp �encoder_ )r r& r� �name�expected_known�u�expected_unknownr r r rq 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- r] r
r� z%s %src �
r� zPSU-Members-List�& � No access for you!)r� r� )r r$ r% r r� r� r� r� r� r� r\ r r� r_ ) 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 �} d
} | j | j t d t d � | � | j � d S) Nz
AUTH PLAINr- r] r
r� �� �
plain auth okr r
)r� r� )
r r� r r$ r% r r r� �sim_authr )r r& Zexpected_auth_okr r r �testAUTH_PLAIN3 s
!$zSMTPSimTests.testAUTH_PLAINc C s� | j j d � t j t | j d d d d �} y | j t d t d � Wn> t j k
r� } z | j t
t | � � WYd d } ~ Xn X| j � d S)Nz
AUTH LOGINr- r] r
r� r r
)
r r� r r$ r% r r� r� �SMTPAuthenticationError�assertIn�sim_auth_login_password�strr )r r& �errr r r �testAUTH_LOGINC s ! )zSMTPSimTests.testAUTH_LOGINc C s� | j j d � t j t | j d d d d �} y | j t d t d � WnB t j k
r� } z | j t
d t | � � WYd d } ~ Xn X| j � d S) Nz
AUTH CRAM-MD5r- r] r
r� r r
zcram-md5)
r r� r r$ r% r r� r� r r r� r r )r r& r r r r �testAUTH_CRAM_MD5K s ! -z SMTPSimTests.testAUTH_CRAM_MD5c C s� | j j d � t j t | j d d d d �} y | j t d t d � Wn> t j k
r� } z | j t
t | � � WYd d } ~ Xn X| j � d S)Nz AUTH BOGUS PLAIN LOGIN CRAM-MD5r- r] r
r� r r
)
r r� r r$ r% r r� r� r r r r r )r r& r r r r �testAUTH_multipleT s ! )z SMTPSimTests.testAUTH_multiplec 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- r] r
r� rc r� ) r r$ r% r rj r r r� r_ Z assertNotInZconnectZehlo_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 QX| j t j | j d � t j t | j � � } | j � Wd QX| j t j | j d � d S)Nrc s foo)
r r$ r% r re r r� r� r r )r r& r r
r r r �test_with_statementl s z SMTPSimTests.test_with_statementc C s� | j t j � �> } t j t | j � � } | j � d | j j _ Wd QXWd QX| j
| j j d � | j
| j j
d � d S)Nz421 QUIT FAILEDi� s QUIT FAILED)r� r r� r$ r% r re r r� r� r � exceptionZ smtp_codeZ
smtp_error)r �errorr& r r r � test_with_statement_QUIT_failureu 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 QX| j | j
� d S)
Nr- r] r
r� z 451 Requested action abortedTrw rx z test message)r r$ r% r re r r� r� r� r� �SMTPSenderRefusedry r0 r3 )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 QX| j | j � | j
| j j j d � d S)
Nr- r] r
r� z421 closing connectionrw rx z test messager )r r$ r% r re r r� r� r� r ry r0 r3 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 QX| j | j � | j
| j j j d � | j | j j d i d d 6� d S)Nr- r] r
r� z 250 acceptedz 421 closingrw rx ZFrankZGeorgez test messager � � closing)r r )r r$ r% r re r r� r� r� ZSMTPRecipientsRefusedry r0 r3 r r� ZassertDictEqualr
rF )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 QX| 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)Z
smtp_stateZDATAr� r� �found_terminator)r )r� r r r � s zNSMTPSimTests.test_421_from_data_cmd.<locals>.MySimSMTPChannel.found_terminator)r8 r9 r: r r r )r� r �MySimSMTPChannel� s r r- r] r
r� z John@foo.orgz
Sally@foo.orgz test messager )r� r r� r r$ r% r re r� Z
SMTPDataErrorry r0 r3 r r� r� )r r r& r r r �test_421_from_data_cmd� s !
z#SMTPSimTests.test_421_from_data_cmdN)r8 r9 r: r r r` r� rq r� r r r r r r r r r r r r r r r r� � s"
r� c C s t j t t t t t t � d S)N)r Z run_unittestr rC r� r� r� r� )�verboser r r � test_main� s r �__main__)/r= Zemail.mime.textr� Z email.utilsr rR r rM r� rJ r> r rb ZunittestZtestr r rP � ImportErrorr% �platformr Z SMTPChannelr ZTestCaser rB r{ r| Z
skipUnlessrC r� r� r� r� r� r� r� r r� r� r� r� r� Z reap_threadsr r8 r r r r �<module> sl
J � e
b �