3
\:
@ 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)supportZwin32)
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 */usr/lib64/python3.6/asyncio/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 r0 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 )r4 formatargsr r r log_messagel s z$SilentWSGIRequestHandler.log_messageN)__name__
__module__ __qualname__r5 r8 r r r r r1 g s r1 c s( e Zd ZdZ fddZdd Z ZS )SilentWSGIServer c s" t j \}}|j| j ||fS )N)super get_request
settimeoutrequest_timeout)r4 request client_addr) __class__r r r? t s z SilentWSGIServer.get_requestc C s d S )Nr )r4 rB client_addressr r r handle_errory s z SilentWSGIServer.handle_error)r9 r: r; rA r? rF
__classcell__r r )rD 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)r4 rB rE ZkeyfileZcertfilecontextZssockr r r finish_request s z!SSLWSGIServerMixin.finish_requestN)r9 r: r; rM r r r r rH } s rH c @ s e Zd ZdS )
SSLWSGIServerN)r9 r: r; r r r r rN s rN 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)rP rQ r )environZstart_responseZstatusZheadersr 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)
r1 Zset_appZserver_addressaddress threadingZThreadstartshutdownZ server_closer )rW rO
server_clsserver_ssl_clsrS Z server_classZ
server_threadr )rT 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)r4 r r r rb s zUnixHTTPServer.server_bindN)r9 r: r; rb 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^ rb Z
setup_environ)r4 r r r rb s
zUnixWSGIServer.server_bindc s" t j \}}|j| j |dfS )N 127.0.0.1 )rd re )r> r? r@ rA )r4 rB rC )rD r r r? s zUnixWSGIServer.get_request)r9 r: r; rA rb r? rG r r )rD r rc s rc c @ s e Zd Zdd ZdS )SilentUnixWSGIServerc C s d S )Nr )r4 rB rE r r r rF s z!SilentUnixWSGIServer.handle_errorN)r9 r: r; rF r r r r rf s rf c @ s e Zd ZdS )UnixSSLWSGIServerN)r9 r: r; r r r r rg s rg c C s t j } | jS Q R X d S )N)tempfileZNamedTemporaryFilename)filer r r gen_unix_socket_path s
rk c c s< t } z
| V W d ytj| W n tk
r4 Y nX X d S )N)rk r unlinkrK )r r r r unix_socket_path s
rm c
c s, t }t|| ttdE d H W d Q R X d S )N)rW rO r[ r\ )rm r] rf rg )rO r r r r run_test_unix_server s rn z 127.0.0.1)hostportrO c c s t | |f|ttdE d H d S )N)rW rO r[ r\ )r] r< rN )ro rp rO r r r run_test_server s
rq c C sP i }x4t | D ](}|jdr(|jdr(qtd d||< qW td| f| j | S )N__) return_valueZ TestProtocol)dir
startswithendswith MockCallbacktype __bases__)baseZdctri 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)r4 r r r __init__ s zTestSelector.__init__Nc C s t j|d||}|| j|< |S )Nr )r Z SelectorKeyr} )r4 fileobjr datakeyr r r register s
zTestSelector.registerc C s | j j|S )N)r} pop)r4 r r r r
unregister s zTestSelector.unregisterc C s g S )Nr )r4 r- r r r select s zTestSelector.selectc C s | j S )N)r} )r4 r r r get_map s zTestSelector.get_map)N)r9 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)r4 r' )rD r r r~ ( s
zTestLoop.__init__c C s | j S )N)r )r4 r r r r+ ? s z
TestLoop.timec C s |r| j |7 _ d S )N)r )r4 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)r4 )rD r r r% G s
zTestLoop.closec G s t j||| | j|< d S )N)r Handler )r4 fdcallbackr7 r r r _add_readerQ s zTestLoop._add_readerc C s0 | j | d7 < || jkr(| j|= dS dS d S )Nr TF)remove_reader_countr )r4 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)r4 r r r7 handler r r
assert_reader\ s
zTestLoop.assert_readerc C s || j krtd| dd S )Nzfd z is registered)r r )r4 r r r r assert_no_readerg s
zTestLoop.assert_no_readerc G s t j||| | j|< d S )N)r r r )r4 r r r7 r r r _add_writerk s zTestLoop._add_writerc C s0 | j | d7 < || jkr(| j|= dS dS d S )Nr TF)remove_writer_countr )r4 r r r r _remove_writern s
zTestLoop._remove_writerc G s | j | }d S )N)r )r4 r r r7 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 RuntimeErrorr6 )r4 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 )r4 r r r7 r r r
add_reader s
zTestLoop.add_readerc C s | j | | j|S )N)r r )r4 r r r r
remove_reader s
zTestLoop.remove_readerc G s | j | | j||f| S )N)r r )r4 r r r7 r r r
add_writer s
zTestLoop.add_writerc C s | j | | j|S )N)r r )r4 r r r r
remove_writer s
zTestLoop.remove_writerc C s t jt| _t jt| _d S )N) collections defaultdictintr r )r4 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 )r4 whenr )rD r r r s
zTestLoop._run_oncec s | j j| t j||f| S )N)r appendr> call_at)r4 r r r7 )rD r r r s zTestLoop.call_atc C s d S )Nr )r4 Z
event_listr r r _process_events s zTestLoop._process_eventsc C s d S )Nr )r4 r r r _write_to_self s zTestLoop._write_to_self)N)r9 r: r; r~ r+ r r% r r r r r r r r r r r r r r r r r rG r r )rD r r s*
r c K s t jf ddgi| S )Nspec__call__)r ZMock)kwargsr r r rw s rw c @ s e Zd Zdd ZdS ) MockPatternc C s t tjt| |tjS )N)boolresearchstrS)r4 otherr r r __eq__ s zMockPattern.__eq__N)r9 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_executorrZ r% )r& Zexecutorr r r
close_loop s zTestCase.close_loopT)cleanupc C s t jd |r | j| j| d S )N)r set_event_loopZ
addCleanupr )r4 r& r r r r r s
zTestCase.set_event_loopNc C s t |}| j| |S )N)r r )r4 r' r& r r r
new_test_loop s
zTestCase.new_test_loopc C s | j t_ d S )N)_get_running_loopr )r4 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 rU s z TestCase.setUp..)r r r Zthreading_setup_thread_cleanup)r4 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)r4 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 )r4 r r r __enter__ s z+TestCase.subTest..EmptyCM.__enter__c W s d S )Nr )r4 excr r r __exit__ s z*TestCase.subTest..EmptyCM.__exit__N)r9 r: r; r r r r r r EmptyCM s r r )r4 r7 r r r r r subTest s zTestCase.subTest)N)
r9 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 MagicMocksocketprotorx familyZ
gettimeoutrs )r rx r Zsockr r r mock_nonblocking_socket s r c C s t jdddS )Nz'asyncio.sslproto._is_sslproto_availableF)rs )r Zpatchr r r r force_legacy_ssl_support s r )r* )Mr
contextlibr2 r r r r r` r rh rX r+ Zunittestr r Z http.serverr Zwsgiref.simple_serverr r r ImportErrorre r r r r
r r Z
coroutinesr
logr r r platformZ
windows_utilsr r rJ rI r" r) r. r0 r1 r< rH rN r] r ra r^ rc rf rg rk contextmanagerrm rn rq r{ Z BaseSelectorr| Z
BaseEventLoopr rw r r r r r Z IPPROTO_TCPZ SOCK_STREAMZAF_INETr r r r r r s
4