Your IP : 216.73.216.85
��Yfٍ � @ s# d d l m Z e j d � Z d d l m Z d d l Z d d l Z d d l Z d d l
Z
d d l Z d d l Z d d l
m Z m Z m Z m Z m Z d d l Z d d l m Z d d l m Z m Z m Z y d d l Z Wn e k
rd Z Yn Xe j j e j j e � p#e j d � Z e j j e j j e � pMe j d � Z Gd
d � d e j! � Z" e r�Gd d
� d
e j# � Z$ e j% Z% n Gd d
� d
� Z$ d Z% Gd d � d e j& � Z' Gd d � d � Z( Gd d � d e( e j! � Z) e j* e d � Gd d � d e( e j! � � Z+ Gd d � d e j! � Z, e j* e d � Gd d � d e, � � Z- e j* e j. d � d � Gd d � d e j! � � Z/ e j* e d � e j* e j. d � d � Gd d! � d! e/ � � � Z0 e j* e d � Gd" d# � d# e/ � � Z1 e2 d$ k r e j3 � d S)%� )�support� threading)�contextmanagerN)� reap_threads�verbose�transient_internet� run_with_tz�run_with_locale)�mock)�datetime�timezone� timedeltaz keycert3.pemz pycacert.pemc @ sv e Z d Z d d � Z e d � d d � � Z d d � Z e d d d
� e d � d d
� � � Z d d � Z d S)� TestImaplibc
C s� t j d
� } t j d � } | j t j | � | � t j d � } | j t j | � | � t j d � } | j t j | � | � d S) N�� � 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")���r r ) r r r r r r r r r )�calendarZtimegm�imaplib�Internaldate2tuple� assertEqual�timeZmktime)�selfZt0Ztt� r �6/opt/alt/python35/lib64/python3.5/test/test_imaplib.py�test_Internaldate2tuple s z#TestImaplib.test_Internaldate2tuplezMST+07MDT,M4.1.0,M10.5.0c C s& | j t j d � t j 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= d d t j d � d d t j d t t d d � � � d g S)Ni �5wg e��A�� � � �! � r r � �< z "18-May-2033 05:33:20 +0200"r r r ) r r r r r r r r r r r ) r r r r r r r r r �x i )r Z localtimer Z
fromtimestampr r
)r r r r �
timevalues2 s zTestImaplib.timevalues�LC_ALLZde_DEZfr_FRzSTD-1DST,M3.2.0,M11.1.0c C s@ d } x3 | j � D]% } t j | � } | j | | � q Wd S)Nz "18-May-2033 05:33:20 +0200")r$ r �Time2Internaldater )r Zexpected�tZinternalr r r �test_Time2Internaldate: s z"TestImaplib.test_Time2Internaldatec C s( x! | j � D] } t j | � q
Wd S)N)r$ r r& )r r' r r r �,test_that_Time2Internaldate_returns_a_resultE s z8TestImaplib.test_that_Time2Internaldate_returns_a_resultN)
�__name__�
__module__� __qualname__r r r r$ r r( r) r r r r r s r c @ s e Z d Z d d � Z d S)�SecureTCPServerc C s: | j j � \ } } t j | d d d t �} | | f S)NZ server_sideT�certfile)�socketZaccept�sslZ wrap_socket�CERTFILE)r Z newsocketZfromaddrZ
connstreamr r r � get_requestP s
zSecureTCPServer.get_requestN)r* r+ r, r2 r r r r r- N s r- c @ s e Z d Z d S)r- N)r* r+ r, r r r r r- [ s c s� e Z d Z d Z d Z d Z � f d d � Z d d � Z d d � Z d
d � Z d d
� Z
d d � Z d d � Z d d � Z
d d � Z � S)�SimpleIMAPHandlerr N� c s t � j � d | j _ d S)N)�super�setup�server�logged)r )� __class__r r r6 f s
zSimpleIMAPHandler.setupc C s. t r t d | j � � | j j | � d S)NzSENT: %r)r �print�strip�wfile�write)r �messager r r �_sendj s zSimpleIMAPHandler._sendc C s | j | d � d S)Ns
)r? )r r> r r r �
_send_lineo s z SimpleIMAPHandler._send_linec C s | j | j d � � d S)N�ASCII)r@ �encode)r r> r r r �_send_textliner s z SimpleIMAPHandler._send_textlinec C s# | j d j | | | f � � d S)N� )rC �join)r �tag�coder> r r r � _send_taggedu s z SimpleIMAPHandler._send_taggedc C sq | j d � x]d } x] y0 | j j d � } | d k r> d S| | 7} Wn t k
ra d SYn X| j d � r Pq Wt r� t d | j � � | j r� y | j j | � Wq t
k
r� d | _ Yq Xq | j d � j � } | d } | d } | d d � } t
| d | � rRt | d | � | | � } | ri| | _ t | � q | j | d
| d � q Wd S) Nz* OK IMAP4rev1� r s
zGOT: %rrA r r! Zcmd_ZBADz unknown)rC Zrfile�read�OSError�endswithr r: r; � continuation�send�
StopIteration�decode�split�hasattr�getattr�nextrH )r �line�partZ splitlinerF �cmd�argsrM r r r �handlex s>
zSimpleIMAPHandler.handlec C sD | j r d | j n d } | j d | � | j | d d � d S)Nz
IMAP4rev1 Z IMAP4rev1z
* CAPABILITY �OKzCAPABILITY completed)� capabilitiesrC rH )r rF rX Zcapsr r r �cmd_CAPABILITY� s z SimpleIMAPHandler.cmd_CAPABILITYc C s0 d | j _ | j d � | j | d d � d S)Nz"* BYE IMAP4ref1 Server logging outrZ zLOGOUT completed)r7 r8 rC rH )r rF rX r r r �
cmd_LOGOUT� s
z SimpleIMAPHandler.cmd_LOGOUTc C s' | d | j _ | j | d d � d S)Nr rZ zLOGIN completed)r7 r8 rH )r rF rX r r r � cmd_LOGIN� s zSimpleIMAPHandler.cmd_LOGIN)r* r+ r, ZtimeoutrM r[ r6 r? r@ rC rH rY r\ r] r^ r r )r9 r r3 a s * r3 c @ s? e Z d Z d Z d d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z e j d# d$ � d% d& � � Z d' d( � Z d) d* � Z d+ d, � Z d- d. � Z d/ d0 � Z d1 d2 � Z d S)3�NewIMAPTestsMixinNTc C s� Gd d � d | j � } | j | j � | j t j d f | � | _ t j d | j d d | j j d d d i � | _
d
| j
_ | j
j � | r� | j
| j j � | _ | j | j f S) 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 Z d Z d d � Z d S)z/NewIMAPTestsMixin._setup.<locals>.TestTCPServerc S s | j | � | j � � d S)zP
End request and raise the error if one occurs.
N)�
close_request� server_close)r �request�client_addressr r r � handle_error� s
z<NewIMAPTestsMixin._setup.<locals>.TestTCPServer.handle_errorN)r* r+ r, rd r r r r �
TestTCPServer� s re r �namez-server�target�kwargs�
poll_intervalg{�G�z�?T)� server_classZ
addCleanup�_cleanupr �HOSTr7 r �ThreadZ_testMethodName�
serve_forever�thread�daemon�start�
imap_class�server_address�client)r Z imap_handler�connectre r r r �_setup� s
zNewIMAPTestsMixin._setupc C s\ | j d k r. | j j d k r. | j j � | j j � | j j � | j j d � d S)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 @)rt �state�shutdownr7 ra ro rE )r r r r rk � s
!
zNewIMAPTestsMixin._cleanupc C sW Gd d � d t j � } | j | d d �\ } } | j t j j | j | j � d S)Nc @ s e Z d Z d d � Z d S)zONewIMAPTestsMixin.test_EOF_without_complete_welcome_message.<locals>.EOFHandlerc S s | j j d � d S)Ns * OK)r< r= )r r r r rY � s zVNewIMAPTestsMixin.test_EOF_without_complete_welcome_message.<locals>.EOFHandler.handleN)r* r+ r, rY r r r r �
EOFHandler� s rz ru F) � socketserver�StreamRequestHandlerrv � assertRaisesr �IMAP4�abortrr rs )r rz �_r7 r r r �)test_EOF_without_complete_welcome_message� s z;NewIMAPTestsMixin.test_EOF_without_complete_welcome_messagec C sT Gd d � d t � } | j | d d �\ } } | j t j j | j | j � d S)Nc @ s e Z d Z d d � Z d S)zBNewIMAPTestsMixin.test_line_termination.<locals>.BadNewlineHandlerc S s$ | j d � | j | d d � d S)Ns * CAPABILITY IMAP4rev1 AUTH
rZ zCAPABILITY completed)r? rH )r rF rX r r r r\ � s
zQNewIMAPTestsMixin.test_line_termination.<locals>.BadNewlineHandler.cmd_CAPABILITYN)r* r+ r, r\ r r r r �BadNewlineHandler� s r� ru F)r3 rv r} r r~ r rr rs )r r� r� r7 r r r �test_line_termination� s z'NewIMAPTestsMixin.test_line_terminationc C s| Gd d � d t � } | j | � \ } } | j | j � | j t j j d � � | j d � Wd QRX| j | j � d S)Nc @ s e Z d Z d Z d S)zMNewIMAPTestsMixin.test_enable_raises_error_if_not_AUTH.<locals>.EnableHandlerzAUTH ENABLE UTF8=ACCEPTN)r* r+ r, r[ r r r r �
EnableHandler� s r� zENABLE.*NONAUTH�foo) r3 rv � assertFalse� utf8_enabled�assertRaisesRegexr r~ �error�enable)r r� rt r� r r r �$test_enable_raises_error_if_not_AUTH� s z6NewIMAPTestsMixin.test_enable_raises_error_if_not_AUTHc C sF | j t � \ } } | j t j j d � � | j d � Wd QRXd S)Nzdoes not support ENABLEr� )rv r3 r� r r~ r� r� )r rt r� r r r �)test_enable_raises_error_if_no_capability s
z;NewIMAPTestsMixin.test_enable_raises_error_if_no_capabilityc C sn | j t � \ } } | j d d � \ } } | j | d � | j t j j d � � | j d � Wd QRXd S)N�user�passrZ zdoes not support ENABLEz UTF8=ACCEPT) rv r3 �loginr r� r r~ r� r� )r rt 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 Gd d � d t � } | j | � \ } } | j | j d � | j d d d � � \ } } | j | d � | j | j d � | j d � \ } } | j | d � | j | j d
� d } | j d d d | j d
� � \ } } | j | d � | j | j d | j d
� � d S)
Nc @ s: e Z d Z d Z d d � Z d d � Z d d � Z d S) zHNewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.UTF8AppendServerzENABLE UTF8=ACCEPTc S s | j | d d � d S)NrZ zENABLE successful)rH )r rF rX r r r �
cmd_ENABLE s zSNewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_ENABLEc s s1 | j d � d V| j _ | j | d d � d S)N�+rZ zFAKEAUTH successful)rC r7 �responserH )r rF rX r r r �cmd_AUTHENTICATE s
zYNewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_AUTHENTICATEc s s1 | j d � d V| j _ | j | d d � d S)Nr� rZ �okay)rC r7 r� rH )r rF rX r r r �
cmd_APPEND s
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 s d S)Ns faker )�xr r r �<lambda> s z@NewIMAPTestsMixin.test_enable_UTF8_True_append.<locals>.<lambda>rZ s
ZmFrZQ==
z UTF8=ACCEPTzutf-8u Subject: üñí©öðéz UTF8 (%s)
) r3 rv r � _encoding� authenticater� r� �appendrB ) r r� rt r7 rG r� �
msg_stringr� r� r r r � test_enable_UTF8_True_append s ' z.NewIMAPTestsMixin.test_enable_UTF8_True_appendc
C s� Gd d � d t � } | j | � \ } } | j d d d � � \ } } | j | d � | j d � \ } } | j | d � | j | j � | j t j j
d � � | j d d
� Wd QRXd S) Nc @ s. e Z d Z d Z d d � Z d d � Z d S)zPNewIMAPTestsMixin.test_search_disallows_charset_in_utf8_mode.<locals>.UTF8ServerzAUTH ENABLE UTF8=ACCEPTc S s | j | d d � d S)NrZ zENABLE successful)rH )r rF rX r r r r� , s z[NewIMAPTestsMixin.test_search_disallows_charset_in_utf8_mode.<locals>.UTF8Server.cmd_ENABLEc s s1 | j d � d V| j _ | j | d d � d S)Nr� rZ zFAKEAUTH successful)rC r7 r� rH )r rF rX r r r r� . s
zaNewIMAPTestsMixin.test_search_disallows_charset_in_utf8_mode.<locals>.UTF8Server.cmd_AUTHENTICATEN)r* r+ r, r[ r� r� r r r r �
UTF8Server* s r� r� c S s d S)Ns faker )r� r r r r� 3 s zNNewIMAPTestsMixin.test_search_disallows_charset_in_utf8_mode.<locals>.<lambda>rZ z UTF8=ACCEPTz
charset.*UTF8r� �bar) r3 rv r� r r� �
assertTruer� r� r r~ r� �search)r r� rt r� r� r r r �*test_search_disallows_charset_in_utf8_mode) s z<NewIMAPTestsMixin.test_search_disallows_charset_in_utf8_modec C se Gd d � d t � } | j | � \ } } | j t j j d � � | j d d d � � Wd QRXd S)Nc @ s e Z d Z d d � Z d S)z6NewIMAPTestsMixin.test_bad_auth_name.<locals>.MyServerc S s$ | j | d d j | d � � d S)N�NOz#unrecognized authentication type {}r )rH �format)r rF rX r r r r� = s zGNewIMAPTestsMixin.test_bad_auth_name.<locals>.MyServer.cmd_AUTHENTICATEN)r* r+ r, r� r r r r �MyServer< s r� z'unrecognized authentication type METHOD�METHODc S s d S)Nr r r r r r r� C s z6NewIMAPTestsMixin.test_bad_auth_name.<locals>.<lambda>)r3 rv r� r r~ r� r� )r r� rt r� r r r �test_bad_auth_name; s
z$NewIMAPTestsMixin.test_bad_auth_namec C se Gd d � d t � } | j | � \ } } | j t j j d � � | j d d d � � Wd QRXd S)Nc @ s e Z d Z d d � Z d S)z?NewIMAPTestsMixin.test_invalid_authentication.<locals>.MyServerc s s. | j d � d V| _ | j | d d � d S)Nr� r� z [AUTHENTICATIONFAILED] invalid)rC r� rH )r rF rX r r r r� G s
zPNewIMAPTestsMixin.test_invalid_authentication.<locals>.MyServer.cmd_AUTHENTICATEN)r* r+ r, r� r r r r r� F s r� z \[AUTHENTICATIONFAILED\] invalidr� c S s d S)Ns faker )r� r r r r� N s z?NewIMAPTestsMixin.test_invalid_authentication.<locals>.<lambda>)r3 rv r� r r~ r� r� )r r� rt r� r r r �test_invalid_authenticationE s
z-NewIMAPTestsMixin.test_invalid_authenticationc C sp Gd d � d t � } | j | � \ } } | j d d d � � \ } } | j | d � | j | j d � d S)Nc @ s e Z d Z d d � Z d S)zCNewIMAPTestsMixin.test_valid_authentication_bytes.<locals>.MyServerc s s1 | j d � d V| j _ | j | d d � d S)Nr� rZ zFAKEAUTH successful)rC r7 r� rH )r rF rX r r r r� R s
zTNewIMAPTestsMixin.test_valid_authentication_bytes.<locals>.MyServer.cmd_AUTHENTICATEN)r* r+ r, r� r r r r r� Q s r� r� c S s d S)Ns faker )r� r r r r� W s zCNewIMAPTestsMixin.test_valid_authentication_bytes.<locals>.<lambda>rZ s
ZmFrZQ==
)r3 rv r� r r� )r r� rt r7 rG r� r r r � test_valid_authentication_bytesP s
z1NewIMAPTestsMixin.test_valid_authentication_bytesc C sp Gd d � d t � } | j | � \ } } | j d d d � � \ } } | j | d � | j | j d � d S)Nc @ s e Z d Z d d � Z d S)zHNewIMAPTestsMixin.test_valid_authentication_plain_text.<locals>.MyServerc s s1 | j d � d V| j _ | j | d d � d S)Nr� rZ zFAKEAUTH successful)rC r7 r� rH )r rF rX r r r r� ] s
zYNewIMAPTestsMixin.test_valid_authentication_plain_text.<locals>.MyServer.cmd_AUTHENTICATEN)r* r+ r, r� r r r r r� \ s r� r� c S s d S)N�faker )r� r r r r� b s zHNewIMAPTestsMixin.test_valid_authentication_plain_text.<locals>.<lambda>rZ s
ZmFrZQ==
)r3 rv r� r r� )r r� rt r7 rG r� r r r �$test_valid_authentication_plain_text[ s
z6NewIMAPTestsMixin.test_valid_authentication_plain_textc C sm Gd d � d t � } | j | � \ } } | j d | j k � | j d d � \ } } | j | d � d S)Nc @ s" e Z d Z d Z d d � Z d S)z@NewIMAPTestsMixin.test_login_cram_md5_bytes.<locals>.AuthHandlerzLOGINDISABLED AUTH=CRAM-MD5c s sM | j d � d V} | d k r6 | j | d d � n | j | d d � d S)Nz:+ PDE4OTYuNjk3MTcwOTUyQHBvc3RvZmZpY2UucmVzdG9uLm1jaS5uZXQ=s2 dGltIGYxY2E2YmU0NjRiOWVmYTFjY2E2ZmZkNmNmMmQ5ZjMy
rZ zCRAM-MD5 successfulr� z No access)rC rH )r rF rX �rr r r r� i s
zQNewIMAPTestsMixin.test_login_cram_md5_bytes.<locals>.AuthHandler.cmd_AUTHENTICATEN)r* r+ r, r[ r� r r r r � AuthHandlerg s r� z
AUTH=CRAM-MD5�tims tanstaaftanstaafrZ )r3 rv r� r[ �login_cram_md5r )r r� rt r� �retr r r �test_login_cram_md5_bytesf s
z+NewIMAPTestsMixin.test_login_cram_md5_bytesc C sm Gd d � d t � } | j | � \ } } | j d | j k � | j d d � \ } } | j | d � d S)Nc @ s" e Z d Z d Z d d � Z d S)zENewIMAPTestsMixin.test_login_cram_md5_plain_text.<locals>.AuthHandlerzLOGINDISABLED AUTH=CRAM-MD5c s sM | j d � d V} | d k r6 | j | d d � n | j | d d � d S)Nz:+ PDE4OTYuNjk3MTcwOTUyQHBvc3RvZmZpY2UucmVzdG9uLm1jaS5uZXQ=s2 dGltIGYxY2E2YmU0NjRiOWVmYTFjY2E2ZmZkNmNmMmQ5ZjMy
rZ zCRAM-MD5 successfulr� z No access)rC rH )r rF rX r� r r r r� z s
zVNewIMAPTestsMixin.test_login_cram_md5_plain_text.<locals>.AuthHandler.cmd_AUTHENTICATEN)r* r+ r, r[ r� r r r r r� x s r� z
AUTH=CRAM-MD5r� �tanstaaftanstaafrZ )r3 rv r� r[ r� r )r r� rt r� r� r r r � test_login_cram_md5_plain_textw s
z0NewIMAPTestsMixin.test_login_cram_md5_plain_textc C se Gd d � d t � } | j | � \ } } | j t j j d � � | j d d d � � Wd QRXd S)Nc @ s e Z d Z d d � Z d S)z?NewIMAPTestsMixin.test_aborted_authentication.<locals>.MyServerc s sS | j d � d V| _ | j d k r< | j | d d � n | j | d d � d S)Nr� s *
r� z [AUTHENTICATIONFAILED] abortedrZ zMYAUTH successful)rC r� rH )r rF rX r r r r� � s
zPNewIMAPTestsMixin.test_aborted_authentication.<locals>.MyServer.cmd_AUTHENTICATEN)r* r+ r, r� r r r r r� � s r� z \[AUTHENTICATIONFAILED\] abortedr� c S s d S)Nr )r� r r r r� � s z?NewIMAPTestsMixin.test_aborted_authentication.<locals>.<lambda>)r3 rv r� r r~ r� r� )r r� rt r� r r r �test_aborted_authentication� s
z-NewIMAPTestsMixin.test_aborted_authenticationzimaplib._MAXLINE�
c C se Gd d � d t � } | j | d d �\ } } | j t j j d � � | j | j � Wd QRXd S)Nc @ s e Z d Z d d � Z d S)z:NewIMAPTestsMixin.test_linetoolong.<locals>.TooLongHandlerc S s | j j d � d S)Ns * OK � � xs
s xxxxxxxxxxxs * OK xxxxxxxxxxxs * OK xxxxxxxxxxx
)r< r= )r r r r rY � s zANewIMAPTestsMixin.test_linetoolong.<locals>.TooLongHandler.handleN)r* r+ r, rY r r r r �TooLongHandler� s r� ru Fzgot more than 10 bytes)r3 rv r� r r~ r� rr rs )r r� r� r7 r r r �test_linetoolong� s
z"NewIMAPTestsMixin.test_linetoolongc C s9 | j t d d �\ } } | j | j � � Wd QRXd S)Nru F)rv r3 rr rs )r r� r7 r r r �test_simple_with_statement� s z,NewIMAPTestsMixin.test_simple_with_statementc
C sn | j t d d �\ } } | j | j � �* } | j d d � | j | j d � Wd QRX| j | j � d S)Nru Fr� r� )rv r3 rr rs r� r r8 � assertIsNone)r r� r7 �imapr r r �test_with_statement� s
z%NewIMAPTestsMixin.test_with_statementc
C s� | j t d d �\ } } | j | j � �D } | j d d � | j | j d � | j � | j | j � Wd QRX| j | j � d S)Nru Fr� r� ) rv r3 rr rs r� r r8 �logoutr� )r r� r7 r� r r r �test_with_statement_logout� s
z,NewIMAPTestsMixin.test_with_statement_logoutc C sh | j t � \ } } | j d d � \ } } | j | d � | j | d d � | j | j d � d S)Nr� r� rZ r s LOGIN completed�AUTH)rv r3 r� r rx )r rt r� r� r� r r r �
test_login� s
z NewIMAPTestsMixin.test_loginc C s� | j t � \ } } | j d d � \ } } | j | d � | j | d d � | j � \ } } | j | d � | j | d d � | j | j d � d S) Nr� r� rZ r s LOGIN completed�BYEs IMAP4ref1 Server logging outrw )rv r3 r� r r� rx )r rt r� r� r� r r r � test_logout� s z NewIMAPTestsMixin.test_logoutc C su Gd d � d t � } | j | � \ } } | j d d � | j � \ } } | j | d � | j | d d � d S)Nc @ s e Z d Z d d � Z d S)z,NewIMAPTestsMixin.test_lsub.<locals>.LsubCmdc S s | j d � | j | d d � S)Nz* LSUB () "." directoryArZ zLSUB completed)rC rH )r rF rX r r r �cmd_LSUB� s
z5NewIMAPTestsMixin.test_lsub.<locals>.LsubCmd.cmd_LSUBN)r* r+ r, r� r r r r �LsubCmd� s r� r� r� rZ r s () "." directoryA)r3 rv r� Zlsubr )r r� rt r� r� r� r r r � test_lsub� s zNewIMAPTestsMixin.test_lsub) r* r+ r, rt rv rk r� r� r� r� r� r� r� r� r� r� r� r� r� r� r
Zpatchr� r� r� r� r� r� r� r r r r r_ � s0 !
r_ c @ s" e Z d Z e j Z e j Z d S)� NewIMAPTestsN) r* r+ r, r r~ rr r{ � TCPServerrj r r r r r� � s r� zSSL not availablec @ s4 e Z d Z e Z e Z d d � Z d d � Z d S)�NewIMAPSSLTestsc C s� t j t j � } t j | _ d | _ | j t � | j t j d � �< | j
t � \ } } | j | j
d | � } | j � Wd QRXd S)NTz.hostname '127.0.0.1' doesn't match 'localhost'� ssl_context)r0 �
SSLContext�PROTOCOL_SSLv23�
CERT_REQUIRED� verify_mode�check_hostname�load_verify_locations�CAFILEr� �CertificateErrorrv r3 rr rs ry )r r� r� r7 rt r r r �test_ssl_raises� s
z NewIMAPSSLTests.test_ssl_raisesc C sv t j t j � } t j | _ d | _ | j t � | j t � \ } } | j
d | j d d | �} | j � d S)NT� localhostr r� )
r0 r� r� r� r� r� r� r� rv r3 rr rs ry )r r� r� r7 rt r r r �test_ssl_verified� s
z!NewIMAPSSLTests.test_ssl_verifiedN) r* r+ r, � IMAP4_SSLrr r- rj r� r� r r r r r� � s
r� c @ s� e Z d Z e j Z e j Z d d � Z d d � Z
e d d � � Z e d d � � Z
e d d
� � Z e d d � � Z e d
d � � Z Gd d � d e � Z e d d � � Z e d d � � Z e d d � � Z e d d � � Z e d d � � Z e d d � � Z e d d � � Z e d d � � Z e d! d" � � Z e d# d$ � � Z d% d& � Z e d' d( � � Z e d) d* � � Z e d+ d, � � Z! d- S).�ThreadedNetworkedTestsc C s� Gd d � d | j � } t r) t d � | | | � } | j | j | j j � � t r� t d � t d | � t d | j � t d | j � t j d d | j d
| j
d d d
i � } d | _ | j � t r� t d � | | f S)Nc @ s e Z d Z d d � Z d S)z4ThreadedNetworkedTests.make_server.<locals>.MyServerc S s | j | � | j � � d S)N)r` ra )r rb rc r r r rd s
zAThreadedNetworkedTests.make_server.<locals>.MyServer.handle_errorN)r* r+ r, rd r r r r r� s r� zcreating serverzserver createdzADDR =zCLASS =zHDLR =rf z
%s servingrg rh ri g{�G�z�?Tzserver running)
rj r r: r rs r/ Z getsocknameZRequestHandlerClassr rm rn rp rq )r Zaddr�hdlrr� r7 r' r r r � make_server� s&
z"ThreadedNetworkedTests.make_serverc C sB t r t d � | j � | j � | j � t r> t d � d S)Nzwaiting for server�done)r r: ry ra rE )r r7 ro r r r � reap_server s
z"ThreadedNetworkedTests.reap_serverc c sB | j t j d f | � \ } } z | VWd | j | | � Xd S)Nr )r� r rl r� )r r� r7 ro r r r �
reaped_server( s ! z$ThreadedNetworkedTests.reaped_serverc c sL | j | � �6 } | j | j � } z | | f VWd | j � XWd QRXd S)N)r� rr rs r� )r r� r7 rt r r r � reaped_pair0 s
z"ThreadedNetworkedTests.reaped_pairc C s9 | j t � �# } | j | j � } | j � Wd QRXd S)N)r� r3 rr rs ry )r r7 rt r r r � test_connect9 s z#ThreadedNetworkedTests.test_connectc C sU Gd d � d t j � } | j | � �& } | j t j j | j | j � Wd QRXd S)Nc @ s e Z d Z d d � Z d S)z9ThreadedNetworkedTests.test_issue5949.<locals>.EOFHandlerc S s | j j d � d S)Ns * OK)r< r= )r r r r rY C s z@ThreadedNetworkedTests.test_issue5949.<locals>.EOFHandler.handleN)r* r+ r, rY r r r r rz B s rz ) r{ r| r� r} r r~ r rr rs )r rz r7 r r r �test_issue5949? s z%ThreadedNetworkedTests.test_issue5949c C sR Gd d � d t � } | j | � �& } | j t j j | j | j � Wd QRXd S)Nc @ s e Z d Z d d � Z d S)zGThreadedNetworkedTests.test_line_termination.<locals>.BadNewlineHandlerc S s$ | j d � | j | d d � d S)Ns * CAPABILITY IMAP4rev1 AUTH
rZ zCAPABILITY completed)r? rH )r rF rX r r r r\ P s
zVThreadedNetworkedTests.test_line_termination.<locals>.BadNewlineHandler.cmd_CAPABILITYN)r* r+ r, r\ r r r r r� N s r� )r3 r� r} r r~ r rr rs )r r� r7 r r r r� K s z,ThreadedNetworkedTests.test_line_terminationc @ s. e Z d Z d Z d d � Z d d � Z d S)z!ThreadedNetworkedTests.UTF8ServerzAUTH ENABLE UTF8=ACCEPTc C s | j | d d � d S)NrZ zENABLE successful)rH )r rF rX r r r r� [ s z,ThreadedNetworkedTests.UTF8Server.cmd_ENABLEc c s1 | j d � d V| j _ | j | d d � d S)Nr� rZ zFAKEAUTH successful)rC r7 r� rH )r rF rX r r r r� ^ s
z2ThreadedNetworkedTests.UTF8Server.cmd_AUTHENTICATEN)r* r+ r, r[ r� r� r r r r r� X s r� c C sb | j | j � �I \ } } | j | j � | j t j j | j d � | j | j � Wd QRXd S)Nr� ) r� r� r� r� r} r r~ r� r� )r r7 rt r r r r� c s z;ThreadedNetworkedTests.test_enable_raises_error_if_not_AUTHc C sX Gd d � d | j � } | j | � �) \ } } | j t j j | j d � Wd QRXd S)Nc @ s e Z d Z d Z d S)zXThreadedNetworkedTests.test_enable_raises_error_if_no_capability.<locals>.NoEnableServerr� N)r* r+ r, r[ r r r r �NoEnableServern s r� r� )r� r� r} r r~ r� r� )r r� r7 rt r r r r� l s z@ThreadedNetworkedTests.test_enable_raises_error_if_no_capabilityc C s� Gd d � d t � } | j t j j � �Y | j | � �B \ } } | j d d � \ } } | j | d � | j d � Wd QRXWd QRXd S)Nc @ s e Z d Z d S)z\ThreadedNetworkedTests.test_enable_UTF8_raises_error_if_not_supported.<locals>.NonUTF8ServerN)r* r+ r, r r r r �
NonUTF8Serveru s r� r� r� rZ z UTF8=ACCEPT) r3 r} r r~ r� r� r� r r� )r r� r7 rt r� r� r r r r� s s
zEThreadedNetworkedTests.test_enable_UTF8_raises_error_if_not_supportedc
C s% Gd d � d | j � } | j | � �� \ } } | j | j d � | j d d d � � \ } } | j | d � | j | j d � | j d � \ } } | j | d � | j | j d
� d } | j d d d | j d
� � \ } } | j | d � | j | j d | j d
� � Wd QRXd S)
Nc @ s e Z d Z d d � Z d S)zMThreadedNetworkedTests.test_enable_UTF8_True_append.<locals>.UTF8AppendServerc s s1 | j d � d V| j _ | j | d d � d S)Nr� rZ r� )rC r7 r� rH )r rF rX r r r r� � s
zXThreadedNetworkedTests.test_enable_UTF8_True_append.<locals>.UTF8AppendServer.cmd_APPENDN)r* r+ r, r� r r r r r� � s r� r� r� c S s d S)Ns faker )r� r r r r� � s zEThreadedNetworkedTests.test_enable_UTF8_True_append.<locals>.<lambda>rZ s
ZmFrZQ==
z UTF8=ACCEPTzutf-8u Subject: üñí©öðéz UTF8 (%s)
) r� r� r r� r� r� r� r� rB ) r r� r7 rt rG r� r� r� r� r r r r� ~ s" !z3ThreadedNetworkedTests.test_enable_UTF8_True_appendc C s� | j | j � �� \ } } | j d d d � � \ } } | j | d � | j d � \ } } | j | d � | j | j � | j t j j
| j d d � Wd QRXd S)Nr� c S s d S)Ns faker )r� r r r r� � s zSThreadedNetworkedTests.test_search_disallows_charset_in_utf8_mode.<locals>.<lambda>rZ z UTF8=ACCEPTr� r� ) r� r� r� r r� r� r� r} r r~ r� r� )r r7 rt r� r� r r r r� � s zAThreadedNetworkedTests.test_search_disallows_charset_in_utf8_modec C sl Gd d � d t � } | j | � �@ \ } } | j t j j � � | j d d d � � Wd QRXWd QRXd S)Nc @ s e Z d Z d d � Z d S)z;ThreadedNetworkedTests.test_bad_auth_name.<locals>.MyServerc S s$ | j | d d j | d � � d S)Nr� z#unrecognized authentication type {}r )rH r� )r rF rX r r r r� � s zLThreadedNetworkedTests.test_bad_auth_name.<locals>.MyServer.cmd_AUTHENTICATEN)r* r+ r, r� r r r r r� � s r� r� c S s d S)Nr r r r r r r� � s z;ThreadedNetworkedTests.test_bad_auth_name.<locals>.<lambda>)r3 r� r} r r~ r� r� )r r� r7 rt r r r r� � s z)ThreadedNetworkedTests.test_bad_auth_namec C st Gd d � d t � } | j | � �H \ } } | j t j j � �# | j d d d � � \ } } Wd QRXWd QRXd S)Nc @ s e Z d Z d d � Z d S)zDThreadedNetworkedTests.test_invalid_authentication.<locals>.MyServerc s s. | j d � d V| _ | j | d d � d S)Nr� r� z [AUTHENTICATIONFAILED] invalid)rC r� rH )r rF rX r r r r� � s
zUThreadedNetworkedTests.test_invalid_authentication.<locals>.MyServer.cmd_AUTHENTICATEN)r* r+ r, r� r r r r r� � s r� r� c S s d S)Ns faker )r� r r r r� � s zDThreadedNetworkedTests.test_invalid_authentication.<locals>.<lambda>)r3 r� r} r r~ r� r� )r r� r7 rt rG r� r r r r� � s z2ThreadedNetworkedTests.test_invalid_authenticationc C s� Gd d � d t � } | j | � �N \ } } | j d d d � � \ } } | j | d � | j | j d � Wd QRX| j | � �N \ } } | j d d d � � \ } } | j | d � | j | j d � Wd QRXd S) Nc @ s e Z d Z d d � Z d S)zBThreadedNetworkedTests.test_valid_authentication.<locals>.MyServerc s s1 | j d � d V| j _ | j | d d � d S)Nr� rZ zFAKEAUTH successful)rC r7 r� rH )r rF rX r r r r� � s
zSThreadedNetworkedTests.test_valid_authentication.<locals>.MyServer.cmd_AUTHENTICATEN)r* r+ r, r� r r r r r� � s r� r� c S s d S)Ns faker )r� r r r r� � s zBThreadedNetworkedTests.test_valid_authentication.<locals>.<lambda>rZ s
ZmFrZQ==
c S s d S)Nr� r )r� r r r r� � s )r3 r� r� r r� )r r� r7 rt rG r� r r r �test_valid_authentication� s z0ThreadedNetworkedTests.test_valid_authenticationc C s� Gd d � d t � } | j | � �K \ } } | j d | j k � | j d d � \ } } | j | d � Wd QRX| j | � �K \ } } | j d | j k � | j d d � \ } } | j | d � Wd QRXd S)Nc @ s" e Z d Z d Z d d � Z d S)z?ThreadedNetworkedTests.test_login_cram_md5.<locals>.AuthHandlerzLOGINDISABLED AUTH=CRAM-MD5c s sM | j d � d V} | d k r6 | j | d d � n | j | d d � d S)Nz:+ PDE4OTYuNjk3MTcwOTUyQHBvc3RvZmZpY2UucmVzdG9uLm1jaS5uZXQ=s2 dGltIGYxY2E2YmU0NjRiOWVmYTFjY2E2ZmZkNmNmMmQ5ZjMy
rZ zCRAM-MD5 successfulr� z No access)rC rH )r rF rX r� r r r r� � s
zPThreadedNetworkedTests.test_login_cram_md5.<locals>.AuthHandler.cmd_AUTHENTICATEN)r* r+ r, r[ r� r r r r r� � s r� z
AUTH=CRAM-MD5r� r� rZ s tanstaaftanstaaf)r3 r� r� r[ r� r )r r� r7 rt r� r� r r r �test_login_cram_md5� s z*ThreadedNetworkedTests.test_login_cram_md5c C st Gd d � d t � } | j | � �H \ } } | j t j j � �# | j d d d � � \ } } Wd QRXWd QRXd S)Nc @ s e Z d Z d d � Z d S)zDThreadedNetworkedTests.test_aborted_authentication.<locals>.MyServerc s sS | j d � d V| _ | j d k r< | j | d d � n | j | d d � d S)Nr� s *
r� z [AUTHENTICATIONFAILED] abortedrZ zMYAUTH successful)rC r� rH )r rF rX r r r r� � s
zUThreadedNetworkedTests.test_aborted_authentication.<locals>.MyServer.cmd_AUTHENTICATEN)r* r+ r, r� r r r r r� � s r� r� c S s d S)Nr )r� r r r r� s zDThreadedNetworkedTests.test_aborted_authentication.<locals>.<lambda>)r3 r� r} r r~ r� r� )r r� r7 rt rG r� r r r r� � s z2ThreadedNetworkedTests.test_aborted_authenticationc C sR Gd d � d t � } | j | � �& } | j t j j | j | j � Wd QRXd S)Nc @ s e Z d Z d d � Z d S)z?ThreadedNetworkedTests.test_linetoolong.<locals>.TooLongHandlerc S s# | j j d t j d d � d S)Ns * OK r� s
)r< r= r Z_MAXLINE)r r r r rY s zFThreadedNetworkedTests.test_linetoolong.<locals>.TooLongHandler.handleN)r* r+ r, rY r r r r r� s r� )r3 r� r} r r~ r� rr rs )r r� r7 r r r r� s z'ThreadedNetworkedTests.test_linetoolongc C s7 | j t � �! } | j | j � � Wd QRXWd QRXd S)N)r� r3 rr rs )r r7 r r r r� s z1ThreadedNetworkedTests.test_simple_with_statementc C sl | j t � �V } | j | j � �* } | j d d � | j | j d � Wd QRX| j | j � Wd QRXd S)Nr� r� )r� r3 rr rs r� r r8 r� )r r7 r� r r r r� s
z*ThreadedNetworkedTests.test_with_statementc C s� | j t � �p } | j | j � �D } | j d d � | j | j d � | j � | j | j � Wd QRX| j | j � Wd QRXd S)Nr� r� ) r� r3 rr rs r� r r8 r� r� )r r7 r� r r r r� s
z1ThreadedNetworkedTests.test_with_statement_logoutN)"r* r+ r, r{ r� rj r r~ rr r� r� r r� r� r r� r� r� r3 r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s0
r� c @ s. e Z d Z e Z e Z e d d � � Z d S)�ThreadedNetworkedTestsSSLc C s� t j t j � } t j | _ d | _ | j t � | j t j d � �@ | j
t � �) } | j | j
d | � } | j � Wd QRXWd QRX| j
t � �0 } | j d | j
d d | �} | j � Wd QRXd S)NTz.hostname '127.0.0.1' doesn't match 'localhost'r� r� r )r0 r� r� r� r� r� r� r� r� r� r� r3 rr rs ry )r r� r7 rt r r r r� 0 s
z+ThreadedNetworkedTestsSSL.test_ssl_verifiedN) r* r+ r, r- rj r� rr r r� r r r r r� + s r� Znetworkznetwork resource disabledc @ sa e Z d Z d Z d Z d Z d Z e j Z d d � Z
d d � Z d d
� Z d d � Z
d
S)�RemoteIMAPTestzcyrus.andrew.cmu.edu� Z anonymousr� c
C s6 t | j � � | j | j | j � | _ Wd QRXd S)N)r �hostrr �portr7 )r r r r �setUpN s zRemoteIMAPTest.setUpc C s7 | j d k r3 t | j � � | j j � Wd QRXd S)N)r7 r r� r� )r r r r �tearDownR s zRemoteIMAPTest.tearDownc C s� t | j � �� x$ | j j D] } | j | t � q W| j d | j j � | j d | j j � | j j | j | j � } | j
| d d � Wd QRXd S)N�
LOGINDISABLEDzAUTH=ANONYMOUSr rZ ) r r� r7 r[ �assertIsInstance�str�assertInr� �username�passwordr )r �cap�rsr r r �test_logincapaW s z RemoteIMAPTest.test_logincapac
C sG t | j � �1 | j j � } d | _ | j | d d � Wd QRXd S)Nr r� )r r� r7 r� r )r r� r r r r� ` s zRemoteIMAPTest.test_logoutN)r* r+ r, r� r� r� r� r r~ rr r� r� r� r� r r r r r� E s r� c s. e Z d Z � f d d � Z d d � Z � S)�RemoteIMAP_STARTTLSTestc
sK t � j � t | j � �( | j j � } | j | d d � Wd QRXd S)Nr rZ )r5 r� r r� r7 Zstarttlsr )r r� )r9 r r r� l s
z RemoteIMAP_STARTTLSTest.setUpc C sA x$ | j j D] } | j | t � q
W| j d | j j � d S)Nr� )r7 r[ r� r� � assertNotIn)r r� r r r r� r s z&RemoteIMAP_STARTTLSTest.test_logincapa)r* r+ r, r� r� r r )r9 r r� g s r� c @ s| e Z d Z d Z e Z d d � Z d d � Z d d � Z d d � Z d
d � Z
d d
� Z d d � Z d d � Z
d S)�RemoteIMAP_SSLTesti� c C s d S)Nr )r r r r r� } s zRemoteIMAP_SSLTest.setUpc C s d S)Nr )r r r r r� � s zRemoteIMAP_SSLTest.tearDownc C s# t j t j � } | j t � | S)N)r0 r� r� Zload_cert_chainr1 )r r� r r r �create_ssl_context� s
z%RemoteIMAP_SSLTest.create_ssl_contextc
C s� zz x! | j D] } | j | t � q
W| j d | j � | j d | j � | j | j | j � } | j | d d � Wd | j � Xd S)Nr� z
AUTH=PLAINr rZ )
r[ r� r� r� r� r� r� r� r r� )r r7 r� r� r r r �check_logincapa� s z"RemoteIMAP_SSLTest.check_logincapac
C s@ t | j � �* | j | j | j � } | j | � Wd QRXd S)N)r r� rr r� r� )r �_serverr r r r� � s z!RemoteIMAP_SSLTest.test_logincapac
C sS t | j � �= | j | j | j � } | j � } | j | d d � Wd QRXd S)Nr r� )r r� rr r� r� r )r r� r� r r r r� � s z RemoteIMAP_SSLTest.test_logoutc C sL t | j � �6 | j t | j | j | j d t d | j � �Wd QRXd S)Nr. r� )r r� r} �
ValueErrorrr r� r1 r� )r r r r �#test_ssl_context_certfile_exclusive� s z6RemoteIMAP_SSLTest.test_ssl_context_certfile_exclusivec C sL t | j � �6 | j t | j | j | j d t d | j � �Wd QRXd S)NZkeyfiler� )r r� r} r� rr r� r1 r� )r r r r �"test_ssl_context_keyfile_exclusive� s z5RemoteIMAP_SSLTest.test_ssl_context_keyfile_exclusiveN)r* r+ r, r� r� rr r� r� r� r� r� r� r r r r r r r� x s r� �__main__)4Ztestr �
import_moduler �
contextlibr r Zos.path�osr{ r r �inspectZ test.supportr r r r r Zunittestr
r r r
r0 � ImportError�pathrE �dirname�__file__�curdirr1 r� ZTestCaser r� r- r� r| r3 r_ r� Z
skipUnlessr� r� r� Zis_resource_enabledr� r� r� r* �mainr r r r �<module> sZ (
**0 R� ' � 1 2