python (3.12.0)
ΑeF d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z ddl
m Z m
Z
ddl
mZ ddl
mZ ddl
mZ e j" j% d e j" j' d d
Zej* Z eed Z ej0 ed Ze j" j4 Z ej0 ed
Z d Z ej Z e j" j> fdZ e j" jC ejD d Z# G d dejH Z% G d dejH Z& G d de jN Z( G d de jR Z* G d de jV e( Z,er G d d e jZ e( Z. G d d ejH Z/ G d d ejH Z0e1d!k( r ejd yy)"z
Test suite for socketserver.
N)
reap_childrenverbose) os_helper)
socket_helper)threading_helpernetworkT)modules hello world
AF_UNIXzrequires Unix socketszrequires forkingc P t t d rt j | yy)z7Call signal.alarm when it exists (i.e. not on Windows).alarmN)hasattrsignalr )ns J/BuggyBox/python/3.12.0/bootstrap/lib/python3.12/test/test_socketserver.py signal_alarmr # s vw Q c n t | gg g | \ }}}| |v r| j | S t d| )Nz
timed out on ) _real_selectrecv RuntimeError)sockr timeoutrwxs r receiver + s= D62r73GAq! qyyy|677r c # K t j }|dk( rt j d d t j j |d y# xY w# t j j |d w xY ww)zBTests that a custom child process is not waited on (Issue 1540386)r H N)exitcode)osfork_exittestsupport wait_process)testcasepids r simple_subprocessr( 3 si
'')C
ax
4
!!#!3
!!#!3s' /BA "BA A #BBc : e Zd ZdZd Zd Zd Zd Zej d Z
d Z d Z d Z
d
Zed Zed Zed
Zeed Zd Zd Zed Zed Zed Zeed Zej d Zd Z d Z d Z y)SocketServerTestzTest all socket servers.c 6 t d d| _ g | _ y )N< r )r port_seed
test_filesselfs r setUpzSocketServerTest.setUpF s R r c t d t | j D ] } t j | g | j d d y # t
$ r Y 6w xY wNr )r r r. r removeOSError)r0 fns r tearDownzSocketServerTest.tearDownK sU Q// B
"
s A
AAc |t j k( rt dfS t j }| j
j
| |S r3 )socketAF_INETHOSTr create_unix_domain_namer. append)r0 protor6 s r pickaddrzSocketServerTest.pickaddrV s? FNN "!9 668B OO " "2 &Ir c 0 G d d| } G d d| }t r t d ||| }| j j
|j j |S # t $ r$}| j d|d|d| Y d }~^d }~ww xY w) Nc e Zd Zd Zy).SocketServerTest.make_server.<locals>.MyServerc & | j | N)
close_requestr0 requestclient_addresss r handle_errorz;SocketServerTest.make_server.<locals>.MyServer.handle_errorb s ""7+r N)__name__
__module__ __qualname__rI r r MyServerrB a s
r rN c e Zd Zd Zy)/SocketServerTest.make_server.<locals>.MyHandlerc n | j j }| j j | y rD )rfilereadlinewfilewrite)r0 lines r handlez6SocketServerTest.make_server.<locals>.MyHandler.handleg s& zz**,
&r NrJ rK rL rW rM r r MyHandlerrP f s
'r rY zcreating serverzCannot create server (z, z): )r printPermissionErrorskipTest assertEqualserver_addressr9 getsockname)r0 addrsvrclshdlrbaserN rY serveres r make_serverz SocketServerTest.make_server` s v
' '
E+, - dI.F
..
0I0I0KL
- MM!4,
-
- -s A( ( B1BBc ( | j | j |j || }|j }t rt d| t d| t
j d|z |j ddi }d|_ |j t r t d t d D ]' }t r t d
| ||j | ) t r t d |j |j |j | j d |j j# t$ r5t' |t( j* r| j- |j. t r t d
y y )NzADDR =zCLASS =z
%s serving
poll_interval{Gz?nametargetkwargsTzserver running z test clientzwaiting for serverdone)re r? address_familyr^ r rZ threadingThread
serve_foreverdaemonstartrangeshutdownjoin server_closer] r9 fileno HAVE_FORKING
isinstance socketserver ForkingMixIn assertFalseactive_children)r0 ra rb testfuncrc r` tis r
run_serverzSocketServerTest.run_serveru s? !!$--0E0E"F"((4 $$ (D ! )V $ &'' $D)
+ E*+q 2AmQ/ V**D 1 2 E./ V]] 1 1 34 Jv|/H/H I
V 3 3 4 E&M7r c F t j |t j 5 }|j | |j t t |d x}}|r d|vrt |d }||z
}|rd|vr| j
|t d d d y # 1 sw Y y xY wNd
)r9 SOCK_STREAMconnectsendallTEST_STRr r] r0 r> r` sbufdatas r stream_examinez SocketServerTest.stream_examine s
]]5&"4"4
5 ,
IIdO
IIh C (C$5+ q#t 5+
S( + , , ,s AB8BB c t j |t j 5 }t r3|t j k( r |j | j | |j
t | t |d x}}|r d|vrt |d }||z
}|rd|vr| j |t d d d y # 1 sw Y y xY wr )
r9
SOCK_DGRAMHAVE_UNIX_SOCKETSr
bindr? sendtor r r] r s r
dgram_examinez SocketServerTest.dgram_examine s
]]5&"3"3
4 , Ufnn%<t}}U+,
HHXt $ C (C$5+ q#t 5+
S( + , , ,s A;C !C C c v | j t j t j | j y rD )r r} TCPServerStreamRequestHandlerr r/ s r test_TCPServerz SocketServerTest.test_TCPServer s( ..$99 ++ -r c v | j t j t j | j y rD )r r} ThreadingTCPServerr r r/ s r test_ThreadingTCPServerz(SocketServerTest.test_ThreadingTCPServer s( 77$99 ++ -r c t | 5 | j t j t j | j
d d d y # 1 sw Y y xY wrD )r( r r} ForkingTCPServerr r r/ s r test_ForkingTCPServerz&SocketServerTest.test_ForkingTCPServer sE
t
$ 1 OOL 9 9 ( = = / /
1 1 1 1 :AAc v | j t j t j | j y rD )r r} UnixStreamServerr r r/ s r test_UnixStreamServerz&SocketServerTest.test_UnixStreamServer s( 55$99 ++ -r c v | j t j t j | j y rD )r r} ThreadingUnixStreamServerr r r/ s r test_ThreadingUnixStreamServerz/SocketServerTest.test_ThreadingUnixStreamServer s( >>$99 ++ -r c t | 5 | j t j t j | j
d d d y # 1 sw Y y xY wrD )r( r r} ForkingUnixStreamServerr r r/ s r test_ForkingUnixStreamServerz-SocketServerTest.test_ForkingUnixStreamServer sG t
$ 1 OOL @ @ ( = = / /
1 1 1 1r c v | j t j t j | j y rD )r r} UDPServerDatagramRequestHandlerr r/ s r test_UDPServerz SocketServerTest.test_UDPServer s( ..$;; ** ,r c v | j t j t j | j y rD )r r} ThreadingUDPServerr r r/ s r test_ThreadingUDPServerz(SocketServerTest.test_ThreadingUDPServer s( 77$;; ** ,r c t | 5 | j t j t j | j
d d d y # 1 sw Y y xY wrD )r( r r} ForkingUDPServerr r r/ s r test_ForkingUDPServerz&SocketServerTest.test_ForkingUDPServer sE
t
$ 0 OOL 9 9 ( ? ? . .
0 0 0 0r c v | j t j t j | j y rD )r r} UnixDatagramServerr r r/ s r test_UnixDatagramServerz(SocketServerTest.test_UnixDatagramServer s( 77$;; ** ,r c v | j t j t j | j y rD )r r} ThreadingUnixDatagramServerr r r/ s r test_ThreadingUnixDatagramServerz1SocketServerTest.test_ThreadingUnixDatagramServer s( @@$;; ** ,r c v | j t j t j | j y rD )r r} ForkingUnixDatagramServerr r r/ s r test_ForkingUnixDatagramServerz/SocketServerTest.test_ForkingUnixDatagramServer s*
>>$;; ** ,r c G d dt j } G d dt j }g }t d D ]O } |t df| }t j d|j dd i
}d |_ |j ||f Q |D ]% \ }}|j |j ' |D ]% \ }}|j |j ' y ) Nc e Zd Zy)0SocketServerTest.test_shutdown.<locals>.MyServerNrJ rK rL rM r r rN r r rN c e Zd Zy)1SocketServerTest.test_shutdown.<locals>.MyHandlerNr rM r r rY r r r rY r zMyServer servingrg rh ri T)r} r r rv r; rq rr rs rt r= ru rw rx ry )r0 rN rY threadsr r r s r
test_shutdownz SocketServerTest.test_shutdown s |-- 99 r #A$I.A ''-/A AH NNAq6 " # DAq
GGI
JJL DAq
FFH
NN r c G d dt j t j } |t dfd }|j y )Nc e Zd Zy)9SocketServerTest.test_close_immediately.<locals>.MyServerNr rM r r rN r r r rN r c y rD rM rM r r <lambda>z9SocketServerTest.test_close_immediately.<locals>.<lambda> s r )r} ThreadingMixInr r; ry )r0 rN rc s r test_close_immediatelyz'SocketServerTest.test_close_immediately s: |22L4J4J 4)\2 r c t d D ]J }| j t 5 t j t
dft j d d d L y # 1 sw Y WxY w)Ni rn )rv assertRaises
OverflowErrorr} r r; r )r0 r s r test_tcpserver_bind_leakz)SocketServerTest.test_tcpserver_bind_leak sd
t JA""=1
J &&bz'3'H'HJ
J
J J
J
Js +AA# c t j t dft j 5 } d d d | j dj
j
y # 1 sw Y 4xY w)Nr rn )r} r r; r r] r9 rz )r0 rc s r test_context_managerz%SocketServerTest.test_context_manager! sZ
#
#T1I$0$E$EG JP
V]] 1 1 34 s A A)N) rJ rK rL __doc__r1 r7 r? re r reap_threadsr r r r r requires_forkingr requires_unix_socketsr r r r r r r r r r r r r rM r r r* r* C s, "
* "" " # "D, ,-
-
1 1 - -
- -
1 1 ,
,
0 0 , ,
, ,
, ,
"" # 2 J5r r* c T e Zd ZdZd Zd Zd Zd Zd Ze d Z
e d Z d Z y
) ErrorHandlerTestzTest that the servers pass normal exceptions from the handler to
handle_error(), and that exiting exceptions like SystemExit and
KeyboardInterrupt are not passed.c J t j t j y rD )r unlinkTESTFNr/ s r r7 zErrorHandlerTest.tearDown- s ))*r c F t t | j d y NThandled)BaseErrorTestServer
ValueError check_resultr/ s r test_sync_handledz"ErrorHandlerTest.test_sync_handled0 s J' $'r c | j t 5 t t d d d | j d y # 1 sw Y xY wNFr )r
SystemExitr r r/ s r test_sync_not_handledz&ErrorHandlerTest.test_sync_not_handled4 s>
z
* ,
+ , %( , ,s AA
c F t t | j d y r )ThreadingErrorTestServerr r r/ s r test_threading_handledz'ErrorHandlerTest.test_threading_handled9 s , $'r c t j 5 }t t | j d | j |j t d d d y # 1 sw Y y xY wr )r catch_threading_exceptionr r r assertIsexc_type)r0 cms r test_threading_not_handledz+ErrorHandlerTest.test_threading_not_handled= sP
7
7
9 3R $Z 0 e , MM"++z 2 3 3 3s AA A)c F t t | j d y r )ForkingErrorTestServerr r r/ s r test_forking_handledz%ErrorHandlerTest.test_forking_handledD s z* $'r c F t t | j d y r )r r r r/ s r test_forking_not_handledz)ErrorHandlerTest.test_forking_not_handledI s z* %(r c t t j 5 }dd|z z }| j |j | d d d y # 1 sw Y y xY w)NHandler called
Error handled
)openr r r] read)r0 r logexpecteds r r z ErrorHandlerTest.check_resultN sM
)""
# 3s),=,GGH SXXZ 2 3 3 3s )A AN)
rJ rK rL r r7 r r r r r r r r rM r r r r ( sM )+()
(3 ( ( ) )3r r c * e Zd Z fdZd Zd Z xZS )r c > || _ t | t dft t j | j 5 d d d | j | j | j y # 1 sw Y ;xY w# | j w xY wr3 ) exceptionsuper__init__r;
BadHandlerr9 create_connectionr^ handle_requestry wait_done)r0 r __class__s r r z BaseErrorTestServer.__init__U s "
$J/
%
%d&9&9
: !
s A>
B
>B
B c t t j d 5 }|j d d d d y # 1 sw Y y xY w)Nar )r r r rU )r0 rG rH r s r rI z BaseErrorTestServer.handle_error` s6
)""C
( )C II' ( ) ) )s 6?c y rD rM r/ s r r z BaseErrorTestServer.wait_doned s r )rJ rK rL r rI r
__classcell__r s @r r r T s )
r r c e Zd Zd Zy)r c t t j d 5 }|j d d d d | j j d # 1 sw Y $xY w)Nr r z
Test error)r r r rU rc r )r0 r s r rW zBadHandler.handlei sK
)""C
( *C II( ) *kk##L11 * *s AANrX rM r r r r h s 2r r c . e Zd Z fdZ fdZd Z xZS )r c V t j | _ t | |i | y rD )rq Eventro r r r0 poskwr s r r z!ThreadingErrorTestServer.__init__q s# OO%
#$$r c X t | |i | | j j y rD )r shutdown_requestro setr s r r z)ThreadingErrorTestServer.shutdown_requestu s"
#,,
r c 8 | j j y rD )ro waitr/ s r r z"ThreadingErrorTestServer.wait_doney s r )rJ rK rL r r r r
r s @r r r o s %r r c e Zd Zy)r Nr rM r r r r ~ s r r c e Zd Zd Zd Zy)SocketWriterTestc B G d dt j }t j t df| }| j |j
t
j |j t j t j }|5 |j |j d d d |j | j |j t j | j# |j$ |j& y # 1 sw Y jxY w)Nc e Zd Zd Zy)-SocketWriterTest.test_basics.<locals>.Handlerc | j | j _ | j j | j _ | j j | j _ y rD )rT rc rz wfile_filenorG request_filenor/ s r rW z4SocketWriterTest.test_basics.<locals>.Handler.handle sD $(JJ !+/::+<+<+> (-1\\-@-@-B *r NrX rM r r Handlerr! s
Cr r% r )r} r r r;
addCleanupry r9 rp r IPPROTO_TCPr r^ r assertIsInstancerT ioBufferedIOBaser] r# r$ )r0 r% rc r s r test_basicsz SocketWriterTest.test_basics s Cl77 C ''q 7; ++, MM ! !6#5#5v7I7I
K
-
IIf++ , - fllB,=,=> ,,f.C.CD - -s DD c
t j j t d G d dt j
}t j t df| | j j t j fd}t j t j | }| j t j t j | d
d t j
fd}t j | }|j j! |j# | j% j& t)
| j%
d | j% j* d | j% j, t j j. | j% t j j. d
z
y ) N pthread_killc e Zd Zd Zy),SocketWriterTest.test_write.<locals>.Handlerc 8 | j j d | j _ | j j | j _ dt j j z }| j j | | j _
y )N write data
) rT rU rc sent1rR rS receivedr# r$
SOCK_MAX_SIZEsent2)r0 big_chunks r rW z3SocketWriterTest.test_write.<locals>.Handler.handle se $(JJ$4$4_$E !'+zz':':'< $ !DLL$>$> > $(JJ$4$4Y$? !r NrX rM r r r% r/ s
@r r% r c & j y rD )r )signumframe interrupteds r signal_handlerz3SocketWriterTest.test_write.<locals>.signal_handler s
OO r c 0 t j j t j t j } | 5 | j d 5 }| j j |j | j d |j d t j j t d rn4t |j d d d d d d y # 1 sw Y xY w# 1 sw Y y xY w)Nrb client response
r )r )r9 rp r r' makefiler r^ rS r r r SIGUSR1r floatlen)r readerr; main_threadr- received2 response1rc s r
run_clientz/SocketWriterTest.test_write.<locals>.run_client s
f 3 3V5G5G""$A
/AJJt$
/ &//0 "OO - ./ C fnn="''a'9
. #
/
/
/
/
/
/s% D B D /D D D D)rk r1 r? r )r# r$
get_attributer r} r r r; r& ry rq r rB get_identrr ru r rx r] r3 rD r4 r6 r5 ) r0 r% r< originalrI
backgroundr; rF r- rG rH rc s @@@@@@r
test_writezSocketWriterTest.test_write sh ||11&.I @l77 @ ''q 7; ++, oo' == @
v~~x@ ))+ / /, %%Z8
s9~6 O4 *>? t||'A'AB DLL$>$>$DEr N)rJ rK rL r+ rN rM r r r r s E"8Fr r c e Zd Zd Zd Zd Zy) MiscTestCasec g }t t D ]F }|j d rt t | }t |dd dk( s6|j | H | j t j | y )N_rK r} )dirr}
startswithgetattrr= assertCountEqual__all__)r0 r rj
mod_objects r test_allzMiscTestCase.test_all sh % *D??3' $\4 8
:|T:nL OOD) *
l 2 2H=r c G d dt j } |t dft j }t j |j
t j }|j |j |j |j | j |j d |j y )Nc e Zd Zd ZdZd Zy)SMiscTestCase.test_shutdown_request_called_if_verify_request_false.<locals>.MyServerc y)NFrM rF s r verify_requestzbMiscTestCase.test_shutdown_request_called_if_verify_request_false.<locals>.MyServer.verify_request s r r c n | xj dz
c_ t j j | | y )Nr@ )shutdown_calledr} r r )r0 rG s r r zdMiscTestCase.test_shutdown_request_called_if_verify_request_false.<locals>.MyServer.shutdown_request s) $$)$ &&77gFr N)rJ rK rL r^ r` r rM r r rN r\ s
O
Gr rN r r@ )r} r r; r r9 rp r r r^ closer r] r` ry )r0 rN rc r s r 4test_shutdown_request_called_if_verify_request_falsezAMiscTestCase.test_shutdown_request_called_if_verify_request_false s G|-- G 4)\%F%FG MM&//1C1C D &''( //3 r c G d dt j t j } |t dft j }t d D ]: }t
j |j 5 |j ddd < | j t |j d |j y# 1 sw Y |xY w)z
In #37193, users reported a memory leak
due to the saving of every request thread. Ensure that
not all threads are kept forever.
c e Zd Zy)2MiscTestCase.test_threads_reaped.<locals>.MyServerNr rM r r rN re r r rN r
N)r} r r r; r rv r9 r r^ r
assertLessrD _threadsry )r0 rN rc r s r test_threads_reapedz MiscTestCase.test_threads_reaped s |22L4J4J 4)\%F%FGr (A))&*?*?@
(%%'
(
( (
FOO,b1
(
(s 3CC
N)rJ rK rL rY rb ri rM r r rP rP s > * r rP __main__)3r
contextlibr) r selectr r9 rq unittestr} test.supportr# r r r r r r$ requiresrequires_working_socketr r; r
r
skipUnlessr has_fork_supportr{ r r r
SHORT_TIMEOUTr
requires_forkcontextmanagerr( TestCaser* r r r BaseRequestHandlerr r r r~ r r rP rJ mainrM r r <module>ry s
/ " & ) i $ $D $ 1 FI. +++,=,CE ||,, &8&&|5GH }} !\\ 7 7 8 4 4 b5x(( b5J)3x(( )3X
,00
(2 00 2 | : :
!:!:<O
JFx(( JFZ/ 8$$ / d zHMMO r 
Browse
Usage
Versions