Your IP : 216.73.216.85
��Yf�2 � @ sP 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 e d k rLe j � d S)� N)�support� threading� c C s" | j d k o! | j � | j k S)Nr )�maxsize�qsize)�q� r �4/opt/alt/python35/lib64/python3.5/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 � sQ | j d | � | j j d � | j j � r� | j d | � | 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 | � | j j j � s� | j d � 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 s- | j � r t d � � | 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 � qTW| 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� )r6 )Tr )r6 )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 s[ xT | j � } | d k r) | j � d S| j � | j | 7_ Wd QRX| 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 rJ r r@ r; r r> �sum)r r rE 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)� type2testrH �
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)rP rO rH rQ r )r r r r r �test_queue_join� s
z"BaseQueueTestMixin.test_queue_joinc C s- | j t � } | j | � | j | � d S)N)rP rA rG )r r r r r �test_simple_queue� s
z$BaseQueueTestMixin.test_simple_queuec C sg | j t � } | j t � � | j d d d �Wd QRX| j t � � | j d d d �Wd QRXd S)Nr1 r5 rM rM )rP rA � assertRaisesrQ 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 QRXx t t � D] } | j � qg W| j t j � � | j � Wd QRXd S)Nr1 ) rP rA r@ �
put_nowaitrU rB rC Z
get_nowaitrD )r r rE 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 QRX| j | j � d � d | _ | j t j � � | j d � Wd QRXd S)Nr3 r1 r2 � ) rP r; rU rB rC rW r> r r )r r r r r �test_shrinking_queue� s
z'BaseQueueTestMixin.test_shrinking_queueN)
r r r r+ rG rJ rO rR rS rT rV rX rZ r r r r r( X s 9
r( c @ s e Z d Z e j Z d S)� QueueTestN)r r r rB r. rP r r r r r[ � s r[ c @ s e Z d Z e j Z d S)�
LifoQueueTestN)r r r rB r/ rP r r r r r\ � s r\ c @ s e Z d Z e j Z d S)�PriorityQueueTestN)r r r rB r0 rP 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_getrB r. r )r r
r r r r � s zFailingQueue.__init__c C s1 | j r d | _ t d � � t j j | | � S)NFzYou Lose)r` r^ rB r. �_put)r �itemr r r rb � s zFailingQueue._putc C s. | j r d | _ t d � � t j j | � S)NFzYou Lose)ra r^ rB r. �_get)r r r r rd s zFailingQueue._getN)r r r r rb rd 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 � � 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
r1Yn 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
rgYn 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�������?rF zQueue should be fullr: r zQueue should not be fullzQueue should be empty�firstzQueue should not be emptyr6 )r: )r: Tr )r: )r6 )r r8 r@ rA r; r` r r^ r9 r
r& r= r' ra )r r rE r r r �failing_queue_test s�
z#FailingQueueTest.failing_queue_testc C s* t t � } | j | � | j | � d S)N)r_ rA rg )r r r r r �test_failing_queue] s
z#FailingQueueTest.test_failing_queueN)r r r rg rh r r r r re
s Qre �__main__)rB r ZunittestZtestr �
import_moduler rA r
r r r r( ZTestCaser[ r\ r] � Exceptionr^ r. r_ re r �mainr r r r �<module> s" )� [