Your IP : 216.73.216.196

Current Path : /opt/alt/python37/lib64/python3.7/test/__pycache__/
Upload File :
Current File : //opt/alt/python37/lib64/python3.7/test/__pycache__/test_imaplib.cpython-37.opt-1.pyc

B

� fK��@sfddlmZddlmZddlZddlZddlZddlZddl	Z	ddl
Z
ddlZddlZddl
mZmZmZmZmZmZddlZddlmZddlmZmZmZyddlZWnek
r�dZYnXej�ej�e�p�ejd�Z ej�ej�e�p�ejd�Z!Gd	d
�d
ej"�Z#e�r0Gdd�dej$�Z%ej&Z&nGd
d�d�Z%dZ&Gdd�dej'�Z(Gdd�d�Z)Gdd�de)ej"�Z*e�+ed�Gdd�de)ej"��Z,Gdd�dej"�Z-e�+ed�Gdd�de-��Z.e�+e�/d�d�e�0d�Gdd�dej"���Z1e�+ed�e�+e�/d�d�e�0d�Gd d!�d!e1����Z2e�+ed�e�0d�Gd"d#�d#e1���Z3e4d$k�rbe�5�dS)%�)�support)�contextmanagerN)�reap_threads�verbose�transient_internet�run_with_tz�run_with_locale�cpython_only)�mock)�datetime�timezone�	timedeltazkeycert3.pemzpycacert.pemc@sXeZdZdd�Zed�dd��Zdd�Zedd	d
�ed�dd
���Zdd�Z	dd�Z
dS)�TestImaplibcCsbt�d�}t�d�}|�t�|�|�t�d�}|�t�|�|�t�d�}|�t�|�|�dS)N)	i��rrrr���rrs.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�timeZmktime)�selfZt0Ztt�r�6/opt/alt/python37/lib64/python3.7/test/test_imaplib.py�test_Internaldate2tuples
z#TestImaplib.test_Internaldate2tuplezMST+07MDT,M4.1.0,M10.5.0cCs|�t�d�t�d��dS)Ns.25 (INTERNALDATE "02-Apr-2000 02:30:00 +0000")s.25 (INTERNALDATE "02-Apr-2000 03:30:00 +0000"))ZassertNotEqualrr)rrrr�"test_Internaldate2tuple_issue10941)sz.TestImaplib.test_Internaldate2tuple_issue10941cCs*ddt�d�ddt�dttdd���dgS)Ni�5wge��A)	i���r�!�rrr)	i�rrrrrrrrri z"18-May-2033 05:33:20 +0200")rZ	localtimerZ
fromtimestamprr
)rrrr�
timevalues0szTestImaplib.timevalues�LC_ALLZde_DEZfr_FRzSTD-1DST,M3.2.0,M11.1.0cCs0d}x&|��D]}t�|�}|�||�qWdS)Nz"18-May-2033 05:33:20 +0200")rr�Time2Internaldater)rZexpected�tZinternalrrr�test_Time2Internaldate8s
z"TestImaplib.test_Time2InternaldatecCs x|��D]}t�|�q
WdS)N)rrr!)rr"rrr�,test_that_Time2Internaldate_returns_a_resultCsz8TestImaplib.test_that_Time2Internaldate_returns_a_resultc	Cs�t���<}y|�dtjf�|�d�Wntjk
r>YnXWdQRXt��}|�t	��}t�
�WdQRX|�|jj
|�dS)N�z3Cannot run the test with local IMAP server running.)�socket�connectrZ
IMAP4_PORTZskipTest�errorrZget_socket_conn_refused_errs�assertRaises�OSError�IMAP4�assertInZ	exception�errno)r�sZexpected_errnos�cmrrr�test_imap4_host_default_valueJs

z)TestImaplib.test_imap4_host_default_valueN)�__name__�
__module__�__qualname__rrrrrr#r$r0rrrrrs
rc@seZdZdd�ZdS)�SecureTCPServercCs6|j��\}}t��}|�t�|j|dd�}||fS)NT)Zserver_side)r&Zaccept�ssl�
SSLContext�load_cert_chain�CERTFILEZwrap_socket)rZ	newsocketZfromaddr�contextZ
connstreamrrr�get_request^s

zSecureTCPServer.get_requestN)r1r2r3r:rrrrr4\sr4c@seZdZdS)r4N)r1r2r3rrrrr4iscsheZdZdZdZdZ�fdd�Zdd�Zdd	�Zd
d�Z	dd
�Z
dd�Zdd�Zdd�Z
dd�Z�ZS)�SimpleIMAPHandlerrNr%cst���d|j_dS)N)�super�setup�server�logged)r)�	__class__rrr=ts
zSimpleIMAPHandler.setupcCs$trtd|���|j�|�dS)NzSENT: %r)r�print�strip�wfile�write)r�messagerrr�_sendxszSimpleIMAPHandler._sendcCs|�|d�dS)Ns
)rF)rrErrr�
_send_line}szSimpleIMAPHandler._send_linecCs|�|�d��dS)N�ASCII)rG�encode)rrErrr�_send_textline�sz SimpleIMAPHandler._send_textlinecCs|�d�|||f��dS)N� )rJ�join)r�tag�coderErrr�_send_tagged�szSimpleIMAPHandler._send_taggedcCs$|�d��xd}xJy$|j�d�}|dkr.dS||7}Wntk
rLdSX|�d�rPqWtrrtd|���|jr�y|j�	|�Wqt
k
r�d|_YqXq|�d���}|d}|d}|dd�}t
|d	|��r
t|d	|�||�}|�r||_t|�q|�|d
|d�qWdS)Nz* OK IMAP4rev1�rs
zGOT: %rrHr�Zcmd_ZBADz unknown)rJZrfile�readr*�endswithrrArB�continuation�send�
StopIteration�decode�split�hasattr�getattr�nextrO)r�line�partZ	splitlinerM�cmd�argsrTrrr�handle�s>


