U
'7` O @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z m Z ddl
mZ ddlmZmZmZmZmZmZmZmZ ddl mZ ddlmZmZ ddl m Z ddl Z dd l m Z m!Z!m"Z" d
d l#m$Z$m%Z% d
d l
m&Z& d
d
l'm Z d
dl(m)Z) d
dl*m+Z+ d
dl,m-Z-m.Z. d
dl/m0Z0 d
dl1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9 d
dl:m;Z; erxddl<m=Z= ndZ=ej>dkoej?dkZ@eAejdddZBeAeCejddd ZDeCd d d ZEG d d! d!e ZFG d"d# d#eFZGG d$d% d%eFZHG d&d' d'ZIG d(d) d)e jJZKeeeed*d+d,ZLeg ejMf ZNejOejPd-feNeQeejM d.d/d0ZRejPfeNejMd1d2d3ZSdEejMeQdd4d5d6ZTejUd d7d8ZVdFee= ejWd9d:d;ZXdGe+e-d
d
d-de+e+e+e+dd<d=d> eAeAeee-eQeeeeeee= eCee5d?d@dAZYe+e+feeedBdCdDZZdS )HzUtilities shared by tests. N)ABCabstractmethod)
TracebackType)
TYPE_CHECKINGAnyCallableIteratorListOptionalTypeUnion)mock) CIMultiDictCIMultiDictProxy)URL)ClientResponse_RequestContextManager_WSRequestContextManager )
ClientSessionhdrs)AbstractCookieJar)r )ClientWebSocketResponse)sentinel) HttpVersionRawRequestMessage)Signal) Application AppRunner
BaseRunnerRequestServer ServerRunnerSockSiteUrlMappingMatchInfo)_RequestHandler)
SSLContextposixcygwin)hostreturnc C s
t | dS )Nr )get_port_socket)r) r, E/opt/alt/python38/lib64/python3.8/site-packages/aiohttp/test_utils.pyget_unused_port_socket7 s r. )r) portr* c C s8 t t jt j}tr&|t jt jd || |f |S )Nr )socketAF_INET SOCK_STREAM
REUSE_ADDRESSZ
setsockoptZ
SOL_SOCKETZ SO_REUSEADDRbind)r) r/ sr, r, r- r+ ; s
r+ r* c
C s> t t jt j$} | d | d W 5 Q R S Q R X dS )z1Return a port that is unused on the current host.) 127.0.0.1r r N)r0 r1 r2 r4 getsockname)r5 r, r, r- unused_portF s
r9 c @ s0 e Zd ZdZedddddeeef ee j
eee e e
ddddZd#ee j
e
ddd d
Zee
ed d d
ZeedddZee dddZee dddZeedddZddddZddddZee e ee ee dd d d Z d dd d Z ee e ee ee dd d!d"Z!dS )$BaseTestServerFNr7 )schemeloopr) r/ skip_url_asserts)r; r< r) r/ r= kwargsr* c K s4 || _ d | _d | _|| _|| _d| _|| _|| _d S )NF)_looprunner_rootr) r/ _closedr; r= )selfr; r< r) r/ r= r> r, r, r- __init__P s
zBaseTestServer.__init__)r< r> r* c s | j r
d S || _|dd | _| jf |I d H | _ | j I d H | jsNd| _t| j| j}| d d \| _| _t
| j || jd}| I d H |j }|d k st
|j}|d k st
|d d | _| jtkr| jrd}nd}|| _t| j d| j d | j | _d S )
Nsslr )ZsockZ ssl_contextr Zhttpshttpz://:)r@ r? popZ_ssl _make_runnerZsetupr/ r+ r) r8 r# start_serverAssertionErrorsocketsr; r r rA )rC r< r> Z_sockZsiteserverrN r; r, r, r- start_serverc s.
zBaseTestServer.start_serverr> r* c s d S Nr, rC r> r, r, r- rJ s zBaseTestServer._make_runnerpathr* c C sJ | j d k stt|}| js4| r(t| j |S tt| j | S d S rR )rA rM r r= Z is_absolutejoinstr)rC rU Zurlr, r, r- make_url s zBaseTestServer.make_urlr6 c C s
| j d k S rR )r@ rC r, r, r- started s zBaseTestServer.startedc C s | j S rR )rB rY r, r, r- closed s zBaseTestServer.closedc C s& | j }|d k st|jd k s t|jS rR )r@ rM rO )rC r@ r, r, r- handler s zBaseTestServer.handlerc s@ | j r<| js<| jdk st| j I dH d| _d| _d| _dS )a` Close all fixtures created by the test client.
After that point, the TestClient is no longer usable.
This is an idempotent function: running close multiple times
will not have any additional effects.
close is also run when the object is garbage collected, and on
exit when used as a context manager.
NT)rZ r[ r@ rM ZcleanuprA r/ rB rY r, r, r- close s zBaseTestServer.closec C s t dd S NzUse async with instead TypeErrorrY r, r, r- __enter__ s zBaseTestServer.__enter__)exc_type exc_value tracebackr* c C s d S rR r, rC rb rc rd r, r, r- __exit__ s zBaseTestServer.__exit__c s | j | jdI d H | S Nr< )rP r? rY r, r, r-
__aenter__ s zBaseTestServer.__aenter__c s | I d H d S rR r] re r, r, r- __aexit__ s zBaseTestServer.__aexit__)N)"__name__
__module__ __qualname____test__r r rW objectr
asyncioAbstractEventLoopintboolr rD rP r r rJ r rX propertyrZ r[ r! r\ r] ra r
BaseExceptionr rf ri rk r, r, r, r- r: M sV
r: c sP e Zd Zedddeeeef eee e
d fddZ e
e ddd Z
ZS )
TestServerr7 Nr; r) r/ )appr; r) r/ r> c s$ || _ t jf |||d| d S Nrx )ry superrD )rC ry r; r) r/ r> __class__r, r- rD s zTestServer.__init__rQ c s t | jf|S rR )r ry rS r, r, r- rJ s zTestServer._make_runner)rl rm rn r r r rW rp r
rs r rD r rJ
__classcell__r, r, r| r- rw s
rw c sV e Zd Zedddeeeef eee e
dd fddZ d e e
e
dd d
Z ZS )
RawTestServerr7 Nrx )r\ r; r) r/ r> r* c s$ || _ t jf |||d| d S rz )_handlerr{ rD )rC r\ r; r) r/ r> r| r, r- rD s zRawTestServer.__init__T)debugr> r* c s. t | jf| j|d|}t|fd|i|S )N)r< r r )r! r r? r" )rC r r> Zsrvr, r, r- rJ s zRawTestServer._make_runner)T)rl rm rn r r% r rW rp r
rs r rD rt r" rJ r~ r, r, r| r- r s
r c @ s e Zd ZdZdZdddeee eej e
ddddZ ddd d
Z e
edd d Ze
ee dd
dZe
edddZe
edddZe
edddZeedddZeee
edddZeee
eddd Z ee
ed d d Z ee
ed d d!Z ee
ed d"d#Z ee
ed d$d%Z ee
ed d&d'Z!ee
ed d(d)Z"ee
ed d*d+Z#ee
e$d d,d-Z%ee
e&d d.d/Z'ddd0d1Z(ddd2d3Z)ee*e+ ee+ ee, dd4d5d6Z-d dd7d8Z.ee*e+ ee+ ee, dd4d9d:Z/dS );
TestClientzb
A test client implementation.
To write functional tests for aiohttp based servers.
FN)
cookie_jarr< )rO r r< r> r* c K sh t |tstdt| || _|| _|d kr<tjd|d}tf ||d|| _ d| _
g | _ g | _ d S )Nz2server must be TestServer instance, found type: %rT)Zunsafer< )r< r F)
isinstancer: r` typerL r? aiohttpZ CookieJarr _sessionrB
_responses _websockets)rC rO r r< r> r, r, r- rD s
zTestClient.__init__r6 c s | j j| jdI d H d S rg )rL rP r? rY r, r, r- rP s zTestClient.start_serverc C s | j jS rR )rL r) rY r, r, r- r) s zTestClient.hostc C s | j jS rR )rL r/ rY r, r, r- r/ s zTestClient.portc C s | j S rR )rL rY r, r, r- rO s zTestClient.serverc C s t | jdd S )Nry )getattrrL rY r, r, r- ry s zTestClient.appc C s | j S )zAn internal aiohttp.ClientSession.
Unlike the methods on the TestClient, client session requests
do not automatically include the host in the url queried, and
will require an absolute path to the resource.
)r rY r, r, r- session s zTestClient.sessionrT c C s | j |S rR )rL rX )rC rU r, r, r- rX ) s zTestClient.make_url)methodrU r> r* c s. | j j|| |f|I d H }| j| |S rR )r requestrX r append)rC r rU r> respr, r, r- _request, s zTestClient._requestc K s t | j||f|S )zRoutes a request to tested http server.
The interface is identical to aiohttp.ClientSession.request,
except the loop kwarg is overridden by the instance used by the
test server.
)r r )rC r rU r> r, r, r- r 2 s zTestClient.request)rU r> r* c K s t | jtj|f|S )z Perform an HTTP GET request.)r r r ZMETH_GETrC rU r> r, r, r- get< s zTestClient.getc K s t | jtj|f|S )z Perform an HTTP POST request.)r r r Z METH_POSTr r, r, r- post@ s zTestClient.postc K s t | jtj|f|S )z Perform an HTTP OPTIONS request.)r r r Z METH_OPTIONSr r, r, r- optionsD s zTestClient.optionsc K s t | jtj|f|S )z Perform an HTTP HEAD request.)r r r Z METH_HEADr r, r, r- headH s zTestClient.headc K s t | jtj|f|S )z Perform an HTTP PUT request.)r r r ZMETH_PUTr r, r, r- putL s zTestClient.putc K s t | jtj|f|S z Perform an HTTP PATCH request.)r r r Z
METH_PATCHr r, r, r- patchP s zTestClient.patchc K s t | jtj|f|S r )r r r Z METH_DELETEr r, r, r- deleteT s zTestClient.deletec K s t | j|f|S )zjInitiate websocket connection.
The api corresponds to aiohttp.ClientSession.ws_connect.
)r _ws_connectr r, r, r-
ws_connectX s zTestClient.ws_connectc s, | j j| |f|I d H }| j| |S rR )r r rX r r )rC rU r> wsr, r, r- r ` s zTestClient._ws_connectc s^ | j sZ| jD ] }| q | jD ]}| I dH q | j I dH | j I dH d| _ dS )aH Close all fixtures created by the test client.
After that point, the TestClient is no longer usable.
This is an idempotent function: running close multiple times
will not have any additional effects.
close is also run on exit when used as a(n) (asynchronous)
context manager.
NT)rB r r] r r rL )rC r r r, r, r- r] e s
zTestClient.closec C s t dd S r^ r_ rY r, r, r- ra z s zTestClient.__enter__)rb exctbr* c C s d S rR r, rC rb r r r, r, r- rf } s zTestClient.__exit__c s | I d H | S rR )rP rY r, r, r- ri s zTestClient.__aenter__c s | I d H d S rR rj r r, r, r- rk s zTestClient.__aexit__)0rl rm rn __doc__ro r: r
r rq rr r rD rP ru rW r) rs r/ rO r ry r r r rX r r r r r r r r r r r r r r r r] ra r rv r rf ri rk r, r, r, r- r s^
r c @ s e Zd ZdZedddZedddZdddd Zddd
d Zddd d
Z ddddZ
ee dddZ e e
dddZdS )AioHTTPTestCasea A base class to allow for unittest web applications using
aiohttp.
Provides the following:
* self.client (aiohttp.test_utils.TestClient): an aiohttp test client.
* self.loop (asyncio.BaseEventLoop): the event loop in which the
application and server are running.
* self.app (aiohttp.web.Application): the application returned by
self.get_application()
Note that the TestClient's methods are asynchronous: you have to
execute function on the test client using asynchronous methods.
r6 c s | S )zy
This method should be overridden
to return the aiohttp.web.Application
object to test.
)get_apprY r, r, r- get_application s z AioHTTPTestCase.get_applicationc C s t ddS )zqObsolete method used to constructing web application.
Use .get_application() coroutine instead
z+Did you forget to define get_application()?N) RuntimeErrorrY r, r, r- r s zAioHTTPTestCase.get_appNc C sl t | _| j| | _| j| | j| _| j| | j| _| j| j | j|
d S rR ) setup_test_loopr< run_until_completer ry
get_serverrO
get_clientclientrP
setUpAsyncrY r, r, r- setUp s zAioHTTPTestCase.setUpc s d S rR r, rY r, r, r- r s zAioHTTPTestCase.setUpAsyncc C s0 | j | | j | j t| j d S rR )r< r
tearDownAsyncr r] teardown_test_looprY r, r, r- tearDown s zAioHTTPTestCase.tearDownc s d S rR r, rY r, r, r- r s z AioHTTPTestCase.tearDownAsync)ry r* c s t || jdS )z Return a TestServer instance.rh )rw r< )rC ry r, r, r- r s zAioHTTPTestCase.get_server)rO r* c s t || jdS )z Return a TestClient instance.rh )r r< )rC rO r, r, r- r s zAioHTTPTestCase.get_client)rl rm rn r r r r r r r r rw r r r r, r, r, r- r s r )funcargsr> r* c s. t j f||ttttd fdd }|S )zA decorator dedicated to use with asynchronous methods of an
AioHTTPTestCase.
Handles executing an asynchronous function, using
the self.loop of the AioHTTPTestCase.
)rC
inner_args inner_kwargsr* c s | j | f||S rR )r< r )rC r r r r, r- new_func s z#unittest_run_loop.<locals>.new_func) functoolswrapsr )r r r> r r, r r- unittest_run_loop s r F) loop_factoryfastr* c c s t | }|V t||d dS )zzA contextmanager that creates an event_loop, for test purposes.
Handles the creation and cleanup of a test loop.
)r N)r r )r r r< r, r, r- loop_context s r )r r* c C s | }z|j j}d|k}W n tk
r2 d}Y nX t| tjdkr|st }t }| | t
t |
| W 5 Q R X |S )zCreate and return an asyncio.BaseEventLoop
instance.
The caller should also call teardown_test_loop,
once they are done with the loop.
ZuvloopTZwin32)r} rm AttributeErrorrq set_event_loopsysplatformZget_event_loop_policyZSafeChildWatcherZ attach_loop
contextlibsuppressNotImplementedErrorZset_child_watcher)r r< moduleZ skip_watcherZpolicyZwatcherr, r, r- r s
r )r< r r* c C sB | }|s(| | j | | |s4t td dS )zHTeardown and cleanup an event_loop created
by setup_test_loop.
N) Z is_closedZ call_soonstopZ run_foreverr] gcZcollectrq r )r< r r[ r, r, r- r s r c C s^ t tt ddd} t tt d ddd}t }i |_| |_||_d|_t||_ |j
|S )N)ry keyr* c S s
| j | S rR
__app_dict)ry r r, r, r- get_dict s z"_create_app_mock.<locals>.get_dict)ry r valuer* c S s || j |< d S rR r )ry r r r, r, r- set_dict! s z"_create_app_mock.<locals>.set_dictF) r rW r
MagicMockr __getitem__ __setitem__Z_debugr Zon_response_prepareZfreeze)r r ry r, r, r- _create_app_mock s
r )
sslcontextr* c s, t }ttt d fdd }||j_|S )N)r r* c s | dkr S d S d S )Nr r, )r r r, r- get_extra_info2 s z)_create_transport.<locals>.get_extra_info)r
MockrW r
r& r Z side_effect)r transportr r, r r- _create_transport/ s r i .)
match_infoversionclosingry writerprotocolr payloadr client_max_sizer< )r rU headersr r r ry r r r r r r r< r* c C st t }|
dkr t }
d|
j_|tddk r2d}|rZtt|}tdd | D }ntt }d}d| t
j d k}t
| |||||d d |t|
}|d
krt }| tkrt| } |tkrt }| |_|tkrt }td
|_td
|_td
|_td
|_| |_| |_||_|
tkr.t }
t||
||||
| d }t|tkrTi n|t }|| ||_ |S ) zCreates mocked web.Request testing purposes.
Useful in unit tests, when spinning full web server is overkill or
specific conditions and errors are hard to trigger.
.r, r Tc s s& | ] \}}| d | d fV qdS )zutf-8N)encode).0kvr, r, r- <genexpr>^ s z&make_mocked_request.<locals>.<genexpr>chunked FN)r ) r
r Z
create_future return_valuer r r tupleitemsr r ZTRANSFER_ENCODINGlowerr r r r r r make_mocked_coroZ
write_headerswriteZ write_eofZdrainr r r$ Zadd_appZ _match_info)r rU r r r r ry r r r r r r r< ZtaskZraw_hdrsr messageZreqr, r, r- make_mocked_request< sx
r )r raise_exceptionr* c s$ t t t d fdd }tj|dS )zCreates a coroutine mock.)r r> r* c s( t k r tsS I d H d S rR )r inspectZ isawaitable)r r> r r r, r- mock_coro s
z#make_mocked_coro.<locals>.mock_coro)r )r r
r )r r r r, r r- r s r )F)N)N)[r rq r r r r osr0 r Zunittestabcr r typesr typingr r r r r r
r r r
Z multidictr r Zyarlr r Zaiohttp.clientr r r r r r r Z
client_reqrepZ client_wsr Zhelpersr rG r r Zsignalsr Zwebr r r r r! r" r# r$ Z web_protocolr% rE r& namer r3 rW r. rs r+ r9 r: rw r r ZTestCaser r rr Z
_LOOP_FACTORYcontextmanagerZnew_event_looprt r r r r r r r r r r, r, r, r- <module> s ( (
| (@
]