Your IP : 216.73.217.129
��Yf�@ � @ s" d 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
d d l m Z
e
j d � Z e
j Z d Z d Z e e d � rd d l Z d Z e j j e j j e � p� e j d � Z e j j e j j e � pe j d
� Z e
e d � Z d Z d
Z Gd d � d e j � Z Gd d � d e j! e j" � Z# Gd d � d e � Z$ e r�d d l% m& Z& Gd d � d e& e � Z' e Gd d � d e$ � � Z( e Gd d � d e$ � � Z) Gd d � d e � Z* d d � Z+ e, d k r e+ � d S) z Test script for poplib module.� N)�TestCase�
skipUnless)�support� threadingF�POP3_SSLTz keycert3.pemz pycacert.pemzSSL not supporteds 1 1
2 2
3 3
4 4
5 5
.
st From: postmaster@python.org
Content-Type: text/plain
MIME-Version: 1.0
Subject: Dummy
line1
line2
line3
.
c @ sE e Z d Z d g d d g i 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 e Z d d � Z e Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z e rAd+ d, � Z d- d. � Z d/ d0 � Z d1 S)2�DummyPOP3HandlerZUIDL�IMPLEMENTATIONzpython-testlib-pop-serverFc C sL t j j | | � | j d � g | _ | j d � d | _ d | _ d S)Ns
z(+OK dummy pop3 server ready. <timestamp>F)�asynchat�
async_chat�__init__�set_terminator� in_buffer�push�
tls_active� tls_starting)�self�conn� r �5/opt/alt/python35/lib64/python3.5/test/test_poplib.pyr 1 s
zDummyPOP3Handler.__init__c C s | j j | � d S)N)r
�append)r �datar r r �collect_incoming_data9 s z&DummyPOP3Handler.collect_incoming_datac C s� d j | j � } t | d � } g | _ | j d � d j � } | j d � } | d k ru | | d d � } n d } t | d | � r� t | d | � } | | � n | j d | � d S)
N� z
ISO-8859-1� r � � Zcmd_z$-ERR unrecognized POP3 command "%s".���) �joinr
�str�split�lower�find�hasattr�getattrr )r �line�cmdZspace�arg�methodr r r �found_terminator< s
z!DummyPOP3Handler.found_terminatorc C s � d S)Nr )r r r r � handle_errorL s z DummyPOP3Handler.handle_errorc C s$ t j j | | j d � d � d S)Nz
ISO-8859-1s
)r r
r �encode)r r r r r r O s zDummyPOP3Handler.pushc C s | j | � d S)N)r )r r& r r r �cmd_echoR s zDummyPOP3Handler.cmd_echoc C s* | d k r | j d � | j d � d S)N�guidoz-ERR no such userz+OK password required)r )r r& r r r �cmd_userV s
zDummyPOP3Handler.cmd_userc C s* | d k r | j d � | j d � d S)N�pythonz-ERR wrong passwordz+OK 10 messages)r )r r& r r r �cmd_pass[ s
zDummyPOP3Handler.cmd_passc C s | j d � d S)Nz
+OK 10 100)r )r r& r r r �cmd_stat` s zDummyPOP3Handler.cmd_statc C sD | r | j d | | f � n | j d � t j j | t � d S)Nz +OK %s %sz+OK)r r r
� LIST_RESP)r r& r r r �cmd_listc s
zDummyPOP3Handler.cmd_listc C s. | j d t t � � t j j | t � d S)Nz +OK %s bytes)r �len� RETR_RESPr r
)r r& r r r �cmd_retrl s zDummyPOP3Handler.cmd_retrc C s | j d � d S)Nz +OK message marked for deletion.)r )r r& r r r �cmd_deler s zDummyPOP3Handler.cmd_delec C s | j d � d S)Nz+OK done nothing.)r )r r& r r r �cmd_noopu s zDummyPOP3Handler.cmd_noopc C s | j d � d S)Nz+OK done nothing.)r )r r& r r r �cmd_rpopx s zDummyPOP3Handler.cmd_rpopc C s | j d � d S)Nz+OK done nothing.)r )r r& r r r �cmd_apop{ s zDummyPOP3Handler.cmd_apopc C s | j d � | j � d S)Nz +OK closing.)r Zclose_when_done)r r& r r r �cmd_quit~ s
zDummyPOP3Handler.cmd_quitc C s- t | j � } | j r) t r) g | d <| S)N�STLS)�dict�CAPASr � SUPPORTS_SSL)r Z_capasr r r �
_get_capas� s
zDummyPOP3Handler._get_capasc C s | j d � | j � rn xR | j � j � D]> \ } } | g } | rT | j | � | j d j | � � q, W| j d � d S)Nz+OK Capability list followsr �.)r r? �items�extendr )r r& ZcapZparamsZ_lnr r r �cmd_capa� s
zDummyPOP3Handler.cmd_capac C s | j | j r d n d � d S)Nz+OK I know RFC6856z-ERR What is UTF8?!)r � enable_UTF8)r r& r r r �cmd_utf8� s zDummyPOP3Handler.cmd_utf8c
C s� | j d k r� | j d � t j | j d t d d d d d d �} | j � | j | � d | _ d | _ g | _ | j
� n
| j d � d S) NFz+OK Begin TLS negotiation�certfileZ server_sideTZdo_handshake_on_connectZsuppress_ragged_eofsz*-ERR Command not permitted when TLS active) r r �sslZ wrap_socket�socket�CERTFILEZ del_channelZ
set_socketr r
�_do_tls_handshake)r r& Ztls_sockr r r �cmd_stls� s
zDummyPOP3Handler.cmd_stlsc C s� y | j j � Wn� t j k
r� } zK | j d t j t j f k rL d S| j d t j k rl | j � S� WYd d } ~ XnW t k
r� } z% | j d t
j k r� | j � SWYd d } ~ Xn Xd | _ d | _
d S)Nr TF)rH Z do_handshakerG ZSSLError�argsZSSL_ERROR_WANT_READZSSL_ERROR_WANT_WRITEZ
SSL_ERROR_EOF� handle_close�OSError�errnoZ ECONNABORTEDr r )r �errr r r rJ � s
z"DummyPOP3Handler._do_tls_handshakec C sP | j r | j � n6 y t j j | � Wn t j k
rK | j � Yn Xd S)N)r rJ r r
� handle_readrG Z SSLEOFErrorrM )r r r r rQ � s
z DummyPOP3Handler.handle_readN) �__name__�
__module__� __qualname__r= rD r r r( r) r r+ r- r/ r0 r2 Zcmd_uidlr5 Zcmd_topr6 r7 r8 r9 r: r? rC rE r> rK rJ rQ r r r r r , s6
r c @ s� e Z d Z e Z e j d d � Z d d � Z d d � Z d d � Z
d d
� Z d d � Z e Z
d
d � Z d d � Z d S)�DummyPOP3Serverc C s� t j j | � t j j | � | j | t j � | j | � | j d � d | _
t j � | _ | j j
� d d � \ | _ | _ d | _ d S)N� F� )r �Threadr �asyncore�
dispatcherZ
create_socketrH � SOCK_STREAMZbind�listen�activeZLock� active_lockZ getsockname�host�port�handler_instance)r ZaddressZafr r r r � s
%zDummyPOP3Server.__init__c C s@ | j s t � t j � | _ t j j | � | j j � d S)N)r] �AssertionErrorr �Event�_DummyPOP3Server__flagrX �start�wait)r r r r re � s zDummyPOP3Server.startc C ss d | _ | j j � xF | j r^ t j r^ | j j � t j d d d d � | j j � q Wt j d d � d S)NT�timeoutg�������?�countr Z
ignore_all)
r] rd �setrY Z
socket_mapr^ �acquireZloop�releaseZ close_all)r r r r �run� s
zDummyPOP3Server.runc C s& | j s t � d | _ | j � d S)NF)r] rb r )r r r r �stop� s zDummyPOP3Server.stopc C s | j | � | _ d S)N)�handlerra )r r �addrr r r �handle_accepted� s z DummyPOP3Server.handle_acceptedc C s | j � d S)N)�close)r r r r �handle_connect� s z DummyPOP3Server.handle_connectc C s d S)Nr r )r r r r �writable� s zDummyPOP3Server.writablec C s � d S)Nr )r r r r r) � s z DummyPOP3Server.handle_errorN)rR rS rT r rn rH �AF_INETr re rl rm rp rr rQ rs r) r r r r rU � s rU c @ sN 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) d* � Z d+ d, � Z e d- d. � � Z e d/ d0 � � Z e d1 d2 � � Z d3 S)4�
TestPOP3Classc C s | j | j d � � d S)Ns +OK)�
assertTrue�
startswith)r �respr r r �assertOK� s zTestPOP3Class.assertOKc C sM t t t f � | _ | j j � t j | j j | j j d d �| _ d S)Nrg � )
rU �HOST�PORT�serverre �poplib�POP3r_ r` �client)r r r r �setUp� s
zTestPOP3Class.setUpc C s | j j � | j j � d S)N)r� rq r} rm )r r r r �tearDown� s
zTestPOP3Class.tearDownc C s | j | j j � d � d S)Ns( +OK dummy pop3 server ready. <timestamp>)� assertEqualr� Z
getwelcome)r r r r �test_getwelcome s z TestPOP3Class.test_getwelcomec C s | j t j | j j d � d S)Nz echo -err)� assertRaisesr~ � error_protor� � _shortcmd)r r r r �test_exceptions s z TestPOP3Class.test_exceptionsc C s9 | j | j j d � � | j t j | j j d � d S)Nr, �invalid)ry r� �userr� r~ r� )r r r r � test_user s zTestPOP3Class.test_userc C s9 | j | j j d � � | j t j | j j d � d S)Nr. r� )ry r� Zpass_r� r~ r� r� )r r r r �
test_pass_ s zTestPOP3Class.test_pass_c C s | j | j j � d � d S)N�
�d )r� r� )r� r� �stat)r r r r � test_stat s zTestPOP3Class.test_statc C s^ | j | j j � d d � d d d d d g d f � | j | j j d � j d � � d S)
Nr s 1 1s 2 2s 3 3s 4 4s 5 5� �1s OK 1 1)r� r� �listrv �endswith)r r r r � test_list s
zTestPOP3Class.test_listc C sM d d d d d d d d d g d
f } | j j d � } | j | | � d S) Ns
+OK 116 bytess From: postmaster@python.orgs Content-Type: text/plains MIME-Version: 1.0s Subject: Dummyr s line1s line2s line3�q �foo)r� Zretrr� )r �expectedr� r r r � test_retr s zTestPOP3Class.test_retrc C s/ | j t j | j j d t j d d � d S)Nzecho +%sr� �a)r� r~ r� r� r� Z_MAXLINE)r r r r �test_too_long_lines" s z!TestPOP3Class.test_too_long_linesc C s | j | j j d � � d S)Nr� )ry r� Zdele)r r r r � test_dele& s zTestPOP3Class.test_delec C s | j | j j � � d S)N)ry r� �noop)r r r r � test_noop) s zTestPOP3Class.test_noopc C s | j | j j d � � d S)Nr� )ry r� Zrpop)r r r r � test_rpop, s zTestPOP3Class.test_rpopc C s | j | j j d d � � d S)Nr� Z
dummypassword)ry r� �apop)r r r r �test_apop_normal/ s z TestPOP3Class.test_apop_normalc C sQ d d d } t j | j d | � �$ | j t j | j j d d � Wd QRXd S)Ns +OK� <i@B Zwelcomer� Zkb)� test_supportZ swap_attrr� r� r~ r� r� )r Z evil_welcomer r r �test_apop_REDOS2 s z TestPOP3Class.test_apop_REDOSc C sJ d d d d d d d d d g d
f } | j | j j d d � | � d S) Ns
+OK 116 bytess From: postmaster@python.orgs Content-Type: text/plains MIME-Version: 1.0s Subject: Dummyr s line1s line2s line3r� r )r� r� �top)r r� r r r �test_top< s zTestPOP3Class.test_topc C s! | j j � | j j d � d S)Nr� )r� Zuidl)r r r r � test_uidlD s
zTestPOP3Class.test_uidlc C s, d | j j _ | j t j | j j � d S)NF)r} rn rD r� r~ r� r� �utf8)r r r r � test_utf8_raises_if_unsupportedH s z-TestPOP3Class.test_utf8_raises_if_unsupportedc C s8 d | j j _ d } | j j � } | j | | � d S)NTs +OK I know RFC6856)r} rn rD r� r� r� )r r� �resultr r r � test_utf8L s zTestPOP3Class.test_utf8c C s, | j j � } | j d | j � k � d S)Nr )r� �caparv �keys)r r� r r r � test_capaR s zTestPOP3Class.test_capac C sF | j j � } | j | � | j | j j � | j | j j � d S)N)r� �quitrv � assertIsNone�sock�file)r rx r r r � test_quitV s
zTestPOP3Class.test_quitc C s, | j j � } | j d | j � k � d S)Nr; )r� r� rv r� )r r� r r r �test_stls_capa\ s z TestPOP3Class.test_stls_capac C s) d } | j j � } | j | | � d S)Ns +OK Begin TLS negotiation)r� �stlsr� )r r� rx r r r � test_stlsa s zTestPOP3Class.test_stlsc C s� d } t j t j � } | j t � t j | _ d | _ | j t j � � | j
j d | � } Wd QRXt j
d | j j d d �| _
| j
j d | � } | j | | � d S)Ns +OK Begin TLS negotiationT�contextZ localhostrg rz )rG �
SSLContext�PROTOCOL_TLSv1Zload_verify_locations�CAFILEZ
CERT_REQUIREDZ verify_modeZcheck_hostnamer� ZCertificateErrorr� r� r~ r r} r` r� )r r� �ctxrx r r r �test_stls_contextg s
!z TestPOP3Class.test_stls_contextN) rR rS rT ry r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� � requires_sslr� r� r� r r r r ru � s2
ru )�
SSLConnectionc @ s e Z d Z d d � Z d S)�DummyPOP3_SSLHandlerc C sV t j j | | � | j � | j d � g | _ | j d � d | _ d | _ d S)Ns
z(+OK dummy pop3 server ready. <timestamp>TF) r r
r Zsecure_connectionr r
r r r )r r r r r r z s
z DummyPOP3_SSLHandler.__init__N)rR rS rT r r r r r r� x s r� c @ sR e Z d Z d d � Z d d � Z d d � Z d d � Z e Z d d
� Z d S) �TestPOP3_SSLClassc C sS t t t f � | _ t | j _ | j j � t j | j j | j j
� | _ d S)N) rU r{ r| r} r� rn re r~ r r_ r` r� )r r r r r� � s
zTestPOP3_SSLClass.setUpc C s | j d t j � d S)Nr )ZassertInr~ �__all__)r r r r � test__all__� s z TestPOP3_SSLClass.test__all__c C s4 t j t j � } | j t t j | j j | j j d t
d | �| j t t j | j j | j j d t
d | �| j t t j | j j | j j d t
d t
d | �| j j � t j | j j | j j d | �| _ | j
| j j t j � | j | j j j | � | j | j j � j d � � d S)NZkeyfiler� rF s +OK)rG r� r� r� �
ValueErrorr~ r r} r_ r` rI r� r� ZassertIsInstancer� Z SSLSocketZassertIsr� rv r� rw )r r� r r r � test_context� s
z TestPOP3_SSLClass.test_contextc C s | j t j | j j � d S)N)r� r~ r� r� r� )r r r r r� � s zTestPOP3_SSLClass.test_stlsc C s, | j j � } | j d | j � k � d S)Nr; )r� r� � assertFalser� )r r� r r r r� � s z TestPOP3_SSLClass.test_stls_capaN) rR rS rT r� r� r� r� r� r� r r r r r� � s r� c @ sF e Z d Z d d � Z d d � Z d d � Z e Z d d � Z d S)
�TestPOP3_TLSClassc C sZ t t t f � | _ | j j � t j | j j | j j d d �| _ | j j
� d S)Nrg rz ) rU r{ r| r} re r~ r r_ r` r� r� )r r r r r� � s
'zTestPOP3_TLSClass.setUpc C sk | j j d k rZ | j j d k rZ y | j j � Wn" t j k
rY | j j � Yn X| j j � d S)N) r� r� r� r� r~ r� rq r} rm )r r r r r� � s $zTestPOP3_TLSClass.tearDownc C s | j t j | j j � d S)N)r� r~ r� r� r� )r r r r r� � s zTestPOP3_TLSClass.test_stlsc C s, | j j � } | j d | j � k � d S)Ns STLS)r� r� r� r� )r r� r r r r� � s z TestPOP3_TLSClass.test_stls_capaN)rR rS rT r� r� r� r� r� r r r r r� � s
r� c @ sX e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
S)� TestTimeoutsc C s� t j � | _ t j t j t j � | _ | j j d � t j | j � | _
t j d | j d | j | j f � | _
| j
j d � | j
j � | j j � d S)N�< �targetrL T)r rc �evtrH rt r[ r� Z
settimeoutr� Z bind_portr` rX r} �threadZ setDaemonre rf )r r r r r� � s *
zTestTimeouts.setUpc C s | j j � | ` d S)N)r� r )r r r r r� � s
zTestTimeouts.tearDownc C so | j � | j � zI y- | j � \ } } | j d � | j � Wn t j k
r[ Yn XWd | j � Xd S)Ns
+ Hola mundo
)r\ ri Zaccept�sendrq rH rg )r r� Zservr ro r r r r} � s
zTestTimeouts.serverc C st | j t j � � t j d � z t j t | j � } Wd t j d � X| j | j j
� d � | j j � d S)N� ) r� rH �getdefaulttimeout�setdefaulttimeoutr~ r r{ r` r� r� �
gettimeoutrq )r �popr r r �testTimeoutDefault� s
z TestTimeouts.testTimeoutDefaultc C sw | j t j � � t j d � z t j t | j d d �} Wd t j d � X| j | j j � � | j j
� d S)Nr� rg ) r� rH r� r� r~ r r{ r` r� r� rq )r r� r r r �testTimeoutNone� s
z TestTimeouts.testTimeoutNonec C sE t j t | j d d �} | j | j j � d � | j j � d S)Nrg r� )r~ r r{ r` r� r� r� rq )r r� r r r �testTimeoutValue� s z TestTimeouts.testTimeoutValueN) rR rS rT r� r� r} r� r� r� r r r r r� � s
r� c C sD t t t t g } t j � } z t j | � Wd t j | � Xd S)N)ru r� r� r� r� Zthreading_setupZ run_unittestZthreading_cleanup)Ztests� thread_infor r r � test_main s r� �__main__)-�__doc__r~ rY r rH �osZtimerO Zunittestr r Ztestr r� �
import_moduler r{ r| r> r" rG �pathr �dirname�__file__�curdirrI r� r� r1 r4 r
r rZ rX rU ru Ztest.test_ftplibr� r� r� r� r� r� rR r r r r �<module> sF ** � 1� ' 6