Your IP : 216.73.216.52
3
� fw3 � @ s� d dl Z d dlZd dlZd dlmZ ejd�ZdZdd� ZG dd� dej �Z
G d d
� d
�Z G d d � d e �Z G d
d� de ej
�ZG dd� de ej
�ZG dd� de ej
�ZG dd� de�ZG dd� de j�ZG dd� de ej
�Zedkr�ej� dS )� N)�support� threading� c C s | j dko| j� | j kS )Nr )�maxsize�qsize)�q� r �4/opt/alt/python36/lib64/python3.6/test/test_queue.py�qfull s r
c @ s e Zd Zdd� Zdd� ZdS )�_TriggerThreadc C s& || _ || _tj� | _tjj| � d S )N)�fn�argsr ZEvent� startedEvent�Thread�__init__)�selfr r
r r r r s
z_TriggerThread.__init__c C s$ t jd� | jj� | 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 @ s e Zd Zdd� Zdd� ZdS )�BlockingTestMixinc C sb t ||�}|j� z(||� | _|jj� s6| jd| � | jS |jd� |j� r\| jd| � X d S )Nz,blocking function '%r' appeared not to block�
z,trigger function '%r' appeared to not return)r �start�resultr �is_set�fail�join�is_alive)r �
block_func�
block_args� trigger_func� trigger_args�threadr r r �do_blocking_test1 s
z"BlockingTestMixin.do_blocking_testc C s� t ||�}|j� z6y ||� W n |k
r6 � Y nX | jd| � W d |jd� |j� rj| jd| � |jj� s~| jd� X d 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# Zexpected_exception_classr$ r r r � do_exceptional_blocking_testB s
z.BlockingTestMixin.do_exceptional_blocking_testN)r r r r% r& r r r r r / s r c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�BaseQueueTestMixinc C s d| _ tj� | _d S )Nr )�cumr ZLock�cumlock)r r r r �setUpX s zBaseQueueTestMixin.setUpc % C s| |j � rtd��| j|j� � | j|j� � |jd� |jd� |jd� tdddgdddgdddgd�}|j� |j� |j� g}| j |||j
j d� x.t t
d �D ] }|j|� | j|j � d� q�W | jt|� d � d
t
}dt
}|j|� | jt|�d � | j|j� � | j|j� � y |j|d
d� | jd� W n tjk
�rR Y nX y |j|dd� | jd� W n tjk
�r� Y nX | j|j|f|jf � | j|j|ddf|jf � xt t
�D ]}|j� �q�W | j|j � d� y|jd
d� | jd� W n tjk
�r Y nX y|jdd� | jd� W n tjk
�rN Y nX | j|jf |jd� | j|jd|jd � 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 fullr )�blockz(Didn't appear to block with a full queueg{�G�z�?)�timeoutz+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� )r5 )Tr )r5 )r � RuntimeError�
assertTruer5 Z assertFalse�full�put�dict�get� assertEqual� __class__r �range�
QUEUE_SIZEr
r �queue�Fullr% �Empty)r r Z target_orderZ actual_order�i�lastr9 r r r �simple_queue_test\ sd
z$BaseQueueTestMixin.simple_queue_testc
C sN xH|j � }|dk r |j� d S | j� | j|7 _W d Q R X |j� qW d S )Nr )r< � task_doner) r( )r r �xr r r �worker� s zBaseQueueTestMixin.workerc C s� d| _ g }x0dD ](}tj| j|fd�}|j� |j|� qW xtd�D ]}|j|� qFW |j� | j | j t
td��d� xdD ]}|jd� q~W |j� x|D ] }|j� q�W d S ) Nr r0 )�targetr
�d z0q.join() did not block until all tasks were done)r r0 )r r0 ���) r( r r rI r �appendr? r: r r= �sum)r r ZthreadsrD r$ r r r �queue_join_test� s
z"BaseQueueTestMixin.queue_join_testc C s8 | j � }y |j� W n tk
r( Y n X | jd� d S )Nz(Did not detect task count going negative)� type2testrG �
ValueErrorr )r r r r r �test_queue_task_done� s z'BaseQueueTestMixin.test_queue_task_donec C sL | j � }| j|� | j|� y |j� W n tk
r< Y n X | jd� d S )Nz(Did not detect task count going negative)rP rO rG 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 r@ rF )r r r r r �test_simple_queue� s
z$BaseQueueTestMixin.test_simple_queuec C sV | j t�}| jt�� |jddd� W d Q R X | jt�� |jddd� W d Q R X d S )Nr0 )r4 rL rL )rP r@ � assertRaisesrQ r: r< )r r r r r �&test_negative_timeout_raises_exception� s
z9BaseQueueTestMixin.test_negative_timeout_raises_exceptionc C s� | j t�}xtt�D ]}|jd� qW | jtj�� |jd� W d Q R X xtt�D ] }|j� qRW | jtj�� |j� W d Q R X d S )Nr0 ) rP r@ r? �
put_nowaitrU rA rB Z
get_nowaitrC )r r rD r r r � test_nowait� s
z BaseQueueTestMixin.test_nowaitc C s� | j d�}|jd� |jd� |jd� | jtj�� |jd� W d Q R X | j|j� d� d|_| jtj�� |jd� W d Q R X d S )Nr2 r0 r1 � ) rP r: rU rA rB rW r= r r )r r r r r �test_shrinking_queue� s
z'BaseQueueTestMixin.test_shrinking_queueN)
r r r r* rF rI rO rR rS rT rV rX rZ r r r r r' W s 9
r' c @ s e Zd ZejZdS )� QueueTestN)r r r rA r- rP r r r r r[ � s r[ c @ s e Zd ZejZdS )�
LifoQueueTestN)r r r rA r. rP r r r r r\ � s r\ c @ s e Zd ZejZdS )�PriorityQueueTestN)r r r rA r/ rP r r r r r] � s r] c @ s e Zd ZdS )�FailingQueueExceptionN)r r r r r r r r^ � s r^ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� FailingQueuec G s" d| _ d| _tjj| f|�� d S )NF)�
fail_next_put�
fail_next_getrA r- r )r r
r r r r � s zFailingQueue.__init__c C s" | j rd| _ td��tjj| |�S )NFzYou Lose)r` r^ rA r- �_put)r �itemr r r rb s zFailingQueue._putc C s | j rd| _ td��tjj| �S )NFzYou Lose)ra r^ rA 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 Zd Zdd� Zdd� ZdS )�FailingQueueTestc < C s� |j � rtd��x ttd �D ]}|j|� q W d|_y |jddd� | jd� W n tk
rf Y nX d|_y |jddd � | jd� W n tk
r� Y nX |jd
� | jt |�d � d|_y"| j
|jd|j f � | jd� W n tk
r� Y nX |jd
� d|_y$| j |jd|j f t� | jd� W n tk
�rB Y nX |jd
� | jt |�d � |j � | jt |� d� |jd
� | jt |�d � | j
|jd|j f � xtt�D ]}|j � �q�W | j|j � d� |jd� d|_
y|j � | jd� W n tk
�r Y nX | j|j � d� d|_
y|j dd � | jd� W n tk
�rX Y nX | j|j � d� |j � | j|j � d� d|_
y$| j |j f |jdt� | jd� W n tk
�r� Y nX | j|j � d� |j � | j|j � d� d S )Nz&Call this function with an empty queuer0 TZoopsr )r3 z)The queue didn't fail when it should haveg�������?)r4 rE zQueue should be fullr9 r zQueue should not be fullzQueue should be empty�firstzQueue should not be emptyr5 )r9 )r9 Tr )r9 )r5 )r r7 r? r@ r: r` r r^ r8 r
r% r< r& ra )r r rD r r r �failing_queue_test s�
z#FailingQueueTest.failing_queue_testc C s t t�}| j|� | j|� d S )N)r_ r@ rg )r r r r r �test_failing_queuea s
z#FailingQueueTest.test_failing_queueN)r r r rg rh r r r r re s Qre �__main__)rA r ZunittestZtestr �
import_moduler r@ r
r r r r' ZTestCaser[ r\ r] � Exceptionr^ r- r_ re r �mainr r r r �<module> s$
( [