zSimpleIMAPHandler.handlecCs4|jrd|jnd}|�d|�|�|dd�dS)Nz
IMAP4rev1 Z	IMAP4rev1z
* CAPABILITY �OKzCAPABILITY completed)�capabilitiesrJrO)rrMr_Zcapsrrr�cmd_CAPABILITY�sz SimpleIMAPHandler.cmd_CAPABILITYcCs$d|j_|�d�|�|dd�dS)Nz"* BYE IMAP4ref1 Server logging outrazLOGOUT completed)r>r?rJrO)rrMr_rrr�
cmd_LOGOUT�s
zSimpleIMAPHandler.cmd_LOGOUTcCs|d|j_|�|dd�dS)NrrazLOGIN completed)r>r?rO)rrMr_rrr�	cmd_LOGIN�szSimpleIMAPHandler.cmd_LOGIN)r1r2r3ZtimeoutrTrbr=rFrGrJrOr`rcrdre�
__classcell__rr)r@rr;os*r;c@s�eZdZdZd3dd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd �Zd!d"�Ze�d#d$�d%d&��Zd'd(�Zd)d*�Zd+d,�Zd-d.�Zd/d0�Zd1d2�ZdS)4�NewIMAPTestsMixinNTcCs�Gdd�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@seZdZdd�ZdS)z/NewIMAPTestsMixin._setup.<locals>.TestTCPServercSs|�|�|���dS)zP
                End request and raise the error if one occurs.
                N)�
close_request�server_close)r�request�client_addressrrr�handle_error�s
z<NewIMAPTestsMixin._setup.<locals>.TestTCPServer.handle_errorN)r1r2r3rlrrrr�
TestTCPServer�srmrz-server�
poll_intervalg{�G�z�?)�name�target�kwargsT)�server_classZ
addCleanup�_cleanupr�HOSTr>�	threading�ThreadZ_testMethodName�
serve_forever�thread�daemon�start�
imap_class�server_address�client)rZimap_handlerr'rmrrr�_setup�s	
zNewIMAPTestsMixin._setupcCsL|jdk	r |jjdkr |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�shutdownr>rirZjoin_threadrx)rrrrrs�s


zNewIMAPTestsMixin._cleanupcCsBGdd�dtj�}|j|dd�\}}|jtjj|jf|j��dS)Nc@seZdZdd�ZdS)zONewIMAPTestsMixin.test_EOF_without_complete_welcome_message.<locals>.EOFHandlercSs|j�d�dS)Ns* OK)rCrD)rrrrr`�szVNewIMAPTestsMixin.test_EOF_without_complete_welcome_message.<locals>.EOFHandler.handleN)r1r2r3r`rrrr�
EOFHandler�sr�F)r')	�socketserver�StreamRequestHandlerr~r)rr+�abortr{r|)rr��_r>rrr�)test_EOF_without_complete_welcome_message�sz;NewIMAPTestsMixin.test_EOF_without_complete_welcome_messagecCs@Gdd�dt�}|j|dd�\}}|jtjj|jf|j��dS)Nc@seZdZdd�ZdS)zBNewIMAPTestsMixin.test_line_termination.<locals>.BadNewlineHandlercSs|�d�|�|dd�dS)Ns* CAPABILITY IMAP4rev1 AUTH
razCAPABILITY completed)rFrO)rrMr_rrrrcs
zQNewIMAPTestsMixin.test_line_termination.<locals>.BadNewlineHandler.cmd_CAPABILITYN)r1r2r3rcrrrr�BadNewlineHandler�sr�F)r')r;r~r)rr+r�r{r|)rr�r�r>rrr�test_line_termination�sz'NewIMAPTestsMixin.test_line_terminationc	Cs`Gdd�dt�}|�|�\}}|�|j�|�tjjd��|�d�WdQRX|�|j�dS)Nc@seZdZdZdS)zMNewIMAPTestsMixin.test_enable_raises_error_if_not_AUTH.<locals>.EnableHandlerzAUTH ENABLE UTF8=ACCEPTN)r1r2r3rbrrrr�
EnableHandlersr�zENABLE.*NONAUTH�foo)	r;r~�assertFalse�utf8_enabled�assertRaisesRegexrr+r(�enable)rr�r}r�rrr�$test_enable_raises_error_if_not_AUTHsz6NewIMAPTestsMixin.test_enable_raises_error_if_not_AUTHc	Cs8|�t�\}}|�tjjd��|�d�WdQRXdS)Nzdoes not support ENABLEr�)r~r;r�rr+r(r�)rr}r�rrr�)test_enable_raises_error_if_no_capabilitys
z;NewIMAPTestsMixin.test_enable_raises_error_if_no_capabilityc	CsT|�t�\}}|�dd�\}}|�|d�|�tjjd��|�d�WdQRXdS)N�user�passrazdoes not support ENABLEzUTF8=ACCEPT)	r~r;�loginrr�rr+r(r�)rr}r��typ�datarrr�.test_enable_UTF8_raises_error_if_not_supporteds
z@NewIMAPTestsMixin.test_enable_UTF8_raises_error_if_not_supportedc	Cs�Gdd�dt�}|�|�\}}|�|jd�|�ddd��\}}|�|d�|�|jd�|�d	�\}}|�|d�|�|jd
�d}|�ddd|�d
��\}}|�|d�|�|jd|�d
��dS)
Nc@s(eZdZdZdd�Zdd�Zdd�ZdS)	zHNewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.UTF8AppendServerzENABLE UTF8=ACCEPTcSs|�|dd�dS)NrazENABLE successful)rO)rrMr_rrr�
cmd_ENABLE!szSNewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_ENABLEcss&|�d�dV|j_|�|dd�dS)N�+razFAKEAUTH successful)rJr>�responserO)rrMr_rrr�cmd_AUTHENTICATE#s

zYNewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_AUTHENTICATEcss&|�d�dV|j_|�|dd�dS)Nr�ra�okay)rJr>r�rO)rrMr_rrr�
cmd_APPEND's

zSNewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_APPENDN)r1r2r3rbr�r�r�rrrr�UTF8AppendServersr��ascii�MYAUTHcSsdS)Nsfaker)�xrrr�<lambda>-rPz@NewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.<lambda>ras
ZmFrZQ==
zUTF8=ACCEPTzutf-8uSubject: üñí©öðézUTF8 (%s)
)	r;r~r�	_encoding�authenticater�r��appendrI)	rr�r}r>rNr��
msg_stringr�r�rrr�test_enable_UTF8_True_appendsz.NewIMAPTestsMixin.test_enable_UTF8_True_appendc	Cs�Gdd�dt�}|�|�\}}|�ddd��\}}|�|d�|�d�\}}|�|d�|�|j�|�tj	j
d��|�d	d
�WdQRXdS)Nc@s eZdZdZdd�Zdd�ZdS)zPNewIMAPTestsMixin.test_search_disallows_charset_in_utf8_mode.<locals>.UTF8ServerzAUTH ENABLE UTF8=ACCEPTcSs|�|dd�dS)NrazENABLE successful)rO)rrMr_rrrr�<sz[NewIMAPTestsMixin.test_search_disallows_charset_in_utf8_mode.<locals>.UTF8Server.cmd_ENABLEcss&|�d�dV|j_|�|dd�dS)Nr�razFAKEAUTH successful)rJr>r�rO)rrMr_rrrr�>s

zaNewIMAPTestsMixin.test_search_disallows_charset_in_utf8_mode.<locals>.UTF8Server.cmd_AUTHENTICATEN)r1r2r3rbr�r�rrrr�
UTF8Server:sr�r�cSsdS)Nsfaker)r�rrrr�CrPzNNewIMAPTestsMixin.test_search_disallows_charset_in_utf8_mode.<locals>.<lambda>razUTF8=ACCEPTz
charset.*UTF8r��bar)r;r~r�rr��
assertTruer�r�rr+r(�search)rr�r}r�r�rrr�*test_search_disallows_charset_in_utf8_mode9sz<NewIMAPTestsMixin.test_search_disallows_charset_in_utf8_modec	CsNGdd�dt�}|�|�\}}|�tjjd��|�ddd��WdQRXdS)Nc@seZdZdd�ZdS)z6NewIMAPTestsMixin.test_bad_auth_name.<locals>.MyServercSs|�|dd�|d��dS)N�NOz#unrecognized authentication type {}r)rO�format)rrMr_rrrr�MszGNewIMAPTestsMixin.test_bad_auth_name.<locals>.MyServer.cmd_AUTHENTICATEN)r1r2r3r�rrrr�MyServerLsr�z'unrecognized authentication type METHOD�METHODcSsdS)Nrrrrrrr�SrPz6NewIMAPTestsMixin.test_bad_auth_name.<locals>.<lambda>)r;r~r�rr+r(r�)rr�r}r�rrr�test_bad_auth_nameKs

z$NewIMAPTestsMixin.test_bad_auth_namec	CsNGdd�dt�}|�|�\}}|�tjjd��|�ddd��WdQRXdS)Nc@seZdZdd�ZdS)z?NewIMAPTestsMixin.test_invalid_authentication.<locals>.MyServercss$|�d�dV|_|�|dd�dS)Nr�r�z[AUTHENTICATIONFAILED] invalid)rJr�rO)rrMr_rrrr�Ws
zPNewIMAPTestsMixin.test_invalid_authentication.<locals>.MyServer.cmd_AUTHENTICATEN)r1r2r3r�rrrrr�Vsr�z \[AUTHENTICATIONFAILED\] invalidr�cSsdS)Nsfaker)r�rrrr�^rPz?NewIMAPTestsMixin.test_invalid_authentication.<locals>.<lambda>)r;r~r�rr+r(r�)rr�r}r�rrr�test_invalid_authenticationUs

z-NewIMAPTestsMixin.test_invalid_authenticationcCsPGdd�dt�}|�|�\}}|�ddd��\}}|�|d�|�|jd�dS)Nc@seZdZdd�ZdS)zCNewIMAPTestsMixin.test_valid_authentication_bytes.<locals>.MyServercss&|�d�dV|j_|�|dd�dS)Nr�razFAKEAUTH successful)rJr>r�rO)rrMr_rrrr�bs

