UV ` d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z m Z ddl
mZ ddlmZmZmZmZmZmZmZmZmZ ddlmZmZ ddl m Z ddl m Z m Z dd l!m"Z" ddl#Z#dd
l$m%Z%m&Z& d d l'm(Z(m)Z) d d
l
m*Z* d dl+m,Z, d dl-m.Z. d dl/m0Z0 d dl1m2Z2m3Z3 d dl4m5Z5 d dl6m7Z7m8Z8m9Z9m:Z:m;Z;mZ> d dl?m@Z@ erddlAmBZB ndZBejC dk o
ejD dk ZEejF fdeGdejH dej fdZIdeGd eJdejH dej fd ZKdeJfd ZL G d d e ZM G d! d"eM ZN G d# d$eM ZO G d% d& ZP G d' d(e ZQd)ed*ed+edefd,ZReg ejS f ZTejU ejV d-fd.eTd/eWdeejS fd0 ZXejV fd.eTdejS fd1ZYdJd2ejS d/eWddfd3ZZdej[ fd4Z\dKd5eeB dej] fd6Z^ dKe0 e2d d d-de0e0e0e0dd7d8d9 d:eGd;eGde2d?eWd@edAedBedCedDed5eeB dEeJd2ede:f dFZ_e0e0fdGedHedefdIZ`dS )LzUtilities shared by tests. N)ABCabstractmethod)
TracebackType)
TYPE_CHECKINGAnyCallableIteratorListOptionalTypeUnioncast)IsolatedAsyncioTestCasemock)Signal) CIMultiDictCIMultiDictProxy)URL)_RequestContextManager_WSRequestContextManager )
ClientSessionhdrs)AbstractCookieJar)ClientResponse)ClientWebSocketResponse)sentinel) HttpVersionRawRequestMessage)StrOrURL) Application AppRunner
BaseRunnerRequestServer ServerRunnerSockSiteUrlMappingMatchInfo)_RequestHandler)
SSLContextposixcygwinhostfamilyreturnc $ t | d| S )Nr )get_port_socket)r- r. s f/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/aiohttp/test_utils.pyget_unused_port_socketr3 = s 4F + ++ portc t j |t j }t r+| t j t j d | | |f |S )Nr )socket SOCK_STREAM
REUSE_ADDRESS
setsockopt
SOL_SOCKET SO_REUSEADDRbind)r- r5 r. ss r2 r1 r1 C sZ
ff 011A @
V&(;Q???FFD$< Hr4 c t j t j t j 5 } | d t t
| d cddd S # 1 swxY w Y dS )z1Return a port that is unused on the current host.) 127.0.0.1r r N)r7 AF_INETr8 r= r int getsockname)r> s r2 unused_portrD P s v~v'9 : : -a C +,,- - - - - - - - - - - - - - - - - -s AA99A= A=c e Zd ZdZeddddeddeeef de e
j dede e d e
d
eee ej gej f d ed dfd
Z d de e
j d ed dfdZed ed efd Zded efdZed e
fd Zed e
fd Z ed e fd Z d dZ d dZ de e!e" de e" de e# d dfdZ$d dZ%de e!e" de e" de e# d dfd Z&dS ) BaseTestServerFNr@ )schemeloopr- r5 skip_url_assertssocket_factoryrG rH r- r5 rI rJ kwargsr/ c || _ d | _ d | _ || _ || _ d| _ || _ || _ || _ d S )NF) _looprunner_rootr- r5 _closedrG rI rJ )selfrG rH r- r5 rI rJ rK s r2 __init__zBaseTestServer.__init__Z sK
,0 $(
0 ,r4 c K | j rd S || _ | dd | _ | j dddi| d {V | _ | j d {V | j sd| _ t j | j j
}n# t $ r d}Y nw xY w|dk r t j
n t j }| | j | j | }| d d \ | _ | _ t# | j || j }| d {V |j }|J |j }|J |d d | _ | j t, u r| j rd
} nd } | | _ t/ | j d | j d
| j | _ d S )Nsslhandler_cancellationTr )sock ssl_contextr httpshttpz://: )rN rM pop_ssl _make_runnersetupr5 ipaddress
ip_addressr- version
ValueErrorr7 AF_INET6rA rJ rC r' start_serversocketsrG r r rO )
rQ rH rK re r. _socksiteserverrj rG s
r2 start_serverzBaseTestServer.start_serverq s ; F
JJud++ -D - R R4 R6 R RRRRRRR k !!!!!!!!!y DI *4955=GG GGG $+qLL fn##DIty&AA $ 0 0 2 22A2 6 49 %TYGGGjjll !!!."""AJ**,,Q/ ;( " "y
DKDKCCDICC CCDD
s 6 B B$#B$c
K d S Nr^ rQ rK s r2 ra zBaseTestServer._make_runner s r4 pathc | j J t | }| j s0| rJ | j | S t t | j t | z S rp )rO r rI is_absolutejoinstr)rQ rr urls r2 make_urlzBaseTestServer.make_url sq z%%%$ii$ 4(( ( ((:??3'' 's4:T233 3r4 c | j d uS rp )rN rQ s r2 startedzBaseTestServer.started s {$&&r4 c | j S rp )rP rz s r2 closedzBaseTestServer.closed
|r4 c 8 | j }|J |j J |j S rp )rN rm )rQ rN s r2 handlerzBaseTestServer.handler s- !!!}(((} r4 c K | j rF| j sA| j J | j d{V d| _ d| _ d| _ dS dS 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)r{ r} rN cleanuprO r5 rP rz s r2 closezBaseTestServer.close s| < ;***+%%'' ' ' ' ' ' ' ' DJ DI DLLL r4 c t d NzUse async with instead TypeErrorrz s r2 __enter__zBaseTestServer.__enter__ 0111r4 exc_type exc_value tracebackc d S rp r^ rQ r r r s r2 __exit__zBaseTestServer.__exit__
r4 c L K | | j d {V | S NrH )rn rM rz s r2
__aenter__zBaseTestServer.__aenter__ s6 TZ 000000000 r4 c > K | d {V d S rp r r s r2 __aexit__zBaseTestServer.__aexit__ . jjllr4 rp r/ N)r/ rF )'__name__
__module__ __qualname____test__r r1 r
rv objectr asyncioAbstractEventLooprB boolr r7
AddressFamilyr rR rn r r# ra r r rx propertyr{ r} r% r r r r
BaseExceptionr r r r r^ r4 r2 rF rF W s H
&.48 "!&
- - - c6k"- w01 -
- sm
- - !
#v+ ,fm ;
- -
- - - -0 ;? E E W 67 EJM E
E E E EB
3
:
^
4X 4# 4 4 4 4 ' ' ' ' X' X X &2 2 2 2
4
./
M*
M*
4
./ M* M*
r4 rF c j e Zd Zeddddedeeef dedee de
f
fd Z de
d
e fd Z
xZS )
TestServerr@ NrG r- r5 apprG r- r5 rK c P || _ t j d|||d| d S Nr r^ )r superrR )rQ r rG r- r5 rK __class__s r2 rR zTestServer.__init__ s8 GTGGGGGGGr4 r/ c * K t | j fi |S rp )r" r rq s r2 ra zTestServer._make_runner s ,,V,,,r4 )r r r r r! r
rv r r rB r rR r# ra
__classcell__r s @r2 r r s
&. "
H
H
H
H c6k"
H
H sm
H
H
H
H
H
H
H-3 -: - - - - - - - -r4 r c
t e Zd Zeddddedeeef dedee de
d df fd
Z dd e de
d e
fd
Z xZS )
RawTestServerr@ Nr r rG r- r5 rK r/ c P || _ t j d|||d| d S r )_handlerr rR )rQ r rG r- r5 rK r s r2 rR zRawTestServer.__init__ s8
GTGGGGGGGr4 Tdebugc X K t | j f| j |d|}t |fd|i|S )N)rH r r )r% r rM r& )rQ r rK srvs r2 ra zRawTestServer._make_runner sA T]K5KKFKKC77u7777r4 )T)r r r r r) r
rv r r rB r rR r r& ra r r s @r2 r r s
&. "
H
H
H
H c6k"
H
H sm
H
H
H
H
H
H
H
H8 8 8s 8| 8 8 8 8 8 8 8 8r4 r c e Zd ZdZdZddddedee deej de
d df
d
Z d'd Z e
d efd Ze
d ee fd
Ze
d efd Ze
d ee fd Ze
d efd Zded efdZdedede
d efdZdedede
d e fdZ dede
d e fdZ dede
d e fdZ dede
d e fdZ dede
d e fdZ!dede
d e fdZ"dede
d e fdZ#dede
d e fd Z$dede
d e%fd Z&dede
d e'fd Z(d'd Z)d'd Z*d!ee+e, d"ee, d#ee- d dfd$Z.d(d%Z/d!ee+e, d"ee, d#ee- d dfd&Z0dS ))
TestClientzb
A test client implementation.
To write functional tests for aiohttp based servers.
FN)
cookie_jarrH rm r rH rK r/ c t |t s t dt | z || _ || _ |t
j d| }t d||d|| _ d| _
g | _ g | _ d S )Nz2server must be TestServer instance, found type: %rT)unsaferH )rH r Fr^ )
isinstancerF r typeri rM aiohttp CookieJarr _sessionrP
_responses _websockets)rQ rm r rH rK s r2 rR zTestClient.__init__ s &.11 G$v,,V
*$TBBBJ%Q4JQQ&QQ
02:<r4 c V K | j | j d {V d S r )ri rn rM rz s r2 rn zTestClient.start_server s9 l''TZ'88888888888r4 c | j j S rp )ri r- rz s r2 r- zTestClient.host!
| r4 c | j j S rp )ri r5 rz s r2 r5 zTestClient.port% r r4 c | j S rp )ri rz s r2 rm zTestClient.server) r~ r4 c j t t t t | j dd S )Nr )r r r! getattrri rz s r2 r zTestClient.app- s% H[)74<+M+MNNNr4 c | 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 rz s r2 sessionzTestClient.session1 s } r4 rr c 6 | j | S rp )ri rx )rQ rr s r2 rx zTestClient.make_url< s |$$T***r4 methodc K | j j || | fi | d {V }| j | |S rp )r requestrx r append)rQ r rr rK resps r2 _requestzTestClient._request? s_ +T]*64==3F3FQQ&QQQQQQQQ t$$$ r4 c : t | j ||fi | 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 )rQ r rr rK s r2 r zTestClient.requestG s( &mdmFD&K&KF&K&KLLLr4 c N t | j t j |fi | S )z Perform an HTTP GET request.)r r r METH_GETrQ rr rK s r2 getzTestClient.getS ( %mdmDM4&R&R6&R&RSSSr4 c N t | j t j |fi | S )z Perform an HTTP POST request.)r r r METH_POSTr s r2 postzTestClient.postW ( %mdmDND&S&SF&S&STTTr4 c N t | j t j |fi | S )z Perform an HTTP OPTIONS request.)r r r METH_OPTIONSr s r2 optionszTestClient.options[ s) %mdmD4Et&V&Vv&V&VWWWr4 c N t | j t j |fi | S )z Perform an HTTP HEAD request.)r r r METH_HEADr s r2 headzTestClient.head_ r r4 c N t | j t j |fi | S )z Perform an HTTP PUT request.)r r r METH_PUTr s r2 putzTestClient.putc r r4 c N t | j t j |fi | S z Perform an HTTP PATCH request.)r r r
METH_PATCHr s r2 patchzTestClient.patchg s( %mdmDOT&T&TV&T&TUUUr4 c N t | j t j |fi | S r )r r r METH_DELETEr s r2 deletezTestClient.deletek s) %mdmD4Dd&U&Uf&U&UVVVr4 c 8 t | j |fi | S )zjInitiate websocket connection.
The api corresponds to aiohttp.ClientSession.ws_connect.
)r _ws_connectr s r2
ws_connectzTestClient.ws_connecto s( ((8(8(H(H(H(HIIIr4 c K | j j | | fi | d {V }| j | |S rp )r r rx r r )rQ rr rK wss r2 r zTestClient._ws_connectw s^ ,4=+DMM$,?,?JJ6JJ
J
J
J
J
J
J ### r4 c * K | j s| j D ]}| | j D ] }| d{V | j d{V | j d{V d| _ dS 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)rP r r r r ri )rQ r r s r2 r zTestClient.close~ s |
&
!
!hhjj -%%'' ' ' ' ' ' ' ',$$&& & & & & & & & DLLL r4 c t d r r rz s r2 r zTestClient.__enter__ r r4 r exctbc d S rp r^ rQ r r r s r2 r zTestClient.__exit__ r r4 c > K | d {V | S rp )rn rz s r2 r zTestClient.__aenter__ s/ !!!!!!!!! r4 c > K | d {V d S rp r r s r2 r zTestClient.__aexit__ r r4 r )r/ r )1r r r __doc__r rF r r r r r rR rn r rv r- rB r5 rm r! r r r r r rx r r r r r r r r r r r r r r r r r r r r r r r r^ r4 r2 r r s H 3748 = = = = ./ =
w01 =
=
= = = =*9 9 9 9 !c ! ! ! X! !hsm ! ! ! X! X OXk* O O O XO X +X +# + + + +!)58
M
M!)
M58
M
M
M
M
MT TC T4J T T T TU US U5K U U U UXH X X8N X X X XU US U5K U U U UT TC T4J T T T TV( Vc V6L V V V VW8 Ws W7M W W W WJx J3 J;S J J J J (+ *2 2 2 2
4
./
m
$
] #
4
./ m
$
] #
r4 r c j e Zd ZdZdefdZdefdZddZddZddZ dd Z
d
ede fd Z d e de
fd
ZdS )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.
r/ c . K | S )zGet application.
This method should be overridden
to return the aiohttp.web.Application
object to test.
)get_apprz s r2 get_applicationz AioHTTPTestCase.get_application s ||~~ r4 c t d )zqObsolete method used to constructing web application.
Use .get_application() coroutine instead.
z+Did you forget to define get_application()?) RuntimeErrorrz s r2 r zAioHTTPTestCase.get_app s
HIIIr4 Nc j K t j | _ | d {V S rp )r get_running_looprH
setUpAsyncrz s r2
asyncSetUpzAioHTTPTestCase.asyncSetUp s8 ,.. __&&&&&&&&&r4 c K | d {V | _ | | j d {V | _ | | j d {V | _ | j d {V d S rp )r r
get_serverrm
get_clientclientrn rz s r2 r zAioHTTPTestCase.setUpAsync s --//////// OODH 5 5555555 OODK 8 8888888 k&&(((((((((((r4 c : K | d {V S rp )
tearDownAsyncrz s r2
asyncTearDownz AioHTTPTestCase.asyncTearDown s* '')))))))))r4 c H K | j d {V d S rp )r r rz s r2 r z AioHTTPTestCase.tearDownAsync s2 k !!!!!!!!!!!r4 r c 2 K t || j S )z Return a TestServer instance.r )r rH )rQ r s r2 r zAioHTTPTestCase.get_server s #DI....r4 rm c 2 K t || j S )z Return a TestClient instance.r )r rH )rQ rm s r2 r zAioHTTPTestCase.get_client s &ty1111r4 r )r r r r r! r r r r r r r r r r r^ r4 r2 r r s { J J J J J' ' ' ') ) ) )* * * *" " " "/K /J / / / /2z 2j 2 2 2 2 2 2r4 r funcargsrK c > t j dt d | S )zw
A decorator dedicated to use with asynchronous AioHTTPTestCase test methods.
In 3.8+, this does nothing.
zBDecorator `@unittest_run_loop` is no longer needed in aiohttp 3.8+rX )
stacklevel)warningswarnDeprecationWarning)r r rK s r2 unittest_run_loopr s-
ML
Kr4 F loop_factoryfastc # R K t | }|V t || dS )zzA contextmanager that creates an event_loop, for test purposes.
Handles the creation and cleanup of a test loop.
)r N)setup_test_loopteardown_test_loop)r r rH s r2 loop_contextr s7 < ( (D
JJJt$''''''r4 c B | }t j | |S )zCreate and return an asyncio.BaseEventLoop instance.
The caller should also call teardown_test_loop,
once they are done with the loop.
)r set_event_loop)r rH s r2 r r s% <>>D 4 Kr4 rH c | }|sB| | j | | |st j t j d dS )z>Teardown and cleanup an event_loop created by setup_test_loop.N) is_closed call_soonstop run_foreverr gccollectr r )rH r r} s r2 r r sv
^^
F ty!!!
4 r4 c 4 dt dt dt fd} dt dt dt dd fd}t j t }i |_ | |_ ||_ d|_ t | |_
|j
|S ) Nr keyr/ c | j | S rp
__app_dict)r r( s r2 get_dictz"_create_app_mock..get_dict s ~c""r4 valuec || j |< d S rp r* )r r( r- s r2 set_dictz"_create_app_mock..set_dict" s #sr4 )specF) r rv r MagicMockr! r+ __getitem__ __setitem___debugr on_response_preparefreeze)r, r/ r s r2 _create_app_mockr7 s #c # # # # # #$c $ $C $D $ $ $ $ .k
*
*
*CCN CO COCJ $SkkC""$$$ Jr4
sslcontextc t j }dt dt t f fd }||j _ |S )Nr( r/ c | dk rS d S )Nr8 r^ )r( r8 s r2 get_extra_infoz)_create_transport..get_extra_info3 s , 4r4 )r Mockrv r r* r; side_effect)r8 transportr; s ` r2 _create_transportr? 0 sR IC HZ$8 ,:I ( r4 i .)
match_infore closingr writerprotocolr> payloadr8 client_max_sizerH r rr headersr@ re rA r rB rC r> rD rE c t j }|
du r t j }
d|
j _ |t dd k rd}|rHt t
| }t d | D }n t t
}d}d| t j d v }t | |||||dd |t |
}|t }| t u rt# | } |t u rt j }| |_ |t u rjt j }t' d |_ t' d |_ t' d |_ t' d |_ | |_ | |_ ||_ |
t u rt j }
t3 ||
||||
|
}t5 |t u ri n|t j }| | ||_ |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 3 p K | ]1\ }}| d | d fV 2dS )zutf-8N)encode).0kvs r2 z&make_mocked_request..] sT
7;q!QXXg
1 1 2
r4 chunked NF)rE ) r r<
create_future return_valuer r r tupleitemsr r TRANSFER_ENCODINGlowerr r r7 r r? r> make_mocked_coro
write_headerswrite write_eofdrainrB r$ r( add_app _match_info)r rr rF r@ re rA r rB rC r> rD r8 rE rH taskraw_hdrsrN messagereqs r2 make_mocked_requestra = sD , 9;;D s{{y{{*,'Q"""" ";w#7#788
?F}}
#;==117;;t'=rBBHHJJJG
D G { H %j11 89;; &
/ 5 5 '-- +D11'-- $"H HO()++
(FD$ C %H$$*dikk J s CO Jr4 rQ raise_exceptionc h dt dt dt f fd }t j | S )zCreates a coroutine mock.r rK r/ c ^ K t urt j sS d {V d S rp )r inspect isawaitable)r rK rb rQ s r2 mock_coroz#make_mocked_coro..mock_coro sO ( * *! !"<00 r4 )wraps)r r r< )rQ rb rg s `` r2 rV rV sQ
s c c 99 % % %%r4 )Frp )ar r
contextlibr% re rc osr7 sysr abcr r typesr typingr r r r r
r r r
r unittestr r aiosignalr multidictr r yarlr r aiohttp.clientr r rO r r r
client_reqrepr client_wsr helpersr r\ r r typedefsr webr! r" r# r$ r% r& r' r( web_protocolr) rT r* nameplatformr9 rA rv r r3 rB r1 rD rF r r r r r r
_LOOP_FACTORYcontextmanagernew_event_loopr r r r r1 r7 r< r? ra rV r^ r4 r2 r s
# # # # # # # #
3 2 2 2 2 2 2 2 3 3 3 3 3 3 3 3 K K K K K K K K ! ! ! ! ! ! ! ! " " " " " " ) ) ) ) ) ) . . . . . . 0 0 0 0 0 0 0 0 * ) ) ) ) ) J7"?s|x'?
/5n, ,
, +, ], , , ,
"("6
]
-S - - - -B B B B BS B B BJ- - - - - - - -"8 8 8 8 8N 8 8 8$j j j j j j j jZ62 62 62 62 62- 62 62 62r C s s W 667
")"8u ( ( (@D (
g'( ( ( ( ( #*"8
! !W 6 !d !t ! ! ! ! $. $
(:"6
$)
W
&;q!,, '+" W W W W
W W
W
W W
W
W W W W $W W W
!W W W Wv ! & & &36 & & & & & & &r4