Your IP : 216.73.216.213
�
g f�E �
@ sQ 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 m
Z
m Z m Z m Z m Z d d l Z d d l m Z m Z m Z y d d l Z Wn e k
r� d Z Yn Xd a d a Gd d � d e j � Z e r6Gd d
� d
e j � Z e j Z n Gd d
� d
� Z d Z Gd d
� d
e j � Z Gd d � d e j � Z! Gd d � d e! � Z" e j# e d � Gd d � d e! � � Z$ Gd d � d e j � Z% e j# e d � Gd d � d e% � � Z& e j# e d � Gd d � d e% � � Z' d d � Z( e) d k rMe j* � n d S) � )�support� threading)�contextmanagerN)� reap_threads�verbose�transient_internet� run_with_tz�run_with_locale)�datetime�timezone� timedeltac @ 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/python34/lib64/python3.4/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 �
timevalues1 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_Time2Internaldate9 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_resultD 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+ r1 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 @ sv e Z d Z d Z d Z 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)�SimpleIMAPHandlerr N� c C s1 t r t d | j � � n | j j | � d S)NzSENT: %r)r �print�strip�wfile�write)r �messager r r �_sendg s zSimpleIMAPHandler._sendc C s | j | d � d S)Ns
)r9 )r r8 r r r �
_send_linek s z SimpleIMAPHandler._send_linec C s | j | j d � � d S)N�ASCII)r: �encode)r r8 r r r �_send_textlinen s z SimpleIMAPHandler._send_textlinec C s# | j d j | | | f � � d S)N� )r= �join)r �tag�coder8 r r r � _send_taggedq s z SimpleIMAPHandler._send_taggedc C s} | j d � xid } x` y0 | j j d � } | d k r> d S| | 7} Wn t k
ra d SYn X| j d � r Pq q Wt r� t d | j � � n | j r� y | j j | � Wq t
k
r� d | _ Yq Xq n | j d � j � } | d } | d } | d d � } t
| d | � r^t | d | � | | � } | ru| | _ t | � quq | j | d
| d � q Wd S) Nz* OK IMAP4rev1� r s
zGOT: %rr; r r Zcmd_ZBADz unknown)r= Zrfile�read�OSError�endswithr r4 r5 � continuation�send�
StopIteration�decode�split�hasattr�getattr�nextrB )r �line�partZ splitliner@ �cmd�argsrG r r r �handlet 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)� capabilitiesr= rB )r r@ rR Zcapsr r r �cmd_CAPABILITY� s z SimpleIMAPHandler.cmd_CAPABILITYc C s$ | j d � | j | d d � d S)Nz"* BYE IMAP4ref1 Server logging outrT zLOGOUT completed)r= rB )r r@ rR r r r �
cmd_LOGOUT� s
z SimpleIMAPHandler.cmd_LOGOUT)
r) r* r+ ZtimeoutrG rU r9 r: r= rB rS rV rW r r r r r2 a s ) r2 c @ s� e Z d 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 e d d � � Z e d d � � Z
e d d � � Z e d d � � Z e d d � � Z d d � Z d S) �BaseThreadedNetworkedTestsc C s� Gd d � d | j � } t r, t d � n | | | � } | j | j | j j � � t r� t d � t d | � t d | j � t d | j � n t j d d | j d
| j
d i d d
6� } d | _ | j � t r� t d � n | | f S)Nc @ s e Z d Z d d � Z d S)z8BaseThreadedNetworkedTests.make_server.<locals>.MyServerc S s | j | � | j � � d S)N)Z
close_request� server_close)r ZrequestZclient_addressr r r � handle_error� s
zEBaseThreadedNetworkedTests.make_server.<locals>.MyServer.handle_errorN)r) r* r+ rZ r r r r �MyServer� s r[ zcreating serverzserver createdzADDR =zCLASS =zHDLR =�namez
%s serving�target�kwargsg{�G�z�?Z
poll_intervalTzserver running)
� server_classr r4 r �server_addressr. Z getsocknameZRequestHandlerClassr ZThreadZ
serve_foreverZdaemon�start)r Zaddr�hdlrr[ �serverr& r r r � make_server� s&
z&BaseThreadedNetworkedTests.make_serverc C sH t r t d � n | j � | j � | j � t rD t d � n d S)Nzwaiting for server�done)r r4 �shutdownrY r? )r rc �threadr r r � reap_server� s
z&BaseThreadedNetworkedTests.reap_serverc c sB | j t j d f | � \ } } z | VWd | j | | � Xd S)Nr )rd r ZHOSTrh )r rb rc rg r r r �
reaped_server� s ! z(BaseThreadedNetworkedTests.reaped_serverc c sK | j | � �6 } | j | j � } z | | f VWd | j � XWd QXd S)N)ri �
imap_classr` �logout)r rb rc �clientr r r � reaped_pair� s
z&BaseThreadedNetworkedTests.reaped_pairc C s8 | j t � �# } | j | j � } | j � Wd QXd S)N)ri r2 rj r` rf )r rc rl r r r � test_connect� s z'BaseThreadedNetworkedTests.test_connectc C sT Gd d � d t j � } | j | � �& } | j t j j | j | j � Wd QXd S)Nc @ s e Z d Z d d � Z d S)z=BaseThreadedNetworkedTests.test_issue5949.<locals>.EOFHandlerc S s | j j d � d S)Ns * OK)r6 r7 )r r r r rS � s zDBaseThreadedNetworkedTests.test_issue5949.<locals>.EOFHandler.handleN)r) r* r+ rS r r r r �
EOFHandler� s ro ) � socketserver�StreamRequestHandlerri � assertRaisesr �IMAP4�abortrj r` )r ro rc r r r �test_issue5949� s z)BaseThreadedNetworkedTests.test_issue5949c C sQ Gd d � d t � } | j | � �& } | j t j j | j | j � Wd QXd S)Nc @ s e Z d Z d d � Z d S)zKBaseThreadedNetworkedTests.test_line_termination.<locals>.BadNewlineHandlerc S s$ | j d � | j | d d � d S)Ns * CAPABILITY IMAP4rev1 AUTH
rT zCAPABILITY completed)r9 rB )r r@ rR r r r rV � s
zZBaseThreadedNetworkedTests.test_line_termination.<locals>.BadNewlineHandler.cmd_CAPABILITYN)r) r* r+ rV r r r r �BadNewlineHandler� s rv )r2 ri rr r rs rt rj r` )r rv rc r r r �test_line_termination� s z0BaseThreadedNetworkedTests.test_line_terminationc C sj Gd d � d t � } | j | � �? \ } } | j t j j � � | j d d d � � Wd QXWd QXd S)Nc @ s e Z d Z d d � Z d S)z?BaseThreadedNetworkedTests.test_bad_auth_name.<locals>.MyServerc S s$ | j | d d j | d � � d S)N�NOz#unrecognized authentication type {}r )rB �format)r r@ rR r r r �cmd_AUTHENTICATE s zPBaseThreadedNetworkedTests.test_bad_auth_name.<locals>.MyServer.cmd_AUTHENTICATEN)r) r* r+ rz r r r r r[ s r[ ZMETHODc S s d S)Nr r r r r r �<lambda> s z?BaseThreadedNetworkedTests.test_bad_auth_name.<locals>.<lambda>)r2 rm rr r rs �error� authenticate)r r[ rc rl r r r �test_bad_auth_name� s z-BaseThreadedNetworkedTests.test_bad_auth_namec C sr Gd d � d t � } | j | � �G \ } } | j t j j � �# | j d d d � � \ } } Wd QXWd QXd S)Nc @ s e Z d Z d d � Z d S)zHBaseThreadedNetworkedTests.test_invalid_authentication.<locals>.MyServerc s s. | j d � d V| _ | j | d d � d S)N�+rx z [AUTHENTICATIONFAILED] invalid)r= �responserB )r r@ rR r r r rz s
zYBaseThreadedNetworkedTests.test_invalid_authentication.<locals>.MyServer.cmd_AUTHENTICATEN)r) r* r+ rz r r r r r[ s r[ �MYAUTHc S s d S)Ns faker )�xr r r r{ s zHBaseThreadedNetworkedTests.test_invalid_authentication.<locals>.<lambda>)r2 rm rr r rs r| r} )r r[ rc rl rA �datar r r �test_invalid_authentication s z6BaseThreadedNetworkedTests.test_invalid_authenticationc C s� Gd d � d t � } | j | � �N \ } } | j d d d � � \ } } | j | d � | j | j d � Wd QX| j | � �N \ } } | j d d d � � \ } } | j | d � | j | j d � Wd QXd S) Nc @ s e Z d Z d d � Z d S)zFBaseThreadedNetworkedTests.test_valid_authentication.<locals>.MyServerc s s1 | j d � d V| j _ | j | d d � d S)Nr rT zFAKEAUTH successful)r= rc r� rB )r r@ rR r r r rz s
zWBaseThreadedNetworkedTests.test_valid_authentication.<locals>.MyServer.cmd_AUTHENTICATEN)r) r* r+ rz r r r r r[ s r[ r� c S s d S)Ns faker )r� r r r r{ $ s zFBaseThreadedNetworkedTests.test_valid_authentication.<locals>.<lambda>rT s
ZmFrZQ==
c S s d S)NZfaker )r� r r r r{ * s )r2 rm r} r r� )r r[ rc rl rA r� r r r �test_valid_authentication s
z4BaseThreadedNetworkedTests.test_valid_authenticationc C s� Gd d � d t � } | j | � �K \ } } | j d | j k � | j d d � \ } } | j | d � Wd QX| j | � �K \ } } | j d | j k � | j d d � \ } } | j | d � Wd QXd S)Nc @ s" e Z d Z d Z d d � Z d S)zCBaseThreadedNetworkedTests.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
rT zCRAM-MD5 successfulrx z No access)r= rB )r r@ rR �rr r r rz 6 s
zTBaseThreadedNetworkedTests.test_login_cram_md5.<locals>.AuthHandler.cmd_AUTHENTICATEN)r) r* r+ rU rz r r r r � AuthHandler2 s r� z
AUTH=CRAM-MD5ZtimZtanstaaftanstaafrT s tanstaaftanstaaf)r2 rm Z
assertTruerU Zlogin_cram_md5r )r r� rc rl Zretr� r r r �test_login_cram_md5/ s
z.BaseThreadedNetworkedTests.test_login_cram_md5c C sr Gd d � d t � } | j | � �G \ } } | j t j j � �# | j d d d � � \ } } Wd QXWd QXd S)Nc @ s e Z d Z d d � Z d S)zHBaseThreadedNetworkedTests.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 *
rx z [AUTHENTICATIONFAILED] abortedrT zMYAUTH successful)r= r� rB )r r@ rR r r r rz P s
zYBaseThreadedNetworkedTests.test_aborted_authentication.<locals>.MyServer.cmd_AUTHENTICATEN)r) r* r+ rz r r r r r[ N s r[ r� c S s d S)Nr )r� r r r r{ [ s zHBaseThreadedNetworkedTests.test_aborted_authentication.<locals>.<lambda>)r2 rm rr r rs r| r} )r r[ rc rl rA r� r r r �test_aborted_authenticationK s z6BaseThreadedNetworkedTests.test_aborted_authenticationc C sQ Gd d � d t � } | j | � �& } | j t j j | j | j � Wd QXd S)Nc @ s e Z d Z d d � Z d S)zCBaseThreadedNetworkedTests.test_linetoolong.<locals>.TooLongHandlerc S s# | j j d t j d d � d S)Ns * OK � xs
)r6 r7 r Z_MAXLINE)r r r r rS _ s zJBaseThreadedNetworkedTests.test_linetoolong.<locals>.TooLongHandler.handleN)r) r* r+ rS r r r r �TooLongHandler^ s r� )r2 ri rr r rs r| rj r` )r r� rc r r r �test_linetoolong] s z+BaseThreadedNetworkedTests.test_linetoolongN)r) r* r+ rd rh r ri rm r rn ru rw r~ r� r� r� r� r� r r r r rX � s
rX c @ s" e Z d Z e j Z e j Z d S)�ThreadedNetworkedTestsN) r) r* r+ rp � TCPServerr_ r rs rj r r r r r� h s r� zSSL not availablec @ 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 d | | j
� } | j � Wd QXWd QX| j
t � �0 } | j d | j
d d | �} | j � Wd QXd S)NTz.hostname '127.0.0.1' doesn't match 'localhost'� ssl_contextZ localhostr )r/ �
SSLContext�PROTOCOL_SSLv23Z
CERT_REQUIREDZ verify_modeZcheck_hostnameZload_verify_locations�CAFILEZassertRaisesRegexZCertificateErrorri r2 rj r` rf )r r� rc rl r r r �test_ssl_verifiedt s
z+ThreadedNetworkedTestsSSL.test_ssl_verifiedN) r) r* r+ r, r_ � IMAP4_SSLrj r r� r r r r r� n s r� c @ 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 anonymous�passc
C s5 t | j � � | j | j | j � | _ Wd QXd S)N)r �hostrj �portrc )r r r r �setUp� s zRemoteIMAPTest.setUpc C s9 | j d k r5 t | j � � | j j � Wd QXn d S)N)rc r r� rk )r r r r �tearDown� 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 QXd S)N�
LOGINDISABLEDzAUTH=ANONYMOUSr rT ) r r� rc rU �assertIsInstance�str�assertIn�login�username�passwordr )r �cap�rsr r r �test_logincapa� s z RemoteIMAPTest.test_logincapac
C sF t | j � �1 | j j � } d | _ | j | d d � Wd QXd S)Nr �BYE)r r� rc rk r )r r� r r r � test_logout� s zRemoteIMAPTest.test_logoutN)r) r* r+ r� r� r� r� r rs rj r� r� r� r� r r r r r� � s r� c s. e Z d Z � f d d � Z d d � Z � S)�RemoteIMAP_STARTTLSTestc
sJ t � j � t | j � �( | j j � } | j | d d � Wd QXd S)Nr rT )�superr� r r� rc Zstarttlsr )r r� )� __class__r r r� � s
z RemoteIMAP_STARTTLSTest.setUpc C sA x$ | j j D] } | j | t � q
W| j d | j j � d S)Nr� )rc rU r� r� � assertNotIn)r r� r r r r� � s z&RemoteIMAP_STARTTLSTest.test_logincapa)r) r* r+ r� r� r r )r� r r� � 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)r/ r� r� Zload_cert_chainr0 )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 rT )
rU r� r� r� r� r� r� r� r rk )r rc r� r� r r r �check_logincapa� s z"RemoteIMAP_SSLTest.check_logincapac
C s? t | j � �* | j | j | j � } | j | � Wd QXd S)N)r r� rj r� r� )r �_serverr r r r� � s z!RemoteIMAP_SSLTest.test_logincapac
C sR t | j � �= | j | j | j � } | j � } | j | d d � Wd QXd S)Nr r� )r r� rj r� rk r )r r� r� r r r r� � s z RemoteIMAP_SSLTest.test_logoutc C sK t | j � �6 | j t | j | j | j d t d | j � �Wd QXd S)Nr- r� )r r� rr �
ValueErrorrj r� r0 r� )r r r r �#test_ssl_context_certfile_exclusive� s z6RemoteIMAP_SSLTest.test_ssl_context_certfile_exclusivec C sK t | j � �6 | j t | j | j | j d t d | j � �Wd QXd S)NZkeyfiler� )r r� rr r� rj r� r0 r� )r r r r �"test_ssl_context_keyfile_exclusive� s z5RemoteIMAP_SSLTest.test_ssl_context_keyfile_exclusiveN)r) r* r+ r� r� rj r� r� r� r� r� r� r� r� r r r r r� � s r� c G s� t g } t j d � r� t r� t j j t j j t � p? t j d � a
t j j t
� sl t j d � � n t j j t j j t � p� t j d � a
t j j t
� s� t j d � � q� n | j t t t t t g � n t j d d � | D� � S)NZnetworkz keycert3.pemz Can't read certificate files!z pycacert.pemzCan't read CA file!c S s g | ] } t j | � � q Sr )�unittestZ makeSuite)�.0�testr r r �
<listcomp>� s z load_tests.<locals>.<listcomp>)r
r Zis_resource_enabledr/ �os�pathr? �dirname�__file__�curdirr0 �existsZ
TestFailedr� �extendr� r� r� r� r� r� Z TestSuite)rR Ztestsr r r �
load_tests� s ! ! r� �__main__)+r� r �
import_moduler �
contextlibr r Zos.pathr� rp r r Z test.supportr r r r r r� r
r r r/ � ImportErrorr0 r� ZTestCaser
r� r, r� rq r2 rX r� Z
skipUnlessr� r� r� r� r� r) �mainr r r r �<module> sF (
2 F� 0