zTNewIMAPTestsMixin.test_valid_authentication_bytes.<locals>.MyServer.cmd_AUTHENTICATEN)r1r2r3r�rrrrr�asr�r�cSsdS)Nsfaker)r�rrrr�grPzCNewIMAPTestsMixin.test_valid_authentication_bytes.<locals>.<lambda>ras
ZmFrZQ==
)r;r~r�rr�)rr�r}r>rNr�rrr�test_valid_authentication_bytes`s
z1NewIMAPTestsMixin.test_valid_authentication_bytescCsPGdd�dt�}|�|�\}}|�ddd��\}}|�|d�|�|jd�dS)Nc@seZdZdd�ZdS)zHNewIMAPTestsMixin.test_valid_authentication_plain_text.<locals>.MyServercss&|�d�dV|j_|�|dd�dS)Nr�razFAKEAUTH successful)rJr>r�rO)rrMr_rrrr�ms

zYNewIMAPTestsMixin.test_valid_authentication_plain_text.<locals>.MyServer.cmd_AUTHENTICATEN)r1r2r3r�rrrrr�lsr�r�cSsdS)N�faker)r�rrrr�rrPzHNewIMAPTestsMixin.test_valid_authentication_plain_text.<locals>.<lambda>ras
ZmFrZQ==
)r;r~r�rr�)rr�r}r>rNr�rrr�$test_valid_authentication_plain_textks
z6NewIMAPTestsMixin.test_valid_authentication_plain_textcCsNGdd�dt�}|�|�\}}|�d|jk�|�dd�\}}|�|d�dS)Nc@seZdZdZdd�ZdS)z@NewIMAPTestsMixin.test_login_cram_md5_bytes.<locals>.AuthHandlerzLOGINDISABLED AUTH=CRAM-MD5css:|�d�dV}|dkr(|�|dd�n|�|dd�dS)Nz:+ PDE4OTYuNjk3MTcwOTUyQHBvc3RvZmZpY2UucmVzdG9uLm1jaS5uZXQ=s2dGltIGYxY2E2YmU0NjRiOWVmYTFjY2E2ZmZkNmNmMmQ5ZjMy
razCRAM-MD5 successfulr�z	No access)rJrO)rrMr_�rrrrr�ys

zQNewIMAPTestsMixin.test_login_cram_md5_bytes.<locals>.AuthHandler.cmd_AUTHENTICATEN)r1r2r3rbr�rrrr�AuthHandlerwsr�z
AUTH=CRAM-MD5�timstanstaaftanstaafra)r;r~r�rb�login_cram_md5r)rr�r}r��retrrr�test_login_cram_md5_bytesvs
z+NewIMAPTestsMixin.test_login_cram_md5_bytescCsNGdd�dt�}|�|�\}}|�d|jk�|�dd�\}}|�|d�dS)Nc@seZdZdZdd�ZdS)zENewIMAPTestsMixin.test_login_cram_md5_plain_text.<locals>.AuthHandlerzLOGINDISABLED AUTH=CRAM-MD5css:|�d�dV}|dkr(|�|dd�n|�|dd�dS)Nz:+ PDE4OTYuNjk3MTcwOTUyQHBvc3RvZmZpY2UucmVzdG9uLm1jaS5uZXQ=s2dGltIGYxY2E2YmU0NjRiOWVmYTFjY2E2ZmZkNmNmMmQ5ZjMy
razCRAM-MD5 successfulr�z	No access)rJrO)rrMr_r�rrrr��s

zVNewIMAPTestsMixin.test_login_cram_md5_plain_text.<locals>.AuthHandler.cmd_AUTHENTICATEN)r1r2r3rbr�rrrrr��sr�z
AUTH=CRAM-MD5r��tanstaaftanstaafra)r;r~r�rbr�r)rr�r}r�r�rrr�test_login_cram_md5_plain_text�s
z0NewIMAPTestsMixin.test_login_cram_md5_plain_textc	CsNGdd�dt�}|�|�\}}|�tjjd��|�ddd��WdQRXdS)Nc@seZdZdd�ZdS)z?NewIMAPTestsMixin.test_aborted_authentication.<locals>.MyServercss>|�d�dV|_|jdkr,|�|dd�n|�|dd�dS)Nr�s*
r�z[AUTHENTICATIONFAILED] abortedrazMYAUTH successful)rJr�rO)rrMr_rrrr��s

zPNewIMAPTestsMixin.test_aborted_authentication.<locals>.MyServer.cmd_AUTHENTICATEN)r1r2r3r�rrrrr��sr�z \[AUTHENTICATIONFAILED\] abortedr�cSsdS)Nr)r�rrrr��rPz?NewIMAPTestsMixin.test_aborted_authentication.<locals>.<lambda>)r;r~r�rr+r(r�)rr�r}r�rrr�test_aborted_authentication�s

z-NewIMAPTestsMixin.test_aborted_authenticationzimaplib._MAXLINE�
c	CsNGdd�dt�}|j|dd�\}}|�tjjd��|j|j�WdQRXdS)Nc@seZdZdd�ZdS)z:NewIMAPTestsMixin.test_linetoolong.<locals>.TooLongHandlercSs|j�d�dS)Ns* OK xxxxxxxxxxx
)rCrD)rrrrr`�szANewIMAPTestsMixin.test_linetoolong.<locals>.TooLongHandler.handleN)r1r2r3r`rrrr�TooLongHandler�sr�F)r'zgot more than 10 bytes)r;r~r�rr+r(r{r|)rr�r�r>rrr�test_linetoolong�s

z"NewIMAPTestsMixin.test_linetoolongc	Cs.|jtdd�\}}|j|j��WdQRXdS)NF)r')r~r;r{r|)rr�r>rrr�test_simple_with_statement�sz,NewIMAPTestsMixin.test_simple_with_statementc	CsT|jtdd�\}}|j|j�� }|�dd�|�|jd�WdQRX|�|j�dS)NF)r'r�r�)r~r;r{r|r�rr?�assertIsNone)rr�r>�imaprrr�test_with_statement�s
z%NewIMAPTestsMixin.test_with_statementc	Csh|jtdd�\}}|j|j��4}|�dd�|�|jd�|��|�|j�WdQRX|�|j�dS)NF)r'r�r�)	r~r;r{r|r�rr?�logoutr�)rr�r>r�rrr�test_with_statement_logout�sz,NewIMAPTestsMixin.test_with_statement_logoutcCsL|�t�\}}|�dd�\}}|�|d�|�|dd�|�|jd�dS)Nr�r�rarsLOGIN completed�AUTH)r~r;r�rr�)rr}r�r�r�rrr�
test_login�s
zNewIMAPTestsMixin.test_logincCs�|�t�\}}|�dd�\}}|�|d�|�|dd�|��\}}|�|d||f�|�|dd||f�|�|jd�dS)	Nr�r�rarsLOGIN completed�BYEsIMAP4ref1 Server logging outr)r~r;r�rr�r�)rr}r�r�r�rrr�test_logout�szNewIMAPTestsMixin.test_logoutcCsVGdd�dt�}|�|�\}}|�dd�|��\}}|�|d�|�|dd�dS)Nc@seZdZdd�ZdS)z,NewIMAPTestsMixin.test_lsub.<locals>.LsubCmdcSs|�d�|�|dd�S)Nz* LSUB () "." directoryArazLSUB completed)rJrO)rrMr_rrr�cmd_LSUB�s
z5NewIMAPTestsMixin.test_lsub.<locals>.LsubCmd.cmd_LSUBN)r1r2r3r�rrrr�LsubCmd�sr�r�r�rars() "." directoryA)r;r~r�Zlsubr)rr�r}r�r�r�rrr�	test_lsub�szNewIMAPTestsMixin.test_lsub)T)r1r2r3r}r~rsr�r�r�r�r�r�r�r�r�r�r�r�r�r�r
�patchr�r�r�r�r�r�r�rrrrrg�s0
!			

rgc@seZdZejZejZdS)�NewIMAPTestsN)	r1r2r3rr+r{r��	TCPServerrrrrrrr��sr�zSSL not availablec@s0eZdZeZeZdd�Zdd�Ze	dd��Z
dS)�NewIMAPSSLTestsc	Cszt�tj�}|�|jtj�|�|jd�|�t�|�	tj
d��.|�t�\}}|j
|jd|i�}|��WdQRXdS)NTz=IP address mismatch, certificate is not valid for '127.0.0.1'�ssl_context)r5r6�PROTOCOL_TLS_CLIENTr�verify_modeZ
CERT_REQUIRED�check_hostname�load_verify_locations�CAFILEr��CertificateErrorr~r;r{r|r�)rr�r�r>r}rrr�test_ssl_raises�s

zNewIMAPSSLTests.test_ssl_raisescCsFt�tj�}|�t�|�t�\}}|jd|jd|d�}|�	�dS)N�	localhostr)r�)
r5r6r�r�r�r~r;r{r|r�)rr�r�r>r}rrr�test_ssl_verifieds
z!NewIMAPSSLTests.test_ssl_verifiedcCsft�dtf��Ntj�|jd��2tj�|jd��|jddtd�WdQRXWdQRXWdQRXdS)Nr%�openZ_connectr��)�certfile)rZcheck_warnings�DeprecationWarningr
r��objectr{r8)rrrr�test_certfile_arg_warnsz&NewIMAPSSLTests.test_certfile_arg_warnN)r1r2r3�	IMAP4_SSLr{r4rrr�r�r	r�rrrrr��s
r�c@s$eZdZejZejZdd�Z	dd�Z
edd��Zedd��Z
ed	d
��Zedd��Zed
d��Zedd��ZGdd�de�Zedd��Zedd��Zedd��Zedd��Zedd��Zedd��Zedd ��Zed!d"��Zed#d$��Zed%d&��Zd'd(�Zed)d*��Z ed+d,��Z!ed-d.��Z"d/S)0�ThreadedNetworkedTestscCs�Gdd�d|j�}trtd�|||�}|�|j|j���trjtd�td|�td|j�td|j�tj	d|j|j
d	d
id�}d|_|��tr�td
�||fS)Nc@seZdZdd�ZdS)z4ThreadedNetworkedTests.make_server.<locals>.MyServercSs|�|�|���dS)N)rhri)rrjrkrrrrls
zAThreadedNetworkedTests.make_server.<locals>.MyServer.handle_errorN)r1r2r3rlrrrrr�sr�zcreating serverzserver createdzADDR =zCLASS =zHDLR =z
%s servingrng{�G�z�?)rorprqTzserver running)
rrrrArr|r&ZgetsocknameZRequestHandlerClassrurvrwryrz)rZaddr�hdlrr�r>r"rrr�make_servers&

z"ThreadedNetworkedTests.make_servercCs4trtd�|��|��|��tr0td�dS)Nzwaiting for serverZdone)rrAr�rirL)rr>rxrrr�reap_server7sz"ThreadedNetworkedTests.reap_serverc
cs4|�tjdf|�\}}z
|VWd|�||�XdS)Nr)r�rrtr�)rr�r>rxrrr�
reaped_server@s
z$ThreadedNetworkedTests.reaped_serverc	cs@|�|��,}|j|j�}z||fVWd|��XWdQRXdS)N)r�r{r|r�)rr�r>r}rrr�reaped_pairHs
z"ThreadedNetworkedTests.reaped_pairc	Cs.|�t��}|j|j�}|��WdQRXdS)N)r�r;r{r|r�)rr>r}rrr�test_connectQsz#ThreadedNetworkedTests.test_connectc	Cs�Gdd�dt�}|�|���\}}|�ddd��\}}|�|d�|�|jd�|�d�|�d	d
d�\}\}|�d|�|�d�|�d
�\}\}|�d|�WdQRXdS)Ncs4eZdZ�fdd�Zdd�Zdd�Zdd�Z�ZS)	zEThreadedNetworkedTests.test_bracket_flags.<locals>.BracketFlagHandlercsdddddg|_t���dS)NZAnsweredZFlaggedZDeletedZSeenZDraft)�flagsr<r`)r)r@rrr`bszLThreadedNetworkedTests.test_bracket_flags.<locals>.BracketFlagHandler.handlecss&|�d�dV|j_|�|dd�dS)Nr�razFAKEAUTH successful)rJr>r�rO)rrMr_rrrr�fs

zVThreadedNetworkedTests.test_bracket_flags.<locals>.BracketFlagHandler.cmd_AUTHENTICATEcSs^d�|j�}|�d|�d��|�d�|�d�d|}|�|�d��|�|dd�dS)	Nz \z* FLAGS (%s)r�s
* 2 EXISTSs
* 0 RECENTz-* OK [PERMANENTFLAGS %s \*)] Flags permitted.raz[READ-WRITE] SELECT completed.)rLr�rGrIrO)rrMr_Zflag_msg�msgrrr�
cmd_SELECTks

zPThreadedNetworkedTests.test_bracket_flags.<locals>.BracketFlagHandler.cmd_SELECTcSsf|d�d��d���}|j�|�dd�|j�}d|d|f}|�|�d��|�|d	d
�dS)NrQ�(�)z(FLAGS (%s))z \z
* %s FETCH %srr�razSTORE completed.)rBrXr��extendrLrGrIrO)rrMr_Z	new_flagsZ	flags_msgr�rrr�	cmd_STOREuszOThreadedNetworkedTests.test_bracket_flags.<locals>.BracketFlagHandler.cmd_STORE)r1r2r3r`r�r�r�rfrr)r@r�BracketFlagHandler`s
r�r�cSsdS)Nsfaker)r�rrrr�~rPz;ThreadedNetworkedTests.test_bracket_flags.<locals>.<lambda>ras
ZmFrZQ==
�test�1z+FLAGSz[test]s[test]ZPERMANENTFLAGS)r;r�r�rr�ZselectZstorer,)rr�r>r}rNr�r�rrr�test_bracket_flagsWs	

z)ThreadedNetworkedTests.test_bracket_flagsc	CsFGdd�dtj�}|�|�� }|jtjj|jf|j��WdQRXdS)Nc@seZdZdd�ZdS)z9ThreadedNetworkedTests.test_issue5949.<locals>.EOFHandlercSs|j�d�dS)Ns* OK)rCrD)rrrrr`�sz@ThreadedNetworkedTests.test_issue5949.<locals>.EOFHandler.handleN)r1r2r3r`rrrrr��sr�)	r�r�r�r)rr+r�r{r|)rr�r>rrr�test_issue5949�s
z%ThreadedNetworkedTests.test_issue5949c	CsDGdd�dt�}|�|�� }|jtjj|jf|j��WdQRXdS)Nc@seZdZdd�ZdS)zGThreadedNetworkedTests.test_line_termination.<locals>.BadNewlineHandlercSs|�d�|�|dd�dS)Ns* CAPABILITY IMAP4rev1 AUTH
razCAPABILITY completed)rFrO)rrMr_rrrrc�s
zVThreadedNetworkedTests.test_line_termination.<locals>.BadNewlineHandler.cmd_CAPABILITYN)r1r2r3rcrrrrr��sr�)r;r�r)rr+r�r{r|)rr�r>rrrr��s
z,ThreadedNetworkedTests.test_line_terminationc@s eZdZdZdd�Zdd�ZdS)z!ThreadedNetworkedTests.UTF8ServerzAUTH ENABLE UTF8=ACCEPTcCs|�|dd�dS)NrazENABLE successful)rO)rrMr_rrrr��sz,ThreadedNetworkedTests.UTF8Server.cmd_ENABLEccs&|�d�dV|j_|�|dd�dS)Nr�razFAKEAUTH successful)rJr>r�rO)rrMr_rrrr��s

z2ThreadedNetworkedTests.UTF8Server.cmd_AUTHENTICATEN)r1r2r3rbr�r�rrrrr��sr�c	CsL|�|j��6\}}|�|j�|�tjj|jd�|�|j�WdQRXdS)Nr�)	r�r�r�r�r)rr+r(r�)rr>r}rrrr��sz;ThreadedNetworkedTests.test_enable_raises_error_if_not_AUTHc	CsDGdd�d|j�}|�|��\}}|�tjj|jd�WdQRXdS)Nc@seZdZdZdS)zXThreadedNetworkedTests.test_enable_raises_error_if_no_capability.<locals>.NoEnableServerr�N)r1r2r3rbrrrr�NoEnableServer�sr�r�)r�r�r)rr+r(r�)rr�r>r}rrrr��sz@ThreadedNetworkedTests.test_enable_raises_error_if_no_capabilityc
CsnGdd�dt�}|�tjj��F|�|��0\}}|�dd�\}}|�|d�|�d�WdQRXWdQRXdS)Nc@seZdZdS)z\ThreadedNetworkedTests.test_enable_UTF8_raises_error_if_not_supported.<locals>.NonUTF8ServerN)r1r2r3rrrr�
NonUTF8Server�sr�r�r�razUTF8=ACCEPT)	r;r)rr+r(r�r�rr�)rr�r>r}r�r�rrrr��s
zEThreadedNetworkedTests.test_enable_UTF8_raises_error_if_not_supportedc		Cs�Gdd�d|j�}|�|���\}}|�|jd�|�ddd��\}}|�|d�|�|jd�|�d	�\}}|�|d�|�|jd
�d}|�ddd|�d
��\}}|�|d�|�|jd|�d
��WdQRXdS)
Nc@seZdZdd�ZdS)zMThreadedNetworkedTests.test_enable_UTF8_True_append.<locals>.UTF8AppendServercss&|�d�dV|j_|�|dd�dS)Nr�rar�)rJr>r�rO)rrMr_rrrr��s

zXThreadedNetworkedTests.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_APPENDN)r1r2r3r�rrrrr��sr�r�r�cSsdS)Nsfaker)r�rrrr��rPzEThreadedNetworkedTests.test_enable_UTF8_True_append.<locals>.<lambda>ras
ZmFrZQ==
zUTF8=ACCEPTzutf-8uSubject: üñí©öðézUTF8 (%s)
)	r�r�rr�r�r�r�r�rI)	rr�r>r}rNr�r�r�r�rrrr��s"z3ThreadedNetworkedTests.test_enable_UTF8_True_appendc	Cs||�|j��f\}}|�ddd��\}}|�|d�|�d�\}}|�|d�|�|j�|�tj	j
|jdd�WdQRXdS)Nr�cSsdS)Nsfaker)r�rrrr��rPzSThreadedNetworkedTests.test_search_disallows_charset_in_utf8_mode.<locals>.<lambda>razUTF8=ACCEPTr�r�)r�r�r�rr�r�r�r)rr+r(r�)rr>r}r�r�rrrr��szAThreadedNetworkedTests.test_search_disallows_charset_in_utf8_modec
CsXGdd�dt�}|�|��4\}}|�tjj��|�ddd��WdQRXWdQRXdS)Nc@seZdZdd�ZdS)z;ThreadedNetworkedTests.test_bad_auth_name.<locals>.MyServercSs|�|dd�|d��dS)Nr�z#unrecognized authentication type {}r)rOr�)rrMr_rrrr��szLThreadedNetworkedTests.test_bad_auth_name.<locals>.MyServer.cmd_AUTHENTICATEN)r1r2r3r�rrrrr��sr�r�cSsdS)Nrrrrrrr��rPz;ThreadedNetworkedTests.test_bad_auth_name.<locals>.<lambda>)r;r�r)rr+r(r�)rr�r>r}rrrr��sz)ThreadedNetworkedTests.test_bad_auth_namec
Cs\Gdd�dt�}|�|��8\}}|�tjj��|�ddd��\}}WdQRXWdQRXdS)Nc@seZdZdd�ZdS)zDThreadedNetworkedTests.test_invalid_authentication.<locals>.MyServercss$|�d�dV|_|�|dd�dS)Nr�r�z[AUTHENTICATIONFAILED] invalid)rJr�rO)rrMr_rrrr�s
zUThreadedNetworkedTests.test_invalid_authentication.<locals>.MyServer.cmd_AUTHENTICATEN)r1r2r3r�rrrrr��sr�r�cSsdS)Nsfaker)r�rrrr�rPzDThreadedNetworkedTests.test_invalid_authentication.<locals>.<lambda>)r;r�r)rr+r(r�)rr�r>r}rNr�rrrr��sz2ThreadedNetworkedTests.test_invalid_authenticationc	Cs�Gdd�dt�}|�|��8\}}|�ddd��\}}|�|d�|�|jd�WdQRX|�|��8\}}|�ddd��\}}|�|d�|�|jd�WdQRXdS)	Nc@seZdZdd�ZdS)zBThreadedNetworkedTests.test_valid_authentication.<locals>.MyServercss&|�d�dV|j_|�|dd�dS)Nr�razFAKEAUTH successful)rJr>r�rO)rrMr_rrrr�s

zSThreadedNetworkedTests.test_valid_authentication.<locals>.MyServer.cmd_AUTHENTICATEN)r1r2r3r�rrrrr�
sr�r�cSsdS)Nsfaker)r�rrrr�rPzBThreadedNetworkedTests.test_valid_authentication.<locals>.<lambda>ras
ZmFrZQ==
cSsdS)Nr�r)r�rrrr�rP)r;r�r�rr�)rr�r>r}rNr�rrr�test_valid_authentication
sz0ThreadedNetworkedTests.test_valid_authenticationc	Cs�Gdd�dt�}|�|��6\}}|�d|jk�|�dd�\}}|�|d�WdQRX|�|��6\}}|�d|jk�|�dd�\}}|�|d�WdQRXdS)Nc@seZdZdZdd�ZdS)z?ThreadedNetworkedTests.test_login_cram_md5.<locals>.AuthHandlerzLOGINDISABLED AUTH=CRAM-MD5css:|�d�dV}|dkr(|�|dd�n|�|dd�dS)Nz:+ PDE4OTYuNjk3MTcwOTUyQHBvc3RvZmZpY2UucmVzdG9uLm1jaS5uZXQ=s2dGltIGYxY2E2YmU0NjRiOWVmYTFjY2E2ZmZkNmNmMmQ5ZjMy
razCRAM-MD5 successfulr�z	No access)rJrO)rrMr_r�rrrr�'s

zPThreadedNetworkedTests.test_login_cram_md5.<locals>.AuthHandler.cmd_AUTHENTICATEN)r1r2r3rbr�rrrrr�#sr�z
AUTH=CRAM-MD5r�r�rastanstaaftanstaaf)r;r�r�rbr�r)rr�r>r}r�r�rrr�test_login_cram_md5 sz*ThreadedNetworkedTests.test_login_cram_md5c
Cs\Gdd�dt�}|�|��8\}}|�tjj��|�ddd��\}}WdQRXWdQRXdS)Nc@seZdZdd�ZdS)zDThreadedNetworkedTests.test_aborted_authentication.<locals>.MyServercss>|�d�dV|_|jdkr,|�|dd�n|�|dd�dS)Nr�s*
r�z[AUTHENTICATIONFAILED] abortedrazMYAUTH successful)rJr�rO)rrMr_rrrr�As


zUThreadedNetworkedTests.test_aborted_authentication.<locals>.MyServer.cmd_AUTHENTICATEN)r1r2r3r�rrrrr�?sr�r�cSsdS)Nr)r�rrrr�LrPzDThreadedNetworkedTests.test_aborted_authentication.<locals>.<lambda>)r;r�r)rr+r(r�)rr�r>r}rNr�rrrr�<sz2ThreadedNetworkedTests.test_aborted_authenticationc	CsDGdd�dt�}|�|�� }|jtjj|jf|j��WdQRXdS)Nc@seZdZdd�ZdS)z?ThreadedNetworkedTests.test_linetoolong.<locals>.TooLongHandlercSs|j�dtjdd�dS)Ns* OK �xs
)rCrDrZ_MAXLINE)rrrrr`QszFThreadedNetworkedTests.test_linetoolong.<locals>.TooLongHandler.handleN)r1r2r3r`rrrrr�Psr�)r;r�r)rr+r(r{r|)rr�r>rrrr�Os
z'ThreadedNetworkedTests.test_linetoolongc
Cs2|�t��}|j|j��WdQRXWdQRXdS)N)r�r;r{r|)rr>rrrr�Ysz1ThreadedNetworkedTests.test_simple_with_statementc
CsX|�t��D}|j|j�� }|�dd�|�|jd�WdQRX|�|j�WdQRXdS)Nr�r�)r�r;r{r|r�rr?r�)rr>r�rrrr�`s
z*ThreadedNetworkedTests.test_with_statementc
Csl|�t��X}|j|j��4}|�dd�|�|jd�|��|�|j�WdQRX|�|j�WdQRXdS)Nr�r�)	r�r;r{r|r�rr?r�r�)rr>r�rrrr�hsz1ThreadedNetworkedTests.test_with_statement_logoutN)#r1r2r3r�r�rrrr+r{r�r�rr�r�rr�r�r�r�r;r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�rrrrr�s2 		1
	


