Your IP : 216.73.216.196
o
0�i�� � @ s� 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 dlmZmZmZmZ d dlmZ d dlmZ d dlmZ d dlZd d lmZ d d
lmZmZmZ zd dlZW n eyq dZY nw ej � ej � e �p}ej d �Z!ej � ej � e �p�ej d �Z"G d
d� dej#�Z$er�G dd� de j%�Z&ej'Z'n G 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 � d ej#�Z.e�,ed�G d d � d e.��Z/e�,e�0d �d �e�1d!�G d"d#� d#ej#���Z2e�,ed�e�,e�0d �d �e�1d!�G d$d%� d%e2����Z3e�,ed�e�1d!�G d&d'� d'e2���Z4e5d(k�rAe�6� dS dS ))� )�support)�
socket_helper)�contextmanagerN)�verbose� run_with_tz�run_with_locale� cpython_only)�hashlib_helper)�threading_helper)�warnings_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 �8/opt/alt/python310/lib64/python3.10/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 �
timevalues3 s ��zTestImaplib.timevalues�LC_ALLZde_DEZfr_FRzSTD-1DST,M3.2.0,M11.1.0c C s, d}| � � D ]
}t�|�}| �||� qd S )Nr$ )r& r �Time2Internaldater )r Zexpected�tZinternalr r r �test_Time2Internaldate; 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_resultF s �z8TestImaplib.test_that_Time2Internaldate_returns_a_resultc C s� t � � �"}z|�dtjf� | �d� W n
t jy Y nw W d � n1 s)w Y t�� }| �t �� }t�
� W d � n1 sFw 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_valueM s
��� �
�z)TestImaplib.test_imap4_host_default_valueN) �__name__�
__module__� __qualname__r r r r& r r* r, r8 r r r r r s
r c @ � e Zd Zdd� ZdS )�SecureTCPServerc C s: | j �� \}}t�tj�}|�t� |j|dd�}||fS )NT)Z server_side)r. Zaccept�ssl�
SSLContextZPROTOCOL_TLS_SERVER�load_cert_chain�CERTFILEZ wrap_socket)r Z newsocketZfromaddr�contextZ
connstreamr r r � get_requesta s
zSecureTCPServer.get_requestN)r9 r: r; rC r r r r r= _ � r= c @ � e Zd ZdS )r= N�r9 r: r; r r r r r= l � 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 rK w s
zSimpleIMAPHandler.setupc C s$ t r
td|�� � | j�|� d S )NzSENT: %r)r �print�strip�wfile�write�r �messager r r �_send| s zSimpleIMAPHandler._sendc C s | � |d � d S )N�
)rW rU r r r �
_send_line� � z SimpleIMAPHandler._send_linec C s | � |�d�� d S )N�ASCII)rY �encoderU r r r �_send_textline� s z SimpleIMAPHandler._send_textlinec C s | � d�|||f�� d S )N� )r] �join)r �tag�coderV r r r � _send_tagged� s z SimpleIMAPHandler._send_taggedc C s | � d� d} z| j�d�}|dkrW d S ||7 }W n
ty& Y d S w |�d�r-nq tr8td|�� � | jrQz| j� |� W n t
yO d | _Y nw q|� d�� � }|d }|d }|dd � }t
| d | �r�t| d | �||�}|r�|| _t|� n | �|d
|d � q) Nz* OK IMAP4rev1r � rX zGOT: %rr[ r � Zcmd_�BADz unknown)r] Zrfile�readr2 �endswithr rQ rR � continuation�send�
StopIteration�decode�split�hasattr�getattr�nextrb )r �line�partZ splitliner` �cmd�argsrh r r r �handle� sH
�
�
� ��zSimpleIMAPHandler.handlec C s4 | j rd| j nd}| �d| � | �|dd� d S )Nz
IMAP4rev1 � IMAP4rev1z
* CAPABILITY �OK�CAPABILITY completed)� capabilitiesr] rb )r r` rs Zcapsr r r �cmd_CAPABILITY� s ��z SimpleIMAPHandler.cmd_CAPABILITYc C s$ d | j _| �d� | �|dd� d S )Nz"* BYE IMAP4ref1 Server logging outrv zLOGOUT completed)rL rN r] rb �r r` rs r r r �
cmd_LOGOUT� �
z SimpleIMAPHandler.cmd_LOGOUTc C s |d | j _| �|dd� d S )Nr rv zLOGIN completed)rL rN rb rz r r r � cmd_LOGIN� s zSimpleIMAPHandler.cmd_LOGINc C s$ d| j _| �d� | �|dd� d S )NT�
* 2 EXISTSrv � [READ-WRITE] SELECT completed.)rL rM rY rb rz r r r �
cmd_SELECT� r| z SimpleIMAPHandler.cmd_SELECTc C s4 | j jrd| j _| �|dd� d S | �|dd� d S )NFrv z*Returned to authenticated state. (Success)re zNo mailbox selected)rL rM rb rz r r r � cmd_UNSELECT� s z SimpleIMAPHandler.cmd_UNSELECT)r9 r: r; r �LOOPBACK_TIMEOUT�timeoutrh rx rK rW rY r] rb rt ry r{ r} r� r� �
__classcell__r r rO r rH r s *rH 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
� � |r<| j
| jj� | _| j| jfS ) a
Sets up imap_handler for tests. imap_handler should inherit from either:
- SimpleIMAPHandler - for testing IMAP commands,
- socketserver.StreamRequestHandler - if raw access to stream is needed.
Returns (client, server).
c @ r< )z/NewIMAPTestsMixin._setup.<locals>.TestTCPServerc S � | � |� | �� � )zP
End request and raise the error if one occurs.
�Z
close_request� server_close�r ZrequestZclient_addressr r r � handle_error� s
z<NewIMAPTestsMixin._setup.<locals>.TestTCPServer.handle_errorN�r9 r: r; r� r r r r �
TestTCPServer� � r� r z-server�
poll_interval�{�G�z�?��name�target�kwargsT)� server_classZ
addCleanup�_cleanupr �HOSTrL � 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 dur| j jdkr| j �� | j�� | j�� t�| j� d| _dS )z�
Cleans up the test server. This method should not be called manually,
it is added to the cleanup queue in the _setup method already.
N�LOGOUT)r� �state�shutdownrL 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 @ r< )zONewIMAPTestsMixin.test_EOF_without_complete_welcome_message.<locals>.EOFHandlerc S � | j �d� d S �Ns * OK�rS rT r r r r rt s zVNewIMAPTestsMixin.test_EOF_without_complete_welcome_message.<locals>.EOFHandler.handleN�r9 r: r; rt r r r r �
EOFHandler r� r� F�r/ ) � socketserver�StreamRequestHandlerr� r1 r r3 �abortr� r� )r r� �_rL 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 @ r< )zBNewIMAPTestsMixin.test_line_termination.<locals>.BadNewlineHandlerc S � | � d� | �|dd� d S �Ns * CAPABILITY IMAP4rev1 AUTH
rv rw �rW rb rz r r r ry �
zQNewIMAPTestsMixin.test_line_termination.<locals>.BadNewlineHandler.cmd_CAPABILITYN�r9 r: r; ry r r r r �BadNewlineHandler r� r� Fr� )rH r� r1 r r3 r� r� r� )r r� r� rL r r r �test_line_termination s
�z'NewIMAPTestsMixin.test_line_terminationc C st G dd� dt �}| �|�\}}| �|j� | �tjjd��
|�d� W d � n1 s-w Y | �|j� d S )Nc @ � e Zd ZdZdS )zMNewIMAPTestsMixin.test_enable_raises_error_if_not_AUTH.<locals>.EnableHandler�AUTH ENABLE UTF8=ACCEPTN�r9 r: r; rx r r r r �
EnableHandler � r� zENABLE.*NONAUTH�foo) rH r� � assertFalse� utf8_enabled�assertRaisesRegexr r3 r0 �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 sN | � t�\}}| �tjjd�� |�d� W d � d S 1 s w Y d S )N�does not support ENABLEr� )r� rH r� r r3 r0 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 sj | � t�\}}|�dd�\}}| �|d� | �tjjd�� |�d� W d � d S 1 s.w Y d S )N�user�passrv r� � UTF8=ACCEPT) r� rH �loginr r� r r3 r0 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 � | � |dd� d S �Nrv zENABLE successful�rb rz r r r �
cmd_ENABLE1 rZ zSNewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_ENABLEc s �( � | � d� d V | j_| �|dd� d S �N�+rv zFAKEAUTH successful�r] rL �responserb rz r r r �cmd_AUTHENTICATE3 � �
zYNewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_AUTHENTICATEc s r� �Nr� rv Zokayr� rz r r r �
cmd_APPEND7 r� zSNewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_APPENDN)r9 r: r; rx r� r� r� r r r r �UTF8AppendServer/ s
r� �ascii�MYAUTHc S � dS �Ns faker ��xr r r �<lambda>= � z@NewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.<lambda>rv �
ZmFrZQ==
r� �utf-8� Subject: üñí©öðé� UTF8 (%s)
) rH r� r � _encoding� authenticater� r� �appendr\ ) r r� r� rL ra 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 � d S 1 sLw Y d S ) Nc @ � e Zd ZdZdd� Zdd� ZdS )zPNewIMAPTestsMixin.test_search_disallows_charset_in_utf8_mode.<locals>.UTF8Serverr� c S r� r� r� rz r r r r� L rZ z[NewIMAPTestsMixin.test_search_disallows_charset_in_utf8_mode.<locals>.UTF8Server.cmd_ENABLEc s r� r� r� rz r r r r� N r� zaNewIMAPTestsMixin.test_search_disallows_charset_in_utf8_mode.<locals>.UTF8Server.cmd_AUTHENTICATEN�r9 r: r; rx r� r� r r r r �
UTF8ServerJ s r� r� c S r� r� r r� r r r r� S r� zNNewIMAPTestsMixin.test_search_disallows_charset_in_utf8_mode.<locals>.<lambda>rv r� z
charset.*UTF8r� �bar) rH r� r� r r� �
assertTruer� r� r r3 r0 �search)r r� r� r� r� r r r �*test_search_disallows_charset_in_utf8_modeI s "�z<NewIMAPTestsMixin.test_search_disallows_charset_in_utf8_modec C �d G dd� dt �}| �|�\}}| �tjjd�� |�ddd� � W d � d S 1 s+w Y d S )Nc @ r< )z6NewIMAPTestsMixin.test_bad_auth_name.<locals>.MyServerc S � | � |dd�|d �� d S �N�NOz#unrecognized authentication type {}r �rb �formatrz r r r r� ] s �zGNewIMAPTestsMixin.test_bad_auth_name.<locals>.MyServer.cmd_AUTHENTICATEN�r9 r: r; r� r r r r �MyServer\ r� r� z'unrecognized authentication type METHOD�METHODc S r� �Nr r r r r r r� c r� z6NewIMAPTestsMixin.test_bad_auth_name.<locals>.<lambda>�rH r� r� r r3 r0 r� �r r� r� r� r r r �test_bad_auth_name[ s
�"�z$NewIMAPTestsMixin.test_bad_auth_namec C r� )Nc @ r< )z?NewIMAPTestsMixin.test_invalid_authentication.<locals>.MyServerc s �& � | � d� d V | _| �|dd� d S �Nr� r� z [AUTHENTICATIONFAILED] invalid�r] r� rb rz r r r r� g � �
zPNewIMAPTestsMixin.test_invalid_authentication.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� f r� r� z \[AUTHENTICATIONFAILED\] invalidr� c S r� r� r r� r r r r� n r� z?NewIMAPTestsMixin.test_invalid_authentication.<locals>.<lambda>r� r� r r r �test_invalid_authenticatione s
�"�z-NewIMAPTestsMixin.test_invalid_authenticationc C �P G dd� dt �}| �|�\}}|�ddd� �\}}| �|d� | �|jd� d S )Nc @ r< )zCNewIMAPTestsMixin.test_valid_authentication_bytes.<locals>.MyServerc s r� r� r� rz r r r r� r r� zTNewIMAPTestsMixin.test_valid_authentication_bytes.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� q r� r� r� c S r� r� r r� r r r r� w r� zCNewIMAPTestsMixin.test_valid_authentication_bytes.<locals>.<lambda>rv r� �rH r� r� r r� �r r� r� rL ra r� r r r � test_valid_authentication_bytesp �
z1NewIMAPTestsMixin.test_valid_authentication_bytesc C r )Nc @ r< )zHNewIMAPTestsMixin.test_valid_authentication_plain_text.<locals>.MyServerc s r� r� r� rz r r r r� } r� zYNewIMAPTestsMixin.test_valid_authentication_plain_text.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� | r� r� r� c S r� �NZfaker r� r r r r� � r� zHNewIMAPTestsMixin.test_valid_authentication_plain_text.<locals>.<lambda>rv r� r r r r r �$test_valid_authentication_plain_text{ r z6NewIMAPTestsMixin.test_valid_authentication_plain_text�md5�Zopensslc C �N G dd� dt �}| �|�\}}| �d|jv � |�dd�\}}| �|d� d S )Nc @ � e Zd ZdZdd� ZdS )z@NewIMAPTestsMixin.test_login_cram_md5_bytes.<locals>.AuthHandler�LOGINDISABLED AUTH=CRAM-MD5c s �> � | � d� d V }|dkr| �|dd� d S | �|dd� d S �Nz:+ PDE4OTYuNjk3MTcwOTUyQHBvc3RvZmZpY2UucmVzdG9uLm1jaS5uZXQ=s2 dGltIGYxY2E2YmU0NjRiOWVmYTFjY2E2ZmZkNmNmMmQ5ZjMy
rv zCRAM-MD5 successfulr� z No access�r] rb �r r` rs �rr r r r� � � �
zQNewIMAPTestsMixin.test_login_cram_md5_bytes.<locals>.AuthHandler.cmd_AUTHENTICATEN�r9 r: r; rx r� r r r r � AuthHandler� � r �
AUTH=CRAM-MD5�tim� tanstaaftanstaafrv �rH r� r� rx �login_cram_md5r �r r r� r� �retr r r �test_login_cram_md5_bytes� �
z+NewIMAPTestsMixin.test_login_cram_md5_bytesc C r )Nc @ r
)zENewIMAPTestsMixin.test_login_cram_md5_plain_text.<locals>.AuthHandlerr c s r r r r r r r r� � r zVNewIMAPTestsMixin.test_login_cram_md5_plain_text.<locals>.AuthHandler.cmd_AUTHENTICATENr r r r r r � r r r r �tanstaaftanstaafrv r r r r r � test_login_cram_md5_plain_text� r z0NewIMAPTestsMixin.test_login_cram_md5_plain_textc C r� )Nc @ r< )z?NewIMAPTestsMixin.test_aborted_authentication.<locals>.MyServerc s �B � | � d� d V | _| jdkr| �|dd� d S | �|dd� d S �Nr� s *
r� z [AUTHENTICATIONFAILED] abortedrv zMYAUTH successfulr rz r r r r� � s �
�zPNewIMAPTestsMixin.test_aborted_authentication.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� � r� r� z \[AUTHENTICATIONFAILED\] abortedr� c 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 sd G dd� dt �}| j|dd�\}}| �tjjd�� | j|j� W d � d S 1 s+w Y d S )Nc @ r< )z:NewIMAPTestsMixin.test_linetoolong.<locals>.TooLongHandlerc S r� )Ns * OK xxxxxxxxxxx
r� r r r r rt � � zANewIMAPTestsMixin.test_linetoolong.<locals>.TooLongHandler.handleNr� r r r r �TooLongHandler� r� r) Fr� zgot more than 10 bytes)rH r� r� r r3 r0 r� r� )r r) r� rL r r r �test_linetoolong� s
�"�z"NewIMAPTestsMixin.test_linetoolongc C sD | j tdd�\}}| j|j� � W d � d S 1 sw Y d S )NFr� )r� rH r� r� )r r� rL 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 � d S 1 sOw Y d S )Nr � localhost�r� r ) r� rH r� r� r Zsockr� r� r r� r1 �
ValueError)r r� rL �addrr� r r r �test_imaplib_timeout_test� s
"�z+NewIMAPTestsMixin.test_imaplib_timeout_testc C sh G dd� dt �}| �|�\}}|jd }| �t�� | jd|dd�}W d � d S 1 s-w Y d S )Nc @ r< )zQNewIMAPTestsMixin.test_imaplib_timeout_functionality_test.<locals>.TimeoutHandlerc S s t �d� t�| � d S r� )r �sleeprH rt r r r r rt � �
zXNewIMAPTestsMixin.test_imaplib_timeout_functionality_test.<locals>.TimeoutHandler.handleNr� r r r r �TimeoutHandler� r� r3 r r, g����MbP?r- )rH r� r� r1 � TimeoutErrorr� )r r3 r� rL 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 s'w Y | �|j� d S �NFr� r� r� )r� rH r� r� r� r rN � assertIsNone�r r� rL �imapr r r �test_with_statement� s �z%NewIMAPTestsMixin.test_with_statementc C s| | j tdd�\}}| j|j� � }|�dd� | �|jd� |�� | �|j� W d � n1 s1w Y | �|j� d S r6 ) r� rH r� r� r� r rN �logoutr7 r8 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� rv r � LOGIN completed�AUTH)r� rH 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� rv r r= �BYEs IMAP4ref1 Server logging outr� )r� rH 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 @ r< )z,NewIMAPTestsMixin.test_lsub.<locals>.LsubCmdc S s | � d� | �|dd�S )Nz* LSUB () "." directoryArv zLSUB completedr rz r r r �cmd_LSUB r2 z5NewIMAPTestsMixin.test_lsub.<locals>.LsubCmd.cmd_LSUBN)r9 r: r; rB r r r r �LsubCmd r� rC r� r� rv r s () "." directoryA)rH r� r� Zlsubr )r rC 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� rv r � 2s* Returned to authenticated state. (Success)r> )r� rH r� �selectr Zunselectr� r� r r r �
test_unselect s z NewIMAPTestsMixin.test_unselect)T)"r9 r: r; r� r� r� r� r� r� r� r� r� r� r� r r r r �requires_hashdigestr r" r& r �patchr* r+ r0 r5 r: r< r? rA rD rG r r r r r� � s>
!
r� c @ s e Zd ZejZejZdS )� NewIMAPTestsN) r9 r: r; r r3 r� r� � TCPServerr� r r r r rJ s
rJ 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�� | � t �\}}| j
|jd|i�}|�� W d � d S 1 sAw 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� rH r� r� r� �r rN r� rL 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 �rN )
r> r? rO rR rS r� rH r� r� r� rU r r r �test_ssl_verified6 s
� z!NewIMAPSSLTests.test_ssl_verifiedc
C s� t �dtf��I tj�| jd��( tj�| jd�� | jddtd� W d � n1 s,w Y W d � n1 s;w Y W d � d S W d � d S 1 sSw Y d S )Nr- �openZ_connectr, � )�certfile)r Zcheck_warnings�DeprecationWarningr rI �objectr� rA r r r r �test_certfile_arg_warnA s �� ��"�z&NewIMAPSSLTests.test_certfile_arg_warnN) r9 r: r; � IMAP4_SSLr� r= r� rV rX r r^ r r r r rL # s rL c @ sz e Zd ZejZejZdd� Z dd� Z
e dd� �Z e dd� �Z
ejd d
� �Zejd d � �Zejd
d� �Zejdd� �ZG dd� de�Zejdd� �Zejdd� �Zejdd� �Zejdd� �Zejdd � �Zejd d � �Zejd d � �Z ejd!d"� �Z eje j d#d$d%�d&d'� ��Z ejd(d)� �Z!d*d+� Z"d,d-� Z#ejd.d/� �Z$ejd0d1� �Z%ejd2d3� �Z&eje'e(�)d$d4�d5d6� ���Z*d7S )8�ThreadedNetworkedTestsc C s� G dd� d| j �}trtd� |||�}| �|j|j�� � tr5td� td|� td| j � td|j� tj d| j |j
d d
id �}d |_ |� � trPtd
� ||fS )Nc @ r< )z4ThreadedNetworkedTests.make_server.<locals>.MyServerc S r� r+ r� r� r r r r� O s
zAThreadedNetworkedTests.make_server.<locals>.MyServer.handle_errorNr� r r r r r� N r� r� zcreating serverzserver createdzADDR =zCLASS =zHDLR =z
%s servingr� r� r� Tzserver running)
r� r rQ r r� r. Z getsocknameZRequestHandlerClassr� r� r� r� r� )r r/ �hdlrr� rL r) r r r � make_serverL s(
�z"ThreadedNetworkedTests.make_serverc C s8 t rtd� |�� |�� |�� t rtd� d S d S )Nzwaiting for serverZdone)r rQ r� r� r_ )r rL r� r r r � reap_serverl s �z"ThreadedNetworkedTests.reap_serverc
c s@ � | � tjdf|�\}}z |V W | �||� d S | �||� w )Nr )rb r r� rc )r ra rL r� r r r �
reaped_serveru s
� z$ThreadedNetworkedTests.reaped_serverc c s` � | � |�� }| j|j� }z ||fV W |�� n|�� w W d � d S 1 s)w Y d S r+ )rd r� r� r; )r ra rL r� r r r � reaped_pair} s � "�z"ThreadedNetworkedTests.reaped_pairc C sD | � t��}| j|j� }|�� W d � d S 1 sw Y d S r+ )rd rH r� r� r� �r rL r� r r r � test_connect� s
"�z#ThreadedNetworkedTests.test_connectc C s� G dd� dt �}| �|��J\}}|�ddd� �\}}| �|d� | �|jd� |�d� |�d d
d �\}\}| �d |� |�d� |�d
�\}\}| �d |� W d � d S 1 sZw 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)�flagsrJ rt r rO r r rt � r2 zLThreadedNetworkedTests.test_bracket_flags.<locals>.BracketFlagHandler.handlec s r� r� r� rz r r r r� � r� 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� r~ s
* 0 RECENTz-* OK [PERMANENTFLAGS %s \*)] Flags permitted.rv r )r_ rh rY r\ rb )r r` rs Zflag_msg�msgr r r r� � 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 ) Nrd �(�)z (FLAGS (%s))ri z
* %s FETCH %sr r� rv zSTORE completed.)rR rl rh �extendr_ rY r\ rb )r r` rs Z new_flagsZ flags_msgrj r r r � cmd_STORE� s zOThreadedNetworkedTests.test_bracket_flags.<locals>.BracketFlagHandler.cmd_STORE)r9 r: r; rt r� r� rn r� r r rO r �BracketFlagHandler� s
ro r� c S r� r� r r� r r r r� � r� z;ThreadedNetworkedTests.test_bracket_flags.<locals>.<lambda>rv r� �test� 1z+FLAGSz[test]s [test]ZPERMANENTFLAGS)rH re r� r r� rF Zstorer4 )r ro rL r� ra 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 � d S 1 s(w Y d S )Nc @ r< )z9ThreadedNetworkedTests.test_issue5949.<locals>.EOFHandlerc S r� r� r� r r r r rt � r( z@ThreadedNetworkedTests.test_issue5949.<locals>.EOFHandler.handleNr� r r r r r� � r� r� ) r� r� rd r1 r r3 r� r� r� )r r� rL r r r �test_issue5949� s
�
�"�z%ThreadedNetworkedTests.test_issue5949c C �\ G dd� dt �}| �|��}| jtjj| jg|j�R � W d � d S 1 s'w Y d S )Nc @ r< )zGThreadedNetworkedTests.test_line_termination.<locals>.BadNewlineHandlerc S r� r� r� rz r r r ry � r� zVThreadedNetworkedTests.test_line_termination.<locals>.BadNewlineHandler.cmd_CAPABILITYNr� r r r r r� � rD r� )rH rd r1 r r3 r� r� r� )r r� rL r r r r� � s
�
�"�z,ThreadedNetworkedTests.test_line_terminationc @ r� )z!ThreadedNetworkedTests.UTF8Serverr� c C r� r� r� rz r r r r� � rZ z,ThreadedNetworkedTests.UTF8Server.cmd_ENABLEc c r� r� r� rz r r r r� � r� z2ThreadedNetworkedTests.UTF8Server.cmd_AUTHENTICATENr� r r r r r� � s r� c C sb | � | j��!\}}| �|j� | �tjj|jd� | �|j� W d � d S 1 s*w Y d S )Nr� ) re r� r� r� r1 r r3 r0 r� rf r r r r� � s
"�z;ThreadedNetworkedTests.test_enable_raises_error_if_not_AUTHc C sZ G dd� d| j �}| �|��\}}| �tjj|jd� W d � d S 1 s&w Y d S )Nc @ r� )zXThreadedNetworkedTests.test_enable_raises_error_if_no_capability.<locals>.NoEnableServerr> Nr� r r r r �NoEnableServer� r� ru r� )r� re r1 r r3 r0 r� )r ru rL r� r r r r� � s "�z@ThreadedNetworkedTests.test_enable_raises_error_if_no_capabilityc C s� G dd� dt �}| �tjj��; | �|�� \}}|�dd�\}}| �|d� |�d� W d � n1 s5w Y W d � d S W d � d S 1 sMw Y d S )Nc @ rE )z\ThreadedNetworkedTests.test_enable_UTF8_raises_error_if_not_supported.<locals>.NonUTF8ServerNrF r r r r �
NonUTF8Server� rG rv r� r� rv r� ) rH r1 r r3 r0 re r� r r� )r rv rL 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 � d S 1 sow Y d S )
Nc @ r< )zMThreadedNetworkedTests.test_enable_UTF8_True_append.<locals>.UTF8AppendServerc s r� r� r� rz r r r r� r� zXThreadedNetworkedTests.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_APPENDN)r9 r: r; r� r r r r r� � r� r� r� r� c S r� r� r r� r r r r� r� zEThreadedNetworkedTests.test_enable_UTF8_True_append.<locals>.<lambda>rv r� r� r� r� r� ) r� re r r� r� r� r� r� r\ ) r r� rL r� ra r� r� r� r� r r r r� � s* � � �"�z3ThreadedNetworkedTests.test_enable_UTF8_True_appendc C s� | � | j��9\}}|�ddd� �\}}| �|d� |�d�\}}| �|d� | �|j� | �tj j
|j dd� W d � d S 1 sBw Y d S )Nr� c S r� r� r r� r r r r� r� zSThreadedNetworkedTests.test_search_disallows_charset_in_utf8_mode.<locals>.<lambda>rv r� r� r� ) re r� r� r r� r� r� r1 r r3 r0 r� )r rL r� r� r� r r r r� s "�zAThreadedNetworkedTests.test_search_disallows_charset_in_utf8_modec C s� G dd� dt �}| �|��2\}}| �tjj�� |�ddd� � W d � n1 s*w Y W d � d S W d � d S 1 sBw Y d S )Nc @ r< )z;ThreadedNetworkedTests.test_bad_auth_name.<locals>.MyServerc S r� r� r� rz r r r r� ) s
�zLThreadedNetworkedTests.test_bad_auth_name.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� ' rD r� r� c S r� r� r r r r r r� / r� z;ThreadedNetworkedTests.test_bad_auth_name.<locals>.<lambda>�rH re r1 r r3 r0 r� )r r� rL r� r r r r� $ s ��"�z)ThreadedNetworkedTests.test_bad_auth_namec C � G dd� dt �}| �|��4\}}| �tjj�� |�ddd� �\}}W d � n1 s,w Y W d � d S W d � d S 1 sDw Y d S )Nc @ r< )zDThreadedNetworkedTests.test_invalid_authentication.<locals>.MyServerc s r� r� r rz r r r r� 6 r zUThreadedNetworkedTests.test_invalid_authentication.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� 4 rD r� r� c S r� r� r r� r r r r� = r� zDThreadedNetworkedTests.test_invalid_authentication.<locals>.<lambda>rw �r r� rL r� ra r� r r r r 1 s ��"�z2ThreadedNetworkedTests.test_invalid_authenticationc C s� G dd� dt �}| �|��!\}}|�ddd� �\}}| �|d� | �|jd� W d � n1 s1w Y | �|��"\}}|�ddd� �\}}| �|d� | �|jd� W d � d S 1 s`w Y d S ) Nc @ r< )zBThreadedNetworkedTests.test_valid_authentication.<locals>.MyServerc s r� r� r� rz r r r r� D r� zSThreadedNetworkedTests.test_valid_authentication.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� B rD r� r� c S r� r� r r� r r r r� J r� zBThreadedNetworkedTests.test_valid_authentication.<locals>.<lambda>rv r� c S r� r r r� r r r r� P r� )rH re r� r r� ry 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 s0w Y | �|��!\}}| �d|jv � |�dd�\}}| �|d� W d � d S 1 s^w Y d S )Nc @ r
)z?ThreadedNetworkedTests.test_login_cram_md5.<locals>.AuthHandlerr c s r r r r r r r r� ] r zPThreadedNetworkedTests.test_login_cram_md5.<locals>.AuthHandler.cmd_AUTHENTICATENr r r r r r Y s r r r r! rv r )rH re r� rx r r )r r rL r� r r� r r r �test_login_cram_md5U s �"�z*ThreadedNetworkedTests.test_login_cram_md5c C rx )Nc @ r< )zDThreadedNetworkedTests.test_aborted_authentication.<locals>.MyServerc s r# r$ r rz r r r r� w s �
zUThreadedNetworkedTests.test_aborted_authentication.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� u rD r� r� c S r% r+ r r� r r r r� � r� zDThreadedNetworkedTests.test_aborted_authentication.<locals>.<lambda>rw ry r r r r& r s ��"�z2ThreadedNetworkedTests.test_aborted_authenticationc C rt )Nc @ r< )z?ThreadedNetworkedTests.test_linetoolong.<locals>.TooLongHandlerc S s | j �dtjd d � d S )Ns * OK � xrX )rS rT r Z_MAXLINEr r r r rt � s zFThreadedNetworkedTests.test_linetoolong.<locals>.TooLongHandler.handleNr� r r r r r) � r� r) )rH rd r1 r r3 r0 r� r� )r r) rL r r r r* � s
�
�"�z'ThreadedNetworkedTests.test_linetoolongc s� d� G � fdd�dt �}tdd�D ]P}d|> � | �d� � ���: | �|��%}| �tjj�� | j|j � W d � n1 s>w Y W d � n1 sMw Y W d � n1 s\w Y qd S )Nr c s e Zd Z� fdd�ZdS )zGThreadedNetworkedTests.test_truncated_large_literal.<locals>.BadHandlerc s | � d� � | � d� d S )Nz * OK {%d}ru )r] r ��sizer r rt � s zNThreadedNetworkedTests.test_truncated_large_literal.<locals>.BadHandler.handleNr� r r} r r �
BadHandler� s r � �@ r zsize=2e)
rH �rangeZsubTestrd r1 r r3 r� r� r� )r r ZexponentrL r r} r � test_truncated_large_literal� s �� �� ���z3ThreadedNetworkedTests.test_truncated_large_literalc C sl | � t��'}| j|j� � W d � n1 sw Y W d � d S W d � d S 1 s/w Y d S r+ )rd rH r� r� )r rL r r r r+ � s ��"�z1ThreadedNetworkedTests.test_simple_with_statementc C s� | � t��2}| j|j� �}|�dd� | �|jd� W d � n1 s$w Y | �|j� W d � d S 1 s:w Y d S �Nr� r� )rd rH r� r� r� r rN r7 �r rL r9 r r r r: � s �"�z*ThreadedNetworkedTests.test_with_statementc C s� | � t��<}| j|j� � }|�dd� | �|jd� |�� | �|j� W d � n1 s.w Y | �|j� W d � d S 1 sDw Y d S r� ) rd rH r� r� r� r rN r; r7 r� r r r r< � s �"�z1ThreadedNetworkedTests.test_with_statement_logoutz Won't work if __debug__ is Falsec
C s� ddgi}| � t��H}| j|j� �)}tj�|d��}|�|� |�d� W d � n1 s.w Y W d � n1 s=w Y W d � d S W d � d S 1 sUw Y d S )Nz
READ-WRITErc Z_mesgz)untagged responses dump:READ-WRITE: [b'']) rd rH r� r� r rI r] Z_dump_urZassert_called_with)r Zuntagged_resp_dictrL r9 Z mock_mesgr r r � test_dump_ur� s
� �� ��"�z#ThreadedNetworkedTests.test_dump_urN)+r9 r: r; r� rK r� r r3 r� rb rc r rd re r
� reap_threadsrg rr rs r� rH r� r� r� r� r� r� r� r rz r rH r{ r& r* r� r+ r: r< r �unittest�
skipUnlessr� r r r r r` H sf
0
r` c @ s" e Zd ZeZeZejdd� �Z dS )�ThreadedNetworkedTestsSSLc C s� t �t j�}|�t� | �t jd��* | �t��}| j |j
d|i�}|� � W d � n1 s0w Y W d � n1 s?w Y | �t��}| j d|j
d |d�}|� � W d � d S 1 sdw Y d S )NrM rN r, r rW ) r> r? rO rR rS r� rT rd rH r� r� r� )r rN rL r� r r r rX � s(
�
�
�� � �
"�z+ThreadedNetworkedTestsSSL.test_ssl_verifiedN)
r9 r: r; r= r� r_ r� r
r� rX 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.edurZ Z anonymousr� c C sD t �| j�� | �| j| j�| _W d � d S 1 sw Y d S r+ )r �transient_internet�hostr� �portrL r r r r �setUp� s "�zRemoteIMAPTest.setUpc C sJ | j d ur#t�| j�� | j �� W d � d S 1 s w Y d S d S r+ )rL r r� r� r; r r r r �tearDown� s
"��zRemoteIMAPTest.tearDownc C s� t �| j��7 | jjD ]}| �|t� q | �d| jj� | �d| jj� | j�| j | j
�}| � |d d� W d � d S 1 s@w Y d S )N�
LOGINDISABLEDzAUTH=ANONYMOUSr rv ) r r� r� rL rx �assertIsInstance�strr4 r� �username�passwordr )r �cap�rsr r r �test_logincapa� s "�z RemoteIMAPTest.test_logincapac C sT t �| j�� | j�� }d | _| �|d d|� W d � d S 1 s#w Y d S �Nr r@ )r r� r� rL r; r �r r� r r r rA � s
"�zRemoteIMAPTest.test_logoutN)r9 r: r; r� r� r� r� r r3 r� r� r� r� rA r r r r r� � s r� c s$ e Zd Z� fdd�Zdd� Z� ZS )�RemoteIMAP_STARTTLSTestc sV t � �� t�| j�� | j�� }| �|d d� W d � d S 1 s$w Y d S )Nr rv )rJ r� r r� r� rL Zstarttlsr r� rO r r r� s
"�z RemoteIMAP_STARTTLSTest.setUpc C s. | j jD ]}| �|t� q| �d| j j� d S )Nr� )rL rx r� r� � assertNotIn)r r� r r r r� s z&RemoteIMAP_STARTTLSTest.test_logincapa)r9 r: r; r� r� r� r r rO r r� s r� c @ sT e Zd ZdZeZdd� Zdd� Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dS )�RemoteIMAP_SSLTesti� c C r% r+ r r r r r r� � zRemoteIMAP_SSLTest.setUpc C r% r+ r r r r r r� r� zRemoteIMAP_SSLTest.tearDownc C s( t �t j�}d|_t j|_|�t� |S rI )r> r? rO rQ Z CERT_NONErP r@ rA )r rN r r r �create_ssl_context! s
z%RemoteIMAP_SSLTest.create_ssl_contextc C sn z1|j D ]}| �|t� q| �d|j � | �d|j � |�| j| j�}| �|d d� W |� � d S |� � w )Nr� z
AUTH=PLAINr rv )
rx r� r� r� r4 r� r� r� r r; )r rL r� r� r r r �check_logincapa( s
z"RemoteIMAP_SSLTest.check_logincapac C sL t �| j�� | �| j| j�}| �|� W d � d S 1 s w Y d S r+ )r r� r� r� r� r� )r �_serverr r r r� 3 s "�z!RemoteIMAP_SSLTest.test_logincapac C s\ t �| j�� | �| j| j�}|�� }| �|d d|� W d � d S 1 s'w Y d S r� )r r� r� r� r� r; r )r r� r� r r r rA 8 s
"�z RemoteIMAP_SSLTest.test_logoutc C �R t �| j�� | jt| j| j| jt| �� d� W d � d S 1 s"w Y d S )N)r[ rN � r r� r� r1 r. r� r� rA r� r r r r �#test_ssl_context_certfile_exclusive> � �"�z6RemoteIMAP_SSLTest.test_ssl_context_certfile_exclusivec C r� )N)ZkeyfilerN r� r r r r �"test_ssl_context_keyfile_exclusiveD r� z5RemoteIMAP_SSLTest.test_ssl_context_keyfile_exclusiveN)r9 r: r; r� r_ r� r� r� r� r� r� rA r� r� r r r r r� s r� �__main__)7rp r Z test.supportr �
contextlibr r Zos.path�osr� r r r� r. r r r r r r
r r� r r
r r r> � ImportError�pathr_ �dirname�__file__�curdirrA rS ZTestCaser rK r= r_ r� rH r� rJ r� rL r` r� Zis_resource_enabled�skipr� r� r� r9 �mainr r r r �<module> sx � @ _ O
$
�
�
4 