��]�8�@sddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddlZddlZddl Z ddlmZddlmZddlmZmZyddlZWnek rdZYnXddlmZddlmZddlmZdd lmZdd lmZddlmZddlmZdd lm Z ddl!m"Z"ej#dkr�ddl$m%Z%nddlm%Z%dd�Z&dd�Z'ddd�Z(dd�Z)Gdd�de�Z*Gdd�de�Z+Gdd�d�Z,Gd d!�d!e,e+�Z-d"d#d$d%�Z.e/ed&�rGd'd(�d(ej0e�Z1Gd)d*�d*e1e�Z2Gd+d,�d,e2�Z3Gd-d.�d.e,e3�Z4d/d0�Z5ej6d1d2��Z7ej6d"d#d3d4��Z8ej6d5d6d7dd"d#d8d9��Z9d:d;�Z:Gd<d=�d=ej;�Z<Gd>d?�d?ej=�Z>d@dA�Z?GdBdC�dCe@�ZAdDdE�ZBGdFdG�dGejC�ZCej6dHdI��ZDejEejFejGdJdK�ZHdLdM�ZIdS)N�N)�mock)� HTTPServer)�WSGIRequestHandler� WSGIServer�)�base_events)�compat)�events)�futures)� selectors)�tasks)� coroutine)�logger)�support�win32)� socketpaircCs$tdkrdStjtj�SdS)N)�sslZ SSLContextZPROTOCOL_SSLv23�rr�//opt/alt/python35/lib64/python3.5/test_utils.py�dummy_ssl_context-src CsVtdd��}|�}|j|�}d|_z|j|�Wd|j�XdS)NcSsdS)Nrrrrr�once5szrun_briefly..onceF)r Zcreate_taskZ_log_destroy_pending�run_until_complete�close)�loopr�gen�trrr�run_briefly4s r�cCsttj�|}x]|�so|dk rP|tj�}|dkrPtj��|jtjdd|��qWdS)Nrg��MbP?r)�timer �TimeoutErrorrr�sleep)rZpred�timeout�deadlinerrr� run_untilCsr#cCs|j|j�|j�dS)N)Z call_soon�stopZrun_forever)rrrr�run_onceMsr%c@s(eZdZdd�Zdd�ZdS)�SilentWSGIRequestHandlercCs tj�S)N)�io�StringIO)�selfrrr� get_stderrZsz#SilentWSGIRequestHandler.get_stderrcGsdS)Nr)r)�format�argsrrr�log_message]sz$SilentWSGIRequestHandler.log_messageN)�__name__� __module__�__qualname__r*r-rrrrr&Xsr&cs4eZdZdZ�fdd�Zdd�Z�S)�SilentWSGIServer�cs/t�j�\}}|j|j�||fS)N)�super�get_request� settimeout�request_timeout)r)�request�client_addr)� __class__rrr4eszSilentWSGIServer.get_requestcCsdS)Nr)r)r7�client_addressrrr�handle_errorjszSilentWSGIServer.handle_error)r.r/r0r6r4r;rr)r9rr1asr1c@seZdZdd�ZdS)�SSLWSGIServerMixincCs�tjjtjjt�dd�}tjj|�s]tjjtjjtj�dd�}tjj|d�}tjj|d�}tj|d|d|d d �}y!|j|||�|j �Wnt k r�YnXdS)Nz..Ztests�testZtest_asynciozssl_key.pemzssl_cert.pem�keyfile�certfileZserver_sideT)�os�path�join�dirname�__file__�isdirrZwrap_socketZRequestHandlerClassr�OSError)r)r7r:�herer>r?Zssockrrr�finish_requestps$ z!SSLWSGIServerMixin.finish_requestN)r.r/r0rHrrrrr<nsr<c@seZdZdS)� SSLWSGIServerN)r.r/r0rrrrrI�srI�use_sslFc #s�dd�}|r|n|}||t��j|��j�_tjd�fdd��}|j�z �VWd�j��j�|j �XdS)NcSs#d}dg}|||�dgS)Nz200 OK�Content-type� text/plainsTest message)rKrLr)�environZstart_response�statusZheadersrrr�app�s z_run_test_server..app�targetcs�jdd�S)NZ poll_intervalg��?)Z serve_foreverr)�httpdrr��sz"_run_test_server..) r&Zset_appZserver_address�address� threading�Thread�start�shutdownZserver_closerB)rSrJ� server_cls�server_ssl_clsrOZserver_classZ server_threadr)rQr�_run_test_server�s rZ�AF_UNIXc@seZdZdd�ZdS)�UnixHTTPServercCs&tjj|�d|_d|_dS)Nz 127.0.0.1�P)�socketserver�UnixStreamServer�server_bindZserver_nameZserver_port)r)rrrr`�s zUnixHTTPServer.server_bindN)r.r/r0r`rrrrr\�sr\cs4eZdZdZdd�Z�fdd�Z�S)�UnixWSGIServerr2cCstj|�|j�dS)N)r\r`Z setup_environ)r)rrrr`�s zUnixWSGIServer.server_bindcs/t�j�\}}|j|j�|dfS)N� 127.0.0.1�)rbrc)r3r4r5r6)r)r7r8)r9rrr4�szUnixWSGIServer.get_request)r.r/r0r6r`r4rr)r9rra�srac@seZdZdd�ZdS)�SilentUnixWSGIServercCsdS)Nr)r)r7r:rrrr;�sz!SilentUnixWSGIServer.handle_errorN)r.r/r0r;rrrrrd�srdc@seZdZdS)�UnixSSLWSGIServerN)r.r/r0rrrrre�srec Cs!tj��}|jSWdQRXdS)N)�tempfile�NamedTemporaryFile�name)�filerrr�gen_unix_socket_path�srjccs@t�}z |VWdytj|�Wntk r:YnXXdS)N)rjr@�unlinkrF)rArrr�unix_socket_path�s rlccs;t��+}td|d|dtdt�EdHWdQRXdS)NrSrJrXrY)rlrZrdre)rJrArrr�run_test_unix_server�srm�hostz 127.0.0.1�portc cs.td||fd|dtdt�EdHdS)NrSrJrXrY)rZr1rI)rnrorJrrr�run_test_server�srpcCsni}xHt|�D]:}|jd�r:|jd�r:qtdd�||.genFTrg��&�.>)r3r~�_check_on_close�_gen�next�_timeZ_clock_resolution�_timersr|� _selector�readers�writers�reset_counters�weakref�WeakValueDictionary�_transports)r)r)r9rrr~s zTestLoop.__init__cCs|jS)N)r�)r)rrrr4sz TestLoop.timecCs|r|j|7_dS)N)r�)r)�advancerrr�advance_time7szTestLoop.advance_timecsOt�j�|jrKy|jjd�Wntk r>Yn Xtd��dS)NrzTime generator is not finished)r3rr�r��send� StopIteration�AssertionError)r))r9rrr<s zTestLoop.closecGs tj|||�|j|.)r r�rZthreading_setup�_thread_cleanup)r)rrr�setUp�szTestCase.setUpcCsU|j�tjd�|jtj�d�|j�tj|j �tj �dS)N)NNN)r�r r�ZassertEqual�sys�exc_infoZ doCleanupsrZthreading_cleanupr�Z reap_children)r)rrr�tearDown�s zTestCase.tearDowncOsGdd�d�}|�S)Nc@s(eZdZdd�Zdd�ZdS)z!TestCase.subTest..EmptyCMcSsdS)Nr)r)rrr� __enter__�sz+TestCase.subTest..EmptyCM.__enter__cWsdS)Nr)r)�excrrr�__exit__�sz*TestCase.subTest..EmptyCM.__exit__N)r.r/r0r�r�rrrr�EmptyCM�sr�r)r)r,r�r�rrr�subTest�szTestCase.subTest)r.r/r0r�r�r�r�r�rZPY34r�rrrrr��s r�ccs;tj}ztjtjd�dVWdtj|�XdS)Nr)r�level�setLevel�logging�CRITICAL)Z old_levelrrr�disable_logger�s r�cCs=tjtj�}||_||_||_d|j_|S)Ng)rZ MagicMock�socket�protorw�family� gettimeoutrr)r�rwr��sockrrr�mock_nonblocking_socket�s r�cCstjddd�S)Nz'asyncio.sslproto._is_sslproto_availablerrF)rZpatchrrrr�force_legacy_ssl_support�sr�)Jr�� contextlibr'r�r@r�r�r^r�rfrTrZunittestr�rZhttp.serverrZwsgiref.simple_serverrrr�ImportErrorrcrrr r rrZ coroutinesr �logrr=r�platformZ windows_utilsrrrr#r%r&r1r<rIrZ�hasattrr_r\rardrerj�contextmanagerrlrmrpr{�BaseSelectorr|Z BaseEventLoopr�rvr�r�r�r�r��IPPROTO_TCP�SOCK_STREAM�AF_INETr�r�rrrr�s~ � -