r�c@s eZdZeZeZedd��ZdS)�ThreadedNetworkedTestsSSLc
Cs�t�tj�}|�t�|�tjd��6|�t�� }|j	|j
d|i�}|��WdQRXWdQRX|�t��$}|j	d|j
d|d�}|��WdQRXdS)Nz=IP address mismatch, certificate is not valid for '127.0.0.1'r�r�r)r�)r5r6r�r�r�r�r�r�r;r{r|r�)rr�r>r}rrrr�ys

z+ThreadedNetworkedTestsSSL.test_ssl_verifiedN)	r1r2r3r4rrr�r{rr�rrrrr�tsr�Znetworkznetwork resource disabledz'cyrus.andrew.cmu.edu blocks connectionsc@sBeZdZdZdZdZdZejZ	dd�Z
dd�Zd	d
�Zdd�Z
d
S)�RemoteIMAPTestzcyrus.andrew.cmu.edur�Z	anonymousr�c	Cs,t|j��|�|j|j�|_WdQRXdS)N)r�hostr{�portr>)rrrr�setUp�szRemoteIMAPTest.setUpc	Cs.|jdk	r*t|j��|j��WdQRXdS)N)r>rr�r�)rrrr�tearDown�s
zRemoteIMAPTest.tearDownc	Cszt|j��fx|jjD]}|�|t�qW|�d|jj�|�d|jj�|j�|j|j	�}|�
|dd�WdQRXdS)N�
LOGINDISABLEDzAUTH=ANONYMOUSrra)rr�r>rb�assertIsInstance�strr,r��username�passwordr)r�cap�rsrrr�test_logincapa�szRemoteIMAPTest.test_logincapac	Cs<t|j��(|j��}d|_|�|dd|�WdQRXdS)Nrr�)rr�r>r�r)rrrrrr��s
zRemoteIMAPTest.test_logoutN)r1r2r3r�r�rrrr+r{rrr	r�rrrrr��s	r�cs$eZdZ�fdd�Zdd�Z�ZS)�RemoteIMAP_STARTTLSTestc	s>t���t|j�� |j��}|�|dd�WdQRXdS)Nrra)r<rrr�r>Zstarttlsr)rr)r@rrr�s

