Sindbad~EG File Manager
a
��glE � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddlm Z m
Z
ddl m Z
ddlmZ ddlmZ ejZdZdZeed�r�ddlZd Zej�ej�e�p�ejd
�Zej�ej�e�p�ejd �Z e
ed �Z d
Z dZ G dd� dej �Z!G dd� dej"ej#�Z$G dd� de �Z%e�rBddl&m'Z' G dd� de'e!�Z(e G dd� de%��Z)e G dd� de%��Z*G d d � d e �Z+d d � Z,e-d k�r�e�.� dS )!z Test script for poplib module.� N)�TestCase�
skipUnless)�support)�hashlib_helper)�
socket_helperF�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 )N�
�(+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.9/test/test_poplib.pyr 3 s
zDummyPOP3Handler.__init__c C s | j �|� d S �N)r �append�r �datar r r �collect_incoming_data; 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� �
ISO-8859-1� r ���� � Zcmd_z$-ERR unrecognized POP3 command "%s".) �joinr �str�split�lower�find�hasattr�getattrr )r �line�cmd�space�arg�methodr r r �found_terminator> s
z!DummyPOP3Handler.found_terminatorc C s � d S r r �r r r r � handle_errorN s z DummyPOP3Handler.handle_errorc C s t j�| |�d�d � d S )Nr r
)r r
r �encoder r r r r Q s zDummyPOP3Handler.pushc C s | � |� d S r �r �r r. r r r �cmd_echoT s zDummyPOP3Handler.cmd_echoc C s |dkr| � d� | � d� d S )N�guidoz-ERR no such userz+OK password requiredr4 r5 r r r �cmd_userX s
zDummyPOP3Handler.cmd_userc C s |dkr| � d� | � d� d S )N�pythonz-ERR wrong passwordz+OK 10 messagesr4 r5 r r r �cmd_pass] s
zDummyPOP3Handler.cmd_passc C s | � d� d S )Nz
+OK 10 100r4 r5 r r r �cmd_statb s zDummyPOP3Handler.cmd_statc C s4 |r| � d||f � n| � d� tj� | t� d S )Nz +OK %s %sz+OK)r r r
� LIST_RESPr5 r r r �cmd_liste s
zDummyPOP3Handler.cmd_listc C s$ | � dtt� � tj� | t� d S )Nz +OK %s bytes)r �len� RETR_RESPr r
r5 r r r �cmd_retrn s zDummyPOP3Handler.cmd_retrc C s | � d� d S )Nz +OK message marked for deletion.r4 r5 r r r �cmd_delet s zDummyPOP3Handler.cmd_delec C s | � d� d S �Nz+OK done nothing.r4 r5 r r r �cmd_noopw s zDummyPOP3Handler.cmd_noopc C s | � d� d S rB r4 r5 r r r �cmd_rpopz s zDummyPOP3Handler.cmd_rpopc C s | � d� d S rB r4 r5 r r r �cmd_apop} s zDummyPOP3Handler.cmd_apopc C s | � d� | �� d S )Nz +OK closing.)r Zclose_when_doner5 r r r �cmd_quit� s
zDummyPOP3Handler.cmd_quitc C s t | j�}| js tr g |d< |S �NZSTLS)�dict�CAPASr � SUPPORTS_SSL)r Z_capasr r r �
_get_capas� s
zDummyPOP3Handler._get_capasc C sZ | � d� | �� rL| �� �� D ],\}}|g}|r:|�|� | � d�|�� q | � d� d S )Nz+OK Capability list followsr �.)r rK �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_UTF8r5 r r r �cmd_utf8� s
��zDummyPOP3Handler.cmd_utf8c C sv | j du rh| �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 z| j �� W n� tjy� } z�|jd tjtjfv rDW Y d }~d S |jd tjkrh| �� W Y d }~S d|jd v s�d|jd v r�| �� W Y d }~S � W Y d }~n\d }~0 t y� } z0|jd t
j kr�| �� W Y d }~S W Y d }~nd }~0 0 d| _ d| _
d S )Nr ZSSLV3_ALERT_BAD_CERTIFICATEr" Z SSLV3_ALERT_CERTIFICATE_UNKNOWNTF)rV Z do_handshakerS ZSSLError�argsZSSL_ERROR_WANT_READZSSL_ERROR_WANT_WRITEZ
SSL_ERROR_EOF� handle_close�OSError�errnoZ ECONNABORTEDr r )r �errr r r rW � s& � �*z"DummyPOP3Handler._do_tls_handshakec C sB | j r| �� n.ztj�| � W n tjy< | �� Y n0 d S r )r rW r r
� handle_readrS Z SSLEOFErrorr[ r1 r r r r_ � s
z DummyPOP3Handler.handle_readN) �__name__�
__module__� __qualname__rI rQ r r r0 r2 r r6 r8 r: r; r= Zcmd_uidlr@ Zcmd_toprA rC rD rE rF rK rP rR rJ rY rW r_ 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_socketrV � SOCK_STREAM�bind�listen�active�Lock� active_lock� getsockname�host�port�handler_instance)r �address�afr r r r � s
zDummyPOP3Server.__init__c C s. | j r
J �t�� | _tj�| � | j�� d S r )rn rf �Event�_DummyPOP3Server__flagrg �start�waitr1 r r r ry � s
zDummyPOP3Server.startc C sv d| _ | j�� zR| j rTtjrT| j� tjddd� W d � q1 sH0 Y qW tjdd� ntjdd� 0 d S )NTg�������?r" )�timeout�count)Z
ignore_all)rn rx �setrh Z
socket_maprp ZloopZ close_allr1 r r r �run� s
0zDummyPOP3Server.runc C s | j s
J �d| _ | �� d S �NF)rn r$ r1 r r r �stop� s
zDummyPOP3Server.stopc C s | � |�| _d S r )�handlerrt )r r �addrr r r �handle_accepted� s z DummyPOP3Server.handle_acceptedc C s | � � d S r )�closer1 r r r �handle_connect� s z DummyPOP3Server.handle_connectc C s dS )Nr r r1 r r r �writable� s zDummyPOP3Server.writablec C s � d S r r r1 r r r r2 � s z DummyPOP3Server.handle_errorN)r` ra rb r r� rV �AF_INETr ry r~ r� r� r� r_ r� r2 r r r r rc � s
rc 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 � Zejd d d �d d!� �Zejd d d �d"d#� �Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zed0d1� �Z ed2d3� �Z ed4d5� �Z d6S )7�
TestPOP3Classc C s | � |�d�� d S )N� +OK)�
assertTrue�
startswith�r �respr r r �assertOK� s zTestPOP3Class.assertOKc C s8 t ttf�| _| j�� tj| jj| jjt j
d�| _ d S �N�r{ ) rc �HOST�PORT�serverry �poplib�POP3rr rs � test_support�LOOPBACK_TIMEOUT�clientr1 r r r �setUp s
�zTestPOP3Class.setUpc C s | j �� | j�� d | _d S r )r� r� r� r� r1 r r r �tearDown s
zTestPOP3Class.tearDownc C s | � | j�� d� d S )Ns( +OK dummy pop3 server ready. <timestamp>)� assertEqualr� Z
getwelcomer1 r r r �test_getwelcome s �z TestPOP3Class.test_getwelcomec C s | � tj| jjd� d S )Nz echo -err)� assertRaisesr� � error_protor� � _shortcmdr1 r r r �test_exceptions s z TestPOP3Class.test_exceptionsc C s* | � | j�d�� | �tj| jjd� d S )Nr7 �invalid)r� r� �userr� r� r� r1 r r r � test_user s zTestPOP3Class.test_userc C s* | � | j�d�� | �tj| jjd� d S )Nr9 r� )r� r� Zpass_r� r� r� r� r1 r r r �
test_pass_ s zTestPOP3Class.test_pass_c C s | � | j�� d� d S )N)�
�d )r� r� �statr1 r r r � test_stat s zTestPOP3Class.test_statc C s> | � | j�� dd � g d�df� | �| j�d��d�� d S )Nr" )s 1 1s 2 2s 3 3s 4 4s 5 5� �1s OK 1 1)r� r� �listr� �endswithr1 r r r � test_list s ��zTestPOP3Class.test_listc C s* dg d�df}| j �d�}| �||� d S )N�
+OK 116 bytes�s 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_MAXLINEr1 r r r �test_too_long_lines/ s �z!TestPOP3Class.test_too_long_linesc C s | � | j�d�� d S �Nr� )r� r� Zdeler1 r r r � test_dele3 s zTestPOP3Class.test_delec C s | � | j�� � d S r )r� r� �noopr1 r r r � test_noop6 s zTestPOP3Class.test_noopc C s | � | j�d�� d S r� )r� r� Zrpopr1 r r r � test_rpop9 s zTestPOP3Class.test_rpop�md5T)Zopensslc C s | � | j�dd�� d S )Nr� Z
dummypassword)r� r� �apopr1 r r r �test_apop_normal< s z TestPOP3Class.test_apop_normalc C sV ddd }t �| jd|��& | �tj| jjdd� W d � n1 sH0 Y d S )Nr� � <i@B Zwelcomer� Zkb)r� Z swap_attrr� r� r� r� r� )r Z evil_welcomer r r �test_apop_REDOS@ s z TestPOP3Class.test_apop_REDOSc C s( dg d�df}| � | j�dd�|� d S )Nr� r� r� r" )r� r� �top)r r� r r r �test_topK s
�zTestPOP3Class.test_topc C s | j �� | j �d� d S r� )r� Zuidlr1 r r r � test_uidlS s
zTestPOP3Class.test_uidlc C s d| j j_| �tj| jj� d S r )r� r� rQ r� r� r� r� �utf8r1 r r r � test_utf8_raises_if_unsupportedW s
z-TestPOP3Class.test_utf8_raises_if_unsupportedc C s( d| j j_d}| j�� }| �||� d S )NTs +OK I know RFC6856)r� r� rQ r� r� r� )r r� �resultr r r � test_utf8[ s
zTestPOP3Class.test_utf8c C s | j �� }| �d|�� v � d S )Nr �r� �capar� �keys�r r� r r r � test_capaa s
zTestPOP3Class.test_capac C s4 | j �� }| �|� | �| j j� | �| j j� d S r )r� �quitr� � assertIsNone�sock�filer� r r r � test_quite s
zTestPOP3Class.test_quitc C s | j �� }| �d|�� v � d S rG r� r� r r r �test_stls_capak s
z TestPOP3Class.test_stls_capac C s d}| j �� }| �||� d S )N� +OK Begin TLS negotiation)r� �stlsr� )r r� r� r r r � test_stlsp s
zTestPOP3Class.test_stlsc C s� d}t �t j�}|�t� | �|jt j� | �|jd� | � t j
�� | j j |d�}W d � n1 sh0 Y t
jd| jjtjd�| _ | j j |d�}| �||� d S )Nr� T�rX � localhostr� )rS rT �PROTOCOL_TLS_CLIENTZload_verify_locations�CAFILEr� � verify_modeZ
CERT_REQUIRED�check_hostnamer� ZCertificateErrorr� r� r� r� r� rs r� r� )r r� �ctxr� r r r �test_stls_contextv s
, �z TestPOP3Class.test_stls_contextN) r` ra rb r� r� r� r� r� r� r� r� r� r� r� r� r� r� r Zrequires_hashdigestr� r� r� r� r� r� r� r� � requires_sslr� r� r� r r r r r� � s<
r� )�
SSLConnectionc @ s e Zd Zdd� ZdS )�DummyPOP3_SSLHandlerc C s@ t j�| |� | �� | �d� g | _| �d� d| _d| _d S )Nr
r TF) r r
r Zsecure_connectionr r r r r r r r r r � s
z DummyPOP3_SSLHandler.__init__N)r` ra rb 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 r ) rc r� r� r� r� r� ry r� r rr rs r� r1 r r r r� � s
zTestPOP3_SSLClass.setUpc C s | � dtj� d S )Nr )ZassertInr� �__all__r1 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)�keyfilerX )�certfilerX )r� r� rX r� r� )rS rT r� r� Z CERT_NONEr� r� �
ValueErrorr� r r� rr rs rU r� r� ZassertIsInstancer� Z SSLSocketZassertIsrX r� r� r� )r r� r r r � test_context� s(
�
��
�z TestPOP3_SSLClass.test_contextc C s | � tj| jj� d S r �r� r� r� r� r� r1 r r r r� � s zTestPOP3_SSLClass.test_stlsc C s | j �� }| �d|�� v � d S rG �r� r� Z assertFalser� r� r r r r� � s
z TestPOP3_SSLClass.test_stls_capaN) r` ra rb 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 sB t ttf�| _| j�� tj| jj| jjt j
d�| _ | j � � d S r� )
rc r� r� r� ry r� r� rr rs r� r� r� r� r1 r r r r� � s
�zTestPOP3_TLSClass.setUpc C sZ | j jd urF| j jd urFz| j �� W n tjyD | j �� Y n0 | j�� d | _d S r ) r� r� r� r� r� r� r� r� r� r1 r r r r� � s
zTestPOP3_TLSClass.tearDownc C s | � tj| jj� d S r r� r1 r r r r� � s zTestPOP3_TLSClass.test_stlsc C s | j �� }| �d|�� v � d S )Ns STLSr� r� r r r r� � s
z TestPOP3_TLSClass.test_stls_capaN)r` ra rb 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�< )�targetrZ T)rf rw �evtrV r� rk r� �
settimeoutr Z bind_portrs rg r� �threadrj ry rz r1 r r r r� � s
zTestTimeouts.setUpc C s | j �� d | _ d S r )r� r$ r1 r r r r� � s
zTestTimeouts.tearDownc C sd |� � |�� zDz"|�� \}}|�d� |�� W n tjyH Y n0 W |�� n
|�� 0 d S )Ns
+ Hola mundo
)rm r} �accept�sendr� rV r{ )r r� Zservr r� r r r r� � s
zTestTimeouts.serverc C sd | � t�� � t�tj� z t�t| j �}W t�d � n t�d � 0 | �
|j � � tj� |�
� d S r )r� rV �getdefaulttimeout�setdefaulttimeoutr� r� r� r� r� rs r� r� �
gettimeoutr� �r �popr r r �testTimeoutDefault� s z TestTimeouts.testTimeoutDefaultc C sb | � t�� � t�d� z tjt| jd d�}W t�d � n t�d � 0 | � |j� � � |�
� d S )N� r� ) r� rV r� r� r� r� r� rs r� r� r� r� r r r �testTimeoutNone s
z TestTimeouts.testTimeoutNonec C sp t jt| jtjd�}| �|j�� tj� |� � | �
t ��" t jt| jdd� W d � n1 sb0 Y d S )Nr� r ) r� r� r� rs r� r� r� r� r� r� r� r� r� r r r �testTimeoutValue s
z TestTimeouts.testTimeoutValueN) r` ra rb r� r� r� r� r� r� r r r r r� � s
r� c C s t �� } tjt jg| �R � d S r )r� Zthreading_setup�unittestZaddModuleCleanupZthreading_cleanup)� thread_infor r r � setUpModule s r� �__main__)/�__doc__r� rh r rV �osr] rf r� r r �testr r� Z test.supportr r r� r� rJ r) rS �pathr$ �dirname�__file__�curdirrU r� r� r< r? r
r ri rg rc r� Ztest.test_ftplibr� r� r� r� r� r� r` �mainr r r r �<module> sN
3 )!9
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists