File: //usr/local/lib/python3.7/test/__pycache__/test_poplib.cpython-37.opt-1.pyc
B
��g�C � @ sv d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z m
Z
ddl m Z
e
jZdZdZeed�r�ddlZdZej�ej�e�p�ejd�Zej�ej�e�p�ejd �Ze
ed
�Zd Zd Z G d
d� 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� de �Z(d d � Z)e*d k�rre)� dS ) z Test script for poplib module.� N)�TestCase�
skipUnless)�supportF�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 j�| |� | �d� g | _| �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 �,/usr/local/lib/python3.7/test/test_poplib.pyr
0 s
zDummyPOP3Handler.__init__c C s | j �|� d S )N)r �append)r �datar r r �collect_incoming_data8 s z&DummyPOP3Handler.collect_incoming_datac C s� d� | j�}t|d�}g | _|�d�d �� }|�d�}|dkrR||d d � }nd}t| d| �r|t| d| �}||� n| �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�| |�d�d � d S )Nz
ISO-8859-1s
)r r r
�encode)r r r r r r
N s zDummyPOP3Handler.pushc C s | � |� d S )N)r
)r r% r r r �cmd_echoQ s zDummyPOP3Handler.cmd_echoc C s |dkr| � d� | � 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| � d� | � 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 | � d� d S )Nz
+OK 10 100)r
)r r% r r r �cmd_stat_ s zDummyPOP3Handler.cmd_statc C s4 |r| � d||f � n| � d� tj� | 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$ | � dtt� � tj� | 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 | � d� d S )Nz +OK message marked for deletion.)r
)r r% r r r �cmd_deleq s zDummyPOP3Handler.cmd_delec C s | � d� d S )Nz+OK done nothing.)r
)r r% r r r �cmd_noopt s zDummyPOP3Handler.cmd_noopc C s | � d� d S )Nz+OK done nothing.)r
)r r% r r r �cmd_rpopw s zDummyPOP3Handler.cmd_rpopc C s | � d� d S )Nz+OK done nothing.)r
)r r% r r r �cmd_apopz s zDummyPOP3Handler.cmd_apopc C s | � d� | �� d S )Nz +OK closing.)r
Zclose_when_done)r r% r r r �cmd_quit} s
zDummyPOP3Handler.cmd_quitc C s t | j�}| js tr g |d< |S )N�STLS)�dict�CAPASr � SUPPORTS_SSL)r Z_capasr r r �
_get_capas� s
zDummyPOP3Handler._get_capasc C s^ | � d� | �� rPx<| �� �� D ],\}}|g}|r<|�|� | � d�|�� q W | � d� d S )Nz+OK Capability list followsr �.)r
r> �items�extendr )r r% Zcap�paramsZ_lnr r r �cmd_capa� s
zDummyPOP3Handler.cmd_capac C s | � | 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| �d� t�� }|�t� |j| jdddd�}| �� | � |� d| _ d| _
g | _ | � � n
| �d� 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 �� W n� tjk
r� } z\|jd tjtjfkr<d S |jd tjkrT| �� S d|jd kspd|jd krx| �� S � W d d }~X Y nF t k
r� } z |jd t
j kr�| �� S W d d }~X Y nX d| _ d| _
d S )Nr ZSSLV3_ALERT_BAD_CERTIFICATEr Z SSLV3_ALERT_CERTIFICATE_UNKNOWNTF)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| �� n0ytj�| � W n tjk
r> | �� 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. r/ r1 Zcmd_uidlr4 Zcmd_topr5 r6 r7 r8 r9 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 st t j�| � tj�| � d| _| �|tj� | � |� | �
d� d| _ t � � | _
| j�� d d� \| _| _d | _d S )NT� F� )� threading�Threadr
�asyncore�
dispatcher�daemonZ
create_socketrI � SOCK_STREAMZbind�listen�activeZLock� active_lockZ getsockname�host�port�handler_instance)r Zaddress�afr r r r
� s
zDummyPOP3Server.__init__c C s$ t �� | _t j�| � | j�� d S )N)rY �Event�_DummyPOP3Server__flagrZ �start�wait)r r r r rh � s
zDummyPOP3Server.startc C sZ d| _ | j�� 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` rg �setr[ Z
socket_mapra ZloopZ close_all)r r r r �run� s
zDummyPOP3Server.runc C s d| _ | �� d S )NF)r` r )r r r r �stop� s zDummyPOP3Server.stopc C s | � |�| _d S )N)�handlerrd )r r �addrr r r �handle_accepted� s z DummyPOP3Server.handle_acceptedc C s | � � 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 ro rI �AF_INETr
rh rm rn rq rs rR rt 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 | � |�d�� d S )Ns +OK)�
assertTrue�
startswith)r �respr r r �assertOK� s zTestPOP3Class.assertOKc C s6 t ttf�| _| j�� tj| jj| jjdd�| _ d S )N� )rj )
rV �HOST�PORT�serverrh �poplib�POP3rb rc �client)r r r r �setUp� s
zTestPOP3Class.setUpc C s | j �� | j�� d | _d S )N)r� rr r~ rn )r r r r �tearDown s
zTestPOP3Class.tearDownc C s | � | 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 | � 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�d�� | �tj| jjd� d S )Nr+ �invalid)rz r� �userr� r r� )r r r r � test_user s zTestPOP3Class.test_userc C s* | � | j�d�� | �tj| jjd� d S )Nr- r� )rz r� Zpass_r� r r� r� )r r r r �
test_pass_ s zTestPOP3Class.test_pass_c C s | � | j�� d� d S )N)�
�d )r� r� �stat)r r r r � test_stat s zTestPOP3Class.test_statc C sD | � | j�� dd � dddddgdf� | �| j�d��d �� d S )
Nr s 1 1s 2 2s 3 3s 4 4s 5 5� �1s OK 1 1)r� r� �listrw �endswith)r r r r � test_list s zTestPOP3Class.test_listc C s6 ddddddddd gd
f}| j �d �}| �||� 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& | � 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�d�� d S )Nr� )rz r� Zdele)r r r r � test_dele/ s zTestPOP3Class.test_delec C s | � | j�� � d S )N)rz r� �noop)r r r r � test_noop2 s zTestPOP3Class.test_noopc C s | � | j�d�� d S )Nr� )rz r� Zrpop)r r r r � test_rpop5 s zTestPOP3Class.test_rpopc C s | � | j�dd�� d S )Nr� Z
dummypassword)rz r� �apop)r r r r �test_apop_normal8 s z TestPOP3Class.test_apop_normalc C sB ddd }t �| jd|�� | �tj| 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_REDOS; s z TestPOP3Class.test_apop_REDOSc C s4 ddddddddd gd
f}| � | 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_topE s zTestPOP3Class.test_topc C s | j �� | j �d� d S )Nr� )r� Zuidl)r r r r � test_uidlM s
zTestPOP3Class.test_uidlc C s d| j j_| �tj| jj� d S )NF)r~ ro rD r� r r� r� �utf8)r r r r � test_utf8_raises_if_unsupportedQ s
z-TestPOP3Class.test_utf8_raises_if_unsupportedc C s( d| j j_d}| j�� }| �||� d S )NTs +OK I know RFC6856)r~ ro rD r� r� r� )r r� �resultr r r � test_utf8U s
zTestPOP3Class.test_utf8c C s | j �� }| �d|�� k� d S )Nr )r� �caparw �keys)r r� r r r � test_capa[ s
zTestPOP3Class.test_capac C s4 | j �� }| �|� | �| j j� | �| j j� d S )N)r� �quitrw � assertIsNone�sock�file)r ry r r r � test_quit_ s
zTestPOP3Class.test_quitc C s | j �� }| �d|�� k� d S )Nr: )r� r� rw r� )r r� r r r �test_stls_capae s
z TestPOP3Class.test_stls_capac C s d}| j �� }| �||� d S )Ns +OK Begin TLS negotiation)r� �stlsr� )r r� ry r r r � test_stlsj s
zTestPOP3Class.test_stlsc C s� d}t �t j�}|�t� | �|jt j� | �|jd� | � t j
�� | j j |d�}W d Q R X t
jd| jjdd�| _ | j j |d�}| �||� d S )Ns +OK Begin TLS negotiationT)rK Z localhostr{ )rj )rF rG �PROTOCOL_TLS_CLIENTZload_verify_locations�CAFILEr� � verify_modeZ
CERT_REQUIRED�check_hostnamer� ZCertificateErrorr� r� r r� r~ rc )r r� �ctxry r r r �test_stls_contextp s
z TestPOP3Class.test_stls_contextN) rS rT rU rz 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 rv � s2
rv )�
SSLConnectionc @ s e Zd Zdd� ZdS )�DummyPOP3_SSLHandlerc C s@ t j�| |� | �� | �d� g | _| �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
� 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_| j�� t�| jj | jj
�| _ d S )N) rV r| r} r~ r� ro rh r r rb rc r� )r r r r r� � s
zTestPOP3_SSLClass.setUpc C s | � dtj� d S )Nr )ZassertInr �__all__)r r r r � test__all__� s z TestPOP3_SSLClass.test__all__c C s� t �t j�}d|_t j|_| jttj | j
j | j
j t
|d� | jttj | j
j | j
j t
|d� | jttj | j
j | j
j t
t
|d� | j�� tj | j
j | j
j |d�| _| �| jjt j� | �| jjj|� | �| j�� �d�� d S )NF)�keyfilerK )�certfilerK )r� r� rK )rK s +OK)rF rG r� r� Z CERT_NONEr� r� �
ValueErrorr r r~ rb rc rH r� r� ZassertIsInstancer� Z SSLSocketZassertIsrK rw r� rx )r r� r r r � test_context� s
z TestPOP3_SSLClass.test_contextc C s | � tj| jj� d S )N)r� r r� r� r� )r r r r r� � s zTestPOP3_SSLClass.test_stlsc C s | j �� }| �d|�� 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�| _| j�� tj| jj| jjdd�| _ | j �
� d S )Nr{ )rj ) rV r| r} r~ rh r r� rb rc r� r� )r r r r r� � s
zTestPOP3_TLSClass.setUpc C s\ | j jd k rH| j jd k rHy| j �� W n tjk
rF | j �� Y nX | j�� d | _d S )N) r� r� r� r� r r� rr r~ rn )r r r r r� � s
zTestPOP3_TLSClass.tearDownc C s | � tj| jj� d S )N)r� r r� r� r� )r r r r r� � s zTestPOP3_TLSClass.test_stlsc C s | j �� }| �d|�� 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 sp t �� | _t�tjtj�| _| j�d� t� | j�| _
t j | j | j| jfd�| _
d| j
_| j
�� | j�� d S )N�< )�targetrM T)rY rf �evtrI ru r^ r� Z
settimeoutr� Z bind_portrc rZ r~ �threadr] rh ri )r r r r r� � s
zTestTimeouts.setUpc C s | j �� d | _ d S )N)r� r )r r r r r� � s
zTestTimeouts.tearDownc C s^ |� � |�� z>y"|�� \}}|�d� |�� W n tjk
rJ Y nX W d |�� X d S )Ns
+ Hola mundo
)r_ rl Zaccept�sendrr rI rj )r r� Zservr rp r r r r~ � s
zTestTimeouts.serverc C sV | � t�� � t�d� zt�t| j�}W d t�d � X | �|j �
� d� |� � d S )N� ) r� rI �getdefaulttimeout�setdefaulttimeoutr r� r| rc r� r� �
gettimeoutrr )r �popr r r �testTimeoutDefault� s
z TestTimeouts.testTimeoutDefaultc C sX | � t�� � t�d� ztjt| jd d�}W d t�d � X | � |j� � � |�
� d S )Nr� )rj ) r� rI r� r� r r� r| rc r� r� rr )r r� r r r �testTimeoutNone� s
z TestTimeouts.testTimeoutNonec C s0 t jt| jdd�}| �|j�� d� |�� d S )Nr� )rj )r r� r| rc r� r� r� rr )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} t�� }ztj| � W d tj|� X d S )N)rv r� r� r� r� Zthreading_setupZ run_unittestZthreading_cleanup)Ztests� thread_infor r r � test_main s r� �__main__)+�__doc__r r[ r rI �osrP rY Zunittestr r Ztestr r� r| r} r= r! rF �pathr �dirname�__file__�curdirrH r� r� r0 r3 r r r\ rZ rV rv Ztest.test_ftplibr� r� r� r� r� r� rS r r r r �<module> sH
3 ) 7