zRemoteIMAP_STARTTLSTest.setUpcCs2x|jjD]}|�|t�q
W|�d|jj�dS)Nr)r>rbrr�assertNotIn)rrrrrr	�sz&RemoteIMAP_STARTTLSTest.test_logincapa)r1r2r3rr	rfrr)r@rr
�sr
c@sTeZdZdZeZdd�Zdd�Zdd�Zdd	�Z	d
d�Z
dd
�Zdd�Zdd�Z
dS)�RemoteIMAP_SSLTesti�cCsdS)Nr)rrrrr�szRemoteIMAP_SSLTest.setUpcCsdS)Nr)rrrrr�szRemoteIMAP_SSLTest.tearDowncCs(t�tj�}d|_tj|_|�t�|S)NF)r5r6r�r�Z	CERT_NONEr�r7r8)rr�rrr�create_ssl_context�s

z%RemoteIMAP_SSLTest.create_ssl_contextcCslz\x|jD]}|�|t�q
W|�d|j�|�d|j�|�|j|j�}|�|dd�Wd|�	�XdS)Nrz
AUTH=PLAINrra)
rbrrrr,r�rrrr�)rr>rrrrr�check_logincapa�sz"RemoteIMAP_SSLTest.check_logincapac	Cs4t|j�� |�|j|j�}|�|�WdQRXdS)N)rr�r{r�r)r�_serverrrrr	�sz!RemoteIMAP_SSLTest.test_logincapac	CsDt|j��0|�|j|j�}|��}|�|dd|�WdQRXdS)Nrr�)rr�r{r�r�r)rrrrrrr��szRemoteIMAP_SSLTest.test_logoutc	Cs:t|j��&|jt|j|j|jt|��d�WdQRXdS)N)r�r�)rr�r)�
ValueErrorr{r�r8r
)rrrr�#test_ssl_context_certfile_exclusive�sz6RemoteIMAP_SSLTest.test_ssl_context_certfile_exclusivec	Cs:t|j��&|jt|j|j|jt|��d�WdQRXdS)N)Zkeyfiler�)rr�r)rr{r�r8r
)rrrr�"test_ssl_context_keyfile_exclusive�sz5RemoteIMAP_SSLTest.test_ssl_context_keyfile_exclusiveN)r1r2r3r�r�r{rrr
rr	r�rrrrrrr�sr�__main__)6r�r�
contextlibrr-rZos.path�osr�rrrur&Ztest.supportrrrrrr	Zunittestr
rrr
r5�ImportError�pathrL�dirname�__file__�curdirr8r�ZTestCaserr�r4r�r�r;rgr�Z
skipUnlessr�r�r�Zis_resource_enabled�skipr�r
rr1�mainrrrr�<module>sf 
@	R*$c 


4