Your IP : 216.73.216.213
3
��2a�G � @ s� d dl Z d dl mZ d dlZd dlmZ G dd� dej�ZG dd� de�ZG dd � d e�ZG d
d � d e�ZG d d
� d
e�Z G dd� de�Z
G dd� d�Z G dd� de e�Z G dd� de e�Z
G dd� de e�Zedkr�e j� dS )� N)�mock)�
test_utilsc s e Zd Z� fdd�Z� ZS )�_QueueTestBasec s t � j� | j� | _d S )N)�super�setUp�
new_test_loop�loop)�self)� __class__� �0/opt/alt/python36/lib64/python3.6/test_queues.pyr s
z_QueueTestBase.setUp)�__name__�
__module__� __qualname__r �
__classcell__r r )r
r r
s r c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�QueueBasicTestsc s� �fdd�}�j |��tj�d�}�j� |�jd�� |�� tt|��� |�k}�j||� tj� ��fdd��}�j |� � tj� ��fdd��}�j |� � tj�d�}|j
d � �jd
� |�k� d S ) Nc 3 s. d V } � j d| � dV } � j d| � dV d S )Ng�������?g�������?)�assertAlmostEqual)�when)r r r �gen s
z.QueueBasicTests._test_repr_or_str.<locals>.gen)r z<Queuec 3 sR t j�d�} t j| j� �d� t jd�d�E d H �jd� | �k� | jd� d S )N)r 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 jd�d�} | jd� t j| jd��d� t jd�d�E d H �jd� | �k� | j� d S )N� )�maxsizer � )r g�������?z _putters[1])r r r r �putr r �
get_nowait)r )r r r r r �
add_putter4 s
z5QueueBasicTests._test_repr_or_str.<locals>.add_putterr z
_queue=[1]) 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 sD t j� }tj|d�}| j|j|� tj| jd�}| j|j| j� d S )N)r )r ZMockr r �assertIs�_loopr )r r r r r r �test_ctor_loopF s
z QueueBasicTests.test_ctor_loopc C s( 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_noloopN s z QueueBasicTests.test_ctor_noloopc C s | j td� d S )NT)r+ �repr)r r r r � test_reprS s zQueueBasicTests.test_reprc C s | j td� d S )NF)r+ �str)r r r r �test_strV s zQueueBasicTests.test_strc C sV t j| jd�}| j|j� � |jd� | j|j� � | jd|j� � | j|j� � d S )N)r r ) r r r r �emptyr � assertFalser( r# )r r r r r �
test_emptyY s
zQueueBasicTests.test_emptyc C sH t j| jd�}| j|j� � t jd| jd�}|jd� | j|j� � d S )N)r r )r r )r r r r5 �fullr r )r r r r r � test_fulla s
zQueueBasicTests.test_fullc sR t j| jd�� xdD ]}� j|� qW � fdd�td�D �}| jdddg|� d S )N)r r � r! c s g | ] }� j � �qS r )r# )�.0�_)r r r �
<listcomp>n s z.QueueBasicTests.test_order.<locals>.<listcomp>)r r9 r! )r r r r �ranger( )r �i�itemsr )r r �
test_orderi s
zQueueBasicTests.test_orderc s� �fdd�}�j |��tjd�d���jd�j� g � tj� �fdd���tj� ����fdd��}�j|� � �jd �j� � d S )
Nc 3 s. d V } � j d| � dV } � j d| � dV d S )Ng{�G�z�?g{�G�z�?)r )r )r r r r s s
z)QueueBasicTests.test_maxsize.<locals>.genr! )r r c 3 s0 x*t d�D ] } �j| �E d H � j| � q
W dS )Nr9 T)r= r" �append)r> )�
have_been_putr r r �putter� s z,QueueBasicTests.test_maxsize.<locals>.putterc 3 s� t j�� �d�} t jd�d�E d H �jddg� � �jd�j� � t jd�d�E d H �jdddg� � �jd�j� � �jd�j� � �j| j� � �j| j� � d S )N)r g{�G�z�?r r r! )r r r r( r# r �done�result)�t)rB r rC 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 rG r )rB r rC r r r � test_maxsizeq s
z QueueBasicTests.test_maxsizeN) r
r r r+ r. r/ r1 r3 r6 r8 r@ rI r r r r r s 3r c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�
QueueGetTestsc sH t j| jd�� � jd� t j� fdd��}| jj|� �}| jd|� d S )N)r r c 3 s � j � E d H S )N)r r )r r r � queue_get� s z2QueueGetTests.test_blocking_get.<locals>.queue_get)r r r r r) r* r( )r rK �resr )r r �test_blocking_get� s
z QueueGetTests.test_blocking_getc C sp t jd| jd�}|jd� t j| jd�}|jj|� | jj|j� �}| j d|� | j
|j � � | j |j
� � d S )Nr )r )r r r r ZFutureZ_puttersrA r* r r( r rD Z assertIsNonerE )r r ZwaiterrL r r r �test_get_with_putters� s
z#QueueGetTests.test_get_with_puttersc s� �fdd�}�j |��tj�d��tj�d��d� tj� ��fdd���tj� �����fdd��}�j|� �}�jd |� �jd
�j� � d S ) Nc 3 s d V } � j d| � dV d S )Ng{�G�z�?)r )r )r r r r � s z1QueueGetTests.test_blocking_get_wait.<locals>.gen)r Fc 3 s �j � �j� E d H } d� | S )NT)�setr )rL )�finishedr �startedr r rK � s z7QueueGetTests.test_blocking_get_wait.<locals>.queue_getc 3 sP �j d�jd� tj�� �d�} �j� E d H �j� � | E d H }�j� � |S )Ng{�G�z�?r )r )�
call_laterr r r �waitr5 r )Zqueue_get_taskrL )rP r r rK r rQ r r � queue_put� s
z7QueueGetTests.test_blocking_get_wait.<locals>.queue_putr g{�G�z�?) r r r �Eventr) r* r( r rH )r r rT rL r )rP r r rK r rQ r �test_blocking_get_wait� s
z$QueueGetTests.test_blocking_get_waitc C s, t j| jd�}|jd� | jd|j� � d S )N)r 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| jd�}| jt j|j� d S )N)r )r r r � assertRaisesZ
QueueEmptyr# )r r r r r � test_nonblocking_get_exception� s z,QueueGetTests.test_nonblocking_get_exceptionc st �fdd�}�j |�� tj� d��tj� �fdd���tj� ��fdd��}�jd� j|� �� �jd � j� � d S )
Nc 3 s. d V } � j d| � dV } � j d| � dV d S )Ng{�G�z�?g� Zd;�?g�������?)r )r )r r r r � s
z-QueueGetTests.test_get_cancelled.<locals>.gen)r c 3 s t j�j� d� d�E d H S )Ng�&1� �?)r )r �wait_forr r )r r r r rK � s z3QueueGetTests.test_get_cancelled.<locals>.queue_getc 3 s8 t j�� � d�} t jd� d�E d H �jd� | E d H S )N)r g{�G�z�?r )r r r r )Zget_task)r r rK r r rG � s
z.QueueGetTests.test_get_cancelled.<locals>.testr g� ��Q��?)r r r r) r( r* r rH )r r rG r )r r rK r r �test_get_cancelled� s
z QueueGetTests.test_get_cancelledc C s� t j| jd�}t j|j� | jd�}t j|j� | jd�}tj| j� |j� tj| j� | j|j � � |j
d� tj| j� | j |j � d� d S )N)r �a)
r r r r r r � run_briefly�cancelr rD r r( rE )r r �t1�t2r r r �test_get_cancelled_race� s
z%QueueGetTests.test_get_cancelled_racec C s| t j| jdd�}t j|jd�| jd� t j|jd�| jd� tj| j� | j| jj|j � �d� | j| jj|j � �d� d S )Nr )r r r\ )r �b)
r r r r r" r r] r( r* r )r r r r r � test_get_with_waiting_putters s z+QueueGetTests.test_get_with_waiting_puttersc C s^ t jdd� �}t jdd� �}d}d}t j|| jd�}| jjt j|||�|||�| jd�� d S )Nc s s$ x t |�D ]}| j� E d H q
W d S )N)r= r )�queueZ num_expectedr; r r r �consumer s z<QueueGetTests.test_why_are_getters_waiting.<locals>.consumerc s s& x t |�D ]}| j|�E d H q
W d S )N)r= r" )rd 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 re rf Z
queue_sizeZproducer_num_itemsr r r r � test_why_are_getters_waiting s z*QueueGetTests.test_why_are_getters_waitingc sd dd� }� j |�� _tj� fdd��}tj� jdd�}� jj� jj||��� � jt|j �d� d S )Nc s s dV dV d S )Ng�������?g�������?r r r r r � a_generator+ s zXQueueGetTests.test_cancelled_getters_not_being_held_in_self_getters.<locals>.a_generatorc 3 s< y t j| j� d� jd�E d H }W n t jk
r6 Y nX d S )Ng�������?)r )r rZ r r � TimeoutError)rd �item)r r r re 0 s zUQueueGetTests.test_cancelled_getters_not_being_held_in_self_getters.<locals>.consumerrg )r r r )
r r r r) r r* � create_taskr( �lenZ_getters)r rj re rd r )r r �5test_cancelled_getters_not_being_held_in_self_getters* s zCQueueGetTests.test_cancelled_getters_not_being_held_in_self_gettersN)
r
r r rM rN rV rW rY r[ ra rc ri ro r r r r rJ � s #rJ c @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�
QueuePutTestsc s2 t j| jd�� t j� fdd��}| jj|� � d S )N)r c 3 s � j d�E d H d S )Nr )r" r )r r r rT A s z2QueuePutTests.test_blocking_put.<locals>.queue_put)r r r r) r* )r rT r )r r �test_blocking_put> s z QueuePutTests.test_blocking_putc s� �fdd�}�j |��tjd�d��tj�d��d� tj� ��fdd���tj� �����fd d
��}�j|� � �jd �j� � d S ) Nc 3 s d V } � j d| � dV d S )Ng{�G�z�?)r )r )r r r r J s z1QueuePutTests.test_blocking_put_wait.<locals>.genr )r r )r Fc 3 s0 �j � �jd�E d H �jd�E d H d� d S )Nr r! T)rO r" r )rP r rQ r r rT U s z7QueuePutTests.test_blocking_put_wait.<locals>.queue_putc 3 sN �j d�j� tj�� �d�} �j� E d H �j� � | E d H �j� � d S )Ng{�G�z�?)r )rR r# r r rS r5 r )Zqueue_put_task)rP r r rT r rQ r r rK ] s
z7QueuePutTests.test_blocking_put_wait.<locals>.queue_getg{�G�z�?)r r r rU r) r* r rH )r r rK r )rP r r rT r rQ r �test_blocking_put_waitH s
z$QueuePutTests.test_blocking_put_waitc C s, t j| jd�}|jd� | jd|j� � d S )N)r r )r r r r r( r# )r r r r r �test_nonblocking_puti s
z"QueuePutTests.test_nonblocking_putc C s� dd� }| j |�}tj|d�}|j|j� �}|jtjd|d�� |jd� |jd� |j� y|j|� W n. tj k
r� |j|j� �}|j|� Y nX |j
� }| j d|� d S )Nc s s dV dV d S )Ng{�G�z�?g�������?r r r r r r o s zBQueuePutTests.test_get_cancel_drop_one_pending_reader.<locals>.gen)r g{�G�z�?r r! ) r r r rm r r* r r r^ �CancelledErrorrE r( )r r r r �readerrE r r r �'test_get_cancel_drop_one_pending_readern s
z5QueuePutTests.test_get_cancel_drop_one_pending_readerc C s� dd� }| j |�}|jd� tj|d�}|j|j� �}|j|j� �}|j|j� �}|jtjd|d�� |jd� |jd� |j � y|j|� W n tj
k
r� Y nX |j|� | j |j � |j � hddh� d S )Nc s s dV dV d S )Ng{�G�z�?g�������?r r r r r r � s zDQueuePutTests.test_get_cancel_drop_many_pending_readers.<locals>.genT)r g{�G�z�?r r! )
r Z set_debugr r rm r r* r r r^ rt r( rE )r r r r Zreader1Zreader2Zreader3r r r �)test_get_cancel_drop_many_pending_readers� s"
z7QueuePutTests.test_get_cancel_drop_many_pending_readersc C s� dd� }| j |�}tjd|d�}|jd� |j|jd��}|jtjd|d�� |j� }| j |d� |j
� y|j|� W n0 tj k
r� |j|jd��}|j|� Y nX |j� }| j |d� | j |j � d� d S )Nc s s dV dV d 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 rm r" r* r r# r( r^ rt �qsize)r r r r �writerZvalue1Zvalue2r r r �test_put_cancel_drop� s"
z"QueuePutTests.test_put_cancel_dropc C s0 t jd| jd�}|jd� | jt j|jd� d S )Nr )r r r! )r r r r rX � QueueFull)r r r r r � test_nonblocking_put_exception� s
z,QueuePutTests.test_nonblocking_put_exceptionc sz t jd�jd�� � jd� � jd� �j� j� � �jt j� jd� t jd�jd�� t j� �fdd��}�jj |� � d S )Ng�������?)r r r r! r9 c 3 s2 � j d�E d H � j d�E d H �j� j� � d S )Nr r! )r" r r7 r )r r r r rT � s z3QueuePutTests.test_float_maxsize.<locals>.queue_put)
r r r r r r7 rX r{ r) r* )r rT r )r r r �test_float_maxsize� s
z QueuePutTests.test_float_maxsizec sz t j| jd�� t j� fdd��}t j� fdd��}t j|� | jd�}| jd| jj|� �� | j|j� � | j|j � � d S )N)r c 3 s � j d�E d H dS )Nr T)r" r )r r r rT � s z3QueuePutTests.test_put_cancelled.<locals>.queue_putc 3 s � j � E d H S )N)r r )r r r rG � s z.QueuePutTests.test_put_cancelled.<locals>.testr )
r r r r) r r( r* r rD rE )r rT rG rF r )r r �test_put_cancelled� s z QueuePutTests.test_put_cancelledc C s� t j| jdd�}t j|jd�| jd�}t j|jd�| jd�}t j|jd�| jd�}tj| j� | j|j� � | j |j� � |j
� tj| j� | j|j� � | j |j � d� tj| j� | j |j � d� | jj
|� d S )Nr )r r r\ )r rb �X)r r r r r" r r] r rD r5 r^ r( r# r* )r r Zput_aZput_bZput_cr r r �test_put_cancelled_race� s z%QueuePutTests.test_put_cancelled_racec C sX t j| jd�}t j|j� | jd�}tj| j� | jj|jd�� | j | jj|�d� d S )N)r r\ )
r r r r r r r] r* r" r( )r r rF r r r � test_put_with_waiting_getters� s
z+QueuePutTests.test_put_with_waiting_gettersc sz t jd| jd�� t j� fdd��}t j� fdd��}|d�}|d�}|d�}|d �}| jjt j|� ||||| jd�� d S )
Nr! )r c 3 s � j | �E d H d S )N)r" )rl )rd r r rC s z:QueuePutTests.test_why_are_putters_waiting.<locals>.putterc 3 s, d V � j � } xt| �D ] }� j� }qW d S )N)rx r= r# )Znumr; rl )rd r r �getter s z:QueuePutTests.test_why_are_putters_waiting.<locals>.getterr r r9 )r r r r) r* rh )r rC r� Zt0r_ r` Zt3r )rd r � test_why_are_putters_waiting s z*QueuePutTests.test_why_are_putters_waitingN)r
r r rq rr rs rv rw rz r| r} r~ r� r� r� r r r r rp < s
! rp c @ s e Zd Zdd� ZdS )�LifoQueueTestsc sR t j| jd�� xdD ]}� j|� qW � fdd�td�D �}| jdddg|� d S )N)r r r9 r! c s g | ] }� j � �qS r )r# )r: r; )r r r r< % s z-LifoQueueTests.test_order.<locals>.<listcomp>)r r9 r! )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 Zd Zdd� ZdS )�PriorityQueueTestsc sR t j| jd�� xdD ]}� j|� qW � fdd�td�D �}| jdddg|� d S )N)r r r9 r! c s g | ] }� j � �qS r )r# )r: r; )r r r r< 0 s z1PriorityQueueTests.test_order.<locals>.<listcomp>)r r9 r! )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 @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � ZdS )
�_QueueJoinTestMixinNc C s | j | jd�}| jt|j� d S )N)r )�q_classr rX �
ValueError� task_done)r r r r r �test_task_done_underflow8 s z,_QueueJoinTestMixin.test_task_done_underflowc s� �j �jd��xtd�D ]}�j|� qW d� d�tj� ��fdd���tj���fdd��}�jj|� �}�jttd��� � d �x tt |��D ]}�jd� q�W �jjtj
|�jd�� d S )
N)r �d r Tc 3 s, x&�r&�j � E d H } � | 7 � �j� qW d S )N)r r� )rl )� accumulatorr �runningr r �workerG s z2_QueueJoinTestMixin.test_task_done.<locals>.workerc 3 s* ��fdd�t d�D �} � j� E d H | S )Nc s g | ]}t j�� � jd ��qS ))r )r r r )r: �index)r r� r r r< R s zD_QueueJoinTestMixin.test_task_done.<locals>.test.<locals>.<listcomp>r! )r= �join)�tasks)r r r� r r rG P s z0_QueueJoinTestMixin.test_task_done.<locals>.testF) r� r r= r r r) r* r( �sumrn rS )r r> rG r� r )r� r r� r r� r �test_task_done< s z"_QueueJoinTestMixin.test_task_donec s2 | j | jd�� tj� fdd��}| jj|� � d S )N)r c 3 s � j � E d H � j � E d H d S )N)r� r )r r r r� g s z7_QueueJoinTestMixin.test_join_empty_queue.<locals>.join)r� r r r) r* )r r� r )r r �test_join_empty_queuea s z)_QueueJoinTestMixin.test_join_empty_queuec C s8 | j | jd�}| j|j� d� d|_| j|j� d� d S )N)r z maxsize=0r! zmaxsize=0 tasks=2)r� r r( �_formatZ_unfinished_tasks)r r r r r � test_formatn s z _QueueJoinTestMixin.test_format)r
r r r� r� r� r� r� r r r r r� 4 s
%
r� c @ s e Zd ZejZdS )�QueueJoinTestsN)r
r r r r r� r r r r r� v s r� c @ s e Zd ZejZdS )�LifoQueueJoinTestsN)r
r r r r� r� r r r r r� z s r� c @ s e Zd ZejZdS )�PriorityQueueJoinTestsN)r
r r r r� r� r r r r r� ~ s r� �__main__)Zunittestr r r ZTestCaser r rJ rp r� r� r� r� r� r� r
�mainr r r r �<module> s$
c B