Your IP : 216.73.216.213
�
f f`E � @ 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 r<e j � n d S)zTests for queues.py� N)�mock)�
test_utilsc @ s e Z d Z d d � Z d S)�_QueueTestBasec C s | j � | _ d S)N)�
new_test_loop�loop)�self� r �B/opt/alt/python34/lib64/python3.4/test/test_asyncio/test_queues.py�setUp s z_QueueTestBase.setUpN)�__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 � �Dd 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 � �Dd 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_putter3 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_loopE 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_noloopM s z QueueBasicTests.test_ctor_noloopc C s | j t d � d S)NT)r( �repr)r r r r � test_reprR s zQueueBasicTests.test_reprc C s | j t d � d S)NF)r( �str)r r r r �test_strU 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_emptyX 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 r2 �fullr r )r r r r r � test_full` 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>m s z.QueueBasicTests.test_order.<locals>.<listcomp>)r r r r �ranger% )r �i�itemsr )r r �
test_orderh 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 r s
z)QueueBasicTests.test_maxsize.<locals>.genr r r c 3 s: x3 t d � D]% } � j | � Dd H� j | � q
Wd S)Nr6 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 � �Dd H� j d d g � � � j d � j � � t j d d � �Dd 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)r? r r@ 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 rD r )r? r r@ r r r � test_maxsizep s 'z QueueBasicTests.test_maxsizeN) r r r
r( r+ r, r. r0 r3 r5 r= rF 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 � Dd 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 rH �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 rA Z assertIsNonerB )r r ZwaiterrI 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 � Dd H} d � | S)NT)�setr )rI )�finishedr �startedr r rH � s
z7QueueGetTests.test_blocking_get_wait.<locals>.queue_getc 3 sf � j d � j d � t j � � d � �} � j � Dd H� j � � | Dd H} � j � � | S)Ng{�G�z�?r r )�
call_laterr r r �waitr2 r )Zqueue_get_taskrI )rM r r rH r rN r r � queue_put� s
z7QueueGetTests.test_blocking_get_wait.<locals>.queue_putr g{�G�z�?) r r r �Eventr& r' r% r rE )r r rQ rI r )rM r r rH r rN 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 � �Dd HS)Ng�&1� �?r )r Zwait_forr r )r r r r rH � s z3QueueGetTests.test_get_cancelled.<locals>.queue_getc 3 sF t j � � d � �} t j d d � �Dd H� j d � | Dd HS)Nr g{�G�z�?r )r r r r )Zget_task)r r rH r r rD � s
z.QueueGetTests.test_get_cancelled.<locals>.testr g� ��Q��?)r r r r& r% r' r rE )r r rD r )r r rH 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 rA r r% rB )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 rX �b)
r r r r r r rY 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 � Dd Hq
Wd S)N)r: r )�queueZ num_expectedr8 r r r �consumer s z<QueueGetTests.test_why_are_getters_waiting.<locals>.consumerc s s- x& t | � D] } | j | � Dd Hq
Wd S)N)r: r )r` 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 ra rb Z
queue_sizeZproducer_num_itemsr r r r � test_why_are_getters_waiting s z*QueueGetTests.test_why_are_getters_waitingN) r r r
rJ rK rS rT rV rW r] r_ re r r r r rG � s # rG 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 � Dd Hd S)Nr )r r )r r r rQ / s z2QueuePutTests.test_blocking_put.<locals>.queue_put)r r r r& r' )r rQ 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 8 s z1QueuePutTests.test_blocking_put_wait.<locals>.genr r r Fc 3 s8 � j � � j d � Dd H� j d � Dd Hd � d S)Nr r T)rL r r )rM r rN r r rQ C s
z7QueuePutTests.test_blocking_put_wait.<locals>.queue_putc 3 sa � j d � j � t j � � d � �} � j � Dd H� j � � | Dd H� j � � d S)Ng{�G�z�?r )rO r r r rP r2 r )Zqueue_put_task)rM r r rQ r rN r r rH K s
z7QueuePutTests.test_blocking_put_wait.<locals>.queue_getg{�G�z�?)r r r rR r& r' r rE )r r rH r )rM r r rQ r rN r �test_blocking_put_wait6 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_putW 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 rZ �CancelledErrorrB r% )r r r r �readerrB 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 y s zDQueuePutTests.test_get_cancel_drop_many_pending_readers.<locals>.genTr g{�G�z�?r r )
r Z set_debugr r rj r r' r r rZ rk r% rB )r r r r Zreader1Zreader2Zreader3r r r �)test_get_cancel_drop_many_pending_readersx 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 rj r r' r r r% rZ rk �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 rU � 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 r6 c 3 s; � j d � Dd H� j d � Dd H� j � j � � d S)Nr r )r r r4 r )r r r r rQ � s z3QueuePutTests.test_float_maxsize.<locals>.queue_put)
r r r r r r4 rU rr r& r' )r rQ 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 � Dd Hd S)Nr T)r r )r r r rQ � s z3QueuePutTests.test_put_cancelled.<locals>.queue_putc 3 s � j � Dd HS)N)r r )r r r rD � s z.QueuePutTests.test_put_cancelled.<locals>.testr )
r r r r& r r% r' r rA rB )r rQ rD rC 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 rX r^ �X)r r r r r r rY r rA r2 rZ 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 rX )
r r r r r r rY r' r r% )r r rC 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 | � Dd Hd S)N)r )�item)r` r r r@ � s z:QueuePutTests.test_why_are_putters_waiting.<locals>.putterc 3 s8 d V� j � } x t | � D] } � j � } q Wd S)N)ro r: r )Znumr8 ry )r` r r �getter� s z:QueuePutTests.test_why_are_putters_waiting.<locals>.getterr r r6 )r r r r& r' rd )r r@ rz Zt0r[ r\ Zt3r )r` r � test_why_are_putters_waiting� s z*QueuePutTests.test_why_are_putters_waitingN)r r r
rg rh ri rm rn rq rs rt ru rw rx r{ r r r r rf * s
! rf 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 r6 r c s g | ] } � j � � q Sr )r )r7 r8 )r r r r9 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 r6 r c s g | ] } � j � � q Sr )r )r7 r8 )r r r r9 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 rU �
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 � Dd H} � | 7� � j � q Wd S)N)r r� )ry )� accumulatorr �runningr r �worker5 s
z2_QueueJoinTestMixin.test_task_done.<locals>.workerc 3 s5 � � f d d � t d � D� } � j � Dd H| S)Nc s+ g | ]! } t j � � d � j �� q S)r )r r r )r7 �index)r r� r r r9 @ s zD_QueueJoinTestMixin.test_task_done.<locals>.test.<locals>.<listcomp>r )r: �join)�tasks)r r r� r r rD > s z0_QueueJoinTestMixin.test_task_done.<locals>.testF) r� r r: r r r& r' r% �sum�lenrP )r r; rD 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 � Dd H� j � Dd Hd S)N)r� r )r r r r� U s z7_QueueJoinTestMixin.test_join_empty_queue.<locals>.join)r� r r r& r' )r r� r )r r �test_join_empty_queueO 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� d 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� h 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� l s r� �__main__)�__doc__Zunittestr r r ZTestCaser r rG rf r| r~ r� r� r� r� r �mainr r r r �<module> s ��� B