python (3.12.0)
Browse
Usage
Versions
ΑeuR d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlmZ d dlm Z e j d ej ddg Z ej ddg
Z
ej e
d Zd Zd
Z G d dej$ Z G d d Z G d de Z G d de Z G d deej. Ze G d deej. Z G d de Z G d d eej. Ze G d d eej. Z G d d!e Z G d" d#e ej. Z e G d$ d%e ej. Z G d& d'e Z! G d( d)e Z" G d* d+e"ej. Z#e G d, d-e"ej. Z$ G d. d/ Z% G d0 d1e%ej. Z&e G d2 d3e%ej. Z'e(d4k( r ejR yy)5 N)
gc_collect)
import_helper)threading_helperT)modulequeue_queue)blocked)freshzNo _queue module found c ^ | j dkD xr | j | j k( S Nr )maxsizeqsize)qs C/BuggyBox/python/3.12.0/bootstrap/lib/python3.12/test/test_queue.pyqfullr s$ 99q= 3QWWY!)) 33 c e Zd Zd Zd Zy)_TriggerThreadc || _ || _ t j | _ t j
j
| y N)fnargs threadingEvent startedEventThread__init__)selfr r s r r z_TriggerThread.__init__ s3 %OO -!!$'r c t j d | j j | j | j
y )N皙?)timesleepr setr r r s r runz_TriggerThread.run! s3
3 r N)__name__
__module__ __qualname__r r& r r r r s ( r r c e Zd Zd Zd Zy)BlockingTestMixinc " t || }|j || | _ |j j s| j d|z | j t
j | S # t
j | w xY w)Nz*blocking function %r appeared not to block)r startresultr is_setfailr join_thread)r
block_func
block_args trigger_func trigger_argsthreads r do_blocking_testz"BlockingTestMixin.do_blocking_test< sv l; 1$j1DK&&--/ F$% &;; ( ( 0 ( ( 0s AA7 7Bc t || }|j || | j d|z t j | |j
j
s| j d y y # |$ r w xY w# t j | |j
j
s| j d w w xY w)Nz expected exception of kind %rz(trigger thread ended but event never set)r r. r1 r r2 r r0 )r r3 r4 r5 r6 expected_exception_classr7 s r do_exceptional_blocking_testz.BlockingTestMixin.do_exceptional_blocking_testJ s l; F
;J' 9!9: ;
( ( 0&&--/ DE 0 ,
( ( 0&&--/ DE 0s A; B ;BB ACN)r' r( r) r8 r; r* r r r, r, : s
1 Fr r, c H e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z y ) BaseQueueTestMixinc D d| _ t j | _ y r
)cumr Lockcumlockr% s r setUpzBaseQueueTestMixin.setUp] s ~~' r c |j r t d | j |j | j |j |j
d |j
d |j
d t g dg dg d }|j |j |j g}| j |||j j d t t d
z
D ]3 }|j
| | j |j d 5 | j t | d d
t z }dt z }|j
| | j t | d | j |j | j |j |j
|d | j d |j
|d | j d | j% |j |f|j d | j% |j |ddf|j d t t D ] }|j | j |j d |j d | j d |j d | j d | j% |j d|j d | j% |j d |j d y # | j j" $ r Y sw xY w# | j j" $ r Y nw xY w# | j j& $ r Y w xY w# | j j& $ r Y w xY w) N&Call this function with an empty queueo M )rE rF rG )rG rF rE )rE rG rF )Queue LifoQueue
PriorityQueuez&Didn't seem to queue the correct data! Queue should not be emptyQueue should not be full Queue should be fullr blockz(Didn't appear to block with a full queueg{Gz?timeoutz+Didn't appear to time-out with a full queuer* T
Queue should be emptyz*Didn't appear to block with an empty queuez-Didn't appear to time-out with an empty queueempty)TrU )r RuntimeError
assertTruerX assertFalsefullputdictget assertEqual __class__r' range
QUEUE_SIZEr r1 r Fullr8 Empty)r r target_order actual_orderilastr\ s r basic_queue_testz#BaseQueueTestMixin.basic_queue_testa s 779 GH H " " c
c
c
O(7,;= !%%'2 |AKK4H4H'IA Cz!|$ DA
EE!H OOAGGI'B C D
E!H &@A:~z! d a"89 # !
EE$aE II@ A
EE$E % IIC D
aeedWaeeR8 aeedD"%5quubAz" A
EEG AGGI
'>?
EEEN IIB C
EE$E IIE F
aeeR
; aeeZ
C5 zz
zz zz
zz sH $M $M6 >#N "#N5 M32M36NNN21N25OOc |j }|dk r|j y | j 5 | xj |z
c_ d d d |j a# 1 sw Y xY wr
)r_ task_donerA r? )r r xs r workerzBaseQueueTestMixin.worker sZ A1u
A
KKM
s A##A,c d| _ g }dD ]E }t j | j |f }|j |j | G t
d D ] }|j | |j | j | j t t
d d dD ] }|j d |j |D ] }|j y )Nr )r rK targetr d z0q.join() did not block until all tasks were done) r? r r rn r. appendrb r] joinr` sum)r r threadsrh r7 s r queue_join_testz"BaseQueueTestMixin.queue_join_test s #A %%T[[tDF LLN NN6 " # s A
EE!H 3uSz?K M A
EE"I F KKM r c | j } |j | j d y # t $ r Y y w xY wNz(Did not detect task count going negative) type2testrl r1
ValueErrorr r s r test_queue_task_donez'BaseQueueTestMixin.test_queue_task_done s@ NN B
KKM
II@ A s 4 A A c | j }| j | | j | |j | j d y # t $ r Y y w xY wrz )r{ rx rl r1 r| r} s r test_queue_joinz"BaseQueueTestMixin.test_queue_join s^
NN Q Q B
KKM
II@ A s A A"!A"c r | j t }| j | | j | y r )r{ rc rj r} s r
test_basicz BaseQueueTestMixin.test_basic s.
NN: & a a r c " | j t }| j t 5 |j dd d d d | j t 5 |j dd d d d y # 1 sw Y ;xY w# 1 sw Y y xY wNrK rs rS )r{ rc assertRaisesr| r] r_ r} s r &test_negative_timeout_raises_exceptionz9BaseQueueTestMixin.test_negative_timeout_raises_exception s{ NN: &
z
* !
EE!RE !
z
* !
EE!RE ! ! ! ! ! !s A9 B9BBc | j t }t t D ] }|j d | j | j
j 5 |j d d d d t t D ] }|j | j | j
j 5 |j d d d y # 1 sw Y lxY w# 1 sw Y y xY w)NrK ) r{ rc rb
put_nowaitr r rd
get_nowaitre r r rh s r test_nowaitz BaseQueueTestMixin.test_nowait s NN: &z" A
LLO
tzz
/
LLO z" A
LLN
tzz / /
0
LLN
s C C) C&)C2c | j d }|j d |j d |j d | j | j j 5 |j d d d d | j
|j d d|_ | j | j j 5 |j d d d d y # 1 sw Y pxY w# 1 sw Y y xY w)N rK rN ) r{ r] r r rd r r` r r r} s r test_shrinking_queuez'BaseQueueTestMixin.test_shrinking_queue s NN1 a a a
tzz
/
LLO A&
tzz
/
LLO s *C+C7+C47D N)
r' r( r) rB rj rn rx r~ r r r r r r* r r r= r= \ s8 (6Dr$B B!!
r r= c e Zd Z fdZ xZS ) QueueTestc X | j j | _ t | y r )r rH r{ superrB r ra s r rB zQueueTest.setUp s ))
r r' r( r) rB
__classcell__ra s @r r r
r r c e Zd ZeZy) PyQueueTestNr' r( r) py_queuer r* r r r r Er r c e Zd ZeZy)
CQueueTestNr' r( r) c_queuer r* r r r r Er r c e Zd Z fdZ xZS )
LifoQueueTestc X | j j | _ t | y r )r rI r{ r rB r s r rB zLifoQueueTest.setUp s --
r r r s @r r r r r r c e Zd ZeZy)PyLifoQueueTestNr r* r r r r
r r r c e Zd ZeZy)CLifoQueueTestNr r* r r r r r r r c e Zd Z fdZ xZS )PriorityQueueTestc X | j j | _ t | y r )r rJ r{ r rB r s r rB zPriorityQueueTest.setUp s 11
r r r s @r r r r r r c e Zd ZeZy)PyPriorityQueueTestNr r* r r r r r r r c e Zd ZeZy)CPriorityQueueTestNr r* r r r r r r r c e Zd Zy)FailingQueueExceptionNr' r( r) r* r r r r $ s r r c * e Zd Z fdZd Zd Z xZS )FailingQueueTestc x | j j G fdd }|| _ t | y )Nc , e Zd Z fdZ fdZ fdZy),FailingQueueTest.setUp.<locals>.FailingQueuec H d| _ d| _ j | g| y )NF)
fail_next_put
fail_next_getr )r r rH s r r z5FailingQueueTest.setUp.<locals>.FailingQueue.__init__. s& %*"%*" t+d+r c d | j rd| _ t d j | | S NFzYou Lose)r r _put)r itemrH s r r z1FailingQueueTest.setUp.<locals>.FailingQueue._put2 s0 %%).D&/
;; zz$--r c b | j rd| _ t d j | S r )r r _get)r rH s r r z1FailingQueueTest.setUp.<locals>.FailingQueue._get7 s. %%).D&/
;; zz$''r N)r' r( r) r r r )rH s r FailingQueuer - s
,
.
(r r )r rH r r rB )r r rH ra s @r rB zFailingQueueTest.setUp) s1
(5 ( )
r c \ |j r t d t t dz
D ] }|j | d|_ |j dd | j
d d|_ |j dd | j
d |j d
| j t | d d|_ | j |j d |j d
| j
d |j d
d|_ | j |j d|j d
t | j
d |j d
| j t | d |j | j t | d |j d
| j t | d | j |j d |j d
t t D ] }|j | j |j d |j d d|_
|j | j
d | j |j d d|_
|j d | j
d | j |j d |j | j |j d d|_
| j |j d
|j dt | j
d | j |j d |j | j |j d y # t $ r Y [w xY w# t $ r Y ?w xY w# t $ r Y w xY w# t $ r Y w xY w# t $ r Y w xY w# t $ r Y Ew xY w# t $ r Y w xY w)NrD rK Toopsr rQ z)The queue didn't fail when it should haver! rS ri rP )r\ r* )r\ TrU rM rV firstrL rW )r rY rb rc r] r r1 r rZ r r8 r_ r; r r s r failing_queue_testz#FailingQueueTest.failing_queue_testA sv 779 GH Hz!|$ A
EE!H
EE&E " IIA B
EE&#E & IIA B
f
a"89 ! !!%%AEE2 > IIA B
f
- -aee5GPR.C
E IIA B
f
a"89 E!H &@A f
a"89 aeeYr:z" A
EEG AGGI
'>? g
EEG IIA B
#>?
EE#E IIA B
#>? AGGI
'>? - -aeeR
.C
E IIA B
#>? AGGI
'>?K % % % % * % % % s~ $N? 8$O 9O ">O/ *!O? 3#P />P ? O O O O O,+O,/ O<;O<? P P P P P+*P+c r | j t }| j | | j | y r )r rc r r} s r test_failing_queuez#FailingQueueTest.test_failing_queue s0
j ) " "r )r' r( r) rB r r r r s @r r r ' s 0O@b#r r c e Zd ZeZy)PyFailingQueueTestNr r* r r r r r r r c e Zd ZeZy)CFailingQueueTestNr r* r r r r r r r c Z e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zy)BaseSimpleQueueTestc . | j | _ y r )r{ r r% s r rB zBaseSimpleQueueTest.setUp s !r c |j }|j | |j dkD r&t j
|j dz \# t $ r |j | Y y w xY w)Ng ?MbP?)pop
IndexErrorr] randomr" r# )r r seqrndsentinelvals r feedzBaseSimpleQueueTest.feed sk
ggi
EE#Jzz|c!
3::<$./
h
s A A;:A;c T |j }||k( ry |j | (r )r_ rt r r resultsr r s r consumezBaseSimpleQueueTest.consume s+ %%'Ch NN3 r c |j d } ||k( ry |j | -# | j j $ r t j d Y nw xY wa)NFrQ h㈵>)r_ r re r" r# rt r s r consume_nonblockz$BaseSimpleQueueTest.consume_nonblock si %%e%,C h NN3 zz'' %JJt$% s / .A A c |j d } ||k( ry |j | -# | j j $ r Y nw xY wL)Nr rS )r_ r re rt r s r consume_timeoutz#BaseSimpleQueueTest.consume_timeout s` %%%-C h NN3 zz'' s / A
A c
r g }d }|j }|j t j d } g fd}
t | D cg c]# } t j |
| ||| |f % } } t | D cg c]" } t j |
| |||f $ }
} t j | |
z 5 d d d | j | j |j | j |j d |S c c} w c c} w # 1 sw Y exY w)N* c fd}|S )Nc f | i | y # t $ r}j | Y d }~y d }~ww xY wr )
BaseExceptionrt )r kwargse
exceptionsfs r wrapperzHBaseSimpleQueueTest.run_threads.<locals>.log_exceptions.<locals>.wrapper s6 )t&v&$ ) %%a(()s 0+0r* )r r r s ` r log_exceptionsz7BaseSimpleQueueTest.run_threads.<locals>.log_exceptions s
)
Nr rp r )copyreverser Randomrb r r r
start_threadsr[ rZ rX r` r )r n_threadsr inputs feed_func consume_funcr r r r r rh feeders consumersr s @r run_threadsz BaseSimpleQueueTest.run_threads s1 kkm
mmB
") ,. ##>)+D*+S#x)@B . .
$I .0 %%^L-I,-w+AC 0 0
+
+Gi,?
@
$ " A& .0 s (D#'D(D--D6c | j }| j |j | j |j d |j d | j
|j | j |j d |j d |j d |j d | j
|j | j |j d | j |j d | j |j d | j |j d | j |j d | j |j d d | j
|j | j |j d | j |j d d | j |j | j |j d | j | j j 5 |j d d d d | j | j j 5 |j d
d d d | j | j j 5 |j d d d | j |j | j |j d y # 1 sw Y xY w# 1 sw Y xY w# 1 sw Y axY w) Nr rK rN r r FrQ r! rS r )
r rZ rX r` r r] r[ r r_ r r r re r} s r r z BaseSimpleQueueTest.test_basic sg FF " A& a # A& a Q a # A& !$ A& + A& U+Q/ # A& s+Q/ " A&
tzz / /
0
EEE
tzz / /
0
EE$E
tzz / /
0
LLN " A& s$ 5M5M5M&MM#&M/c | j }|j d | j t 5 |j d d d d y # 1 sw Y y xY wr )r r] r r| r_ r} s r r z:BaseSimpleQueueTest.test_negative_timeout_raises_exception sG FF a
z
*
EE"E s AAc | j }t t d }| j d||| j | j
}| j
|| y )Nrr rK )r listrb r r r r` )r r r r s r
test_orderz BaseSimpleQueueTest.test_order! sK FFeCj!""1aDLLI
&)r c d}| j }t t d }| j |||| j | j
}| j
t | | y N2 ' )r r rb r r r r` sortedr Nr r r s r test_many_threadsz%BaseSimpleQueueTest.test_many_threads* sT FFeEl#""1aDLLI
&1r c d}| j }t t d }| j |||| j | j
}| j
t | | y r )r r rb r r r r` r r s r test_many_threads_nonblockz.BaseSimpleQueueTest.test_many_threads_nonblock5 sZ FFeEl#""1a#'99d.C.CE
&1r c d}| j }t t d }| j |||| j | j
}| j
t | | y )Nr i )r r rb r r r r` r r s r test_many_threads_timeoutz-BaseSimpleQueueTest.test_many_threads_timeout? sZ FFeDk"""1a#'99d.B.BD
&1r c & G d d }d}| j }t | D ] }|j | t | D ]E }t j |j }t
| j | G y )Nc e Zd Zy).BaseSimpleQueueTest.test_references.<locals>.CNr r* r r Cr L s r r
)r rb r] weakrefrefr_ r assertIsNone)r r
r r rh wrs r test_referencesz#BaseSimpleQueueTest.test_referencesI sx
FFq A
EE!#J q $AQUUW%B L bd # $r N)r' r( r) rB r r r r r r r r r r r r r* r r r r sE " 0 @#'J * 222
$r r c " e Zd ZeZ fdZ xZS )PySimpleQueueTestc X | j j | _ t | y r )r _PySimpleQueuer{ r rB r s r rB zPySimpleQueueTest.setUp\ s 22
r )r' r( r) r r rB r r s @r r r Y s E r r c . e Zd ZeZ fdZd Zd Z xZS )CSimpleQueueTestc X | j j | _ t | y r )r SimpleQueuer{ r rB r s r rB zCSimpleQueueTest.setUpf s //
r c | j | j | j j | j | j | j j y r )assertIsr{ r r r% s r test_is_defaultz CSimpleQueueTest.test_is_defaultj s:
dnndjj&<&<=
dnndjj&<&<=r c Z | j t j d}g } G fddt } | }j t ~|j
j |d |k\ rnK| j |t t |dz y )Nr c e Zd Zd Z fdZy)2CSimpleQueueTest.test_reentrancy.<locals>.Circularc | | _ y r )circularr% s r r z;CSimpleQueueTest.test_reentrancy.<locals>.Circular.__init__z s $
r c : j t y r )r] next)r genr s r __del__z:CSimpleQueueTest.test_reentrancy.<locals>.Circular.__del__} s d3i r N)r' r( r) r r' )r& r s r Circularr! y s
%
!r r( rs rK ) r itertoolscountobjectr] r% rt r_ r` r rb )r r r r( or& r s @@r test_reentrancyz CSimpleQueueTest.test_reentrancyn s
FFoo
!v !
A
EE$s) NN1557 #r{a
$uQU|"45r ) r' r( r) r r rB r r- r r s @r r r a s
E>6r r __main__)*r) r r r" unittestr test.supportr r r requires_working_threadingimport_fresh_moduler r
skipUnless need_c_queuerc r r r r, r= r TestCaser r r r r r r r Exceptionr r r r r r r r' mainr* r r <module>r8 s2
# & ) , + +4 8 ,= , ,Wxj I
+-
+
+GH:
F"x""7,DE
4 Y%% @ F FDV * V p" )X . . H - - & mX%6%6 ]H$5$5 * +X->-> *H,=,=
-I ,q#( q#j)8+<+< ((*;*; q$ q$h+X->-> &6*H,=,= &6 &6R zHMMO r