Your IP : 216.73.216.213
3
� fB � @ sx d Z ddlZddlZddlZddlZddlZddlZddlmZm Z ddl
m Z e j
d�Ze jZdZdZeed�r�ddlZdZejjejje�p�ejd �Zejjejje�p�ejd
�Ze ed �Zd Z d
Z G dd� dej �Z G dd� dej ej!�Z"G dd� de�Z#e�r$ddl$m%Z% G dd� de%e �Z&eG dd� de#��Z'eG dd� de#��Z(G dd � d e�Z)d d � Z*e+d k�rte*� dS ) 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 @ s� e Zd Zg dgd�ZdZdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZeZdd� ZeZdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zer�d*d+� Zd,d-� Z d.d/� Z d0S )1�DummyPOP3Handlerzpython-testlib-pop-server)ZUIDL�IMPLEMENTATIONFc C s8 t jj| |� | jd� g | _| jd� 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/python36/lib64/python3.6/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� dj | j�}t|d�}g | _|jd�d j� }|jd�}|d krR||d d � }nd}t| d| �r|t| d| �}||� n| jd| � 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 jj| |jd�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 |dkr| j d� | 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 |dkr| j d� | 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 s4 |r| j d||f � n| j d� tjj | 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 dtt� � tjj | 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 s" t | j�}| j r tr 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� rPx<| j� j� D ],\}}|g}|r<|j|� | j dj|�� 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 | jrdnd� 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 sv | j dkrh| jd� tj� }|jt� |j| jdddd�}| j� | j |� d| _ d| _
g | _ | j � n
| jd� d S )NFz+OK Begin TLS negotiationT)Z server_sideZdo_handshake_on_connectZsuppress_ragged_eofsz*-ERR Command not permitted when TLS active)
r r �ssl�
SSLContextZload_cert_chain�CERTFILEZ wrap_socket�socketZ del_channelZ
set_socketr r
�_do_tls_handshake)r r& �contextZtls_sockr r r �cmd_stls� s
zDummyPOP3Handler.cmd_stlsc C s� y| j j� W n� tjk
rf } z:|jd tjtjfkr<d S |jd tjkrT| j� S � W Y d d }~X nF t k
r� } z |jd t
j kr�| j� S W Y d d }~X nX d| _ d| _
d S )Nr TF)rI Z do_handshakerF 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 sD | j r| j� n0ytjj| � W n tjk
r> | j� Y nX d S )N)r rJ r r
� handle_readrF Z SSLEOFErrorrN )r r r r rR � 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> rL rJ rR r r r r r + s6
r c @ sZ e Zd ZeZejfdd�Zdd� Zdd� Z dd� Z
d d
� Z d d � Z e Z
d
d� Zdd� ZdS )�DummyPOP3Serverc C sn t jj| � tjj| � | j|tj� | j|� | j d� d| _
t j � | _ | jj
� 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 s$ t j� | _t jj| � | jj� d S )N)r �Event�_DummyPOP3Server__flagrY �start�wait)r r r r re � s
zDummyPOP3Server.startc C sZ d| _ | jj� z6x0| j rBtjrB| j� tjddd� W d Q R X qW W d tjdd� X d S )NTg�������?r )�timeout�count)Z
ignore_all)r^ rd �setrZ Z
socket_mapr_ ZloopZ 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)�handlerrb )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 dS )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)rS rT rU r rl rI �AF_INETr re rj rk rn rp rR rq r) r r r r rV � s
rV c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zed-d.� �Zed/d0� �Zed1d2� �Z d3S )4�
TestPOP3Classc C s | j |jd�� d S )Ns +OK)�
assertTrue�
startswith)r �respr r r �assertOK� s zTestPOP3Class.assertOKc C s6 t ttf�| _| jj� tj| jj| jjdd�| _ d S )N� )rg )
rV �HOST�PORT�serverre �poplib�POP3r` ra �client)r r r r �setUp� s
zTestPOP3Class.setUpc C s | j j� | jj� d | _d S )N)r~ ro r{ rk )r r r r �tearDown� s
zTestPOP3Class.tearDownc C s | j | jj� 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 tj| jjd� d S )Nz echo -err)� assertRaisesr| � error_protor~ � _shortcmd)r r r r �test_exceptions s z TestPOP3Class.test_exceptionsc C s* | j | jjd�� | jtj| jjd� d S )Nr, �invalid)rw r~ �userr� r| r� )r r r r � test_user s zTestPOP3Class.test_userc C s* | j | jjd�� | jtj| jjd� d S )Nr. r� )rw r~ Zpass_r� r| r� r� )r r r r �
test_pass_ s zTestPOP3Class.test_pass_c C s | j | jj� d� d S )N�
�d )r� r� )r� r~ �stat)r r r r � test_stat s zTestPOP3Class.test_statc C sD | j | jj� dd � dddddgdf� | j| jjd�jd �� d S )
Nr s 1 1s 2 2s 3 3s 4 4s 5 5� �1s OK 1 1)r� r~ �listrt �endswith)r r r r � test_list s zTestPOP3Class.test_listc C s6 ddddddddd gd
f}| j jd �}| 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 tj| jjdtjd 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 | jjd�� d S )Nr� )rw r~ Zdele)r r r r � test_dele* s zTestPOP3Class.test_delec C s | j | jj� � d S )N)rw r~ �noop)r r r r � test_noop- s zTestPOP3Class.test_noopc C s | j | jjd�� d S )Nr� )rw r~ Zrpop)r r r r � test_rpop0 s zTestPOP3Class.test_rpopc C s | j | jjdd�� d S )Nr� Z
dummypassword)rw r~ �apop)r r r r �test_apop_normal3 s z TestPOP3Class.test_apop_normalc C sB ddd }t j| jd|�� | jtj| jjdd� W d Q R X d S )Ns +OK� <i@B Zwelcomer� Zkb)� test_supportZ swap_attrr~ r� r| r� r� )r Z evil_welcomer r r �test_apop_REDOS6 s z TestPOP3Class.test_apop_REDOSc C s4 ddddddddd gd
f}| j | jjd 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 jd� d S )Nr� )r~ Zuidl)r r r r � test_uidlH s
zTestPOP3Class.test_uidlc C s d| j j_| jtj| jj� d S )NF)r{ rl rD r� r| r� r~ �utf8)r r r r � test_utf8_raises_if_unsupportedL s
z-TestPOP3Class.test_utf8_raises_if_unsupportedc C s( d| j j_d}| jj� }| j||� d S )NTs +OK I know RFC6856)r{ rl rD r~ r� r� )r r� �resultr r r � test_utf8P s
zTestPOP3Class.test_utf8c C s | j j� }| jd|j� k� d S )Nr )r~ �capart �keys)r r� r r r � test_capaV s
zTestPOP3Class.test_capac C s4 | j j� }| j|� | j| j j� | j| j j� d S )N)r~ �quitrt � assertIsNone�sock�file)r rv r r r � test_quitZ s
zTestPOP3Class.test_quitc C s | j j� }| jd|j� k� d S )Nr; )r~ r� rt 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� rv r r r � test_stlse s
zTestPOP3Class.test_stlsc C s� d}t jt j�}|jt� t j|_d|_| jt j �� | j
j |d�}W d Q R X t j
d| jjdd�| _
| j
j |d�}| j||� d S )Ns +OK Begin TLS negotiationT)rK Z localhostrx )rg )rF rG � PROTOCOL_TLSZload_verify_locations�CAFILEZ
CERT_REQUIREDZ verify_modeZcheck_hostnamer� ZCertificateErrorr~ r� r| r} r{ ra r� )r r� �ctxrv r r r �test_stls_contextk s
z TestPOP3Class.test_stls_contextN) rS rT rU rw 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 rs � s2
rs )�
SSLConnectionc @ s e Zd Zdd� ZdS )�DummyPOP3_SSLHandlerc C s@ t jj| |� | j� | jd� g | _| jd� 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 ~ s
z DummyPOP3_SSLHandler.__init__N)rS rT rU r r r r r r� | s r� c @ s8 e Zd Zdd� Zdd� Zdd� Zdd� ZeZd d
� Zd S ) �TestPOP3_SSLClassc C s: t ttf�| _t| j_| jj� tj| jj | jj
�| _ d S )N) rV ry rz r{ r� rl re r| r r` ra r~ )r r r r r � s
zTestPOP3_SSLClass.setUpc C s | j dtj� d S )Nr )ZassertInr| �__all__)r r r r � test__all__� s z TestPOP3_SSLClass.test__all__c C s� t jt j�}| jttj| jj| jj t
|d� | jttj| jj| jj t
|d� | jttj| jj| jj t
t
|d� | j j � tj| jj| jj |d�| _ | j
| j jt j� | j| j jj|� | j| j j� jd�� d S )N)�keyfilerK )�certfilerK )r� r� rK )rK s +OK)rF rG r� r� �
ValueErrorr| r r{ r` ra rH r~ r� ZassertIsInstancer� Z SSLSocketZassertIsrK rt r� ru )r r� r r r � test_context� s
z TestPOP3_SSLClass.test_contextc C s | j tj| jj� d S )N)r� r| r� r~ r� )r r r r r� � s zTestPOP3_SSLClass.test_stlsc C s | j j� }| jd|j� k� d S )Nr; )r~ r� � assertFalser� )r r� r r r r� � s
z TestPOP3_SSLClass.test_stls_capaN) rS rT rU r r� r� r� r� r� r r r r r� � s r� c @ s0 e Zd Zdd� Zdd� Zdd� ZeZdd� Zd S )
�TestPOP3_TLSClassc C s@ t ttf�| _| jj� tj| jj| jjdd�| _ | j j
� d S )Nrx )rg ) rV ry rz r{ re r| r} r` ra r~ r� )r r r r r � s
zTestPOP3_TLSClass.setUpc C s\ | j jd k rH| j jd k rHy| j j� W n tjk
rF | j j� Y nX | jj� d | _d S )N) r~ r� r� r� r| r� ro r{ rk )r r r r r� � s
zTestPOP3_TLSClass.tearDownc C s | j tj| jj� d S )N)r� r| r� r~ r� )r r r r r� � s zTestPOP3_TLSClass.test_stlsc C s | j j� }| jd|j� k� d S )Ns STLS)r~ r� r� r� )r r� r r r r� � s
z TestPOP3_TLSClass.test_stls_capaN)rS rT rU r r� r� r� r� r r r r r� � s
r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )� TestTimeoutsc C st t j� | _tjtjtj�| _| jjd� tj | j�| _
t j | j | j| jfd�| _
| j
jd� | j
j� | jj� d S )N�< )�targetrM T)r rc �evtrI rr r\ r� Z
settimeoutr� Z bind_portra rY r{ �threadZ setDaemonre rf )r r r r r � s
zTestTimeouts.setUpc C s | j j� d | _ d S )N)r� r )r r r r r� � s
zTestTimeouts.tearDownc C s^ |j � |j� z>y"|j� \}}|jd� |j� W n tjk
rJ Y nX W d |j� X d S )Ns
+ Hola mundo
)r] ri Zaccept�sendro rI rg )r r� Zservr rm r r r r{ � s
zTestTimeouts.serverc C sV | j tj� � tjd� ztjt| j�}W d tjd � X | j|j j
� d� |j � d S )N� ) r� rI �getdefaulttimeout�setdefaulttimeoutr| r} ry ra r� r� �
gettimeoutro )r �popr r r �testTimeoutDefault� s
z TestTimeouts.testTimeoutDefaultc C sX | j tj� � tjd� ztjt| jd d�}W d tjd � X | j |jj � � |j
� d S )Nr� )rg ) r� rI r� r� r| r} ry ra r� r� ro )r r� r r r �testTimeoutNone� s
z TestTimeouts.testTimeoutNonec C s0 t jt| jdd�}| j|jj� d� |j� d S )Nr� )rg )r| r} ry ra r� r� r� ro )r r� r r r �testTimeoutValue s z TestTimeouts.testTimeoutValueN) rS rT rU r r� r{ r� r� r� r r r r r� � s
r� c C s4 t tttg} tj� }ztj| � W d tj|� X d S )N)rs r� r� r� r� Zthreading_setupZ run_unittestZthreading_cleanup)Ztests� thread_infor r r � test_main s r� �__main__),�__doc__r| rZ r rI �osrP Zunittestr r Ztestr r� �
import_moduler ry rz r> r" rF �pathr �dirname�__file__�curdirrH r� r� r1 r4 r
r r[ rY rV rs Ztest.test_ftplibr� r� r� r� r� r� rS r r r r �<module> sH
2 ' 7