Your IP : 216.73.216.213
�
f fX3 � @ s\ d d l Z d d l Z d d l Z d d l m Z e j d � Z d Z d d � Z Gd d � d e j � Z
Gd d
� d
� Z Gd d � d e � Z Gd
d � d e e j
� Z Gd d � d e e j
� Z Gd d � d e e j
� Z Gd d � d e � Z Gd d � d e j � Z Gd d � d e e j
� Z d d � Z e d k rXe � n d S) � N)�support� threading� c C s" | j d k o! | j � | j k S)Nr )�maxsize�qsize)�q� r �4/opt/alt/python34/lib64/python3.4/test/test_queue.py�qfull s r
c @ s( e Z d Z d d � Z d d � Z d S)�_TriggerThreadc C s5 | | _ | | _ t j � | _ t j j | � d S)N)�fn�argsr ZEvent� startedEvent�Thread�__init__)�selfr r
r r r r s z_TriggerThread.__init__c C s. t j d � | j j � | j | j � d S)Ng�������?)�timeZsleepr �setr r
)r r r r �run s
z_TriggerThread.runN)�__name__�
__module__� __qualname__r r r r r r r s r c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�BlockingTestMixinc C s
d | _ d S)N)�t)r r r r �tearDown1 s zBlockingTestMixin.tearDownc C s� t | | � | _ | j j � | | � | _ | j j j � sT | j d | � n | j j d � | j j � r� | j d | � n | j S)Nz,blocking function '%r' appeared not to block�
z,trigger function '%r' appeared to not return) r r �start�resultr �is_set�fail�join�is_alive)r �
block_func�
block_args� trigger_func� trigger_argsr r r �do_blocking_test4 s
z"BlockingTestMixin.do_blocking_testc C s� t | | � | _ | j j � z; y | | � Wn | k
rG � Yn X| j d | � Wd | j j d � | j j � r� | j d | � n | j j j � s� | j d � n Xd S)Nz expected exception of kind %rr z,trigger function '%r' appeared to not returnz(trigger thread ended but event never set)r r r r r r! r r )r r" r# r$ r% Zexpected_exception_classr r r � do_exceptional_blocking_testC s
z.BlockingTestMixin.do_exceptional_blocking_testN)r r r r r& 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 d � Z d S)�BaseQueueTestMixinc C s d | _ t j � | _ d S)Nr )�cumr ZLock�cumlock)r r r r �setUpY s zBaseQueueTestMixin.setUpc % C s0 | j � r t d � � n | j | j � � | j | j � � | j d � | j d � | j d � t d d d d g d d d d g d d d d g � } | j � | j � | j � g } | j | | | j
j d � x; t t
d � D]) } | j | � | j | j � d
� q� W| j t | � d � d t
} d t
} | j | � | j t | � d � | j | j � � | j | j � � y$ | j | d d �| j d � Wn t j k
r�Yn Xy$ | j | d d �| j d � Wn t j k
rYn X| j | j | f | j f � | j | j | d d f | j f � x t t
� D] } | j � qWW| j | j � d � y! | j d d � | j d � Wn t j k
r�Yn Xy! | j d d � | j d � Wn t j k
r�Yn X| j | j f | j d � | j | j d | j d � d S) Nz&Call this function with an empty queue�o iM �� �Queue� LifoQueue�
PriorityQueuez&Didn't seem to queue the correct data!� zQueue should not be emptyzQueue should not be full� � zQueue should be full�blockr z(Didn't appear to block with a full queue�timeoutg{�G�z�?z+Didn't appear to time-out with a full queueTr zQueue should be emptyz*Didn't appear to block with an empty queuez-Didn't appear to time-out with an empty queue�empty� )�empty)Tr )r8 )r � RuntimeError�
assertTruer6 Z assertFalse�full�put�dict�get� assertEqual� __class__r �range�
QUEUE_SIZEr
r �queue�Fullr& �Empty)r r Z target_orderZ actual_order�i�lastr; r r r �simple_queue_test] sd
!
% z$BaseQueueTestMixin.simple_queue_testc
C sZ xS | j � } | d k r) | j � d S| j � | j | 7_ Wd QX| j � q Wd S)Nr )r> � task_doner* r) )r r �xr r r �worker� s
zBaseQueueTestMixin.workerc C s� d | _ x0 d D]( } t j d | j d | f � j � q Wx! t d � D] } | j | � qI W| j � | j | j t t d � � d � x d D] } | j d � q� W| j � d S)
Nr r1 �targetr
�d z0q.join() did not block until all tasks were done)r r1 )r r1 ���)
r) r r rK r rA r< r r? �sum)r r rF r r r �queue_join_test� s
&
z"BaseQueueTestMixin.queue_join_testc C s@ | j � } y | j � Wn t k
r. Yn X| j d � d S)Nz(Did not detect task count going negative)� type2testrI �
ValueErrorr )r r r r r �test_queue_task_done� s
z'BaseQueueTestMixin.test_queue_task_donec C sZ | j � } | j | � | j | � y | j � Wn t k
rH Yn X| j d � d S)Nz(Did not detect task count going negative)rQ rP rI rR r )r r r r r �test_queue_join� s
z"BaseQueueTestMixin.test_queue_joinc C s- | j t � } | j | � | j | � d S)N)rQ rB rH )r r r r r �test_simple_queue� s
z$BaseQueueTestMixin.test_simple_queuec C se | j t � } | j t � � | j d d d �Wd QX| j t � � | j d d d �Wd QXd S)Nr1 r5 rN rN )rQ rB � assertRaisesrR r< r> )r r r r r �&test_negative_timeout_raises_exception� s
z9BaseQueueTestMixin.test_negative_timeout_raises_exceptionc C s� | j t � } x! t t � D] } | j d � q W| j t j � � | j d � Wd QXx t t � D] } | j � qf W| j t j � � | j � Wd QXd S)Nr1 ) rQ rB rA �
put_nowaitrV rC rD Z
get_nowaitrE )r r rF r r r � test_nowait� s z BaseQueueTestMixin.test_nowaitc C s� | j d � } | j d � | j d � | j d � | j t j � � | j d � Wd QX| j | j � d � d | _ | j t j � � | j d � Wd QXd S)Nr3 r1 r2 � ) rQ r< rV rC rD rX r? r r )r r r r r �test_shrinking_queue� s
z'BaseQueueTestMixin.test_shrinking_queueN)
r r r r+ rH rK rP rS rT rU rW rY r[ r r r r r( X s 9
r( c @ s e Z d Z e j Z d S)� QueueTestN)r r r rC r. rQ r r r r r\ � s r\ c @ s e Z d Z e j Z d S)�
LifoQueueTestN)r r r rC r/ rQ r r r r r] � s r] c @ s e Z d Z e j Z d S)�PriorityQueueTestN)r r r rC r0 rQ r r r r r^ � s r^ c @ s e Z d Z d S)�FailingQueueExceptionN)r r r r r r r r_ � s r_ c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)� FailingQueuec G s) d | _ d | _ t j j | | � d S)NF)�
fail_next_put�
fail_next_getrC r. r )r r
r r r r � s zFailingQueue.__init__c C s4 | j r! d | _ t d � � n t j j | | � S)NFzYou Lose)ra r_ rC r. �_put)r �itemr r r rc � s zFailingQueue._putc C s1 | j r! d | _ t d � � n t j j | � S)NFzYou Lose)rb r_ rC r. �_get)r r r r re s zFailingQueue._getN)r r r r rc re r r r r r` � s r` c @ s( e Z d Z d d � Z d d � Z d S)�FailingQueueTestc < C s� | j � r t d � � n x% t t d � D] } | j | � q, Wd | _ y$ | j d d d �| j d � Wn t k
r� Yn Xd | _ y$ | j d d d �| j d � Wn t k
r� Yn X| j d
� | j t | � d � d | _ y- | j
| j d | j f � | j d � Wn t k
r4Yn X| j d
� d | _ y0 | j | j d | j f t � | j d � Wn t k
r�Yn X| j d
� | j t | � d � | j � | j t | � d � | j d
� | j t | � d � | j
| j d | j f � x t t � D] } | j � q W| j | j � d � | j d � d | _
y | j � | j d � Wn t k
r�Yn X| j | j � d � d | _
y! | j d d � | j d � Wn t k
r�Yn X| j | j � d � | j � | j | j � d � d | _
y0 | j | j f | j d t � | j d � Wn t k
rjYn X| j | j � d � | j � | j | j � d � d S)Nz&Call this function with an empty queuer1 TZoopsr4 r z)The queue didn't fail when it should haver5 g�������?rG zQueue should be fullr; r zQueue should not be fullzQueue should be empty�firstzQueue should not be emptyr6 )�full)rh Tr )rh )zempty)r r9 rA rB r< ra r r_ r: r
r& r> r' rb )r r rF r r r �failing_queue_test s�
z#FailingQueueTest.failing_queue_testc C s* t t � } | j | � | j | � d S)N)r` rB ri )r r r r r �test_failing_queue] s
z#FailingQueueTest.test_failing_queueN)r r r ri rj r r r r rf
s Qrf c C s t j t t t t � d S)N)r Z run_unittestr\ r] r^ rf r r r r � test_maine s rk �__main__)rC r ZunittestZtestr �
import_moduler rB r
r r r r( ZTestCaser\ r] r^ � Exceptionr_ r. r` rf rk r r r r r �<module> s$ )� [