Your IP : 216.73.216.213
U
i�f͛ � @ sb d dl mZ d dlmZ d dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m
Z
mZmZmZmZmZmZ d dlZd dlmZ d dlmZmZmZ z d dlZW n ek
r� dZY nX ej� ej� e �p�ej d�Z ej� ej� e �p�ej d�Z!G d d
� d
ej"�Z#e�r,G d d � d ej$�Z%ej&Z&nG d
d � d �Z%dZ&G dd� dej'�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� de-��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�r^e�5� dS )%� )�support)�contextmanagerN)� reap_threads�verbose�transient_internet� run_with_tz�run_with_locale� cpython_only�requires_hashdigest)�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/python38/lib64/python3.8/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 �
timevalues0 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_Time2Internaldate8 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_resultC s z8TestImaplib.test_that_Time2Internaldate_returns_a_resultc C s� t � � �<}z |�dtjf� | �d� W n t jk
r> Y nX W 5 Q R X t�� }| �t ��}t�
� W 5 Q R X | � |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_valueJ 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; i s c sh e Zd ZdZdZdZ� fdd�Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Z� ZS )�SimpleIMAPHandlerr Nr, c s t � �� d | j_d S r* )�super�setup�server�loggedr �� __class__r r rE t s
zSimpleIMAPHandler.setupc C s$ t rtd|�� � | j�|� d S )NzSENT: %r)r �print�strip�wfile�write�r �messager r r �_sendx s zSimpleIMAPHandler._sendc C s | � |d � d S )N�
)rP rN r r r �
_send_line} s z SimpleIMAPHandler._send_linec C s | � |�d�� d S )N�ASCII)rR �encoderN r r r �_send_textline� s z SimpleIMAPHandler._send_textlinec C s | � d�|||f�� d S )N� )rU �join)r �tag�coderO r r r � _send_tagged� s z SimpleIMAPHandler._send_taggedc C s | � d� d}z&| j�d�}|dkr*W d S ||7 }W n tk
rJ Y d S X |�d�rqZqtrntd|�� � | jr�z| j� |� W q
t
k
r� d | _Y q
X q
|� d�� � }|d }|d }|dd � }t
| d | ��rt| d | �||�}|�r|| _t|� q
| �|d
|d � q
d S ) Nz* OK IMAP4rev1� r rQ zGOT: %rrS r � Zcmd_ZBADz unknown)rU Zrfile�readr1 �endswithr rJ rK � continuation�send�
StopIteration�decode�split�hasattr�getattr�nextrZ )r �line�partZ splitlinerX �cmd�argsr_ r r r �handle� s:
zSimpleIMAPHandler.handlec C s4 | j rd| j nd}| �d| � | �|dd� d S )Nz
IMAP4rev1 Z IMAP4rev1z
* CAPABILITY �OK�CAPABILITY completed)� capabilitiesrU rZ )r rX rj Zcapsr r r �cmd_CAPABILITY� s � �z SimpleIMAPHandler.cmd_CAPABILITYc C s$ d | j _| �d� | �|dd� d S )Nz"* BYE IMAP4ref1 Server logging outrl zLOGOUT completed)rF rG rU rZ �r rX rj r r r �
cmd_LOGOUT� s
z SimpleIMAPHandler.cmd_LOGOUTc C s |d | j _| �|dd� d S )Nr rl zLOGIN completed)rF rG rZ rp r r r � cmd_LOGIN� s zSimpleIMAPHandler.cmd_LOGIN)r8 r9 r: Ztimeoutr_ rn rE rP rR rU rZ rk ro rq rr �
__classcell__r r rH r rC o s *rC c @ s� e Zd ZdZd4dd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zed �d d � �Zed �d d!� �Zd"d#� Ze�d$d%�d&d'� �Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Z d2d3� Z dS )5�NewIMAPTestsMixinNTc C s� G dd� d| j �}| �| j� | � tjdf|�| _tj| jd | jj ddid�| _
d| j
_ | j
� � |rx| 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 @ s e Zd Zdd� ZdS )z/NewIMAPTestsMixin._setup.<locals>.TestTCPServerc S s | � |� | �� � dS )zP
End request and raise the error if one occurs.
N�Z
close_request� server_close�r ZrequestZclient_addressr r r � handle_error� s
z<NewIMAPTestsMixin._setup.<locals>.TestTCPServer.handle_errorN�r8 r9 r: rx r r r r �
TestTCPServer� s rz r z-server�
poll_interval�{�G�z�?��name�target�kwargsT)� server_classZ
addCleanup�_cleanupr �HOSTrF � threading�ThreadZ_testMethodName�
serve_forever�thread�daemon�start�
imap_class�server_address�client)r Z imap_handlerr. rz r r r �_setup� s �
zNewIMAPTestsMixin._setupc C sL | j dk r | j jdkr | j �� | j�� | j�� t�| jd� 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�LOGOUTg @)r� �state�shutdownrF rv r Z join_threadr� r r r r r� � s
zNewIMAPTestsMixin._cleanupc C sB G dd� dt j�}| j|dd�\}}| jtjj| jf|j�� 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�rL rM r r r r rk � s zVNewIMAPTestsMixin.test_EOF_without_complete_welcome_message.<locals>.EOFHandler.handleN�r8 r9 r: rk r r r r �
EOFHandler� s r� F�r. ) � socketserver�StreamRequestHandlerr� r0 r r2 �abortr� r� )r r� �_rF r r r �)test_EOF_without_complete_welcome_message� s
�z;NewIMAPTestsMixin.test_EOF_without_complete_welcome_messagec C s@ G dd� dt �}| j|dd�\}}| jtjj| jf|j�� d S )Nc @ s e Zd Zdd� ZdS )zBNewIMAPTestsMixin.test_line_termination.<locals>.BadNewlineHandlerc S s | � d� | �|dd� d S �Ns * CAPABILITY IMAP4rev1 AUTH
rl rm �rP rZ rp r r r ro s
zQNewIMAPTestsMixin.test_line_termination.<locals>.BadNewlineHandler.cmd_CAPABILITYN�r8 r9 r: ro r r r r �BadNewlineHandler� s r� Fr� )rC r� r0 r r2 r� r� r� )r r� r� rF r r r �test_line_termination� s
�z'NewIMAPTestsMixin.test_line_terminationc C s` G dd� dt �}| �|�\}}| �|j� | �tjjd�� |�d� W 5 Q R X | �|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: rn 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 s8 | � t�\}}| �tjjd�� |�d� W 5 Q R X 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 sT | � t�\}}|�dd�\}}| �|d� | �tjjd�� |�d� W 5 Q R X d S )N�user�passrl 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 �Nrl zENABLE successful�rZ rp 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�+rl zFAKEAUTH successful�rU rF �responserZ rp r r r �cmd_AUTHENTICATE# s
zYNewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_AUTHENTICATEc s s& | � d� d V | j_| �|dd� d S �Nr� rl Zokayr� rp r r r �
cmd_APPEND' s
zSNewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_APPENDN)r8 r9 r: rn 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>rl �
ZmFrZQ==
r� �utf-8� Subject: üñí©öðé� UTF8 (%s)
) rC r� r � _encoding� authenticater� r� �appendrT ) r r� r� rF rY 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 5 Q R X 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� rp r r r r� < 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� rp r r r r� > s
zaNewIMAPTestsMixin.test_search_disallows_charset_in_utf8_mode.<locals>.UTF8Server.cmd_AUTHENTICATEN�r8 r9 r: rn r� r� r r r r �
UTF8Server: s r� r� c S s dS r� r r� r r r r� C r[ zNNewIMAPTestsMixin.test_search_disallows_charset_in_utf8_mode.<locals>.<lambda>rl 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_mode9 s z<NewIMAPTestsMixin.test_search_disallows_charset_in_utf8_modec C sN G dd� dt �}| �|�\}}| �tjjd�� |�ddd� � W 5 Q R X 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 �rZ �formatrp r r r r� M s �zGNewIMAPTestsMixin.test_bad_auth_name.<locals>.MyServer.cmd_AUTHENTICATEN�r8 r9 r: r� r r r r �MyServerL s r� z'unrecognized authentication type METHOD�METHODc S s dS �Nr r r r r r r� S 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_nameK s
�z$NewIMAPTestsMixin.test_bad_auth_namec C sN G dd� dt �}| �|�\}}| �tjjd�� |�ddd� � W 5 Q R X 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�rU r� rZ rp r r r r� W s
zPNewIMAPTestsMixin.test_invalid_authentication.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� V s r� z \[AUTHENTICATIONFAILED\] invalidr� c S s dS r� r r� r r r r� ^ r[ z?NewIMAPTestsMixin.test_invalid_authentication.<locals>.<lambda>r� r� r r r �test_invalid_authenticationU 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� rp r r r r� b s
zTNewIMAPTestsMixin.test_valid_authentication_bytes.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� a s r� r� c S s dS r� r r� r r r r� g r[ zCNewIMAPTestsMixin.test_valid_authentication_bytes.<locals>.<lambda>rl r� �rC r� r� r r� �r r� r� rF rY r� r r r � test_valid_authentication_bytes` 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� rp r r r r� m s
zYNewIMAPTestsMixin.test_valid_authentication_plain_text.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� l s r� r� c S s dS �NZfaker r� r r r r� r r[ zHNewIMAPTestsMixin.test_valid_authentication_plain_text.<locals>.<lambda>rl r� r� r� r r r �$test_valid_authentication_plain_textk s
z6NewIMAPTestsMixin.test_valid_authentication_plain_text�md5c C sN G dd� dt �}| �|�\}}| �d|jk� |�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
rl zCRAM-MD5 successfulr� z No access�rU rZ �r rX rj �rr r r r� z s
zQNewIMAPTestsMixin.test_login_cram_md5_bytes.<locals>.AuthHandler.cmd_AUTHENTICATEN�r8 r9 r: rn r� r r r r � AuthHandlerx s r� �
AUTH=CRAM-MD5�tim� tanstaaftanstaafrl �rC r� r� rn �login_cram_md5r �r r� r� r� �retr r r �test_login_cram_md5_bytesv s
z+NewIMAPTestsMixin.test_login_cram_md5_bytesc C sN G dd� dt �}| �|�\}}| �d|jk� |�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� �tanstaaftanstaafrl r� r� r r r � test_login_cram_md5_plain_text� s
z0NewIMAPTestsMixin.test_login_cram_md5_plain_textc C sN G dd� dt �}| �|�\}}| �tjjd�� |�ddd� � W 5 Q R X 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] abortedrl zMYAUTH successfulr� rp 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 sN G dd� dt �}| j|dd�\}}| �tjjd�� | j|j� W 5 Q R X 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 rk � 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� rF r r r �test_linetoolong� s
�z"NewIMAPTestsMixin.test_linetoolongc C s. | j tdd�\}}| j|j� � W 5 Q R X d S )NFr� )r� rC r� r� )r r� rF r r r �test_simple_with_statement� s z,NewIMAPTestsMixin.test_simple_with_statementc C sT | j tdd�\}}| j|j� � }|�dd� | �|jd� W 5 Q R X | �|j� d S �NFr� r� r� )r� rC r� r� r� r rG � assertIsNone�r r� rF �imapr r r �test_with_statement� s
z%NewIMAPTestsMixin.test_with_statementc C sh | j tdd�\}}| j|j� �4}|�dd� | �|jd� |�� | �|j� W 5 Q R X | �|j� d S r� ) r� rC r� r� r� r rG �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� rl 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� rl 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 () "." directoryArl zLSUB completedr� rp 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� rl r s () "." directoryA)rC r� r� Zlsubr )r r r� r� r� r� r r r � test_lsub� s zNewIMAPTestsMixin.test_lsub)T) r8 r9 r: r� r� r� r� r� r� r� r� r� r� r� r� r� r� r
r� r� r� r �patchr� r� r r r r r r r r r rt � s6
!
rt 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 sz t �t j�}| �|jt j� | �|jd� |�t� | � t j
d��. | � t �\}}| j
|jd|i�}|�� W 5 Q R X 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� rF 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 )N� localhostr �r )
r< r= r r r r� rC r� r� r� r r r r �test_ssl_verified s
�z!NewIMAPSSLTests.test_ssl_verifiedc C sf t �dtf��N tj�| jd��2 tj�| jd�� | jddtd� W 5 Q R X W 5 Q R X W 5 Q R X 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� r r r r# 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 d#�d$d%� ��Z ed&d'� �Z d(d)� Z ed*d+� �Z!ed,d-� �Z"ed.d/� �Z#d0S )1�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* ru rw r r r rx s
zAThreadedNetworkedTests.make_server.<locals>.MyServer.handle_errorNry r r r r r� s r� zcreating serverzserver createdzADDR =zCLASS =zHDLR =z
%s servingr{ r| r} Tzserver running)
r� r rJ r r� r- Z getsocknameZRequestHandlerClassr� r� r� r� r� )r Zaddr�hdlrr� rF r( r r r � make_server s(
�z"ThreadedNetworkedTests.make_serverc C s4 t r td� |�� |�� |�� t r0td� d S )Nzwaiting for serverZdone)r rJ r� rv rW )r rF r� r r r � reap_server9 s z"ThreadedNetworkedTests.reap_serverc
c s4 | � tjdf|�\}}z
|V W 5 | �||� X d S )Nr )r' r r� r( )r r& rF r� r r r �
reaped_serverB s
z$ThreadedNetworkedTests.reaped_serverc c s@ | � |��,}| j|j� }z||fV W 5 |�� X W 5 Q R X d S r* )r) r� r� r )r r& rF r� r r r � reaped_pairJ s
z"ThreadedNetworkedTests.reaped_pairc C s. | � t��}| j|j� }|�� W 5 Q R X d S r* )r) rC r� r� r� �r rF r� r r r � test_connectS s z#ThreadedNetworkedTests.test_connectc C s� G dd� dt �}| �|���\}}|�ddd� �\}}| �|d� | �|jd� |�d� |�d d
d �\}\}| �d |� |�d� |�d
�\}\}| �d |� W 5 Q R X d S )Nc s4 e Zd Z� fdd�Zdd� Zdd� Zdd� Z� ZS ) zEThreadedNetworkedTests.test_bracket_flags.<locals>.BracketFlagHandlerc s dddddg| _ t� �� d S )NZAnsweredZFlaggedZDeletedZSeenZDraft)�flagsrD rk r rH r r rk d s zLThreadedNetworkedTests.test_bracket_flags.<locals>.BracketFlagHandler.handlec s s& | � d� d V | j_| �|dd� d S r� r� rp r r r r� h 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� s
* 2 EXISTSs
* 0 RECENTz-* OK [PERMANENTFLAGS %s \*)] Flags permitted.rl z [READ-WRITE] SELECT completed.)rW r- rR rT rZ )r rX rj Zflag_msg�msgr r r �
cmd_SELECTm 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))r. z
* %s FETCH %sr r� rl zSTORE completed.)rK rc r- �extendrW rR rT rZ )r rX rj Z new_flagsZ flags_msgr/ r r r � cmd_STOREw s zOThreadedNetworkedTests.test_bracket_flags.<locals>.BracketFlagHandler.cmd_STORE)r8 r9 r: rk r� r0 r4 rs r r rH r �BracketFlagHandlerb s
r5 r� c S s dS r� r r� r r r r� � r[ z;ThreadedNetworkedTests.test_bracket_flags.<locals>.<lambda>rl r� �test� 1z+FLAGSz[test]s [test]ZPERMANENTFLAGS)rC r* r� r r� ZselectZstorer3 )r r5 rF r� rY r� r� r r r �test_bracket_flagsY s
z)ThreadedNetworkedTests.test_bracket_flagsc C sF G dd� dt j�}| �|�� }| jtjj| jf|j�� W 5 Q R X 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 rk � 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� rF r r r �test_issue5949� s
��z%ThreadedNetworkedTests.test_issue5949c C sD G dd� dt �}| �|�� }| jtjj| jf|j�� W 5 Q R X d S )Nc @ s e Zd Zdd� ZdS )zGThreadedNetworkedTests.test_line_termination.<locals>.BadNewlineHandlerc S s | � d� | �|dd� d S r� r� rp r r r ro � 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� rF 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� rp r r r r� � s z,ThreadedNetworkedTests.UTF8Server.cmd_ENABLEc c s& | � d� d V | j_| �|dd� d S r� r� rp r r r r� � s
z2ThreadedNetworkedTests.UTF8Server.cmd_AUTHENTICATENr� r r r r r� � s r� c C sL | � | j��6\}}| �|j� | �tjj|jd� | �|j� W 5 Q R X d S )Nr� ) r* r� r� r� r0 r r2 r/ r� r+ r r r r� � s z;ThreadedNetworkedTests.test_enable_raises_error_if_not_AUTHc C sD G dd� d| j �}| �|�� \}}| �tjj|jd� W 5 Q R X d S )Nc @ s e Zd ZdZdS )zXThreadedNetworkedTests.test_enable_raises_error_if_no_capability.<locals>.NoEnableServerr Nr� r r r r �NoEnableServer� s r: r� )r� r* r0 r r2 r/ r� )r r: rF r� r r r r� � s z@ThreadedNetworkedTests.test_enable_raises_error_if_no_capabilityc
C sn G dd� dt �}| �tjj��F | �|��0\}}|�dd�\}}| �|d� |�d� W 5 Q R X W 5 Q R X 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 r; r� r� rl r� ) rC r0 r r2 r/ r* r� r r� )r r; rF 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 5 Q R X 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� rp 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>rl r� r� r� r� r� ) r� r* r r� r� r� r� r� rT ) r r� rF r� rY r� r� r� r� r r r r� � s. � � �z3ThreadedNetworkedTests.test_enable_UTF8_True_appendc C s| | � | j��f\}}|�ddd� �\}}| �|d� |�d�\}}| �|d� | �|j� | �tj j
|j dd� W 5 Q R X 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>rl r� r� r� ) r* r� r� r r� r� r� r0 r r2 r/ r� )r rF r� r� r� r r r r� � s zAThreadedNetworkedTests.test_search_disallows_charset_in_utf8_modec
C sX G dd� dt �}| �|��4\}}| �tjj�� |�ddd� � W 5 Q R X W 5 Q R X d S )Nc @ s e Zd Zdd� ZdS )z;ThreadedNetworkedTests.test_bad_auth_name.<locals>.MyServerc S s | � |dd�|d �� d S r� r� rp 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� rF r� r r r r� � s z)ThreadedNetworkedTests.test_bad_auth_namec
C s\ G dd� dt �}| �|��8\}}| �tjj�� |�ddd� �\}}W 5 Q R X W 5 Q R X d S )Nc @ s e Zd Zdd� ZdS )zDThreadedNetworkedTests.test_invalid_authentication.<locals>.MyServerc s s$ | � d� d V | _| �|dd� d S r� r� rp r r r r� s
zUThreadedNetworkedTests.test_invalid_authentication.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� s r� r� c S s dS r� r r� r r r r�
r[ zDThreadedNetworkedTests.test_invalid_authentication.<locals>.<lambda>r< �r r� rF r� rY r� r r r r� � s z2ThreadedNetworkedTests.test_invalid_authenticationc C s� G dd� dt �}| �|��8\}}|�ddd� �\}}| �|d� | �|jd� W 5 Q R X | �|��8\}}|�ddd� �\}}| �|d� | �|jd� W 5 Q R X 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� rp r r r r� 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� r[ zBThreadedNetworkedTests.test_valid_authentication.<locals>.<lambda>rl r� c S s dS r� r r� r r r r� r[ )rC r* r� r r� r= r r r �test_valid_authentication s � �z0ThreadedNetworkedTests.test_valid_authenticationr� c C s� G dd� dt �}| �|��6\}}| �d|jk� |�dd�\}}| �|d� W 5 Q R X | �|��6\}}| �d|jk� |�dd�\}}| �|d� W 5 Q R X 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� & s r� r� r� r� rl r� )rC r* r� rn r� r )r r� rF r� r� r� r r r �test_login_cram_md5" s z*ThreadedNetworkedTests.test_login_cram_md5c
C s\ G dd� dt �}| �|��8\}}| �tjj�� |�ddd� �\}}W 5 Q R X W 5 Q R X 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� rp r r r r� D s
zUThreadedNetworkedTests.test_aborted_authentication.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� B s r� r� c S s d S r* r r� r r r r� O r[ zDThreadedNetworkedTests.test_aborted_authentication.<locals>.<lambda>r< r= r r r r� ? s z2ThreadedNetworkedTests.test_aborted_authenticationc C sD G dd� dt �}| �|�� }| jtjj| jf|j�� W 5 Q R X d S )Nc @ s e Zd Zdd� ZdS )z?ThreadedNetworkedTests.test_linetoolong.<locals>.TooLongHandlerc S s | j �dtjd d � d S )Ns * OK � xrQ )rL rM r Z_MAXLINEr r r r rk T s zFThreadedNetworkedTests.test_linetoolong.<locals>.TooLongHandler.handleNr� r r r r r� S s r� )rC r) r0 r r2 r/ r� r� )r r� rF r r r r� R s
��z'ThreadedNetworkedTests.test_linetoolongc
C s2 | � t�� }| j|j� � W 5 Q R X W 5 Q R X d S r* )r) rC r� r� )r rF r r r r� \ s z1ThreadedNetworkedTests.test_simple_with_statementc
C sX | � t��D}| j|j� � }|�dd� | �|jd� W 5 Q R X | �|j� W 5 Q R X d S �Nr� r� )r) rC r� r� r� r rG r� �r rF r r r r r c s
z*ThreadedNetworkedTests.test_with_statementc
C sl | � t��X}| j|j� �4}|�dd� | �|jd� |�� | �|j� W 5 Q R X | �|j� W 5 Q R X d S rA ) r) rC r� r� r� r rG r r� rB r r r r k s z1ThreadedNetworkedTests.test_with_statement_logoutN)$r8 r9 r: r� r r� r r2 r� r' r( r r) r* r r, r8 r9 r� rC r� r� r� r� r� r� r� r� r> r
r? r� r� r� r r r r r r r% sZ
0
r% c @ s e Zd ZeZeZedd� �ZdS )�ThreadedNetworkedTestsSSLc
C s� t �t j�}|�t� | �t jd��6 | �t�� }| j |j
d|i�}|� � W 5 Q R X W 5 Q R X | �t��$}| j d|j
d |d�}|� � W 5 Q R X d S )Nr r r r r ) r< r= r r r r� r r) rC r� r� r� )r r rF r� r r r r | s
�
� �z+ThreadedNetworkedTestsSSL.test_ssl_verifiedN) r8 r9 r: r; r� r$ r� r r r r r r rC w s rC 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.edur Z anonymousr� c C s, t | j�� | �| j| j�| _W 5 Q R X d S r* )r �hostr� �portrF r r r r �setUp� s zRemoteIMAPTest.setUpc C s. | j d k r*t| j�� | j �� W 5 Q R X d S r* )rF r rE r r r r r �tearDown� s
zRemoteIMAPTest.tearDownc C sv t | j��b | jjD ]}| �|t� q| �d| jj� | �d| jj� | j�| j| j �}| �
|d d� W 5 Q R X d S )N�
LOGINDISABLEDzAUTH=ANONYMOUSr rl ) r rE rF rn �assertIsInstance�strr3 r� �username�passwordr )r �cap�rsr r r �test_logincapa� s z RemoteIMAPTest.test_logincapac C s< t | j��( | j�� }d | _| �|d d|� W 5 Q R X d S �Nr r )r rE rF r r �r rO r r r r � s
zRemoteIMAPTest.test_logoutN)r8 r9 r: rE rF rL rM r r2 r� rG rH rP r r r r r rD � s rD c s$ e Zd Z� fdd�Zdd� Z� ZS )�RemoteIMAP_STARTTLSTestc s> t � �� t| j�� | j�� }| �|d d� W 5 Q R X d S )Nr rl )rD rG r rE rF Zstarttlsr rR rH r r rG � s
z RemoteIMAP_STARTTLSTest.setUpc C s. | j jD ]}| �|t� q| �d| j j� d S )NrI )rF rn rJ rK � assertNotIn)r rN r r r rP � s z&RemoteIMAP_STARTTLSTest.test_logincapa)r8 r9 r: rG rP rs r r rH r rS � s rS 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 rG � s zRemoteIMAP_SSLTest.setUpc C s d S r* r r r r r rH � s zRemoteIMAP_SSLTest.tearDownc C s( t �t j�}d|_t j|_|�t� |S )NF)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 sh zX|jD ]}| �|t� q| �d|j� | �d|j� |�| j| j�}| � |d d� W 5 |� � X d S )NrI z
AUTH=PLAINr rl )
r rn rJ rK rT r3 r� rL rM r )r rF rN rO r r r �check_logincapa� s
z"RemoteIMAP_SSLTest.check_logincapac C s4 t | j�� | �| j| j�}| �|� W 5 Q R X d S r* )r rE r� rF rW )r �_serverr r r rP � s z!RemoteIMAP_SSLTest.test_logincapac C sD t | j��0 | �| j| j�}|�� }| �|d d|� W 5 Q R X d S rQ )r rE r� rF r r )r rX rO r r r r � s z RemoteIMAP_SSLTest.test_logoutc C s: t | j��& | jt| j| j| jt| �� d� W 5 Q R X d S )N)r r �r rE r0 �
ValueErrorr� rF r? rV r r r r �#test_ssl_context_certfile_exclusive� s �z6RemoteIMAP_SSLTest.test_ssl_context_certfile_exclusivec C s: t | j��& | jt| j| j| jt| �� d� W 5 Q R X d S )N)Zkeyfiler rY r r r r �"test_ssl_context_keyfile_exclusive� s �z5RemoteIMAP_SSLTest.test_ssl_context_keyfile_exclusiveN)r8 r9 r: rF r$ r� rG rH rV rW rP r r[ r\ r r r r rU � s rU �__main__)6r6 r �
contextlibr r Zos.path�osr� r r r� r- Z test.supportr r r r r r r
Zunittestr r r
r r< � ImportError�pathrW �dirname�__file__�curdirr? r ZTestCaser r r; r$ r� rC rt r Z
skipUnlessr r% rC Zis_resource_enabled�skiprD rS rU r8 �mainr r r r �<module> sl $
@ R ,
$ d
�
�
4