shell bypass 403
o
` b>� � @ 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)�self�t0�tt� r �:/usr/local/python-3.10/lib/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_ALL�de_DE�fr_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�assertIn� 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. r0 r= 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)r2 �accept�ssl�
SSLContextZPROTOCOL_TLS_SERVER�load_cert_chain�CERTFILEZ wrap_socket)r Z newsocketZfromaddr�contextZ
connstreamr r r � get_requesta s
zSecureTCPServer.get_requestN)r> r? r@ rI r r r r rB _ � rB c @ � e Zd ZdS )rB N�r> r? r@ r r r r rB 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 )�SimpleIMAPHandlerNr1 c s t � �� d| j_d | j_d S �NF)�super�setup�server� is_selected�loggedr �� __class__r r rQ 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�
)r] r[ r r r �
_send_line� � z SimpleIMAPHandler._send_linec C s | � |�d�� d S )N�ASCII)r_ �encoder[ r r r �_send_textline� s z SimpleIMAPHandler._send_textlinec C s | � d�|||f�� d S )N� )rc �join)r �tag�coder\ 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-q.q 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 � r^ zGOT: %rra r � Zcmd_�BADz unknown)rc Zrfile�readr6 �endswithr rW rX � continuation�send�
StopIteration�decode�split�hasattr�getattr�nextrh )r �line�partZ splitlinerf �cmd�argsrn r r r �handle� sH
�
�
� ��zSimpleIMAPHandler.handlec C s4 | j rd| j nd}| �d| � | �|dd� d S )Nz
IMAP4rev1 Z IMAP4rev1z
* CAPABILITY �OK�CAPABILITY completed)� capabilitiesrc rh )r rf ry Zcapsr r r �cmd_CAPABILITY� s ��z SimpleIMAPHandler.cmd_CAPABILITYc C s$ d | j _| �d� | �|dd� d S )Nz"* BYE IMAP4ref1 Server logging outr{ zLOGOUT completed)rR rT rc rh �r rf ry r r r �
cmd_LOGOUT� �
z SimpleIMAPHandler.cmd_LOGOUTc C s |d | j _| �|dd� d S )Nr r{ zLOGIN completed)rR rT rh r r r r � cmd_LOGIN� s zSimpleIMAPHandler.cmd_LOGINc C s$ d| j _| �d� | �|dd� d S )NT�
* 2 EXISTSr{ � [READ-WRITE] SELECT completed.)rR rS r_ rh r r r r �
cmd_SELECT� r� z SimpleIMAPHandler.cmd_SELECTc C s4 | j jrd| j _| �|dd� d S | �|dd� d S )NFr{ z*Returned to authenticated state. (Success)rk zNo mailbox selected)rR rS rh r r r r � cmd_UNSELECT� s z SimpleIMAPHandler.cmd_UNSELECT)r> r? r@ r �LOOPBACK_TIMEOUT�timeoutrn r} rQ r] r_ rc rh rz r~ r� r� r� r� �
__classcell__r r rU r rN r s *rN 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 @ rA )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�r> r? r@ r� r r r r �
TestTCPServer� � r� r z-server�
poll_interval�{�G�z�?��name�target�kwargsT)� server_classZ
addCleanup�_cleanupr �HOSTrR � threading�ThreadZ_testMethodName�
serve_forever�thread�daemon�start�
imap_class�server_address�client)r Z imap_handlerr3 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�shutdownrR r� r
� 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 @ rA )zONewIMAPTestsMixin.test_EOF_without_complete_welcome_message.<locals>.EOFHandlerc S � | j �d� d S �Ns * OK�rY rZ r r r r rz s zVNewIMAPTestsMixin.test_EOF_without_complete_welcome_message.<locals>.EOFHandler.handleN�r> r? r@ rz r r r r �
EOFHandler r� r� F�r3 ) � socketserver�StreamRequestHandlerr� r5 r r7 �abortr� r� )r r� �_rR 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 @ rA )zBNewIMAPTestsMixin.test_line_termination.<locals>.BadNewlineHandlerc S � | � d� | �|dd� d S �Ns * CAPABILITY IMAP4rev1 AUTH
r{ r| �r] rh r r r r r~ �
zQNewIMAPTestsMixin.test_line_termination.<locals>.BadNewlineHandler.cmd_CAPABILITYN�r> r? r@ r~ r r r r �BadNewlineHandler r� r� Fr� )rN r� r5 r r7 r� r� r� )r r� r� rR 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�r> r? r@ r} r r r r �
EnableHandler � r� zENABLE.*NONAUTH�foo) rN r� � assertFalse� utf8_enabled�assertRaisesRegexr r7 r4 �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� rN r� r r7 r4 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�passr{ r� � UTF8=ACCEPT) r� rN �loginr r� r r7 r4 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 �Nr{ zENABLE successful�rh r r r r �
cmd_ENABLE1 r` zSNewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_ENABLEc s �( � | � d� d V | j_| �|dd� d S �N�+r{ zFAKEAUTH successful�rc rR �responserh r r r r �cmd_AUTHENTICATE3 � �
zYNewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_AUTHENTICATEc s r� �Nr� r{ Zokayr� r r r r �
cmd_APPEND7 r� zSNewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_APPENDN)r> r? r@ r} 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>r{ �
ZmFrZQ==
r� �utf-8� Subject: üñí©öðé� UTF8 (%s)
) rN r� r � _encoding� authenticater� r� �appendrb ) r r� r� rR rg 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� r r r r r� L r` z[NewIMAPTestsMixin.test_search_disallows_charset_in_utf8_mode.<locals>.UTF8Server.cmd_ENABLEc s r� r� r� r r r r r� N r� zaNewIMAPTestsMixin.test_search_disallows_charset_in_utf8_mode.<locals>.UTF8Server.cmd_AUTHENTICATEN�r> r? r@ r} 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>r{ r� z
charset.*UTF8r� �bar) rN r� r� r r� �
assertTruer� r� r r7 r4 �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 @ rA )z6NewIMAPTestsMixin.test_bad_auth_name.<locals>.MyServerc S � | � |dd�|d �� d S �N�NOz#unrecognized authentication type {}r �rh �formatr r r r r� ] s �zGNewIMAPTestsMixin.test_bad_auth_name.<locals>.MyServer.cmd_AUTHENTICATEN�r> 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>�rN r� r� r r7 r4 r� �r r� r� r� r r r �test_bad_auth_name[ s
�"�z$NewIMAPTestsMixin.test_bad_auth_namec C r� )Nc @ rA )z?NewIMAPTestsMixin.test_invalid_authentication.<locals>.MyServerc s �& � | � d� d V | _| �|dd� d S �Nr� r� z [AUTHENTICATIONFAILED] invalid�rc r� rh r 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 @ rA )zCNewIMAPTestsMixin.test_valid_authentication_bytes.<locals>.MyServerc s r� r� r� r 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>r{ r� �rN r� r� r r� �r r� r� rR rg r� r r r � test_valid_authentication_bytesp �
z1NewIMAPTestsMixin.test_valid_authentication_bytesc C r )Nc @ rA )zHNewIMAPTestsMixin.test_valid_authentication_plain_text.<locals>.MyServerc s r� r� r� r 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>r{ 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
r{ zCRAM-MD5 successfulr� z No access�rc rh �r rf ry �rr r r r� � � �
zQNewIMAPTestsMixin.test_login_cram_md5_bytes.<locals>.AuthHandler.cmd_AUTHENTICATEN�r> r? r@ r} r� r r r r � AuthHandler� � r �
AUTH=CRAM-MD5�tim� tanstaaftanstaafr{ �rN r� r� r} �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 �tanstaaftanstaafr{ r! r# r r r � test_login_cram_md5_plain_text� r& z0NewIMAPTestsMixin.test_login_cram_md5_plain_textc C r� )Nc @ rA )z?NewIMAPTestsMixin.test_aborted_authentication.<locals>.MyServerc s �B � | � d� d V | _| jdkr| �|dd� d S | �|dd� d S �Nr� s *
r� z [AUTHENTICATIONFAILED] abortedr{ zMYAUTH successfulr r 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 @ rA )z:NewIMAPTestsMixin.test_linetoolong.<locals>.TooLongHandlerc S r� )Ns * OK xxxxxxxxxxx
r� r r r r rz � � zANewIMAPTestsMixin.test_linetoolong.<locals>.TooLongHandler.handleNr� r r r r �TooLongHandler� r� r/ Fr� zgot more than 10 bytes)rN r� r� r r7 r4 r� r� )r r/ r� rR 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� rN r� r� )r r� rR 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� rN r� r� r �sockr� r� r r� r5 �
ValueError)r r� rR �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 @ rA )zQNewIMAPTestsMixin.test_imaplib_timeout_functionality_test.<locals>.TimeoutHandlerc S s t �d� t�| � d S r )r �sleeprN rz r r r r rz � �
zXNewIMAPTestsMixin.test_imaplib_timeout_functionality_test.<locals>.TimeoutHandler.handleNr� r r r r �TimeoutHandler� r� r: r r2 g����MbP?r3 )rN r� r� r5 � TimeoutErrorr� )r r: r� rR r6 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� rN r� r� r� r rT � assertIsNone�r r� rR �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 r= ) r� rN r� r� r� r rT �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� r{ r � LOGIN completed�AUTH)r� rN 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� r{ r rD �BYEs IMAP4ref1 Server logging outr� )r� rN r� r rB r� r� r r r � test_logout� s z NewIMAPTestsMixin.test_logoutc C sV G dd� dt �}| �|�\}}|�dd� |�� \}}| �|d� | �|d d� d S )Nc @ rA )z,NewIMAPTestsMixin.test_lsub.<locals>.LsubCmdc S s | � d� | �|dd�S )Nz* LSUB () "." directoryAr{ zLSUB completedr r r r r �cmd_LSUB r9 z5NewIMAPTestsMixin.test_lsub.<locals>.LsubCmd.cmd_LSUBN)r> r? r@ rI r r r r �LsubCmd r� rJ r� r� r{ r s () "." directoryA)rN r� r� Zlsubr )r rJ 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� r{ r � 2s* Returned to authenticated state. (Success)rE )r� rN r� �selectr Zunselectr� r� r r r �
test_unselect s z NewIMAPTestsMixin.test_unselect)T)"r> r? r@ r� r� r� r� r� r� r� r� r� r� r r r r r �requires_hashdigestr% r( r, r �patchr0 r1 r7 r< rA rC rF rH rK rN r r r r r� � s>
!
r� c @ s e Zd ZejZejZdS )� NewIMAPTestsN) r> r? r@ r r7 r� r� � TCPServerr� r r r r rQ s
rQ 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)rD rE �PROTOCOL_TLS_CLIENTr � verify_modeZ
CERT_REQUIRED�check_hostname�load_verify_locations�CAFILEr� �CertificateErrorr� rN r� r� r� �r rU r� rR 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 )Nr2 r �rU )
rD rE rV rY rZ r� rN r� r� r� r\ 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 )Nr1 �openZ_connectr2 � )�certfile)r Zcheck_warnings�DeprecationWarningr rP �objectr� rG r r r r �test_certfile_arg_warnA s �� ��"�z&NewIMAPSSLTests.test_certfile_arg_warnN) r> r? r@ � IMAP4_SSLr� rB r� r] r_ r re r r r r rS # s rS c @ sf 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"ejd,d-� �Z#ejd.d/� �Z$ejd0d1� �Z%eje&d2d3� ��Z'd4S )5�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 @ rA )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 rW r r� r2 � getsocknameZRequestHandlerClassr� r� r� r� r� )r r6 �hdlrr� rR r- r r r � make_serverL s(
�z"ThreadedNetworkedTests.make_serverc C s8 t rtd� |�� |�� |�� t rtd� d S d S )Nzwaiting for server�done)r rW r� r� re )r rR r� r r r � reap_serverl s �z"ThreadedNetworkedTests.reap_serverc
c s@ � | � tjdf|�\}}z |V W | �||� d S | �||� w )Nr )rj r r� rl )r ri rR 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/ )rm r� r� rB )r ri rR 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/ )rm rN r� r� r� �r rR 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)�flagsrP rz r rU r r rz � r9 zLThreadedNetworkedTests.test_bracket_flags.<locals>.BracketFlagHandler.handlec s r� r� r� r 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.r{ r� )re rq r_ rb rh )r rf ry 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 ) Nrj �(�)z (FLAGS (%s))rr z
* %s FETCH %sr r� r{ zSTORE completed.)rX rr rq �extendre r_ rb rh )r rf ry Z new_flagsZ flags_msgrs r r r � cmd_STORE� s zOThreadedNetworkedTests.test_bracket_flags.<locals>.BracketFlagHandler.cmd_STORE)r> r? r@ rz r� r� rw r� r r rU r �BracketFlagHandler� s
rx r� c S r� r� r r� r r r r� � r� z;ThreadedNetworkedTests.test_bracket_flags.<locals>.<lambda>r{ r� �test� 1z+FLAGSz[test]s [test]ZPERMANENTFLAGS)rN rn r� r r� rM �storer8 )r rx rR r� rg 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 @ rA )z9ThreadedNetworkedTests.test_issue5949.<locals>.EOFHandlerc S r� r� r� r r r r rz � r. z@ThreadedNetworkedTests.test_issue5949.<locals>.EOFHandler.handleNr� r r r r r� � r� r� ) r� r� rm r5 r r7 r� r� r� )r r� rR 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 @ rA )zGThreadedNetworkedTests.test_line_termination.<locals>.BadNewlineHandlerc S r� r� r� r r r r r~ � r� zVThreadedNetworkedTests.test_line_termination.<locals>.BadNewlineHandler.cmd_CAPABILITYNr� r r r r r� � rJ r� )rN rm r5 r r7 r� r� r� )r r� rR r r r r� � s
�
�"�z,ThreadedNetworkedTests.test_line_terminationc @ r� )z!ThreadedNetworkedTests.UTF8Serverr� c C r� r� r� r r r r r� � r` z,ThreadedNetworkedTests.UTF8Server.cmd_ENABLEc c r� r� r� r 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� ) rn r� r� r� r5 r r7 r4 r� ro 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>.NoEnableServerrE Nr� r r r r �NoEnableServer� r� r r� )r� rn r5 r r7 r4 r� )r r rR 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 @ rK )z\ThreadedNetworkedTests.test_enable_UTF8_raises_error_if_not_supported.<locals>.NonUTF8ServerNrL r r r r �
NonUTF8Server� rM r� r� r� r{ r� ) rN r5 r r7 r4 rn r� r r� )r r� rR 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 @ rA )zMThreadedNetworkedTests.test_enable_UTF8_True_append.<locals>.UTF8AppendServerc s r� r� r� r r r r r� r� zXThreadedNetworkedTests.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_APPENDN)r> 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>r{ r� r� r� r� r� ) r� rn r r� r� r� r� r� rb ) r r� rR r� rg 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>r{ r� r� r� ) rn r� r� r r� r� r� r5 r r7 r4 r� )r rR 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 @ rA )z;ThreadedNetworkedTests.test_bad_auth_name.<locals>.MyServerc S r� r� r� r r r r r� ) s
�zLThreadedNetworkedTests.test_bad_auth_name.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� ' rJ r� r� c S r� r r r r r r r� / r� z;ThreadedNetworkedTests.test_bad_auth_name.<locals>.<lambda>�rN rn r5 r r7 r4 r� )r r� rR 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 @ rA )zDThreadedNetworkedTests.test_invalid_authentication.<locals>.MyServerc s r r r r r r r r� 6 r zUThreadedNetworkedTests.test_invalid_authentication.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� 4 rJ r� r� c S r� r� r r� r r r r� = r� zDThreadedNetworkedTests.test_invalid_authentication.<locals>.<lambda>r� �r r� rR r� rg 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 @ rA )zBThreadedNetworkedTests.test_valid_authentication.<locals>.MyServerc s r� r� r� r r r r r� D r� zSThreadedNetworkedTests.test_valid_authentication.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� B rJ r� r� c S r� r� r r� r r r r� J r� zBThreadedNetworkedTests.test_valid_authentication.<locals>.<lambda>r{ r� c S r� r r r� r r r r� P r� )rN rn r� r r� r� 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' r{ r )rN rn r� r} r" r )r r rR r� r$ r� r r r �test_login_cram_md5U s �"�z*ThreadedNetworkedTests.test_login_cram_md5c C r� )Nc @ rA )zDThreadedNetworkedTests.test_aborted_authentication.<locals>.MyServerc s r) r* r r r r r r� w s �
zUThreadedNetworkedTests.test_aborted_authentication.<locals>.MyServer.cmd_AUTHENTICATENr� r r r r r� u rJ r� r� c S r+ r/ r r� r r r r� � r� zDThreadedNetworkedTests.test_aborted_authentication.<locals>.<lambda>r� r� r r r r, r s ��"�z2ThreadedNetworkedTests.test_aborted_authenticationc C r~ )Nc @ rA )z?ThreadedNetworkedTests.test_linetoolong.<locals>.TooLongHandlerc S s | j �dtjd d � d S )Ns * OK � xr^ )rY rZ r Z_MAXLINEr r r r rz � s zFThreadedNetworkedTests.test_linetoolong.<locals>.TooLongHandler.handleNr� r r r r r/ � r� r/ )rN rm r5 r r7 r4 r� r� )r r/ rR r r r r0 � s
�
�"�z'ThreadedNetworkedTests.test_linetoolongc 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/ )rm rN r� r� )r rR r r r r1 � 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� )rm rN r� r� r� r rT r> �r rR r@ r r r rA � 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� ) rm rN r� r� r� r rT rB r> r� r r r rC � s �"�z1ThreadedNetworkedTests.test_with_statement_logoutc
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-WRITEri Z_mesgz)untagged responses dump:READ-WRITE: [b'']) rm rN r� r� r rP rd Z_dump_urZassert_called_with)r Zuntagged_resp_dictrR r@ Z mock_mesgr r r � test_dump_ur� s
� �� ��"�z#ThreadedNetworkedTests.test_dump_urN)(r> r? r@ r� rR r� r r7 r� rj rl r rm rn r
� reap_threadsrp r| r} r� rN r� r� r� r� r� r� r r r� r rO r� r, r0 r1 rA rC r r� r r r r rg H sb
0
rg 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 )NrT rU r2 r r^ ) rD rE rV rY rZ r� r[ rm rN r� r� r� )r rU rR r� r r r r_ � s(
�
�
�� � �
"�z+ThreadedNetworkedTestsSSL.test_ssl_verifiedN)
r> r? r@ rB r� rf r� r
r� r_ 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.edura Z anonymousr� c C sD t �| j�� | �| j| j�| _W d � d S 1 sw Y d S r/ )r �transient_internet�hostr� �portrR 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/ )rR r r� r� rB 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 r{ ) r r� r� rR r} �assertIsInstance�strr8 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 rG )r r� r� rR rB r �r r� r r r rH � s
"�zRemoteIMAPTest.test_logoutN)r> r? r@ r� r� r� r� r r7 r� r� r� r� rH 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 r{ )rP r� r r� r� rR Zstarttlsr r� rU r r r� � s
"�z RemoteIMAP_STARTTLSTest.setUpc C s. | j jD ]}| �|t� q| �d| j j� d S )Nr� )rR r} r� r� � assertNotIn)r r� r r r r� s z&RemoteIMAP_STARTTLSTest.test_logincapa)r> r? r@ r� r� r� r r rU 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 rO )rD rE rV rX Z CERT_NONErW rF rG )r rU 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 r{ )
r} r� r� r� r8 r� r� r� r rB )r rR 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� $ 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� rB r )r r� r� r r r rH ) 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)rb rU � r r� r� r5 r5 r� r� rG r� r r r r �#test_ssl_context_certfile_exclusive/ � �"�z6RemoteIMAP_SSLTest.test_ssl_context_certfile_exclusivec C r� )N)ZkeyfilerU r� r r r r �"test_ssl_context_keyfile_exclusive5 r� z5RemoteIMAP_SSLTest.test_ssl_context_keyfile_exclusiveN)r> r? r@ r� rf r� r� r� r� r� r� rH r� r� r r r r r� s r� �__main__)7ry r Z test.supportr �
contextlibr r Zos.path�osr� r r r� r2 r r r r r r
r Zunittestr r
r r rD � ImportError�pathre �dirname�__file__�curdirrG rZ ZTestCaser rR rB rf r� rN r� rQ Z
skipUnlessrS rg r� Zis_resource_enabled�skipr� r� r� r> �mainr r r r �<module> sv � @ _ O
$
r
�
�
4 