Your IP : 216.73.216.147
a
XC?h�� � @ sr d dl mZ d dlmZ d dlmZ d dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlmZmZmZmZmZ d dlmZ d dlZd dlmZ d dlmZmZmZ z d dlZW n ey� dZY n0 ej� ej� e �p�ej d �Z ej� ej� e ��pej d
�Z!G d d � d ej"�Z#e�r<G d
d� de j$�Z%ej&Z&nG dd� d�Z%dZ&G dd� de j'�Z(G dd� d�Z)G dd� de)ej"�Z*e�+ed�G dd� de)ej"��Z,G dd� dej"�Z-e�+ed�G dd � d e-��Z.e�+e�/d �d �e�0d �G d d!� d!ej"���Z1e�+ed�e�+e�/d �d �e�0d �G d"d#� d#e1����Z2e�+ed�e�0d �G d$d%� d%e1���Z3e4d&k�rne�5� dS )'� )�support)�
socket_helper)�contextmanagerN)� reap_threads�verbose� run_with_tz�run_with_locale� cpython_only)�hashlib_helper)�mock)�datetime�timezone� timedeltaz keycert3.pemz pycacert.pemc @ sX e Zd Zdd� Zed�dd� �Zdd� Zedd d
�ed �d d
� ��Zdd� Z dd� Z
dS )� TestImaplibc C sb t �d�}t�d�}| �t�|�|� t�d�}| �t�|�|� t�d�}| �t�|�|� d S )N) i� � r r r r ���r r s. 25 (INTERNALDATE "01-Jan-2000 00:00:00 +0000")s. 25 (INTERNALDATE "01-Jan-2000 11:30:00 +1130")s. 25 (INTERNALDATE "31-Dec-1999 12:30:00 -1130"))�calendarZtimegm�imaplib�Internaldate2tuple� assertEqual�time�mktime)�selfZt0Ztt� r �6/opt/alt/python39/lib64/python3.9/test/test_imaplib.py�test_Internaldate2tuple s
���z#TestImaplib.test_Internaldate2tuplezMST+07MDT,M4.1.0,M10.5.0c C s | � t�d�t�d�� d S )Ns. 25 (INTERNALDATE "02-Apr-2000 02:30:00 +0000")s. 25 (INTERNALDATE "02-Apr-2000 03:30:00 +0000"))ZassertNotEqualr r �r r r r �"test_Internaldate2tuple_issue10941* s ���z.TestImaplib.test_Internaldate2tuple_issue10941c C s* ddt �d�ddt�dttdd���dgS )Ni �5wg e��A) �� � � r �! � r r r ) r r r r r! r" r r r r i � "18-May-2033 05:33:20 +0200")r � localtimer Z
fromtimestampr
r r r r r �
timevalues1 s ��zTestImaplib.timevalues�LC_ALLZde_DEZfr_FRzSTD-1DST,M3.2.0,M11.1.0c C s, d}| � � D ]}t�|�}| �||� q d S )Nr# )r% r �Time2Internaldater )r Zexpected�tZinternalr r r �test_Time2Internaldate9 s
z"TestImaplib.test_Time2Internaldatec C s | � � D ]}t�|� qd S �N)r% r r' )r r( r r r �,test_that_Time2Internaldate_returns_a_resultD s z8TestImaplib.test_that_Time2Internaldate_returns_a_resultc C s� t � � �D}z |�dtjf� | �d� W n t jy< Y n0 W d � n1 sR0 Y t�� }| �t ��}t�
� W d � n1 s�0 Y | � |j j
|� d S )N� z3Cannot run the test with local IMAP server running.)�socket�connectr Z
IMAP4_PORTZskipTest�errorr Z get_socket_conn_refused_errs� assertRaises�OSError�IMAP4�assertInZ exception�errno)r �sZexpected_errnos�cmr r r � test_imap4_host_default_valueK s
�$ &z)TestImaplib.test_imap4_host_default_valueN) �__name__�
__module__� __qualname__r r r r% r r) r+ r7 r r r r r s
r c @ s e Zd Zdd� ZdS )�SecureTCPServerc C s6 | j �� \}}t�� }|�t� |j|dd�}||fS )NT)Z server_side)r- Zaccept�ssl�
SSLContext�load_cert_chain�CERTFILEZ wrap_socket)r Z newsocketZfromaddr�contextZ
connstreamr r r � get_request_ s
zSecureTCPServer.get_requestN)r8 r9 r: rA r r r r r; ] s r; c @ s e Zd ZdS )r; N�r8 r9 r: r r r r r; j s c sz e Zd ZejZdZdZ� fdd�Zdd� Z dd� Z
d d
� Z d d � Z d
d� Z
dd� Zdd� Zdd� Zdd� Zdd� Z� ZS )�SimpleIMAPHandlerNr, c s t � �� d| j_d | j_d S �NF)�super�setup�server� is_selected�loggedr �� __class__r r rF u s
zSimpleIMAPHandler.setupc C s$ t rtd|�� � | j�|� d S )NzSENT: %r)r �print�strip�wfile�write�r �messager r r �_sendz s zSimpleIMAPHandler._sendc C s | � |d � d S )N�
)rR rP r r r �
_send_line s z SimpleIMAPHandler._send_linec C s | � |�d�� d S )N�ASCII)rT �encoderP r r r �_send_textline� s z SimpleIMAPHandler._send_textlinec C s | � d�|||f�� d S )N� )rW �join)r �tag�coderQ r r r � _send_tagged� s z SimpleIMAPHandler._send_taggedc C s | � d� d}z&| j�d�}|dkr*W d S ||7 }W n tyH Y d S 0 |�d�rqXqtrltd|�� � | jr�z| j� |� W q
t
y� d | _Y q
0 q
|� d�� � }|d }|d }|dd � }t
| d | ��rt| d | �||�}|�r|| _t|� q
| �|d
|d � q
d S ) Nz* OK IMAP4rev1� r rS zGOT: %rrU r � Zcmd_�BADz unknown)rW Zrfile�readr1 �endswithr rL rM � continuation�send�
StopIteration�decode�split�hasattr�getattr�nextr\ )r �line�partZ splitlinerZ �cmd�argsrb r r r �handle� s:
zSimpleIMAPHandler.handlec C s4 | j rd| j nd}| �d| � | �|dd� d S )Nz
IMAP4rev1 � IMAP4rev1z
* CAPABILITY �OK�CAPABILITY completed)� capabilitiesrW r\ )r rZ rm Zcapsr r r �cmd_CAPABILITY� s � �z SimpleIMAPHandler.cmd_CAPABILITYc C s$ d | j _| �d� | �|dd� d S )Nz"* BYE IMAP4ref1 Server logging outrp zLOGOUT completed)rG rI rW r\ �r rZ rm r r r �
cmd_LOGOUT� s
z SimpleIMAPHandler.cmd_LOGOUTc C s |d | j _| �|dd� d S )Nr rp zLOGIN completed)rG rI r\ rt r r r � cmd_LOGIN� s zSimpleIMAPHandler.cmd_LOGINc C s$ d| j _| �d� | �|dd� d S )NT�
* 2 EXISTSrp � [READ-WRITE] SELECT completed.)rG rH rT r\ rt r r r �
cmd_SELECT� s
z SimpleIMAPHandler.cmd_SELECTc C s2 | j jr d| j _| �|dd� n| �|dd� d S )NFrp z*Returned to authenticated state. (Success)r_ zNo mailbox selected)rG rH r\ rt r r r � cmd_UNSELECT� s z SimpleIMAPHandler.cmd_UNSELECT)r8 r9 r: r �LOOPBACK_TIMEOUT�timeoutrb rr rF rR rT rW r\ rn rs ru rv ry rz �
__classcell__r r rJ r rC p s *rC c @ s
e Zd ZdZd;dd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zejd dd �d d � �Zejd dd �d!d"� �Zd#d$� Ze�d%d&�d'd(� �Zd)d*� Zd+d,� Zd-d.� Zd/d0� Z d1d2� Z d3d4� Z d5d6� Z d7d8� Z d9d:� Z!dS )<�NewIMAPTestsMixinNTc C s� G dd� d| j �}| �| j� | � tjdf|�| _tj| jd | jj ddid�| _
d| j
_ | j
� � |rx| j
| jj� | _| j| jfS ) Nc @ s e Zd Zdd� ZdS )z/NewIMAPTestsMixin._setup.<locals>.TestTCPServerc S s | � |� | �� � d S r* �Z
close_request� server_close�r ZrequestZclient_addressr r r � handle_error� s
z<NewIMAPTestsMixin._setup.<locals>.TestTCPServer.handle_errorN�r8 r9 r: r� r r r r �
TestTCPServer� s r� r z-server�
poll_interval�{�G�z�?��name�target�kwargsT)� server_classZ
addCleanup�_cleanupr �HOSTrG � threading�ThreadZ_testMethodName�
serve_forever�thread�daemon�start�
imap_class�server_address�client)r Z imap_handlerr. r� r r r �_setup� s �
zNewIMAPTestsMixin._setupc C sJ | j d ur | j jdkr | j �� | j�� | j�� t�| j� d | _d S )N�LOGOUT)r� �state�shutdownrG r� r Z join_threadr� r r r r r� � s
zNewIMAPTestsMixin._cleanupc C sD G dd� dt j�}| j|dd�\}}| jtjj| jg|j�R � d S )Nc @ s e Zd Zdd� ZdS )zONewIMAPTestsMixin.test_EOF_without_complete_welcome_message.<locals>.EOFHandlerc S s | j �d� d S �Ns * OK�rN rO r r r r rn s zVNewIMAPTestsMixin.test_EOF_without_complete_welcome_message.<locals>.EOFHandler.handleN�r8 r9 r: rn r r r r �
EOFHandler s r� F�r. ) � socketserver�StreamRequestHandlerr� r0 r r2 �abortr� r� )r r� �_rG r r r �)test_EOF_without_complete_welcome_message s
�z;NewIMAPTestsMixin.test_EOF_without_complete_welcome_messagec C sB G dd� dt �}| j|dd�\}}| jtjj| jg|j�R � d S )Nc @ s e Zd Zdd� ZdS )zBNewIMAPTestsMixin.test_line_termination.<locals>.BadNewlineHandlerc S s | � d� | �|dd� d S �Ns * CAPABILITY IMAP4rev1 AUTH
rp rq �rR r\ rt r r r rs s
zQNewIMAPTestsMixin.test_line_termination.<locals>.BadNewlineHandler.cmd_CAPABILITYN�r8 r9 r: rs r r r r �BadNewlineHandler
s r� Fr� )rC r� r0 r r2 r� r� r� )r r� r� rG r r r �test_line_termination s
�z'NewIMAPTestsMixin.test_line_terminationc C st G dd� dt �}| �|�\}}| �|j� | �tjjd�� |�d� W d � n1 sZ0 Y | �|j� d S )Nc @ s e Zd ZdZdS )zMNewIMAPTestsMixin.test_enable_raises_error_if_not_AUTH.<locals>.EnableHandler�AUTH ENABLE UTF8=ACCEPTN�r8 r9 r: rr r r r r �
EnableHandler s r� zENABLE.*NONAUTH�foo) rC r� � assertFalse� utf8_enabled�assertRaisesRegexr r2 r/ �enable)r r� r� r� r r r �$test_enable_raises_error_if_not_AUTH s (z6NewIMAPTestsMixin.test_enable_raises_error_if_not_AUTHc C sL | � t�\}}| �tjjd�� |�d� W d � n1 s>0 Y d S )N�does not support ENABLEr� )r� rC r� r r2 r/ r� )r r� r� r r r �)test_enable_raises_error_if_no_capability s
�z;NewIMAPTestsMixin.test_enable_raises_error_if_no_capabilityc C sh | � t�\}}|�dd�\}}| �|d� | �tjjd�� |�d� W d � n1 sZ0 Y d S )N�user�passrp r� � UTF8=ACCEPT) r� rC �loginr r� r r2 r/ r� �r r� r� �typ�datar r r �.test_enable_UTF8_raises_error_if_not_supported$ s
�z@NewIMAPTestsMixin.test_enable_UTF8_raises_error_if_not_supportedc C s� G dd� dt �}| �|�\}}| �|jd� |�ddd� �\}}| �|d� | �|jd� |�d �\}}| �|d� | �|jd
� d }|�d d d |�d
��\}}| �|d� | �|jd | �d
�� d S )
Nc @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) zHNewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.UTF8AppendServerzENABLE UTF8=ACCEPTc S s | � |dd� d S �Nrp zENABLE successful�r\ rt r r r �
cmd_ENABLE/ s zSNewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_ENABLEc s s& | � d� d V | j_| �|dd� d S �N�+rp zFAKEAUTH successful�rW rG �responser\ rt r r r �cmd_AUTHENTICATE1 s
zYNewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_AUTHENTICATEc s s& | � d� d V | j_| �|dd� d S �Nr� rp Zokayr� rt r r r �
cmd_APPEND5 s
zSNewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_APPENDN)r8 r9 r: rr r� r� r� r r r r �UTF8AppendServer- s r� �ascii�MYAUTHc S s dS �Ns faker ��xr r r �<lambda>; r] z@NewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.<lambda>rp �
ZmFrZQ==
r� �utf-8� Subject: üñí©öðé� UTF8 (%s)
) rC r� r � _encoding� authenticater� r� �appendrV ) r r� r� rG r[ r� �
msg_stringr� r� r r r � test_enable_UTF8_True_append, s �z.NewIMAPTestsMixin.test_enable_UTF8_True_appendc C s� G dd� dt �}| �|�\}}|�ddd� �\}}| �|d� |�d�\}}| �|d� | �|j� | �tj j
d�� |� d d
� W d � n1 s�0 Y d S ) Nc @ s e Zd ZdZdd� Zdd� ZdS )zPNewIMAPTestsMixin.test_search_disallows_charset_in_utf8_mode.<locals>.UTF8Serverr� c S s | � |dd� d S r� r� rt r r r r� J s z[NewIMAPTestsMixin.test_search_disallows_charset_in_utf8_mode.<locals>.UTF8Server.cmd_ENABLEc s s& | � d� d V | j_| �|dd� d S r� r� rt r r r r� L s
zaNewIMAPTestsMixin.test_search_disallows_charset_in_utf8_mode.<locals>.UTF8Server.cmd_AUTHENTICATEN�r8 r9 r: rr r� r� r r r r �
UTF8ServerH s r� r� c S s dS r� r r� r r r r� Q r] zNNewIMAPTestsMixin.test_search_disallows_charset_in_utf8_mode.<locals>.<lambda>rp r� z
charset.*UTF8r� �bar) rC r� r� r r� �
assertTruer� r� r r2 r/ �search)r r� r� r� r� r r r �*test_search_disallows_charset_in_utf8_modeG s z<NewIMAPTestsMixin.test_search_disallows_charset_in_utf8_modec C sb G dd� dt �}| �|�\}}| �tjjd�� |�ddd� � W d � n1 sT0 Y d S )Nc @ s e Zd Zdd� ZdS )z6NewIMAPTestsMixin.test_bad_auth_name.<locals>.MyServerc S s | � |dd�|d �� d S �N�NOz#unrecognized authentication type {}r �r\ �formatrt r r r r� [ s �zGNewIMAPTestsMixin.test_bad_auth_name.<locals>.MyServer.cmd_AUTHENTICATEN�r8 r9 r: r� r r r r �MyServerZ s r� z'unrecognized authentication type METHOD�METHODc S s dS �Nr r r r r r r� a r] z6NewIMAPTestsMixin.test_bad_auth_name.<locals>.<lambda>�rC r� r� r r2 r/ r� �r r� r� r� r r r �test_bad_auth_nameY s
�z$NewIMAPTestsMixin.test_bad_auth_namec C sb G dd� dt �}| �|�\}}| �tjjd�� |�ddd� � W d � n1 sT0 Y d S )Nc @ s e Zd Zdd� ZdS )z?NewIMAPTestsMixin.test_invalid_authentication.<locals>.MyServerc s s$ | � d� d V | _| �|dd� d S �Nr� r� z [AUTHENTICATIONFAILED] invalid�rW r� r\ rt r r r r� e s
zPNewIMAPTestsMixin.test_invalid_authentication.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� d s r� z \[AUTHENTICATIONFAILED\] invalidr� c S s dS r� r r� r r r r� l r] z?NewIMAPTestsMixin.test_invalid_authentication.<locals>.<lambda>r� r� r r r �test_invalid_authenticationc s
�z-NewIMAPTestsMixin.test_invalid_authenticationc C sP G dd� dt �}| �|�\}}|�ddd� �\}}| �|d� | �|jd� d S )Nc @ s e Zd Zdd� ZdS )zCNewIMAPTestsMixin.test_valid_authentication_bytes.<locals>.MyServerc s s& | � d� d V | j_| �|dd� d S r� r� rt r r r r� p s
zTNewIMAPTestsMixin.test_valid_authentication_bytes.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� o s r� r� c S s dS r� r r� r r r r� u r] zCNewIMAPTestsMixin.test_valid_authentication_bytes.<locals>.<lambda>rp r� �rC r� r� r r� �r r� r� rG r[ r� r r r � test_valid_authentication_bytesn s
z1NewIMAPTestsMixin.test_valid_authentication_bytesc C sP G dd� dt �}| �|�\}}|�ddd� �\}}| �|d� | �|jd� d S )Nc @ s e Zd Zdd� ZdS )zHNewIMAPTestsMixin.test_valid_authentication_plain_text.<locals>.MyServerc s s& | � d� d V | j_| �|dd� d S r� r� rt r r r r� { s
zYNewIMAPTestsMixin.test_valid_authentication_plain_text.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� z s r� r� c S s dS �NZfaker r� r r r r� � r] zHNewIMAPTestsMixin.test_valid_authentication_plain_text.<locals>.<lambda>rp r� r� r� r r r �$test_valid_authentication_plain_texty s
z6NewIMAPTestsMixin.test_valid_authentication_plain_text�md5�Zopensslc C sN G dd� dt �}| �|�\}}| �d|jv � |�dd�\}}| �|d� d S )Nc @ s e Zd ZdZdd� ZdS )z@NewIMAPTestsMixin.test_login_cram_md5_bytes.<locals>.AuthHandler�LOGINDISABLED AUTH=CRAM-MD5c s s: | � d� d V }|dkr(| �|dd� n| �|dd� d S �Nz:+ PDE4OTYuNjk3MTcwOTUyQHBvc3RvZmZpY2UucmVzdG9uLm1jaS5uZXQ=s2 dGltIGYxY2E2YmU0NjRiOWVmYTFjY2E2ZmZkNmNmMmQ5ZjMy
rp zCRAM-MD5 successfulr� z No access�rW r\ �r rZ rm �rr r r r� � s
zQNewIMAPTestsMixin.test_login_cram_md5_bytes.<locals>.AuthHandler.cmd_AUTHENTICATEN�r8 r9 r: rr r� r r r r � AuthHandler� s r� �
AUTH=CRAM-MD5�tim� tanstaaftanstaafrp �rC r� r� rr �login_cram_md5r �r r� r� r� �retr r r �test_login_cram_md5_bytes� s
z+NewIMAPTestsMixin.test_login_cram_md5_bytesc C sN G dd� dt �}| �|�\}}| �d|jv � |�dd�\}}| �|d� d S )Nc @ s e Zd ZdZdd� ZdS )zENewIMAPTestsMixin.test_login_cram_md5_plain_text.<locals>.AuthHandlerr� c s s: | � d� d V }|dkr(| �|dd� n| �|dd� d S r� r� r� r r r r� � s
zVNewIMAPTestsMixin.test_login_cram_md5_plain_text.<locals>.AuthHandler.cmd_AUTHENTICATENr� r r r r r� � s r� r� r� �tanstaaftanstaafrp r� r� r r r � test_login_cram_md5_plain_text� s
z0NewIMAPTestsMixin.test_login_cram_md5_plain_textc C sb G dd� dt �}| �|�\}}| �tjjd�� |�ddd� � W d � n1 sT0 Y d S )Nc @ s e Zd Zdd� ZdS )z?NewIMAPTestsMixin.test_aborted_authentication.<locals>.MyServerc s s> | � d� d V | _| jdkr,| �|dd� n| �|dd� d S �Nr� s *
r� z [AUTHENTICATIONFAILED] abortedrp zMYAUTH successfulr� rt r r r r� � s
�zPNewIMAPTestsMixin.test_aborted_authentication.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� � s r� z \[AUTHENTICATIONFAILED\] abortedr� c S s d S r* r r� r r r r� � r] z?NewIMAPTestsMixin.test_aborted_authentication.<locals>.<lambda>r� r� r r r �test_aborted_authentication� s
�z-NewIMAPTestsMixin.test_aborted_authenticationzimaplib._MAXLINE�
c C sb G dd� dt �}| j|dd�\}}| �tjjd�� | j|j� W d � n1 sT0 Y d S )Nc @ s e Zd Zdd� ZdS )z:NewIMAPTestsMixin.test_linetoolong.<locals>.TooLongHandlerc S s | j �d� d S )Ns * OK xxxxxxxxxxx
r� r r r r rn � s zANewIMAPTestsMixin.test_linetoolong.<locals>.TooLongHandler.handleNr� r r r r �TooLongHandler� s r Fr� zgot more than 10 bytes)rC r� r� r r2 r/ r� r� )r r r� rG r r r �test_linetoolong� s
�z"NewIMAPTestsMixin.test_linetoolongc C sB | j tdd�\}}| j|j� � W d � n1 s40 Y d S )NFr� )r� rC r� r� )r r� rG r r r �test_simple_with_statement� s z,NewIMAPTestsMixin.test_simple_with_statementc C s� | � t�\}}|jd }| jd|d d�}| �|jjd � |�� | jd|tj d�}| �|jjtj � |�� | �
t �� | jd|dd�}W d � n1 s�0 Y d S )Nr � localhost�r| r ) r� rC r� r� r Zsockr| r� r r{ r0 �
ValueError)r r� rG �addrr� r r r �test_imaplib_timeout_test� s
z+NewIMAPTestsMixin.test_imaplib_timeout_testc C sh G dd� dt �}| �|�\}}|jd }| �tj�� | jd|dd�}W d � n1 sZ0 Y d S )Nc @ s e Zd Zdd� ZdS )zQNewIMAPTestsMixin.test_imaplib_timeout_functionality_test.<locals>.TimeoutHandlerc S s t �d� t�| � d S r� )r �sleeprC rn r r r r rn � s
zXNewIMAPTestsMixin.test_imaplib_timeout_functionality_test.<locals>.TimeoutHandler.handleNr� r r r r �TimeoutHandler� s r r r g����MbP?r
)rC r� r� r0 r- r| r� )r r r� rG r r� r r r �'test_imaplib_timeout_functionality_test� s
z9NewIMAPTestsMixin.test_imaplib_timeout_functionality_testc C sh | j tdd�\}}| j|j� �*}|�dd� | �|jd� W d � n1 sN0 Y | �|j� d S �NFr� r� r� )r� rC r� r� r� r rI � assertIsNone�r r� rG �imapr r r �test_with_statement� s
,z%NewIMAPTestsMixin.test_with_statementc C s| | j tdd�\}}| j|j� �>}|�dd� | �|jd� |�� | �|j� W d � n1 sb0 Y | �|j� d S r ) r� rC r� r� r� r rI �logoutr r r r r �test_with_statement_logout� s *z,NewIMAPTestsMixin.test_with_statement_logoutc C sL | � t�\}}|�dd�\}}| �|d� | �|d d� | �|jd� d S )Nr� r� rp r � LOGIN completed�AUTH)r� rC r� r r� r� r r r �
test_login� s
z NewIMAPTestsMixin.test_loginc C s� | � t�\}}|�dd�\}}| �|d� | �|d d� |�� \}}| �|d||f� | �|d d||f� | �|jd� d S ) Nr� r� rp r r �BYEs IMAP4ref1 Server logging outr� )r� rC r� r r r� r� r r r � test_logout� s z NewIMAPTestsMixin.test_logoutc C sV G dd� dt �}| �|�\}}|�dd� |�� \}}| �|d� | �|d d� d S )Nc @ s e Zd Zdd� ZdS )z,NewIMAPTestsMixin.test_lsub.<locals>.LsubCmdc S s | � d� | �|dd�S )Nz* LSUB () "." directoryArp zLSUB completedr� rt r r r �cmd_LSUB s
z5NewIMAPTestsMixin.test_lsub.<locals>.LsubCmd.cmd_LSUBN)r8 r9 r: r r r r r �LsubCmd s r r� r� rp r s () "." directoryA)rC r� r� Zlsubr )r r r� r� r� r� r r r � test_lsub s zNewIMAPTestsMixin.test_lsubc C s| | � t�\}}|�dd� |�� \}}| �|d� | �|d d� |�� \}}| �|d� | �|d d� | �|jd� d S )Nr� r� rp r � 2s* Returned to authenticated state. (Success)r )r� rC r� �selectr Zunselectr� r� r r r �
test_unselect s z NewIMAPTestsMixin.test_unselect)T)"r8 r9 r: r� r� r� r� r� r� r� r� r� r� r� r� r� r� r
�requires_hashdigestr r r r �patchr r r
r r r r r r r" r r r r r~ � s<
!
r~ c @ s e Zd ZejZejZdS )� NewIMAPTestsN) r8 r9 r: r r2 r� r� � TCPServerr� r r r r r% s r% zSSL not availablec @ s0 e Zd ZeZeZdd� Zdd� Ze dd� �Z
dS )�NewIMAPSSLTestsc C s� t �t j�}| �|jt j� | �|jd� |�t� | � t j
d��8 | � t �\}}| j
|jd|i�}|�� W d � n1 s�0 Y d S )NT�=IP address mismatch, certificate is not valid for '127.0.0.1'� ssl_context)r< r= �PROTOCOL_TLS_CLIENTr � verify_modeZ
CERT_REQUIRED�check_hostname�load_verify_locations�CAFILEr� �CertificateErrorr� rC r� r� r� �r r) r� rG r� r r r �test_ssl_raises& s
�
�z NewIMAPSSLTests.test_ssl_raisesc C sF t �t j�}|�t� | �t�\}}| jd|jd |d�}|� � d S )Nr r �r) )
r< r= r* r- r. r� rC r� r� r� r0 r r r �test_ssl_verified4 s
�z!NewIMAPSSLTests.test_ssl_verifiedc
C s� t �dtf��� tj�| jd��P tj�| jd�� | jddtd� W d � n1 sX0 Y W d � n1 sv0 Y W d � n1 s�0 Y d S )Nr, �openZ_connectr � )�certfile)r Zcheck_warnings�DeprecationWarningr r$ �objectr� r? r r r r �test_certfile_arg_warn? s z&NewIMAPSSLTests.test_certfile_arg_warnN) r8 r9 r: � IMAP4_SSLr� r; r� r1 r3 r r9 r r r r r' ! s r' c @ s> e Zd ZejZejZdd� Z dd� Z
e dd� �Z e dd� �Z
ed d
� �Zed d � �Zed
d� �Zedd� �ZG dd� de�Zedd� �Zedd� �Zedd� �Zedd� �Zedd � �Zed d � �Zed d � �Zed!d"� �Z ee j d#d$d%�d&d'� ��Z ed(d)� �Z d*d+� Z!ed,d-� �Z"ed.d/� �Z#ed0d1� �Z$ed2d3� �Z%d4S )5�ThreadedNetworkedTestsc C s� G dd� d| j �}tr td� |||�}| �|j|j�� � trjtd� td|� td| j � td|j� tj d| j |j
d d
id �}d |_ |� � tr�td
� ||fS )Nc @ s e Zd Zdd� ZdS )z4ThreadedNetworkedTests.make_server.<locals>.MyServerc S s | � |� | �� � d S r* r r� r r r r� M s
zAThreadedNetworkedTests.make_server.<locals>.MyServer.handle_errorNr� r r r r r� L s r� zcreating serverzserver createdzADDR =zCLASS =zHDLR =z
%s servingr� r� r� Tzserver running)
r� r rL r r� r- Z getsocknameZRequestHandlerClassr� r� r� r� r� )r r �hdlrr� rG r( r r r � make_serverJ s(
�z"ThreadedNetworkedTests.make_serverc C s4 t r td� |�� |�� |�� t r0td� d S )Nzwaiting for serverZdone)r rL r� r� rY )r rG r� r r r � reap_serverj s z"ThreadedNetworkedTests.reap_serverc
c s@ | � tjdf|�\}}z|V W | �||� n| �||� 0 d S )Nr )r= r r� r> )r r< rG r� r r r �
reaped_servers s z$ThreadedNetworkedTests.reaped_serverc c s\ | � |��>}| j|j� }z||fV W |�� n
|�� 0 W d � n1 sN0 Y d S r* )r? r� r� r )r r< rG r� r r r � reaped_pair{ s
z"ThreadedNetworkedTests.reaped_pairc C sB | � t��$}| j|j� }|�� W d � n1 s40 Y d S r* )r? rC r� r� r� �r rG r� r r r � test_connect� s z#ThreadedNetworkedTests.test_connectc C s� G dd� dt �}| �|���\}}|�ddd� �\}}| �|d� | �|jd� |�d� |�d d
d �\}\}| �d |� |�d� |�d
�\}\}| �d |� W d � n1 s�0 Y d S )Nc s4 e Zd Z� fdd�Zdd� Zdd� Zdd� Z� ZS ) zEThreadedNetworkedTests.test_bracket_flags.<locals>.BracketFlagHandlerc s g d�| _ t� �� d S )N)ZAnsweredZFlaggedZDeletedZSeenZDraft)�flagsrE rn r rJ r r rn � s
zLThreadedNetworkedTests.test_bracket_flags.<locals>.BracketFlagHandler.handlec s s& | � d� d V | j_| �|dd� d S r� r� rt r r r r� � s
zVThreadedNetworkedTests.test_bracket_flags.<locals>.BracketFlagHandler.cmd_AUTHENTICATEc S s^ d� | j�}| �d| �d�� | �d� | �d� d| }| �|�d�� | �|dd� d S ) N� \z * FLAGS (%s)r� rw s
* 0 RECENTz-* OK [PERMANENTFLAGS %s \*)] Flags permitted.rp rx )rY rC rT rV r\ )r rZ rm Zflag_msg�msgr r r ry � s
�zPThreadedNetworkedTests.test_bracket_flags.<locals>.BracketFlagHandler.cmd_SELECTc S sf |d � d�� d��� }| j�|� dd�| j� }d|d |f }| �|�d�� | �|d d
� d S ) Nr^ �(�)z (FLAGS (%s))rD z
* %s FETCH %sr r� rp zSTORE completed.)rM rf rC �extendrY rT rV r\ )r rZ rm Z new_flagsZ flags_msgrE r r r � cmd_STORE� s zOThreadedNetworkedTests.test_bracket_flags.<locals>.BracketFlagHandler.cmd_STORE)r8 r9 r: rn r� ry rI r} r r rJ r �BracketFlagHandler� s
rJ r� c S s dS r� r r� r r r r� � r] z;ThreadedNetworkedTests.test_bracket_flags.<locals>.<lambda>rp r� �test� 1z+FLAGSz[test]s [test]ZPERMANENTFLAGS)rC r@ r� r r� r! Zstorer3 )r rJ rG r� r[ r� r� r r r �test_bracket_flags� s
z)ThreadedNetworkedTests.test_bracket_flagsc C s\ G dd� dt j�}| �|��,}| jtjj| jg|j�R � W d � n1 sN0 Y d S )Nc @ s e Zd Zdd� ZdS )z9ThreadedNetworkedTests.test_issue5949.<locals>.EOFHandlerc S s | j �d� d S r� r� r r r r rn � s z@ThreadedNetworkedTests.test_issue5949.<locals>.EOFHandler.handleNr� r r r r r� � s r� ) r� r� r? r0 r r2 r� r� r� )r r� rG r r r �test_issue5949� s
��z%ThreadedNetworkedTests.test_issue5949c C sZ G dd� dt �}| �|��,}| jtjj| jg|j�R � W d � n1 sL0 Y d S )Nc @ s e Zd Zdd� ZdS )zGThreadedNetworkedTests.test_line_termination.<locals>.BadNewlineHandlerc S s | � d� | �|dd� d S r� r� rt r r r rs � s
zVThreadedNetworkedTests.test_line_termination.<locals>.BadNewlineHandler.cmd_CAPABILITYNr� r r r r r� � s r� )rC r? r0 r r2 r� r� r� )r r� rG r r r r� � s
��z,ThreadedNetworkedTests.test_line_terminationc @ s e Zd ZdZdd� Zdd� ZdS )z!ThreadedNetworkedTests.UTF8Serverr� c C s | � |dd� d S r� r� rt r r r r� � s z,ThreadedNetworkedTests.UTF8Server.cmd_ENABLEc c s& | � d� d V | j_| �|dd� d S r� r� rt r r r r� � s
z2ThreadedNetworkedTests.UTF8Server.cmd_AUTHENTICATENr� r r r r r� � s r� c C s` | � | j��@\}}| �|j� | �tjj|jd� | �|j� W d � n1 sR0 Y d S )Nr� ) r@ r� r� r� r0 r r2 r/ r� rA r r r r� � s z;ThreadedNetworkedTests.test_enable_raises_error_if_not_AUTHc C sX G dd� d| j �}| �|��(\}}| �tjj|jd� W d � n1 sJ0 Y d S )Nc @ s e Zd ZdZdS )zXThreadedNetworkedTests.test_enable_raises_error_if_no_capability.<locals>.NoEnableServerr Nr� r r r r �NoEnableServer� s rO r� )r� r@ r0 r r2 r/ r� )r rO rG r� r r r r� � s z@ThreadedNetworkedTests.test_enable_raises_error_if_no_capabilityc C s� G dd� dt �}| �tjj��d | �|��:\}}|�dd�\}}| �|d� |�d� W d � n1 sj0 Y W d � n1 s�0 Y d S )Nc @ s e Zd ZdS )z\ThreadedNetworkedTests.test_enable_UTF8_raises_error_if_not_supported.<locals>.NonUTF8ServerNrB r r r r �
NonUTF8Server� s rP r� r� rp r� ) rC r0 r r2 r/ r@ r� r r� )r rP rG r� r� r� r r r r� � s
zEThreadedNetworkedTests.test_enable_UTF8_raises_error_if_not_supportedc C s� G dd� d| j �}| �|���\}}| �|jd� |�ddd� �\}}| �|d� | �|jd� |�d �\}}| �|d� | �|jd
� d }|�d d d |�d
��\}}| �|d� | �|jd | �d
�� W d � n1 s�0 Y d S )
Nc @ s e Zd Zdd� ZdS )zMThreadedNetworkedTests.test_enable_UTF8_True_append.<locals>.UTF8AppendServerc s s& | � d� d V | j_| �|dd� d S r� r� rt r r r r� � s
zXThreadedNetworkedTests.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_APPENDN)r8 r9 r: r� r r r r r� � s r� r� r� c S s dS r� r r� r r r r� r] zEThreadedNetworkedTests.test_enable_UTF8_True_append.<locals>.<lambda>rp r� r� r� r� r� ) r� r@ r r� r� r� r� r� rV ) r r� rG r� r[ r� r� r� r� r r r r� � s( � � �z3ThreadedNetworkedTests.test_enable_UTF8_True_appendc C s� | � | j��p\}}|�ddd� �\}}| �|d� |�d�\}}| �|d� | �|j� | �tj j
|j dd� W d � n1 s�0 Y d S )Nr� c S s dS r� r r� r r r r� r] zSThreadedNetworkedTests.test_search_disallows_charset_in_utf8_mode.<locals>.<lambda>rp r� r� r� ) r@ r� r� r r� r� r� r0 r r2 r/ r� )r rG r� r� r� r r r r� s zAThreadedNetworkedTests.test_search_disallows_charset_in_utf8_modec C s� G dd� dt �}| �|��R\}}| �tjj�� |�ddd� � W d � n1 sT0 Y W d � n1 sr0 Y d S )Nc @ s e Zd Zdd� ZdS )z;ThreadedNetworkedTests.test_bad_auth_name.<locals>.MyServerc S s | � |dd�|d �� d S r� r� rt r r r r� ' s �zLThreadedNetworkedTests.test_bad_auth_name.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� % s r� r� c S s dS r� r r r r r r� - r] z;ThreadedNetworkedTests.test_bad_auth_name.<locals>.<lambda>�rC r@ r0 r r2 r/ r� )r r� rG r� r r r r� " s z)ThreadedNetworkedTests.test_bad_auth_namec C s� G dd� dt �}| �|��V\}}| �tjj��$ |�ddd� �\}}W d � n1 sX0 Y W d � n1 sv0 Y d S )Nc @ s e Zd Zdd� ZdS )zDThreadedNetworkedTests.test_invalid_authentication.<locals>.MyServerc s s$ | � d� d V | _| �|dd� d S r� r� rt r r r r� 4 s
zUThreadedNetworkedTests.test_invalid_authentication.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� 2 s r� r� c S s dS r� r r� r r r r� ; r] zDThreadedNetworkedTests.test_invalid_authentication.<locals>.<lambda>rQ �r r� rG r� r[ r� r r r r� / s z2ThreadedNetworkedTests.test_invalid_authenticationc C s� G dd� dt �}| �|��B\}}|�ddd� �\}}| �|d� | �|jd� W d � n1 sb0 Y | �|��B\}}|�ddd� �\}}| �|d� | �|jd� W d � n1 s�0 Y d S ) Nc @ s e Zd Zdd� ZdS )zBThreadedNetworkedTests.test_valid_authentication.<locals>.MyServerc s s& | � d� d V | j_| �|dd� d S r� r� rt r r r r� B s
zSThreadedNetworkedTests.test_valid_authentication.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� @ s r� r� c S s dS r� r r� r r r r� H r] zBThreadedNetworkedTests.test_valid_authentication.<locals>.<lambda>rp r� c S s dS r� r r� r r r r� N r] )rC r@ r� r r� rR r r r �test_valid_authentication= s �" �z0ThreadedNetworkedTests.test_valid_authenticationr� Tr� c C s� G dd� dt �}| �|��@\}}| �d|jv � |�dd�\}}| �|d� W d � n1 s`0 Y | �|��@\}}| �d|jv � |�dd�\}}| �|d� W d � n1 s�0 Y d S )Nc @ s e Zd ZdZdd� ZdS )z?ThreadedNetworkedTests.test_login_cram_md5.<locals>.AuthHandlerr� c s s: | � d� d V }|dkr(| �|dd� n| �|dd� d S r� r� r� r r r r� [ s
zPThreadedNetworkedTests.test_login_cram_md5.<locals>.AuthHandler.cmd_AUTHENTICATENr� r r r r r� W s r� r� r� r rp r� )rC r@ r� rr r� r )r r� rG r� r� r� r r r �test_login_cram_md5S s *z*ThreadedNetworkedTests.test_login_cram_md5c C s� G dd� dt �}| �|��V\}}| �tjj��$ |�ddd� �\}}W d � n1 sX0 Y W d � n1 sv0 Y d S )Nc @ s e Zd Zdd� ZdS )zDThreadedNetworkedTests.test_aborted_authentication.<locals>.MyServerc s s> | � d� d V | _| jdkr,| �|dd� n| �|dd� d S r r� rt r r r r� u s
zUThreadedNetworkedTests.test_aborted_authentication.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� s s r� r� c S s d S r* r r� r r r r� � r] zDThreadedNetworkedTests.test_aborted_authentication.<locals>.<lambda>rQ rR r r r r p s z2ThreadedNetworkedTests.test_aborted_authenticationc C sZ G dd� dt �}| �|��,}| jtjj| jg|j�R � W d � n1 sL0 Y d S )Nc @ s e Zd Zdd� ZdS )z?ThreadedNetworkedTests.test_linetoolong.<locals>.TooLongHandlerc S s | j �dtjd d � d S )Ns * OK � xrS )rN rO r Z_MAXLINEr r r r rn � s zFThreadedNetworkedTests.test_linetoolong.<locals>.TooLongHandler.handleNr� r r r r r � s r )rC r? r0 r r2 r/ r� r� )r r rG r r r r � s
��z'ThreadedNetworkedTests.test_linetoolongc s� d� G � fdd�dt �}tdd�D ]�}d|> � | �d� � ���t | �|��J}| �tjj�� | j|j � W d � n1 s|0 Y W d � n1 s�0 Y W d � q"1 s�0 Y q"d S )Nr c s e Zd Z� fdd�ZdS )zGThreadedNetworkedTests.test_truncated_large_literal.<locals>.BadHandlerc s | � d� � | � d� d S )Nz * OK {%d}ro )rW r ��sizer r rn � s zNThreadedNetworkedTests.test_truncated_large_literal.<locals>.BadHandler.handleNr� r rV r r �
BadHandler� s rX � �@ r zsize=2e)
rC �rangeZsubTestr? r0 r r2 r� r� r� )r rX ZexponentrG r rV r � test_truncated_large_literal� s z3ThreadedNetworkedTests.test_truncated_large_literalc C sZ | � t��<}| j|j� � W d � n1 s.0 Y W d � n1 sL0 Y d S r* )r? rC r� r� )r rG r r r r � s z1ThreadedNetworkedTests.test_simple_with_statementc C s� | � t��b}| j|j� �*}|�dd� | �|jd� W d � n1 sH0 Y | �|j� W d � n1 sr0 Y d S �Nr� r� )r? rC r� r� r� r rI r �r rG r r r r r � s
,z*ThreadedNetworkedTests.test_with_statementc C s� | � t��v}| j|j� �>}|�dd� | �|jd� |�� | �|j� W d � n1 s\0 Y | �|j� W d � n1 s�0 Y d S r] ) r? rC r� r� r� r rI r r r^ r r r r � s *z1ThreadedNetworkedTests.test_with_statement_logoutN)&r8 r9 r: r� r& r� r r2 r� r= r> r r? r@ r rB rM rN r� rC r� r� r� r� r� r� r� r� rS r
r# rT r r r\ r r r r r r r r; F s^
0
r; c @ s e Zd ZeZeZedd� �ZdS )�ThreadedNetworkedTestsSSLc C s� t �t j�}|�t� | �t jd��T | �t��*}| j |j
d|i�}|� � W d � n1 s`0 Y W d � n1 s~0 Y | �t��.}| j d|j
d |d�}|� � W d � n1 s�0 Y d S )Nr( r) r r r2 ) r< r= r* r- r. r� r/ r? rC r� r� r� )r r) rG r� r r r r3 � s
�
�D �z+ThreadedNetworkedTestsSSL.test_ssl_verifiedN) r8 r9 r: r; r� r: r� r r3 r r r r r_ � s r_ Znetworkznetwork resource disabledz'cyrus.andrew.cmu.edu blocks connectionsc @ sB e Zd ZdZdZdZdZejZ dd� Z
dd� Z d d
� Z d d � Z
d
S )�RemoteIMAPTestzcyrus.andrew.cmu.edur5 Z anonymousr� c C sB t �| j��" | �| j| j�| _W d � n1 s40 Y d S r* )r �transient_internet�hostr� �portrG r r r r �setUp� s zRemoteIMAPTest.setUpc C sD | j d ur@t�| j�� | j �� W d � n1 s60 Y d S r* )rG r ra rb r r r r r �tearDown� s
zRemoteIMAPTest.tearDownc C s� t �| j��l | jjD ]}| �|t� q| �d| jj� | �d| jj� | j�| j | j
�}| � |d d� W d � n1 s~0 Y d S )N�
LOGINDISABLEDzAUTH=ANONYMOUSr rp ) r ra rb rG rr �assertIsInstance�strr3 r� �username�passwordr )r �cap�rsr r r �test_logincapa� s z RemoteIMAPTest.test_logincapac C sR t �| j��2 | j�� }d | _| �|d d|� W d � n1 sD0 Y d S �Nr r )r ra rb rG r r �r rl r r r r � s
zRemoteIMAPTest.test_logoutN)r8 r9 r: rb rc ri rj r r2 r� rd re rm r r r r r r` � s r` c s$ e Zd Z� fdd�Zdd� Z� ZS )�RemoteIMAP_STARTTLSTestc sT t � �� t�| j��* | j�� }| �|d d� W d � n1 sF0 Y d S )Nr rp )rE rd r ra rb rG Zstarttlsr ro rJ r r rd � s
z RemoteIMAP_STARTTLSTest.setUpc C s. | j jD ]}| �|t� q| �d| j j� d S )Nrf )rG rr rg rh � assertNotIn)r rk r r r rm � s z&RemoteIMAP_STARTTLSTest.test_logincapa)r8 r9 r: rd rm r} r r rJ r rp � s rp c @ sT e Zd ZdZeZdd� Zdd� Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dS )�RemoteIMAP_SSLTesti� c C s d S r* r r r r r rd s zRemoteIMAP_SSLTest.setUpc C s d S r* r r r r r re s zRemoteIMAP_SSLTest.tearDownc C s( t �t j�}d|_t j|_|�t� |S rD )r< r= r* r, Z CERT_NONEr+ r> r? )r r) r r r �create_ssl_context s
z%RemoteIMAP_SSLTest.create_ssl_contextc C sp z`|j D ]}| �|t� q| �d|j � | �d|j � |�| j| j�}| �|d d� W |� � n
|� � 0 d S )Nrf z
AUTH=PLAINr rp )
rr rg rh rq r3 r� ri rj r r )r rG rk rl r r r �check_logincapa s
z"RemoteIMAP_SSLTest.check_logincapac C sJ t �| j��* | �| j| j�}| �|� W d � n1 s<0 Y d S r* )r ra rb r� rc rt )r �_serverr r r rm # s z!RemoteIMAP_SSLTest.test_logincapac C sZ t �| j��: | �| j| j�}|�� }| �|d d|� W d � n1 sL0 Y d S rn )r ra rb r� rc r r )r ru rl r r r r ( s z RemoteIMAP_SSLTest.test_logoutc C sP t �| j��0 | jt| j| j| jt| �� d� W d � n1 sB0 Y d S )N)r6 r) � r ra rb r0 r r� rc r? rs r r r r �#test_ssl_context_certfile_exclusive. s
�z6RemoteIMAP_SSLTest.test_ssl_context_certfile_exclusivec C sP t �| j��0 | jt| j| j| jt| �� d� W d � n1 sB0 Y d S )N)Zkeyfiler) rv r r r r �"test_ssl_context_keyfile_exclusive4 s
�z5RemoteIMAP_SSLTest.test_ssl_context_keyfile_exclusiveN)r8 r9 r: rc r: r� rd re rs rt rm r rw rx r r r r rr s rr �__main__)6rK r Z test.supportr �
contextlibr r Zos.path�osr� r r r� r- r r r r r r
Zunittestr r r
r r< � ImportError�pathrY �dirname�__file__�curdirr? r. ZTestCaser r& r; r: r� rC r~ r% Z
skipUnlessr' r; r_ Zis_resource_enabled�skipr` rp rr r8 �mainr r r r �<module> sl
@ _ O
$ s
�
�
4