File: //usr/local/lib/python3.7/test/__pycache__/test_queue.cpython-37.pyc
B
��gGL � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
y d dl Z W n e k
rx dZ Y nX dZ
dd� ZG dd� dej�ZG dd � d �ZG d
d � d e�ZG d d
� d
eej�ZG dd� deej�ZG dd� deej�ZG dd� de�ZG dd� dej�ZG dd� deej�ZG dd� d�Z G dd� de ej�Z e� e dkd �G d d � d e ej��Z e d k�r�e�!� dS ) � N)�support� c C s | j dko| �� | j kS )Nr )�maxsize�qsize)�q� r �+/usr/local/lib/python3.7/test/test_queue.py�qfull s r c @ s e Zd Zdd� Zdd� ZdS )�_TriggerThreadc C s& || _ || _t�� | _tj�| � d S )N)�fn�args� threadingZEvent� startedEvent�Thread�__init__)�selfr r r r r r s
z_TriggerThread.__init__c C s$ t �d� | j�� | j| j� d S )Ng�������?)�time�sleepr �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 sN t ||�}|�� z(||� | _|j�� s6| �d| � | jS t�|d� X d S )Nz*blocking function %r appeared not to block�
)r
�start�resultr �is_set�failr � join_thread)r �
block_func�
block_args� trigger_func� trigger_args�threadr r r �do_blocking_test< s
z"BlockingTestMixin.do_blocking_testc
C sp t ||�}|�� z6y ||� W n |k
r6 � Y nX | �d| � W d t�|d� |j�� sj| �d� X d S )Nz expected exception of kind %rr z(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_testJ 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| _ t�� | _d S )Nr )�cumr
ZLock�cumlock)r r r r �setUp] s zBaseQueueTestMixin.setUpc C s| |� � rtd��| �|�� � | �|�� � |�d� |�d� |�d� tdddgdddgdddgd�}|�� |�� |�� g}| � |||j
j d� x.t t
d �D ] }|�|� | �|� � d� q�W | �t|� d � d
t
}d t
}|�|� | �t|�d � | �|�� � | �|�� � y |j|d
d� | �d� W n tjk
�rR Y nX y |j|dd� | �d� W n tjk
�r� Y nX | �|j|f|jd� | �|j|ddf|jd� xt t
�D ]}|�� �q�W | �|� � d� y|jd
d� | �d� W n tjk
�r Y nX y|jdd� | �d� W n tjk
�rN Y nX | �|jd|jd� | �|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 queuer Tr zQueue should be emptyz*Didn't appear to block with an empty queuez-Didn't appear to time-out with an empty queue)�empty)Tr )r � RuntimeError�
assertTruer5 � 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 �basic_queue_testa sd
z#BaseQueueTestMixin.basic_queue_testc C sN xH|� � }|dk r |�� d S | j� | j|7 _W d Q R X |�� 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�}|�� |�|� qW xtd�D ]}|�|� qFW |�� | � | j t
td��d� xdD ]}|�d� q~W |�� x|D ] }|�� q�W d S )Nr )r r0 )�targetr �d z0q.join() did not block until all tasks were done���) r( r
r rI r �appendr? r: �joinr= �sum)r r ZthreadsrD r$ r r r �queue_join_test� s
z"BaseQueueTestMixin.queue_join_testc C s8 | � � }y |�� W n tk
r( Y n X | �d� 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 | � � }| �|� | �|� y |�� W n tk
r< Y n X | �d� d S )Nz(Did not detect task count going negative)rQ rP rG rR r )r r r r r �test_queue_join� s
z"BaseQueueTestMixin.test_queue_joinc C s" | � t�}| �|� | �|� d S )N)rQ r@ rF )r r r r r �
test_basic� s
z BaseQueueTestMixin.test_basicc C sV | � t�}| �t�� |jddd� W d Q R X | �t�� |jddd� W d Q R X d S )Nr0 rL )r4 )rQ r@ � assertRaisesrR r: r< )r r r r r �&test_negative_timeout_raises_exception� s
z9BaseQueueTestMixin.test_negative_timeout_raises_exceptionc C s� | � t�}xtt�D ]}|�d� qW | �tj�� |�d� W d Q R X xtt�D ] }|�� qRW | �tj�� |�� W d Q R X d S )Nr0 ) rQ r@ r? �
put_nowaitrV rA rB �
get_nowaitrC )r r rD r r r � test_nowait� s
z BaseQueueTestMixin.test_nowaitc C s� | � d�}|�d� |�d� |�d� | �tj�� |�d� W d Q R X | �|�� d� d|_| �tj�� |�d� W d Q R X d S )N� r0 r1 � ) rQ r: rV rA rB rX r= r r )r r r r r �test_shrinking_queue� s
z'BaseQueueTestMixin.test_shrinking_queueN)
r r r r* rF rI rP rS rT rU rW rZ r] r r r r r' \ s 9
r' c @ s e Zd ZejZdS )� QueueTestN)r r r rA r- rQ r r r r r^ � s r^ c @ s e Zd ZejZdS )�
LifoQueueTestN)r r r rA r. rQ r r r r r_ � s r_ c @ s e Zd ZejZdS )�PriorityQueueTestN)r r r rA r/ rQ r r r r r` � s r` c @ s e Zd ZdS )�FailingQueueExceptionN)r r r r r r r ra s ra 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��tj�| |�S )NFzYou Lose)rc ra rA r- �_put)r �itemr r r re s zFailingQueue._putc C s | j rd| _ td��tj�| �S )NFzYou Lose)rd ra rA r- �_get)r r r r rg
s zFailingQueue._getN)r r r r re rg r r r r rb s rb c @ s e Zd Zdd� Zdd� ZdS )�FailingQueueTestc C s� |� � rtd��x ttd �D ]}|�|� q W d|_y |jddd� | �d� W n tk
rf Y nX d|_y |jddd � | �d� W n tk
r� Y nX |�d
� | �t |�d � d|_y"| �
|jd |j d
� | �d� W n tk
r� Y nX |�d
� d|_y$| � |jd|j d
t� | �d� W n tk
�rB Y nX |�d
� | �t |�d � |� � | �t |� d� |�d
� | �t |�d � | �
|jd |j d
� xtt�D ]}|� � �q�W | �|� � d� |�d� d|_
y|� � | �d� W n tk
�r Y nX | �|� � d� d|_
y|j dd � | �d� W n tk
�rX Y nX | �|� � d� |� � | �|� � d� d|_
y$| � |j d
|jdt� | �d� W n tk
�r� Y nX | �|� � d� |� � | �|� � 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 full)r9 r )r9 Tr zQueue should not be fullzQueue should be empty�firstzQueue should not be empty)r5 )r r6 r? r@ r: rc r ra r7 r r% r< r& rd )r r rD r r r �failing_queue_test s�
z#FailingQueueTest.failing_queue_testc C s t t�}| �|� | �|� d S )N)rb r@ rj )r r r r r �test_failing_queuef s
z#FailingQueueTest.test_failing_queueN)r r r rj rk r r r r rh s Qrh c @ st e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS ) �BaseSimpleQueueTestc C s | � � | _d S )N)rQ r )r r r r r* p s zBaseSimpleQueueTest.setUpc C sT xNy |� � }W n tk
r" d S X |�|� |�� dkrt�|�� d � qW d S )Ng �?g����MbP?)�pop�
IndexErrorr: �randomr r )r r �seq�rnd�valr r r �feeds s
zBaseSimpleQueueTest.feedc C s( x"|� � }||krd S |�|� qW d S )N)r< rM )r r �results�sentinelrr r r r �consume} s
zBaseSimpleQueueTest.consumec C sZ xTx8y|j dd�}W n tjk
r4 t�d� Y qX P qW ||krHd S |�|� qW d S )NF)r3 g�h㈵��>)r< rA rC r r rM )r r rt ru rr r r r �consume_nonblock� s z$BaseSimpleQueueTest.consume_nonblockc C sP xJx.y|j dd�}W n tjk
r* Y qX P qW ||kr>d S |�|� qW d S )Ng�h㈵��>)r4 )r< rA rC rM )r r rt ru rr r r r �consume_timeout� s z#BaseSimpleQueueTest.consume_timeoutc s� g �d �|�g| ��� � t�d��g ��fdd�������fdd�t|�D �}� ����fdd�t|�D �}t�|| �� W d Q R X | ��� | ���� � | � ��
� d� �S )N�* c s �� fdd�}|S )Nc
s@ y�| |� W n, t k
r: } z� �|� W d d }~X Y nX d S )N)�
BaseExceptionrM )r �kwargs�e)�
exceptions�fr r �wrapper� s zHBaseSimpleQueueTest.run_threads.<locals>.log_exceptions.<locals>.wrapperr )r~ r )r} )r~ r �log_exceptions� s z7BaseSimpleQueueTest.run_threads.<locals>.log_exceptionsc s$ g | ] }t j�� ����fd ��qS ))rJ r )r
r )�.0rD )� feed_funcr� r rq rp r r �
<listcomp>� s z3BaseSimpleQueueTest.run_threads.<locals>.<listcomp>c s$ g | ] }t j�� ����fd ��qS ))rJ r )r
r )r� rD )� consume_funcr� r rt ru r r r� � s r ) �reversero ZRandomr? r Z
start_threadsr8 r7 r5 r= r ) r Z n_feedersZ n_consumersr �inputsr� r� ZfeedersZ consumersr ) r� r} r� r� r rt rq ru rp r � run_threads� s"
z BaseSimpleQueueTest.run_threadsc C s� | j }| �|�� � | �|�� d� |�d� | �|�� � | �|�� d� |�d� |�d� |�d� | �|�� � | �|�� d� | �|�� d� | �|�� d� | �|� � d� | �|�� d� | �|jdd�d� | �|�� � | �|�� d� | �|jdd �d� | �|�� � | �|�� d� | �
t j �� |jdd� W d Q R X | �
t j �� |jd
d � W d Q R X | �
t j �� |� � W d Q R X | �|�� � | �|�� d� d S ) Nr r0 r1 r[ r\ F)r3 g�������?)r4 g����MbP?)
r r7 r5 r= r r: r8 rX r< rY rV rA rC )r r r r r rU � s:
z BaseSimpleQueueTest.test_basicc C s6 | j }|�d� | �t�� |jdd� W d Q R X d S )Nr0 rL )r4 )r r: rV rR r< )r r r r r rW � s
z:BaseSimpleQueueTest.test_negative_timeout_raises_exceptionc C s: | j }ttd��}| �dd||| j| j�}| �||� d S )NrK r0 )r �listr? r� rs rv r= )r r r� rt r r r �
test_order� s z BaseSimpleQueueTest.test_orderc C sB d}| j }ttd��}| �||||| j| j�}| �t|�|� d S )N�2 i' )r r� r? r� rs rv r= �sorted)r �Nr r� rt r r r �test_many_threads� s
z%BaseSimpleQueueTest.test_many_threadsc C sB d}| j }ttd��}| �||||| j| j�}| �t|�|� d S )Nr� i' )r r� r? r� rs rw r= r� )r r� r r� rt r r r �test_many_threads_nonblock� s z.BaseSimpleQueueTest.test_many_threads_nonblockc C sB d}| j }ttd��}| �||||| j| j�}| �t|�|� d S )Nr� i� )r r� r? r� rs rx r= r� )r r� r r� rt r r r �test_many_threads_timeout s z-BaseSimpleQueueTest.test_many_threads_timeoutc C sf G dd� d�}d}| j }x t|�D ]}|�|� � q"W x*t|�D ] }t�|�� �}| �|� � q@W d S )Nc @ s e Zd ZdS )z.BaseSimpleQueueTest.test_references.<locals>.CN)r r r r r r r �C s r� � )r r? r: �weakref�refr< Z assertIsNone)r r� r� r rD Zwrr r r �test_references s z#BaseSimpleQueueTest.test_referencesN)r r r r* rs rv rw rx r� rU rW r� r� r� r� r� r r r r rl n s
!%
rl c @ s e Zd ZejZdS )�PySimpleQueueTestN)r r r rA Z_PySimpleQueuerQ r r r r r� ! s r� zNo _queue module foundc s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )�CSimpleQueueTestc s t j| _t� �� d S )N)�_queue� SimpleQueuerQ �superr* )r )r> r r r* ( s zCSimpleQueueTest.setUpc C s | � | jtj� d S )N)ZassertIsrQ rA r� )r r r r �test_is_default, s z CSimpleQueueTest.test_is_defaultc s� | j �t�� � d}g }G � �fdd�dt�}x6|� }��t� �� ~|���� � |d |kr.P q.W | �|t t
|d ��� d S )Ni' c s" e Zd Zdd� Z� �fdd�ZdS )z2CSimpleQueueTest.test_reentrancy.<locals>.Circularc S s
| | _ d S )N)Zcircular)r r r r r ; s z;CSimpleQueueTest.test_reentrancy.<locals>.Circular.__init__c s �� t� �� d S )N)r: �next)r )�genr r r �__del__> s z:CSimpleQueueTest.test_reentrancy.<locals>.Circular.__del__N)r r r r r� r )r� r r r �Circular: s r� rL r0 ) r � itertools�count�objectr: r� rM r< r= r� r? )r r� rt r� �or )r� r r �test_reentrancy/ s z CSimpleQueueTest.test_reentrancy)r r r r* r� r� �
__classcell__r r )r> r r� % s r� �__main__)"� collectionsr� rA ro �sysr
r Zunittestr� Ztestr r� � ImportErrorr@ r r r
r r' ZTestCaser^ r_ r` � Exceptionra r- rb rh rl r� ZskipIfr� r �mainr r r r �<module> s@
" [ 4 &