Your IP : 216.73.216.213
�
g f�? � @ 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 � r d 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 n 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�Gd d � d e � Z% n e Gd d � d e$ � � Z& e Gd d � d e$ � � Z' Gd d � d e � Z( d d � Z) e* d k re) � n 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 @ s8 e Z d Z i g d 6d g d 6Z 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 e r4d( d) � Z d* d+ � Z d, d- � Z n d. S)/�DummyPOP3HandlerZUIDLzpython-testlib-pop-server�IMPLEMENTATIONc 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/python34/lib64/python3.4/test/test_poplib.pyr 0 s
zDummyPOP3Handler.__init__c C s | j j | � d S)N)r
�append)r �datar r r �collect_incoming_data8 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_errorK 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 N s zDummyPOP3Handler.pushc C s | j | � d S)N)r )r r& r r r �cmd_echoQ s zDummyPOP3Handler.cmd_echoc C s- | d k r | j d � n | j d � d S)N�guidoz-ERR no such userz+OK password required)r )r r& r r r �cmd_userU s zDummyPOP3Handler.cmd_userc C s- | d k r | j d � n | j d � d S)N�pythonz-ERR wrong passwordz+OK 10 messages)r )r r& r r r �cmd_passZ 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_listb 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_retrk s zDummyPOP3Handler.cmd_retrc C s | j d � d S)Nz +OK message marked for deletion.)r )r r& r r r �cmd_deleq s zDummyPOP3Handler.cmd_delec C s | j d � d S)Nz+OK done nothing.)r )r r& r r r �cmd_noopt s zDummyPOP3Handler.cmd_noopc C s | j d � d S)Nz+OK done nothing.)r )r r& r r r �cmd_rpopw s zDummyPOP3Handler.cmd_rpopc C s | j d � d S)Nz+OK done nothing.)r )r r& r r r �cmd_apopz 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 s0 t | j � } | j r, t r, g | d <n | S)N�STLS)�dict�CAPASr � SUPPORTS_SSL)r Z_capasr r r �
_get_capas� s
zDummyPOP3Handler._get_capasc C s� | j d � | j � rt xX | j � j � D]A \ } } | g } | rW | j | � n | j d j | � � q, Wn | 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 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�certfile� server_sideT�do_handshake_on_connectZsuppress_ragged_eofsz*-ERR Command not permitted when TLS active) r r �ssl� wrap_socket�socket�CERTFILE� del_channel�
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)rI Z do_handshakerG ZSSLError�argsZSSL_ERROR_WANT_READZSSL_ERROR_WANT_WRITEZ
SSL_ERROR_EOF� handle_close�OSError�errnoZ ECONNABORTEDr r )r �errr r r rM � 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 rM r r
� handle_readrG Z SSLEOFErrorrP )r r r r rT � s
z DummyPOP3Handler.handle_readN) �__name__�
__module__� __qualname__r= r r r( r) r r+ r- r/ r0 r2 Zcmd_uidlr5 Zcmd_topr6 r7 r8 r9 r: r? rC r> rN rM rT r r r r r , s2
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_socketrI � SOCK_STREAMZbind�listen�activeZLock� active_lockZ getsockname�host�port�handler_instance)r ZaddressZafr r r r � s
%zDummyPOP3Server.__init__c C s0 t j � | _ t j j | � | j j � d S)N)r �Event�_DummyPOP3Server__flagr[ �start�wait)r r r r rg � 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` rf �setr\ Z
socket_mapra �acquireZloop�releaseZ close_all)r r r r �run� s
zDummyPOP3Server.runc C s d | _ | j � d S)NF)r` r )r r r r �stop� s zDummyPOP3Server.stopc C s | j | � | _ d S)N)�handlerrd )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)rU rV rW r rp rI �AF_INETr rg rn ro rr rt rT ru r) r r r r rX � s rX c @ s6 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 e d) d* � � Z e d+ d, � � Z e d- d. � � Z d/ S)0�
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)Nri � )
rX �HOST�PORT�serverrg �poplib�POP3rb rc �client)r r r r �setUp� s
zTestPOP3Class.setUpc C s | j j � | j j � d S)N)r� rs r ro )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)r{ 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� )r{ 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� �listrx �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� )r{ r� Zdele)r r r r � test_dele s zTestPOP3Class.test_delec C s | j | j j � � d S)N)r{ r� �noop)r r r r � test_noop# s zTestPOP3Class.test_noopc C s | j | j j d � � d S)Nr� )r{ r� Zrpop)r r r r � test_rpop& s zTestPOP3Class.test_rpopc C s | j | j j d d � � d S)Nr� Z
dummypassword)r{ r� �apop)r r r r �test_apop_normal) s z TestPOP3Class.test_apop_normalc C sP d d d } t j | j d | � �$ | j t j | j j d d � Wd QXd S)Ns +OK� <i@B Zwelcomer� Zkb)� test_supportZ swap_attrr� r� r� r� r� )r Z evil_welcomer r r �test_apop_REDOS, 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_top6 s zTestPOP3Class.test_topc C s! | j j � | j j d � d S)Nr� )r� Zuidl)r r r r � test_uidl> s
zTestPOP3Class.test_uidlc C s, | j j � } | j d | j � k � d S)Nr )r� �caparx �keys)r r� r r r � test_capaB s zTestPOP3Class.test_capac C sF | j j � } | j | � | j | j j � | j | j j � d S)N)r� �quitrx � assertIsNone�sock�file)r rz r r r � test_quitF s
zTestPOP3Class.test_quitc C s, | j j � } | j d | j � k � d S)Nr; )r� r� rx r� )r r� r r r �test_stls_capaL s z TestPOP3Class.test_stls_capac C s) d } | j j � } | j | | � d S)Ns +OK Begin TLS negotiation)r� �stlsr� )r r� rz r r r � test_stlsQ s zTestPOP3Class.test_stlsc C s� d } t j t j � } | j t � t j | _ d | _ | j t j � � | j
j d | � } Wd QXt j
d | j j d d �| _
| j
j d | � } | j | | � d S)Ns +OK Begin TLS negotiationT�contextZ localhostri r| )rG �
SSLContext�PROTOCOL_TLSv1Zload_verify_locations�CAFILEZ
CERT_REQUIREDZ verify_modeZcheck_hostnamer� ZCertificateErrorr� r� r� r� r rc r� )r r� �ctxrz r r r �test_stls_contextW s
!z TestPOP3Class.test_stls_contextN)rU rV rW 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 rw � s.
rw c @ s e Z d Z d d � Z d S)�DummyPOP3_SSLHandlerc C s� t j j | | � t j | j d t d d d d �} | j � | j | � d | _ d | _
| j � | j d � g | _
| j d � d S)NrD rE TrF Fs
z(+OK dummy pop3 server ready. <timestamp>)r r
r rG rH rI rJ rK rL r r rM r r
r )r r Z
ssl_socketr r r r i s
z DummyPOP3_SSLHandler.__init__N)rU rV rW r r r r r r� g 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) rX r} r~ r r� rp rg r� r rb rc 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� rD s +OK)rG r� r� r� �
ValueErrorr� r r rb rc rJ r� r� ZassertIsInstancer� Z SSLSocketZassertIsr� rx r� ry )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) rU rV rW r� r� r� r� r� r� r r r r r� y 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)Nri r| ) rX r} r~ r rg r� r� rb rc r� r� )r r r r r� � s
'zTestPOP3_TLSClass.setUpc C sn | j j d k r] | j j d k r] y | j j � Wq] t j k
rY | j j � Yq] Xn | j j � d S)N) r� r� r� r� r� r� rs r ro )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)rU rV rW 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�< �targetrO T)r re �evtrI rv r^ r� Z
settimeoutr� Z bind_portrc r[ r �threadZ setDaemonrg rh )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 sr | j d � | j � zI y- | j � \ } } | j d � | j � Wn t j k
r^ Yn XWd | j � Xd S)NrY s
+ Hola mundo
)r_ rk Zaccept�sendrs rI ri )r r� Zservr rq 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� rI �getdefaulttimeout�setdefaulttimeoutr� r� r} rc r� r� �
gettimeoutrs )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� ri ) r� rI r� r� r� r� r} rc r� r� rs )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)Nri r� )r� r� r} rc r� r� r� rs )r r� r r r �testTimeoutValue� s z TestTimeouts.testTimeoutValueN) rU rV rW 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)rw r� r� r� r� Zthreading_setupZ run_unittestZthreading_cleanup)Ztests� thread_infor r r � test_main� s r� �__main__)+�__doc__r� r\ r rI �osZtimerR Zunittestr r Ztestr r� �
import_moduler r} r~ r> r" rG �pathr �dirname�__file__�curdirrJ r� r� r1 r4 r
r r] r[ rX rw r� r� r� r� r� rU r r r r �<module> sD *- � 1w ' 6