3
2a:
@ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dl mZ d dlmZ d dlmZmZ y d dlZW n ek
r dZY nX ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lm Z dd l m Z dd
l m Z d dl!m"Z" ej#dkrDddl$m%Z% n d dlm%Z% dd Z&e&dZ'e&dZ(dd Z)dd Z*dQddZ+d d Z,G d d d eZ-G d d! d!eZ.G d"d# d#Z/G d$d% d%e/e.Z0d&d'd(d)Z1e2ed*rVG d+d, d,ej3eZ4G d-d. d.e4eZ5G d/d0 d0e5Z6G d1d2 d2e/e6Z7d3d4 Z8ej9d5d6 Z:ej9d&d'd7d8Z;ej9d9d d&d:d;d<Z Z=G d?d@ d@ej>Z?G dAdB dBej@ZAdCdD ZBG dEdF dFeCZDdGdH ZEG dIdJ dJe jFZFej9dKdL ZGejHejIejJfdMdNZKdOdP ZLdS )R N)mock)
HTTPServer)WSGIRequestHandler
WSGIServer ) base_events)compat)events)futures) selectors)tasks) coroutine)logger)supportwin32)
socketpairc C s` t tdr*tjjtj| }tjj|r*|S tjjtjjtjd| }tjj|rT|S t | d S )N
TEST_HOME_DIRtest)
hasattrr ospathjoinr isfiledirname__file__FileNotFoundError)filenamefullname r //opt/alt/python36/lib64/python3.6/test_utils.py data_file- s
r z ssl_cert.pemz ssl_key.pemc C s t d kr d S t jt jS d S )N)ssl
SSLContextZPROTOCOL_SSLv23r r r r dummy_ssl_context< s r# c
C s@ t dd }| }| j|}d|_z| j| W d |j X d S )Nc S s d S )Nr r r r r onceD s zrun_briefly..onceF)r
Z create_taskZ_log_destroy_pendingrun_until_completeclose)loopr$ gentr r r run_brieflyC s
r* c C sT t j | }xB| sN|d k r8|t j }|dkr8tj | jtjd| d qW d S )Nr gMbP?)r' )timer
TimeoutErrorr% r Zsleep)r' ZpredtimeoutZdeadliner r r run_untilR s r/ c C s | j | j | j d S )N)Z call_soonstopZ run_forever)r' r r r run_once\ s r1 c @ s e Zd Zdd Zdd ZdS )SilentWSGIRequestHandlerc C s t j S )N)ioStringIO)selfr r r
get_stderri s z#SilentWSGIRequestHandler.get_stderrc G s d S )Nr )r5 formatargsr r r log_messagel s z$SilentWSGIRequestHandler.log_messageN)__name__
__module__ __qualname__r6 r9 r r r r r2 g s r2 c s( e Zd ZdZ fddZdd Z ZS )SilentWSGIServer c s" t j \}}|j| j ||fS )N)super get_request
settimeoutrequest_timeout)r5 request client_addr) __class__r r r@ t s z SilentWSGIServer.get_requestc C s d S )Nr )r5 rC client_addressr r r handle_errory s z SilentWSGIServer.handle_error)r: r; r< rB r@ rG
__classcell__r r )rE r r= p s r= c @ s e Zd Zdd ZdS )SSLWSGIServerMixinc C s^ t }t}tj }|j|| |j|dd}y| j||| |j W n tk
rX Y nX d S )NT)Z server_side) ONLYKEYONLYCERTr! r" Zload_cert_chainZ wrap_socketZRequestHandlerClassr& OSError)r5 rC rF ZkeyfileZcertfilecontextZssockr r r finish_request s z!SSLWSGIServerMixin.finish_requestN)r: r; r< rN r r r r rI } s rI c @ s e Zd ZdS )
SSLWSGIServerN)r: r; r< r r r r rO s rO F)use_sslc
# sv dd }|r|n|}|| t j| j _tj fddd}|j z
V W d j j |j X d S )Nc S s d}dg}||| dgS )Nz200 OK Content-type
text/plains Test message)rQ rR r )environZstart_responsestatusZheadersr r r app s
z _run_test_server..appc s j ddS )Ng?)Z
poll_interval)Z
serve_foreverr )httpdr r s z"_run_test_server..)target)
r2 Zset_appZserver_addressaddress threadingZThreadstartshutdownZ server_closer )rY rP
server_clsserver_ssl_clsrU Z server_classZ
server_threadr )rV r _run_test_server s
r_ ZAF_UNIXc @ s e Zd Zdd ZdS )UnixHTTPServerc C s t jj| d| _d| _d S )Nz 127.0.0.1P ) socketserverUnixStreamServer server_bindZ server_nameZ server_port)r5 r r r rd s zUnixHTTPServer.server_bindN)r: r; r< rd r r r r r` s r` c s( e Zd ZdZdd Z fddZ ZS )UnixWSGIServerr> c C s t j| | j d S )N)r` rd Z
setup_environ)r5 r r r rd s
zUnixWSGIServer.server_bindc s" t j \}}|j| j |dfS )N 127.0.0.1 )rf rg )r? r@ rA rB )r5 rC rD )rE r r r@ s zUnixWSGIServer.get_request)r: r; r< rB rd r@ rH r r )rE r re s re c @ s e Zd Zdd ZdS )SilentUnixWSGIServerc C s d S )Nr )r5 rC rF r r r rG s z!SilentUnixWSGIServer.handle_errorN)r: r; r< rG r r r r rh s rh c @ s e Zd ZdS )UnixSSLWSGIServerN)r: r; r< r r r r ri s ri c C s t j } | jS Q R X d S )N)tempfileZNamedTemporaryFilename)filer r r gen_unix_socket_path s
rm c c s< t } z
| V W d ytj| W n tk
r4 Y nX X d S )N)rm r unlinkrL )r r r r unix_socket_path s
ro c
c s, t }t|| ttdE d H W d Q R X d S )N)rY rP r] r^ )ro r_ rh ri )rP r r r r run_test_unix_server s rp z 127.0.0.1)hostportrP c c s t | |f|ttdE d H d S )N)rY rP r] r^ )r_ r= rO )rq rr rP r r r run_test_server s
rs c C sP i }x4t | D ](}|jdr(|jdr(qtd d||< qW td| f| j | S )N__) return_valueZ TestProtocol)dir
startswithendswith MockCallbacktype __bases__)baseZdctrk r r r make_test_protocol s r} c @ s6 e Zd Zdd Zd ddZdd Zdd Zd
d ZdS )
TestSelectorc C s
i | _ d S )N)keys)r5 r r r __init__ s zTestSelector.__init__Nc C s t j|d||}|| j|< |S )Nr )r Z SelectorKeyr )r5 fileobjr datakeyr r r register s
zTestSelector.registerc C s | j j|S )N)r pop)r5 r r r r
unregister s zTestSelector.unregisterc C s g S )Nr )r5 r. r r r select s zTestSelector.selectc C s | j S )N)r )r5 r r r get_map s zTestSelector.get_map)N)r: r; r< r r r r r r r r r r~ s
r~ c s e Zd Zd, fdd Zdd Zdd Z fdd Zd
d Zd d
Zdd Z dd Z
dd Z dd Z dd Z
dd Zdd Zd d Zd d Zd d! Zd"d# Z fd$d%Z fd&d'Zd(d) Zd*d+ Z ZS )-TestLoopNc sv t j |d kr"dd }d| _nd| _| | _t| j d| _d| _g | _t | _ i | _
i | _ | j t
j | _d S )Nc s s
d V d S )Nr r r r r r( , s z TestLoop.__init__..genFTr g& .>)r? r _check_on_close_gennext_timeZ_clock_resolution_timersr~ Z _selectorreaderswritersreset_countersweakrefWeakValueDictionary _transports)r5 r( )rE r r r ( s
zTestLoop.__init__c C s | j S )N)r )r5 r r r r, ? s z
TestLoop.timec C s |r| j |7 _ d S )N)r )r5 advancer r r advance_timeB s zTestLoop.advance_timec sB t j | jr>y| jjd W n tk
r4 Y n
X tdd S )Nr z Time generator is not finished)r? r& r r send
StopIterationAssertionError)r5 )rE r r r& G s
zTestLoop.closec G s t j||| | j|< d S )N)r Handler )r5 fdcallbackr8 r r r _add_readerQ s zTestLoop._add_readerc C s0 | j | d7 < || jkr(| j|= dS dS d S )Nr TF)remove_reader_countr )r5 r r r r _remove_readerT s
zTestLoop._remove_readerc G sh || j krtd| d| j | }|j|krDtd|j d| |j|krdtd|j d| d S )Nzfd z is not registeredzunexpected callback: z != zunexpected callback args: )r r Z _callbackZ_args)r5 r r r8 handler r r
assert_reader\ s
zTestLoop.assert_readerc C s || j krtd| dd S )Nzfd z is registered)r r )r5 r r r r assert_no_readerg s
zTestLoop.assert_no_readerc G s t j||| | j|< d S )N)r r r )r5 r r r8 r r r _add_writerk s zTestLoop._add_writerc C s0 | j | d7 < || jkr(| j|= dS dS d S )Nr TF)remove_writer_countr )r5 r r r r _remove_writern s
zTestLoop._remove_writerc G s | j | }d S )N)r )r5 r r r8 r r r r
assert_writerv s
zTestLoop.assert_writerc C s8 y| j | }W n tk
r" Y nX tdj||d S )Nz.File descriptor {!r} is used by transport {!r})r KeyError RuntimeErrorr7 )r5 r Z transportr r r _ensure_fd_no_transport~ s z TestLoop._ensure_fd_no_transportc G s | j | | j||f| S )N)r r )r5 r r r8 r r r
add_reader s
zTestLoop.add_readerc C s | j | | j|S )N)r r )r5 r r r r
remove_reader s
zTestLoop.remove_readerc G s | j | | j||f| S )N)r r )r5 r r r8 r r r
add_writer s
zTestLoop.add_writerc C s | j | | j|S )N)r r )r5 r r r r
remove_writer s
zTestLoop.remove_writerc C s t jt| _t jt| _d S )N) collections defaultdictintr r )r5 r r r r s zTestLoop.reset_countersc s: t j x$| jD ]}| jj|}| j| qW g | _d S )N)r? _run_oncer r r r )r5 whenr )rE r r r s
zTestLoop._run_oncec s | j j| t j||f| S )N)r appendr? call_at)r5 r r r8 )rE r r r s zTestLoop.call_atc C s d S )Nr )r5 Z
event_listr r r _process_events s zTestLoop._process_eventsc C s d S )Nr )r5 r r r _write_to_self s zTestLoop._write_to_self)N)r: r; r< r r, r r& r r r r r r r r r r r r r r r r r rH r r )rE r r s*
r c K s t jf ddgi| S )Nspec__call__)r ZMock)kwargsr r r ry s ry c @ s e Zd Zdd ZdS ) MockPatternc C s t tjt| |tjS )N)boolresearchstrS)r5 otherr r r __eq__ s zMockPattern.__eq__N)r: r; r< r r r r r r s r c C s$ t j| }|d kr td| f |S )Nz unable to get the source of %r)r Z_get_function_source
ValueError)funcsourcer r r get_function_source s
r c @ sV e Zd Zedd ZddddZddd Zd
d Zd d
Zdd Z e
j sRdd Z dS )TestCasec C s& | j }|d k r|jdd | j d S )NT)wait)Z_default_executorr\ r& )r' executorr r r
close_loop s zTestCase.close_loopT)cleanupc C s t jd |r | j| j| d S )N)r set_event_loopZ
addCleanupr )r5 r' r r r r r s
zTestCase.set_event_loopNc C s t |}| j| |S )N)r r )r5 r( r' r r r
new_test_loop s
zTestCase.new_test_loopc C s | j t_ d S )N)_get_running_loopr )r5 r r r unpatch_get_running_loop s z!TestCase.unpatch_get_running_loopc C s t j| _dd t _tj | _d S )Nc S s d S )Nr r r r r rW s z TestCase.setUp..)r r r Zthreading_setup_thread_cleanup)r5 r r r setUp s
zTestCase.setUpc C sB | j tjd | jtj d | j tj| j tj
d S )N)NNN) r r r Z assertEqualsysexc_infoZ
doCleanupsr Zthreading_cleanupr Z
reap_children)r5 r r r tearDown s
zTestCase.tearDownc O s G dd d}| S )Nc @ s e Zd Zdd Zdd ZdS )z!TestCase.subTest..EmptyCMc S s d S )Nr )r5 r r r __enter__ s z+TestCase.subTest..EmptyCM.__enter__c W s d S )Nr )r5 excr r r __exit__ s z*TestCase.subTest..EmptyCM.__exit__N)r: r; r< r r r r r r EmptyCM s r r )r5 r8 r r r r r subTest s zTestCase.subTest)N)
r: r; r< staticmethodr r r r r r r ZPY34r r r r r r s
r c c s2 t j} zt jtjd d V W d t j| X d S )Nr )r levelZsetLevelloggingZCRITICAL)Z old_levelr r r disable_logger s
r c C s* t jtj}| |_||_||_d|j_|S )Ng )r Z MagicMocksocketprotorz familyZ
gettimeoutru )r rz r Zsockr r r mock_nonblocking_socket s r c C s t jdddS )Nz'asyncio.sslproto._is_sslproto_availableF)ru )r Zpatchr r r r force_legacy_ssl_support s r )r+ )Mr
contextlibr3 r r r r rb r rj rZ r, Zunittestr r Z http.serverr Zwsgiref.simple_serverr r r! ImportErrorrg r r r r
r r Z
coroutinesr
logr r r platformZ
windows_utilsr r rK rJ r# r* r/ r1 r2 r= rI rO r_ r rc r` re rh ri rm contextmanagerro rp rs r} Z BaseSelectorr~ Z
BaseEventLoopr ry r r r r r Z IPPROTO_TCPZ SOCK_STREAMZAF_INETr r r r r r s
4