Your IP : 216.73.216.213
��YfxE � @ s= d Z d d l Z d d l m Z d d l Z d d l m Z Gd d � d e j � Z Gd d � d e � Z Gd d
� d
e � Z Gd d � d e � Z Gd
d � d e � Z
Gd d � d e � Z Gd d � d � Z Gd d � d e e � Z
Gd d � d e e � Z Gd d � d e e � Z e d k r9e j � d S)zTests for queues.py� N)�mock)�
test_utilsc s" e Z d Z � f d d � Z � S)�_QueueTestBasec s t � j � | j � | _ d S)N)�super�setUp�
new_test_loop�loop)�self)� __class__� �B/opt/alt/python35/lib64/python3.5/test/test_asyncio/test_queues.pyr s
z_QueueTestBase.setUp)�__name__�
__module__� __qualname__r r r )r
r r
s r c @ s| e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d S)�QueueBasicTestsc s$ � f d d � } � j | � � t j d � � } � j � | � j d � � | � � t t | � � � | � k } � j | | � t j � � � f d d � � } � j | � � t j � � � f d d � � } � j | � � t j d � � } | j
d � � j d
� | � k � d S) z�Test Queue's repr or str.
fn is repr or str. expect_id is True if we expect the Queue's id to
appear in fn(Queue()).
c 3 s7 d V} � j d | � d V} � j d | � d Vd S)Ng�������?g�������?)�assertAlmostEqual)�when)r r r �gen s
z.QueueBasicTests._test_repr_or_str.<locals>.genr z<Queuec 3 sm t j d � � } t j | j � d � �t j d d � �Ed H� j d � | � k � | j d � d S)Nr g�������?z _getters[1]r )�asyncio�Queue�Task�get�sleep�
assertTrue�
put_nowait)�q)�fnr r r r �
add_getter' s
z5QueueBasicTests._test_repr_or_str.<locals>.add_getterc 3 s� t j d d d � � } | j d � t j | j d � d � �t j d d � �Ed H� j d � | � k � | j � d S)N�maxsize� r � g�������?z _putters[1])r r r r �putr r �
get_nowait)r )r r r r r �
add_putter4 s
z5QueueBasicTests._test_repr_or_str.<locals>.add_putterr z
_queue=[1]N) r r r r �
startswith�hex�id� assertEqual� coroutine�run_until_completer )r r Z expect_idr r Z
id_is_presentr r# r )r r r r �_test_repr_or_str s % ! !
z!QueueBasicTests._test_repr_or_strc C s` t j � } t j d | � } | j | j | � t j d | j � } | j | j | j � d S)Nr )r ZMockr r �assertIs�_loopr )r r r r r r �test_ctor_loopF s
z QueueBasicTests.test_ctor_loopc C s6 t j | j � t j � } | j | j | j � d S)N)r Zset_event_loopr r r+ r, )r r r r r �test_ctor_noloopN s z QueueBasicTests.test_ctor_noloopc C s | j t d � d S)NT)r* �repr)r r r r � test_reprS s zQueueBasicTests.test_reprc C s | j t d � d S)NF)r* �str)r r r r �test_strV s zQueueBasicTests.test_strc C su t j d | j � } | j | j � � | j d � | j | j � � | j d | j � � | j | j � � d S)Nr r ) r r r r �emptyr � assertFalser' r" )r r r r r �
test_emptyY s
zQueueBasicTests.test_emptyc C sg t j d | j � } | j | j � � t j d d d | j � } | j d � | j | j � � d S)Nr r r )r r r r4 �fullr r )r r r r r � test_fulla s
zQueueBasicTests.test_fullc sx t j d | j � � x$ d d d g D] } � j | � q% W� f d d � t d � D� } | j d d d g | � d S)Nr r � r c s g | ] } � j � � q Sr )r" )�.0�_)r r r �
<listcomp>n s z.QueueBasicTests.test_order.<locals>.<listcomp>)r r r r �ranger' )r �i�itemsr )r r �
test_orderi s
zQueueBasicTests.test_orderc s� � f d d � } � j | � � t j d d d � � � � j d � j � g � t j � � f d d � � � t j � � � � � f d d � � } � j | � � � j d
� j � � d S) Nc 3 s7 d V} � j d | � d V} � j d | � d Vd S)Ng{�G�z�?g{�G�z�?)r )r )r r r r s s
z)QueueBasicTests.test_maxsize.<locals>.genr r r c 3 s: x3 t d � D]% } � j | � Ed H� j | � q
Wd S)Nr8 T)r< r! �append)r= )�
have_been_putr r r �putter� s z,QueueBasicTests.test_maxsize.<locals>.putterc 3 s� t j � � d � �} t j d d � �Ed H� j d d g � � � j d � j � � t j d d � �Ed H� j d d d g � � � j d � j � � � j d � j � � � j | j � � � j | j � � d S)Nr g{�G�z�?r r r )r r r r' r" r �done�result)�t)rA r rB r r r r �test� s z*QueueBasicTests.test_maxsize.<locals>.testg{�G�z�?) r r r r' r r( r) r �time)r r rF r )rA r rB r r r � test_maxsizeq s 'z QueueBasicTests.test_maxsizeN) r
r r r* r- r. r0 r2 r5 r7 r? rH r r r r r s 3 r c @ s| e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d S)�
QueueGetTestsc sf t j d | j � � � j d � t j � f d d � � } | j j | � � } | j d | � d S)Nr r c 3 s � j � Ed HS)N)r r )r r r � queue_get� s z2QueueGetTests.test_blocking_get.<locals>.queue_get)r r r r r( r) r' )r rJ �resr )r r �test_blocking_get� s
z QueueGetTests.test_blocking_getc C s� t j d d | j �} | j d � t j d | j � } | j j | � | j j | j � � } | j d | � | j
| j � � | j | j
� � d S)Nr r )r r r r ZFutureZ_puttersr@ r) r r' r rC Z assertIsNonerD )r r ZwaiterrK r r r �test_get_with_putters� s
z#QueueGetTests.test_get_with_puttersc s� � f d d � } � j | � � t j d � � � t j d � � � d � t j � � � f d d � � � t j � � � � � � f d d � � } � j | � � } � j d | � � j d
� j � � d S) Nc 3 s d V} � j d | � d Vd S)Ng{�G�z�?)r )r )r r r r � s z1QueueGetTests.test_blocking_get_wait.<locals>.genr Fc 3 s% � j � � j � Ed H} d � | S)NT)�setr )rK )�finishedr �startedr r rJ � s
z7QueueGetTests.test_blocking_get_wait.<locals>.queue_getc 3 sf � j d � j d � t j � � d � �} � j � Ed H� j � � | Ed H} � j � � | S)Ng{�G�z�?r r )�
call_laterr r r �waitr4 r )Zqueue_get_taskrK )rO r r rJ r rP r r � queue_put� s
z7QueueGetTests.test_blocking_get_wait.<locals>.queue_putr g{�G�z�?) r r r �Eventr( r) r' r rG )r r rS rK r )rO r r rJ r rP r �test_blocking_get_wait� s !*
z$QueueGetTests.test_blocking_get_waitc C s< t j d | j � } | j d � | j d | j � � d S)Nr r )r r r r r' r" )r r r r r �test_nonblocking_get� s
z"QueueGetTests.test_nonblocking_getc C s/ t j d | j � } | j t j | j � d S)Nr )r r r � assertRaisesZ
QueueEmptyr" )r r r r r � test_nonblocking_get_exception� s z,QueueGetTests.test_nonblocking_get_exceptionc s� � f d d � } � j | � � t j d � � � t j � � f d d � � � t j � � � f d d � � } � j d � j | � � � � j d � j � � d S)
Nc 3 s7 d V} � j d | � d V} � j d | � d Vd S)Ng{�G�z�?g� Zd;�?g�������?)r )r )r r r r � s
z-QueueGetTests.test_get_cancelled.<locals>.genr c 3 s! t j � j � d d � �Ed HS)Ng�&1� �?r )r Zwait_forr r )r r r r rJ � s z3QueueGetTests.test_get_cancelled.<locals>.queue_getc 3 sF t j � � d � �} t j d d � �Ed H� j d � | Ed HS)Nr g{�G�z�?r )r r r r )Zget_task)r r rJ r r rF � s
z.QueueGetTests.test_get_cancelled.<locals>.testr g� ��Q��?)r r r r( r' r) r rG )r r rF r )r r rJ r r �test_get_cancelled� s ! z QueueGetTests.test_get_cancelledc C s� t j d | j � } t j | j � d | j �} t j | j � d | j �} t j | j � | j � t j | j � | j | j � � | j
d � t j | j � | j | j � d � d S)Nr �a)
r r r r r r � run_briefly�cancelr rC r r' rD )r r �t1�t2r r r �test_get_cancelled_race� s
z%QueueGetTests.test_get_cancelled_racec C s� t j d | j d d � } t j | j d � d | j �t j | j d � d | j �t j | j � | j | j j | j � � d � | j | j j | j � � d � d S)Nr r r rZ �b)
r r r r r! r r[ r' r) r )r r r r r � test_get_with_waiting_putters s "z+QueueGetTests.test_get_with_waiting_puttersc C s� t j d d � � } t j d d � � } d } d } t j | d | j �} | j j t j | | | � | | | � d | j �� d S)Nc s s* x# t | � D] } | j � Ed Hq
Wd S)N)r< r )�queueZ num_expectedr: r r r �consumer s z<QueueGetTests.test_why_are_getters_waiting.<locals>.consumerc s s- x& t | � D] } | j | � Ed Hq
Wd S)N)r< r! )rb Z num_itemsr= r r r �producer s z<QueueGetTests.test_why_are_getters_waiting.<locals>.producerr � r )r r( r r r) �gather)r rc rd Z
queue_sizeZproducer_num_itemsr r r r � test_why_are_getters_waiting s z*QueueGetTests.test_why_are_getters_waitingN) r
r r rL rM rU rV rX rY r_ ra rg r r r r rI � s # rI c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d S)�
QueuePutTestsc sG t j d | j � � t j � f d d � � } | j j | � � d S)Nr c 3 s � j d � Ed Hd S)Nr )r! r )r r r rS 0 s z2QueuePutTests.test_blocking_put.<locals>.queue_put)r r r r( r) )r rS r )r r �test_blocking_put- s z QueuePutTests.test_blocking_putc s� � f d d � } � j | � � t j d d d � � � t j d � � � d � t j � � � f d d � � � t j � � � � � � f d d
� � } � j | � � � j d � j � � d S) Nc 3 s d V} � j d | � d Vd S)Ng{�G�z�?)r )r )r r r r 9 s z1QueuePutTests.test_blocking_put_wait.<locals>.genr r r Fc 3 s8 � j � � j d � Ed H� j d � Ed Hd � d S)Nr r T)rN r! r )rO r rP r r rS D s
z7QueuePutTests.test_blocking_put_wait.<locals>.queue_putc 3 sa � j d � j � t j � � d � �} � j � Ed H� j � � | Ed H� j � � d S)Ng{�G�z�?r )rQ r" r r rR r4 r )Zqueue_put_task)rO r r rS r rP r r rJ L s
z7QueuePutTests.test_blocking_put_wait.<locals>.queue_getg{�G�z�?)r r r rT r( r) r rG )r r rJ r )rO r r rS r rP r �test_blocking_put_wait7 s !* z$QueuePutTests.test_blocking_put_waitc C s< t j d | j � } | j d � | j d | j � � d S)Nr r )r r r r r' r" )r r r r r �test_nonblocking_putX s
z"QueuePutTests.test_nonblocking_putc C s� d d � } | j | � } t j d | � } | j | j � � } | j t j d d | �� | j d � | j d � | j � y | j | � Wn7 t j k
r� | j | j � � } | j | � Yn X| j
� } | j d | � d S)Nc s s d Vd Vd S)Ng{�G�z�?g�������?r r r r r r ^ s zBQueuePutTests.test_get_cancel_drop_one_pending_reader.<locals>.genr g{�G�z�?r r ) r r r � create_taskr r) r r r\ �CancelledErrorrD r' )r r r r �readerrD r r r �'test_get_cancel_drop_one_pending_reader] s
z5QueuePutTests.test_get_cancel_drop_one_pending_readerc C s d d � } | j | � } | j d � t j d | � } | j | j � � } | j | j � � } | j | j � � } | j t j d d | �� | j d � | j d � | j � y | j | � Wn t j
k
r� Yn X| j | � | j | j � | j � h d d h � d S)Nc s s d Vd Vd S)Ng{�G�z�?g�������?r r r r r r z s zDQueuePutTests.test_get_cancel_drop_many_pending_readers.<locals>.genTr g{�G�z�?r r )
r Z set_debugr r rl r r) r r r\ rm r' rD )r r r r Zreader1Zreader2Zreader3r r r �)test_get_cancel_drop_many_pending_readersy s"
z7QueuePutTests.test_get_cancel_drop_many_pending_readersc C s d d � } | j | � } t j d d | �} | j d � | j | j d � � } | j t j d d | �� | j � } | j | d � | j
� y | j | � Wn: t j k
r� | j | j d � � } | j | � Yn X| j � } | j | d � | j | j � d � d S)Nc s s d Vd Vd S)Ng{�G�z�?g�������?r r r r r r � s z/QueuePutTests.test_put_cancel_drop.<locals>.genr r r g{�G�z�?r )
r r r r rl r! r) r r" r' r\ rm �qsize)r r r r �writerZvalue1Zvalue2r r r �test_put_cancel_drop� s"
z"QueuePutTests.test_put_cancel_dropc C sE t j d d d | j � } | j d � | j t j | j d � d S)Nr r r r )r r r r rW � QueueFull)r r r r r � test_nonblocking_put_exception� s
z,QueuePutTests.test_nonblocking_put_exceptionc s� t j d d d � j � � � j d � � j d � � j � j � � � j t j � j d � t j d d d � j � � t j � � f d d � � } � j j | � � d S) Nr g�������?r r r r8 c 3 s; � j d � Ed H� j d � Ed H� j � j � � d S)Nr r )r! r r6 r )r r r r rS � s z3QueuePutTests.test_float_maxsize.<locals>.queue_put)
r r r r r r6 rW rt r( r) )r rS r )r r r �test_float_maxsize� s
z QueuePutTests.test_float_maxsizec s� t j d | j � � t j � f d d � � } t j � f d d � � } t j | � d | j �} | j d | j j | � � � | j | j � � | j | j � � d S)Nr c 3 s � j d � Ed Hd S)Nr T)r! r )r r r rS � s z3QueuePutTests.test_put_cancelled.<locals>.queue_putc 3 s � j � Ed HS)N)r r )r r r rF � s z.QueuePutTests.test_put_cancelled.<locals>.testr )
r r r r( r r' r) r rC rD )r rS rF rE r )r r �test_put_cancelled� s z QueuePutTests.test_put_cancelledc C s1 t j d | j d d � } t j | j d � d | j �} t j | j d � d | j �} t j | j d � d | j �} t j | j � | j | j � � | j | j � � | j
� t j | j � | j | j � � | j | j � d � t j | j � | j | j � d � | j j
| � d S)Nr r r rZ r` �X)r r r r r! r r[ r rC r4 r\ r' r" r) )r r Zput_aZput_bZput_cr r r �test_put_cancelled_race� s !!!
z%QueuePutTests.test_put_cancelled_racec C s| t j d | j � } t j | j � d | j �} t j | j � | j j | j d � � | j | j j | � d � d S)Nr rZ )
r r r r r r r[ r) r! r' )r r rE r r r � test_put_with_waiting_getters� s
z+QueuePutTests.test_put_with_waiting_gettersc s� t j d d | j �� t j � f d d � � } t j � f d d � � } | d � } | d � } | d � } | d � } | j j t j | � | | | | d | j �� d S)
Nr r c 3 s � j | � Ed Hd S)N)r! )�item)rb r r rB � s z:QueuePutTests.test_why_are_putters_waiting.<locals>.putterc 3 s8 d V� j � } x t | � D] } � j � } q Wd S)N)rq r< r" )Znumr: r{ )rb r r �getter� s z:QueuePutTests.test_why_are_putters_waiting.<locals>.getterr r r8 )r r r r( r) rf )r rB r| Zt0r] r^ Zt3r )rb r � test_why_are_putters_waiting� s z*QueuePutTests.test_why_are_putters_waitingN)r
r r ri rj rk ro rp rs ru rv rw ry rz r} r r r r rh + s
! rh c @ s e Z d Z d d � Z d S)�LifoQueueTestsc sx t j d | j � � x$ d d d g D] } � j | � q% W� f d d � t d � D� } | j d d d g | � d S)Nr r r8 r c s g | ] } � j � � q Sr )r" )r9 r: )r r r r; s z-LifoQueueTests.test_order.<locals>.<listcomp>)r � LifoQueuer r r< r' )r r= r> r )r r r? s
zLifoQueueTests.test_orderN)r
r r r? r r r r r~
s r~ c @ s e Z d Z d d � Z d S)�PriorityQueueTestsc sx t j d | j � � x$ d d d g D] } � j | � q% W� f d d � t d � D� } | j d d d g | � d S)Nr r r8 r c s g | ] } � j � � q Sr )r" )r9 r: )r r r r; s z1PriorityQueueTests.test_order.<locals>.<listcomp>)r �
PriorityQueuer r r< r' )r r= r> r )r r r? s
z PriorityQueueTests.test_orderN)r
r r r? r r r r r� s r� c @ sF e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d S)
�_QueueJoinTestMixinNc C s, | j d | j � } | j t | j � d S)Nr )�q_classr rW �
ValueError� task_done)r r r r r �test_task_done_underflow' s z,_QueueJoinTestMixin.test_task_done_underflowc s � j d � j � � x! t d � D] } � j | � q" Wd � d � t j � � � f d d � � � t j � � � f d d � � } � j j | � � } � j t t d � � � � d � x' t t | � � D] } � j d � q� W� j j t j
| d � j �� d S)
Nr �d r Tc 3 s6 x/ � r1 � j � Ed H} � | 7� � j � q Wd S)N)r r� )r{ )� accumulatorr �runningr r �worker6 s
z2_QueueJoinTestMixin.test_task_done.<locals>.workerc 3 s5 � � f d d � t d � D� } � j � Ed H| S)Nc s+ g | ]! } t j � � d � j �� q S)r )r r r )r9 �index)r r� r r r; A s zD_QueueJoinTestMixin.test_task_done.<locals>.test.<locals>.<listcomp>r )r< �join)�tasks)r r r� r r rF ? s z0_QueueJoinTestMixin.test_task_done.<locals>.testF) r� r r< r r r( r) r' �sum�lenrR )r r= rF r� r )r� r r� r r� r �test_task_done+ s ! ! z"_QueueJoinTestMixin.test_task_donec sG | j d | j � � t j � f d d � � } | j j | � � d S)Nr c 3 s" � j � Ed H� j � Ed Hd S)N)r� r )r r r r� V s z7_QueueJoinTestMixin.test_join_empty_queue.<locals>.join)r� r r r( r) )r r� r )r r �test_join_empty_queueP s z)_QueueJoinTestMixin.test_join_empty_queuec C sN | j d | j � } | j | j � d � d | _ | j | j � d � d S)Nr z maxsize=0r zmaxsize=0 tasks=2)r� r r' Z_formatZ_unfinished_tasks)r r r r r � test_format] s z _QueueJoinTestMixin.test_format)r
r r r� r� r� r� r� r r r r r� # s
%
r� c @ s e Z d Z e j Z d S)�QueueJoinTestsN)r
r r r r r� r r r r r� e s r� c @ s e Z d Z e j Z d S)�LifoQueueJoinTestsN)r
r r r r r� r r r r r� i s r� c @ s e Z d Z e j Z d S)�PriorityQueueJoinTestsN)r
r r r r� r� r r r r r� m s r� �__main__)�__doc__Zunittestr r r ZTestCaser r rI rh r~ r� r� r� r� r� r
�mainr r r r �<module> s ��� B