7F ; r d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z m Z ddl
mZ ddlmZmZmZmZmZmZmZmZmZ ddlmZ ddlm 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.m/Z/m0Z0 d dl1m2Z2m3Z3 d dl4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ> erddl?m@Z@ ndZ@e/rddlmAZB nddlCmBZB ejD dk o
ejE dk ZFejG fdeHdejI dej fd ZJdeHd eKdejI dej fd ZLdeKfd ZM G d d!e ZN G d" d#eN ZO G d$ d%eN ZP G d& d' ZQ G d( d)eB ZRd*ed+ed,edefd-ZSeg ejT f ZUejV ejW d.fd/eUd0eXdeejT fd1 ZYejW fd/eUdejT fd2ZZdKd3ejT d0eXddfd4Z[dej\ fd5Z]dLd6ee@ dej^ fd7Z_ dLe0 e2d d d.de0e0e0e0dd8d9d: d;eHded?e2d@eXdAedBedCedDedEed6ee@ dFeKd3ede8f dGZ`e0e0fdHedIedefdJZadS )MzUtilities shared by tests. N)ABCabstractmethod)
TracebackType)
TYPE_CHECKINGAnyCallableIteratorListOptionalTypeUnioncast)mock)Signal) CIMultiDictCIMultiDictProxy)URL)_RequestContextManager_WSRequestContextManager )
ClientSessionhdrs)AbstractCookieJar)ClientResponse)ClientWebSocketResponse)PY_38sentinel) HttpVersionRawRequestMessage) Application AppRunner
BaseRunnerRequestServer ServerRunnerSockSiteUrlMappingMatchInfo)_RequestHandler)
SSLContext)IsolatedAsyncioTestCase)TestCaseposixcygwinhostfamilyreturnc $ t | d| S )Nr )get_port_socket)r. r/ s t/builddir/build/BUILD/imunify360-venv-2.6.2/opt/imunify360/venv/lib64/python3.11/site-packages/aiohttp/test_utils.pyget_unused_port_socketr4 A 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. r6 r/ ss r3 r2 r2 G sZ
ff 011A @
V&(;Q???FFD$< Hr5 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)r8 AF_INETr9 r> r int getsockname)r? s r3 unused_portrE T 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 Zed 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 ) BaseTestServerFNrA )schemeloopr. r6 skip_url_assertssocket_factoryrH rI r. r6 rJ rK kwargsr0 c || _ d | _ d | _ || _ || _ d| _ || _ || _ || _ d S )NF) _looprunner_rootr. r6 _closedrH rJ rK )selfrH rI r. r6 rJ rK rL s r3 __init__zBaseTestServer.__init__^ sK
,0 $(
0 ,r5 c K | j rd S || _ | dd | _ | j d i | 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 )
Nsslr )sock ssl_contextr httpshttpz://: )rO rN pop_ssl _make_runnersetupr6 ipaddress
ip_addressr. version
ValueErrorr8 AF_INET6rB rK rD r& start_serversocketsrH r r rP )
rR rI rL re r/ _socksiteserverrj rH s
r3 start_serverzBaseTestServer.start_serveru s ; F
JJud++ -D - 7 7 7 7777777 k !!!!!!!!!y DI *4955=GG GGG $+qLL fn##DIty&AA $ 0 0 2 22A2 6 49 %TYGGGjjll !!!."""AJ**,,Q/ ;( " "y
DKDKCCDICC CCDD
s 4 B B"!B"c
K d S Nr^ rR rL s r3 ra zBaseTestServer._make_runner s r5 pathc | j J t | }| j s0| rJ | j | S t t | j |z S rp )rP r rJ is_absolutejoinstr)rR rr urls r3 make_urlzBaseTestServer.make_url sk z%%%$ii$ /(( ( ((:??3'' 's4:-.. .r5 c | j d uS rp )rO rR s r3 startedzBaseTestServer.started s {$&&r5 c | j S rp )rQ rz s r3 closedzBaseTestServer.closed
|r5 c 8 | j }|J |j J |j S rp )rO rm )rR rO s r3 handlerzBaseTestServer.handler s- !!!}(((} r5 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} rO cleanuprP r6 rQ rz s r3 closezBaseTestServer.close s| < ;***+%%'' ' ' ' ' ' ' ' DJ DI DLLL r5 c t d NzUse async with instead TypeErrorrz s r3 __enter__zBaseTestServer.__enter__ 0111r5 exc_type exc_value tracebackc d S rp r^ rR r r r s r3 __exit__zBaseTestServer.__exit__
r5 c L K | | j d {V | S NrI )rn rN rz s r3
__aenter__zBaseTestServer.__aenter__ s6 TZ 000000000 r5 c > K | d {V d S rp r r s r3 __aexit__zBaseTestServer.__aexit__ . jjllr5 rp r0 N)r0 rG )&__name__
__module__ __qualname____test__r r2 r
rv objectr asyncioAbstractEventLooprC boolr r8
AddressFamilyr rS rn r r" ra r rx propertyr{ r} r$ r r r r
BaseExceptionr r r r r^ r5 r3 rG rG [ s H
&.48 "!&
- - - c6k"- w01 -
- sm
- - !
#v+ ,fm ;
- -
- - - -0 ;? E E W 67 EJM E
E E E EB
3
:
^
/S /S / / / / ' ' ' ' X' X X &2 2 2 2
4
./
M*
M*
4
./ M* M*
r5 rG c j e Zd Zeddddedeeef dedee de
f
fd Z de
d
e fd Z
xZS )
TestServerrA NrH r. r6 apprH r. r6 rL c P || _ t j d|||d| d S Nr r^ )r superrS )rR r rH r. r6 rL __class__s r3 rS zTestServer.__init__ s8 GTGGGGGGGr5 r0 c * K t | j fi |S rp )r! r rq s r3 ra zTestServer._make_runner s ,,V,,,r5 )r r r r r r
rv r r rC r rS r" ra
__classcell__r s @r3 r r s
&. "
H
H
H
H c6k"
H
H sm
H
H
H
H
H
H
H-3 -: - - - - - - - -r5 r c
t e Zd Zeddddedeeef dedee de
d df fd
Z dd e de
d e
fd
Z xZS )
RawTestServerrA Nr r rH r. r6 rL r0 c P || _ t j d|||d| d S r )_handlerr rS )rR r rH r. r6 rL r s r3 rS zRawTestServer.__init__ s8
GTGGGGGGGr5 Tdebugc X K t | j f| j |d|}t |fd|i|S )N)rI r r )r$ r rN r% )rR r rL srvs r3 ra zRawTestServer._make_runner sA T]K5KKFKKC77u7777r5 )T)r r r r r( r
rv r r rC r rS r r% ra r r s @r3 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 8r5 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 efdZ dede
d efdZ dede
d efdZ dede
d efdZ dede
d efdZ dede
d efdZ!dede
d efdZ"dede
d efd 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&Z/dS ))
TestClientzb
A test client implementation.
To write functional tests for aiohttp based servers.
FN)
cookie_jarrI rm r rI rL r0 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)unsaferI )rI r Fr^ )
isinstancerG r typeri rN aiohttp CookieJarr _sessionrQ
_responses _websockets)rR rm r rI rL s r3 rS zTestClient.__init__
s &.11 G$v,,V
*$TBBBJ%Q4JQQ&QQ
02:<r5 c V K | j | j d {V d S r )ri rn rN rz s r3 rn zTestClient.start_server" s9 l''TZ'88888888888r5 c | j j S rp )ri r. rz s r3 r. zTestClient.host%
| r5 c | j j S rp )ri r6 rz s r3 r6 zTestClient.port) r r5 c | j S rp )ri rz s r3 rm zTestClient.server- r~ r5 c j t t t t | j dd S )Nr )r r r getattrri rz s r3 r zTestClient.app1 s% H[)74<+M+MNNNr5 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 r3 sessionzTestClient.session5 s } r5 rr c 6 | j | S rp )ri rx )rR rr s r3 rx zTestClient.make_url@ s |$$T***r5 methodc K | j j || | fi | d {V }| j | |S rp )r requestrx r append)rR r rr rL resps r3 _requestzTestClient._requestC s] *T]*64==3F3FQQ&QQQQQQQQ t$$$ r5 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 )rR r rr rL s r3 r zTestClient.requestI s( &mdmFD&K&KF&K&KLLLr5 c N t | j t j |fi | S )z Perform an HTTP GET request.)r r r METH_GETrR rr rL s r3 getzTestClient.getS ( %mdmDM4&R&R6&R&RSSSr5 c N t | j t j |fi | S )z Perform an HTTP POST request.)r r r METH_POSTr s r3 postzTestClient.postW ( %mdmDND&S&SF&S&STTTr5 c N t | j t j |fi | S )z Perform an HTTP OPTIONS request.)r r r METH_OPTIONSr s r3 optionszTestClient.options[ s) %mdmD4Et&V&Vv&V&VWWWr5 c N t | j t j |fi | S )z Perform an HTTP HEAD request.)r r r METH_HEADr s r3 headzTestClient.head_ r r5 c N t | j t j |fi | S )z Perform an HTTP PUT request.)r r r METH_PUTr s r3 putzTestClient.putc r r5 c N t | j t j |fi | S z Perform an HTTP PATCH request.)r r r
METH_PATCHr s r3 patchzTestClient.patchg s( %mdmDOT&T&TV&T&TUUUr5 c N t | j t j |fi | S r )r r r METH_DELETEr s r3 deletezTestClient.deletek s) %mdmD4Dd&U&Uf&U&UVVVr5 c 8 t | j |fi | S )zjInitiate websocket connection.
The api corresponds to aiohttp.ClientSession.ws_connect.
)r _ws_connectr s r3
ws_connectzTestClient.ws_connecto s( ((8(8(H(H(H(HIIIr5 c K | j j | | fi | d {V }| j | |S rp )r r rx r r )rR rr rL wss r3 r zTestClient._ws_connectw s\ +4=+DMM$,?,?JJ6JJ
J
J
J
J
J
J ### r5 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)rQ r r r r ri )rR r r s r3 r zTestClient.close| s |
&
!
!hhjj -%%'' ' ' ' ' ' ' ',$$&& & & & & & & & DLLL r5 c t d r r rz s r3 r zTestClient.__enter__ r r5 r exctbc d S rp r^ rR r r r s r3 r zTestClient.__exit__ r r5 c > K | d {V | S rp )rn rz s r3 r zTestClient.__aenter__ s/ !!!!!!!!! r5 c > K | d {V d S rp r r s r3 r zTestClient.__aexit__ r r5 r )r0 r )0r r r __doc__r rG r r r r r rS rn r rv r. rC r6 rm 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^ r5 r3 r r s H 3748 = = = = ./ =
w01 =
=
= = = =*9 9 9 9 !c ! ! ! X! !hsm ! ! ! X! X OXk* O O O XO X +S +S + + + +S s ~ Mc M M M@V M M M MT Ts T/E T T T TU U U0F U U U UXC X3 X3I X X X XU U U0F U U U UT Ts T/E T T T TV# V V1G V V V VW3 W# W2H W W W WJs Jc J6N J J J Jc S =T
*2 2 2 2
4
./
m
$
] #
4
./ m
$
] #
r5 r c z e Zd ZdZdefdZdefdZddZddZddZ dd Z
dd
Z dd Z d ede
fd
Zde
defdZdS )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.
r0 c . K | S )zGet application.
This method should be overridden
to return the aiohttp.web.Application
object to test.
)get_apprz s r3 get_applicationz AioHTTPTestCase.get_application s ||~~ r5 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 r3 r zAioHTTPTestCase.get_app s
HIIIr5 Nc t s:t j | d S d S rp )r r get_event_looprun_until_complete
asyncSetUprz s r3 setUpzAioHTTPTestCase.setUp sD K " $ $ 7 78I8I J J J J J K Kr5 c K t j | _ nA# t t f$ r- t j | _ Y nw xY w| d {V S rp )r get_running_looprI AttributeErrorr get_event_loop_policyr
setUpAsyncrz s r3 r zAioHTTPTestCase.asyncSetUp s I 022DII - I I I 577FFHHDIII I __&&&&&&&&&s ;AAc K | d {V | _ | | j d {V | _ | | j d {V | _ | j d {V d S rp )r r
get_serverrm
get_clientclientrn rz s r3 r zAioHTTPTestCase.setUpAsync s --//////// OODH 5 5555555 OODK 8 8888888 k&&(((((((((((r5 c p t s.| j | d S d S rp )r rI r
asyncTearDownrz s r3 tearDownzAioHTTPTestCase.tearDown s< ? I ( (););)=)= > > > > > ? ?r5 c : K | d {V S rp )
tearDownAsyncrz s r3 r z AioHTTPTestCase.asyncTearDown s* '')))))))))r5 c H K | j d {V d S rp )r r rz s r3 r z AioHTTPTestCase.tearDownAsync s2 k !!!!!!!!!!!r5 r c 2 K t || j S )z Return a TestServer instance.r )r rI )rR r s r3 r
zAioHTTPTestCase.get_server s #DI....r5 rm c 2 K t || j S )z Return a TestClient instance.r )r rI )rR rm s r3 r zAioHTTPTestCase.get_client s &ty1111r5 r )r r r r r r r r r r r r r r r
r r r^ r5 r3 r r s { J J J J JK K K K' ' ' ') ) ) )? ? ? ?* * * *" " " "/K /J / / / /2z 2j 2 2 2 2 2 2r5 r funcargsrL 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 rL s r3 unittest_run_loopr s-
ML
Kr5 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! rI s r3 loop_contextr% s7 < ( (D
JJJt$''''''r5 c | } |j j }d|v }n# t $ r d}Y nw xY wt j | t
j dk r|st j } t j }n## t $ r t j }Y nw xY w|
| t j t 5 | | ddd n # 1 swxY w Y |S )zCreate and return an asyncio.BaseEventLoop instance.
The caller should also call teardown_test_loop,
once they are done with the loop.
uvloopTwin32N)r r r r set_event_loopsysplatformr
ThreadedChildWatcherSafeChildWatcher attach_loop
contextlibsuppressNotImplementedErrorset_child_watcher)r rI module skip_watcherpolicywatchers r3 r# r#
sY <>>D * 6) 4
|w | .00 1 244GG 1 1 1 .00GGG 1D!!!
!4
5
5 . . $ $W - - - . . . . . . . . . . . . . . . Ks0 ,,)A= = B B C00C47C4rI 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) )rI r! r} s r3 r$ r$ / sv
^^
F ty!!!
4 r5 c 4 dt dt dt fd} dt dt dt dd fd}t j t }i |_ | |_ ||_ d|_ t | |_
|j
|S ) Nr keyr0 c | j | S rp
__app_dict)r r? s r3 get_dictz"_create_app_mock..get_dict> s ~c""r5 valuec || j |< d S rp rA )r r? rD s r3 set_dictz"_create_app_mock..set_dictA s #sr5 )specF) r rv r MagicMockr rB __getitem__ __setitem___debugr on_response_preparefreeze)rC rF r s r3 _create_app_mockrN = s #c # # # # # #$c $ $C $D $ $ $ $ .k
*
*
*CCN CO COCJ $SkkC""$$$ Jr5
sslcontextc t j }dt dt t f fd }||j _ |S )Nr? r0 c | dk rS d S )NrO r^ )r? rO s r3 get_extra_infoz)_create_transport..get_extra_infoR s , 4r5 )r Mockrv r r) rR side_effect)rO transportrR s ` r3 _create_transportrV O sR IC HZ$8 ,:I ( r5 i .)
match_infore closingr writerprotocolrU payloadrO client_max_sizerI r rr headersrW re rX r rY rZ rU r[ r\ 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 r3 z&make_mocked_request..| sT
7;q!QXXg
1 1 2
r5 chunked NF)r\ ) r rS
create_future return_valuer r r tupleitemsr r TRANSFER_ENCODINGlowerr r rN r rV rU make_mocked_coro
write_headerswrite write_eofdrainrY r# r' add_app _match_info)r rr r] rW re rX r rY rZ rU r[ rO r\ rI taskraw_hdrsre messagereqs r3 make_mocked_requestrx \ 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 Jr5 rh raise_exceptionc h dt dt dt f fd }t j | S )zCreates a coroutine mock.r rL r0 c ^ K t urt j sS d {V d S rp )r inspect isawaitable)r rL ry rh s r3 mock_coroz#make_mocked_coro..mock_coro sO ( * *! !"<00 r5 )wraps)r r rS )rh ry r~ s `` r3 rm rm sQ
s c c 99 % % %%r5 )Frp )br r r/ r< r| rc osr8 r* r abcr r typesr typingr r r r r
r r r
r unittestr aiosignalr multidictr r yarlr r aiohttp.clientr r rf r r r
client_reqrepr client_wsr helpersr r r\ r r webr r! r" r# r$ r% r& r' web_protocolr( rU r) r* r+ asynctestnamer+ r: rB rv r r4 rC r2 rE rG r r r r r r
_LOOP_FACTORYcontextmanagernew_event_loopr r% r# r$ rH rN rS rV rx rm r^ r5 r3 r s
# # # # # # # #
3 3 3 3 3 3 3 3 K K K K K K K K ! ! ! ! ! ! ! ! " " " " " " ) ) ) ) ) ) . . . . . . $ $ $ $ $ $ $ $ 0 0 0 0 0 0 0 0 * ) ) ) ) ) J #<<<<<<<""""""7"?s|x'?
/5n, ,
, +, ], , , ,
"("6
]
-S - - - -B B B B BS B B BJ- - - - - - - -"8 8 8 8 8N 8 8 8$d d d d d d d dNB2 B2 B2 B2 B2h B2 B2 B2J C s s W 667
")"8u ( ( (@D (
g'( ( ( ( ( #*"8 D ! !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 & & & & & & &r5