Your IP : 216.73.216.213
3
��2a�W � @ sl d dl Zd dlZd dlZd dlZd dlZd dlZd dlmZ d dlZd dlm Z d dlm
Z
yd dl m Z W n e
k
r� d dlmZ Y nX dd� Zd d
� Zd d � ZG d
d� d�ZG dd� de j�ZG dd� d�Zejee
d�d�G dd� dee j��ZG dd� dee j�ZG dd� d�Zejee
d�d�G dd � d ee j��ZG d d � d ee j�Z e d k�rhej � dS ) � N)�mock)�
test_utils)�futures)�support)� test_supportc C s | S )N� )�fr r �1/opt/alt/python36/lib64/python3.6/test_futures.py� _fakefunc s r
c C s d S )Nr r r r r �first_cb s r c C s d S )Nr r r r r �last_cb s r c @ s\ e Zd ZdZdZdZdZdd� Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dd� ZdS )�
DuckFutureFNc C s | j � r dS d| _dS )NFT)�done�_DuckFuture__cancelled)�selfr r r �cancel) s zDuckFuture.cancelc C s | j S )N)r )r r r r � cancelled/ s zDuckFuture.cancelledc C s | j p| jd k p| jd k S )N)r �_DuckFuture__result�_DuckFuture__exception)r r r r r 2 s
zDuckFuture.donec C s | j d k r| j �| jS )N)r r )r r r r �result7 s
zDuckFuture.resultc C s | j S )N)r )r r r r � exception= s zDuckFuture.exceptionc C s
|| _ d S )N)r )r r r r r �
set_resultA s zDuckFuture.set_resultc C s
|| _ d S )N)r )r r r r r �
set_exceptionF s zDuckFuture.set_exceptionc c s | j � sd| _| V | j� S )NT)r �_asyncio_future_blockingr )r r r r �__iter__K s zDuckFuture.__iter__)�__name__�
__module__� __qualname__r r r r r r r r r r r r r r r r r
s r
c s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )� DuckTestsc s& t � j� | j� | _| j| jj� d S )N)�super�setUp�
new_test_loop�loop�
addCleanup�close)r )� __class__r r r U s
zDuckTests.setUpc C s t � }tj|�}d S )N)r
�asyncio� wrap_future)r r �gr r r �test_wrap_futureZ s
zDuckTests.test_wrap_futurec C s t � }tj|�}d S )N)r
r& Z
ensure_future)r r r( r r r �test_ensure_future_ s
z DuckTests.test_ensure_future)r r r r r) r* �
__classcell__r r )r% r r S s r c s| e Zd Zdd� Z� fdd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zejd �d d!� �Zejd �d"d#� �Zejd �d$d%� �Zejd �d&d'� �Zejd �d(d)� �Zejd �d*d+� �Zejd �d,d-� �Zd.d/� Zd0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z ejd �d:d;� �Z!d<d=� Z"d>d?� Z#d@dA� Z$dBdC� Z%dDdE� Z&dFdG� Z'� Z(S )H�BaseFutureTestsc O s | j ||�S )N)�cls)r �args�kwargsr r r � _new_futureg s zBaseFutureTests._new_futurec s& t � j� | j� | _| j| jj� d S )N)r r r! r" r# r$ )r )r% r r r j s
zBaseFutureTests.setUpc C s� G dd� d�}| j tj|�� | jtj|� �� | j tjd�� | j tjtj� �� | j| jd�}| jtj|�� | j tjt|��� | jtjtjt|���� |j � d S )Nc @ s e Zd ZdZdd� ZdS )z/BaseFutureTests.test_isfuture.<locals>.MyFutureNc S s
d| _ d S )NF)r )r r r r �__init__s s z8BaseFutureTests.test_isfuture.<locals>.MyFuture.__init__)r r r r r1 r r r r �MyFuturep s r2 � )r" )
� assertFalser& �isfuture�
assertTruer ZMockr0 r" �typer )r r2 r r r r �
test_isfutureo s z BaseFutureTests.test_isfuturec C sD | j | jd�}| j|j� � | j|j� � |j� | j|j� � d S )N)r" )r0 r" r4 r r r r6 )r r r r r �test_initial_state� s
z"BaseFutureTests.test_initial_statec C s( t j| j� | j� }| j|j| j� d S )N)r& Zset_event_loopr" r0 �assertIs�_loop)r r r r r �"test_init_constructor_default_loop� s z2BaseFutureTests.test_init_constructor_default_loopc C s | j t| jd� d S )N�* )� assertRaises� TypeErrorr0 )r r r r �test_constructor_positional� s z+BaseFutureTests.test_constructor_positionalc 6 C s | j j| j | jd�}| jtj|j� | j j| j | jd�}| jtj|j� | j j| j | jd�}| jtt f�� |j
d � W d Q R X | j j| j | jd�}| jtt f�� |j t � W d Q R X | j j| j | jd�}| jtt f�� |j
� W d Q R X | j j| j | jd�}| jtt f�� |jdd� � W d Q R X | j j| j | jd�}| jtt f�� |jdd� � W d Q R X | j j| j | jd�}| jtt f�� |j� W d Q R X | j j| j | jd�}y t|� W n t k
�r� Y nX | j j| j | jd�}|j� |j� t|� d S )N)r" c S s d S )Nr )r r r r �<lambda>� s z4BaseFutureTests.test_uninitialized.<locals>.<lambda>c S s d S )Nr )r r r r rA � s )r- �__new__r" r> r& �InvalidStateErrorr r � RuntimeError�AttributeErrorr r � Exceptionr �add_done_callback�remove_done_callbackZ_schedule_callbacks�reprr r �iter)r �futr r r �test_uninitialized� s> z"BaseFutureTests.test_uninitializedc C s� | j | jd�}| j|j� � | j|j� � | j|j� � | jtj|j � | jtj|j
� | jtj |j d � | jtj |j
d � | j|j� � d S )N)r" )r0 r" r6 r r r r> r& ZCancelledErrorr r rC r r r4 )r r r r r � test_cancel� s zBaseFutureTests.test_cancelc C s� | j | jd�}| jtj|j� |jd� | j|j� � | j |j
� � | j |j� d� | j |j � d � | jtj|jd � | jtj|j
d � | j|j� � d S )N)r" r= )r0 r" r> r& rC r r r4 r r6 r � assertEqualr r r )r r r r r � test_result� s
zBaseFutureTests.test_resultc C s� t � }| j| jd�}| jtj|j� | jtd|j t
� |j |� | j |j � � | j
|j� � | jt |j� | j|j� |� | jtj|jd � | jtj|j d � | j |j� � d S )N)r" z!StopIteration .* cannot be raised)rD r0 r" r> r& rC r ZassertRaisesRegexr? r �
StopIterationr4 r r6 r r rN r r )r �excr r r r �test_exception� s
z BaseFutureTests.test_exceptionc C s, | j | jd�}|jt� | j|j� t� d S )N)r" )r0 r" r rD �assertIsInstancer )r r r r r �test_exception_class� s
z$BaseFutureTests.test_exception_classc sn | j | jd�� � fdd�}|� }| jt|�d� | jt|�� � � jd� | jt|�d� | jt|�d � d S )
N)r" c 3 s2 dV � E d H } d| fV � E d H }d|fV d S )N�A�B�Cr )�x�y)r r r �fixture� s
z6BaseFutureTests.test_yield_from_twice.<locals>.fixturerU r= rV rW )rV r= )rW r= )r0 r" rN �nextr )r rZ r( r )r r �test_yield_from_twice� s
z%BaseFutureTests.test_yield_from_twicec C s4 | j jd� | j| j d�}|jd }| jt|�d|d |d f � |j� | j jd� | j| j d�}| jt|�d� |j� | j| j d�}|j� | jt|�d� | j| j d�}|jd � | jt|�d
� | j|j� d � t � }| j| j d�}|j
|� | jt|�d � | j |j � |� d d
� }| j| j d�} | j
t� |t�}
| jt| �d|
� | j� | jt| �d� | j| j d�} | j
t� | j
t� |t�} |t�}
| jt| �d| |
f � | j| j d�}|j
t� x td�D ]}|j
t� �q�W |j
t� d| |
f }| jt|�d| � |j� | jt|�d� d S )NT)r" r3 z!<Future pending created at %s:%s>r Fz<Future pending>z<Future cancelled>� z<Future finished result=4>z*<Future finished exception=RuntimeError()>c S s( t j| �\}}d| j||f }tj|�S )Nz
%s() at %s:%s)r Zget_function_sourcer �re�escape)�func�filename�lineno�textr r r � func_repr s z3BaseFutureTests.test_future_repr.<locals>.func_reprz<Future pending cb=\[%s\]>z <Future pending cb=\[%s, %s\]>� z%s, <8 more>, %s���)r" � set_debugr0 �_source_tracebackrN rI r r r rD r r: r rG r
� assertRegexr r �range)r Zf_pending_debug�frameZ f_pending� f_cancelledZf_resultrQ � f_exceptionrd Zf_one_callbacksZ fake_reprZf_two_callbacksZ
first_reprZ last_reprZf_many_callbacks�iZcb_regexr r r �test_future_repr� sh
z BaseFutureTests.test_future_reprc C s� ddl m} | j| jd�}|jd� | j| jd�}|||� | j|j� � | j|j� d� | j| jd�}|j t
� � | j| jd�}|||� | j|j� � | j t
|j� | j| jd�}|j � | j| jd�}|||� | j|j
� � d S )Nr )�_copy_future_state)r" �
)Zasyncio.futuresrp r0 r" r r6 r rN r r rD r> r r )r rp r Znewfrm Znewf_exceptionrl Znewf_cancelledr r r �test_copy_state9 s$
z BaseFutureTests.test_copy_statec s> | j | jd���fdd�� � fdd�}| jt|� �j� d S )N)r" c 3 s � E d H d S )Nr r )rK r r �coroV s z'BaseFutureTests.test_iter.<locals>.coroc s � � \} }d S )Nr )Zarg1Zarg2)rs r r �testY s z'BaseFutureTests.test_iter.<locals>.test)r0 r" r> �AssertionErrorr )r rt r )rs rK r � test_iterS s
zBaseFutureTests.test_iterzasyncio.base_events.loggerc C s" | j | jd�}~| j|jj� d S )N)r" )r0 r" r4 �error�called)r �m_logrK r r r �test_tb_logger_abandoned_ s z(BaseFutureTests.test_tb_logger_abandonedc C s6 | j | jd�}|jt� � |j� ~| j|jj� d S )N)r" )r0 r" r rF r r4 rw rx )r ry rK r r r �&test_tb_logger_not_called_after_cancele s
z6BaseFutureTests.test_tb_logger_not_called_after_cancelc C s, | j | jd�}|jd� ~| j|jj� d S )N)r" r= )r0 r" r r4 rw rx )r ry rK r r r �!test_tb_logger_result_unretrievedm s
z1BaseFutureTests.test_tb_logger_result_unretrievedc C s4 | j | jd�}|jd� |j� ~| j|jj� d S )N)r" r= )r0 r" r r r4 rw rx )r ry rK r r r � test_tb_logger_result_retrievedt s
z/BaseFutureTests.test_tb_logger_result_retrievedc C sD | j | jd�}|jtd�� ~tj| j� tj� | j|j j
� d S )N)r" �boom) r0 r" r rD r � run_brieflyr �
gc_collectr6 rw rx )r ry rK r r r �$test_tb_logger_exception_unretrieved| s z4BaseFutureTests.test_tb_logger_exception_unretrievedc C s8 | j | jd�}|jtd�� |j� ~| j|jj� d S )N)r" r~ )r0 r" r rD r r4 rw rx )r ry rK r r r �"test_tb_logger_exception_retrieved� s
z2BaseFutureTests.test_tb_logger_exception_retrievedc C s> | j | jd�}|jtd�� | jt|j� ~| j|jj� d S )N)r" r~ ) r0 r" r rD r> r r4 rw rx )r ry rK r r r �)test_tb_logger_exception_result_retrieved� s
z9BaseFutureTests.test_tb_logger_exception_result_retrievedc C s| dd� }t jjd�}|j|d�}tj|| jd�}| jj|�\}}| jtj |�� | j
|d� | j |t j
� � |jdd� d S )Nc S s | t j� fS )N)� threading� get_ident)�argr r r �run� s z-BaseFutureTests.test_wrap_future.<locals>.runr3 �oi)r" T)�wait)�
concurrentr �ThreadPoolExecutor�submitr& r' r" Zrun_until_completer6 r5 rN ZassertNotEqualr� r� �shutdown)r r� �ex�f1�f2�resZidentr r r r) � s z BaseFutureTests.test_wrap_futurec C s( | j | jd�}tj|�}| j||� d S )N)r" )r0 r" r& r' r: )r r� r� r r r �test_wrap_future_future� s
z'BaseFutureTests.test_wrap_future_futurec
sn t jd��Z}� fdd�|_dd� }tjjd�}|j|d�}tj|�}� j � j
|j � |j dd � W d Q R X d S )
Nzasyncio.futures.eventsc s � j S )N)r" r )r r r rA � s zBBaseFutureTests.test_wrap_future_use_global_loop.<locals>.<lambda>c S s | t j� fS )N)r� r� )r� r r r r� � s z=BaseFutureTests.test_wrap_future_use_global_loop.<locals>.runr3 r� T)r� )
r �patchZget_event_loopr� r r� r� r& r' r: r" r; r� )r Zeventsr� r� r� r� r )r r � test_wrap_future_use_global_loop� s
z0BaseFutureTests.test_wrap_future_use_global_loopc C sN t jj� }tj|| jd�}|j� tj| j� | j |j
� � | j |j
� � d S )N)r" ) r� r �Futurer& r' r" r r r r6 r )r r� r� r r r �test_wrap_future_cancel� s
z'BaseFutureTests.test_wrap_future_cancelc C sh t jj� }tj|| jd�}|jd� |j� tj | j� | j
|j � � | j |j
� d� | j|j � � d S )N)r" r= )r� r r� r& r' r" r r r r r4 r rN r r6 )r r� r� r r r �test_wrap_future_cancel2� s
z(BaseFutureTests.test_wrap_future_cancel2c C sZ | j jd� | j| j d�}tj� jd }| j|jt� | j |jd d d� t
|df� d S )NT)r" r3 � � � test_future_source_traceback���) r" rg r0 �sys� _getframe�f_linenorS rh �listrN �__file__)r �futurerb r r r r� � s z,BaseFutureTests.test_future_source_tracebackc C s� | j j|� dd� }|� }| j| j d�}|j|� d }tj| j � tj� tj d kr|d}t
|�||j f}|j j
tj|d� nd}|j j
tjd d� |j jd
d
}| j|tj|tj�� d S ) Nc S s0 y
t � �W n tk
r* } z| S d } ~ X nX d S )N)� MemoryError�
BaseException)rQ r r r � memory_error� s
zLBaseFutureTests.check_future_exception_never_retrieved.<locals>.memory_error)r" r� r] z'^Future exception was never retrieved\n)�exc_infoz,^Future/Task exception was never retrieved\nFr )r� r] )r" rg r0 r r r r r� r� � version_infor7 �
__traceback__rw Zassert_called_once_withr �ANY� call_argsri r^ �compile�DOTALL) r �debugry r� rQ r� Zregexr� �messager r r �&check_future_exception_never_retrieved� s
z6BaseFutureTests.check_future_exception_never_retrievedc C s | j d� d S )NF)r� )r r r r �%test_future_exception_never_retrieved� s z5BaseFutureTests.test_future_exception_never_retrievedc C s | j d� d S )NT)r� )r r r r �+test_future_exception_never_retrieved_debug� s z;BaseFutureTests.test_future_exception_never_retrieved_debugc C s4 | j | jd�}|j� tj|d� | j|j� � d S )N)r" r� )r0 r" r r Z _set_result_unless_cancelledr6 r )r rK r r r � test_set_result_unless_cancelled� s z0BaseFutureTests.test_set_result_unless_cancelledc C sz | j | jd�}|jd� |j� }d }y|jd � W n, tk
r^ } z|jd }W Y d d }~X n X | jd� | j|d� d S )N)r" r3 r� r zStopIteration was expected)r3 r� )r3 r� ) r0 r" r r �sendrP r. ZfailrN )r rK �fir r� r r r � test_future_stop_iteration_args� s
z/BaseFutureTests.test_future_stop_iteration_argsc C s\ | j | jd�}t|�}| jt|jttd�d� | jt|jtd�td�� | jt|jt� d S )N)r" Zelephant� )r0 r" rJ r> r? �throwrF r� )r rK r� r r r �test_future_iter_throw s
z&BaseFutureTests.test_future_iter_throwc C s> G dd� d�}x*t d�D ] }| j| jd�}|j|� � qW d S )Nc @ s e Zd Zdd� ZdS )z5BaseFutureTests.test_future_del_collect.<locals>.Evilc S s t j� d S )N)�gcZcollect)r r r r �__del__ s z=BaseFutureTests.test_future_del_collect.<locals>.Evil.__del__N)r r r r� r r r r �Evil
s r� �d )r" )rj r0 r" r )r r� rn rK r r r �test_future_del_collect s z'BaseFutureTests.test_future_del_collect))r r r r0 r r8 r9 r< r@ rL rM rO rR rT r\ ro rr rv r r� rz r{ r| r} r� r� r� r) r� r� r� r� r� r� r� r� r� r� r� r� r+ r r )r% r r, e sF !
A
r, �_CFuturez requires the C _asyncio modulec @ s e Zd Zeed�ZdS )� CFutureTestsr� N)r r r �getattrr r- r r r r r� s r� c @ s e Zd ZejZdS )�
PyFutureTestsN)r r r r � _PyFuturer- r r r r r� s r� c sd e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z � Z
S )�BaseFutureDoneCallbackTestsc s t � j� | j� | _d S )N)r r r! r" )r )r% r r r " s
z!BaseFutureDoneCallbackTests.setUpc C s t j| j� d S )N)r r r" )r r r r r & s z'BaseFutureDoneCallbackTests.run_brieflyc s � �fdd�}|S )Nc s � j �� d S )N)�append)r� )�bag�thingr r � bag_appender+ s z@BaseFutureDoneCallbackTests._make_callback.<locals>.bag_appenderr )r r� r� r� r )r� r� r �_make_callback) s z*BaseFutureDoneCallbackTests._make_callbackc C s t �d S )N)�NotImplementedError)r r r r r0 / s z'BaseFutureDoneCallbackTests._new_futurec C sr g }| j � }|j| j|d�� |j| j|d�� | j|g � |jd� | j� | j|ddg� | j|j� d� d S )Nr= � �foo)r0 rG r� rN r r r )r r� r r r r �$test_callbacks_invoked_on_set_result2 s
z@BaseFutureDoneCallbackTests.test_callbacks_invoked_on_set_resultc C sd g }| j � }|j| j|d�� | j|g � t� }|j|� | j� | j|dg� | j|j� |� d S )Nr� )r0 rG r� rN rD r r r )r r� r rQ r r r �'test_callbacks_invoked_on_set_exception@ s
zCBaseFutureDoneCallbackTests.test_callbacks_invoked_on_set_exceptionc C s� g }| j � }| j|d�}| j|d�}| j|d�}|j|� |j|� | j|j|�d� | j|j|�d� |j|� xtd�D ]}|j|� q|W | j|j|�d� | j|g � |jd� | j� | j|dg� | j|j� d� d S )Nr3 r� r� r � � r� ) r0 r� rG rN rH rj r r r )r r� r Zcb1Zcb2Zcb3rn r r r �test_remove_done_callbackN s$
z5BaseFutureDoneCallbackTests.test_remove_done_callbackc sP | j � � � jt� xtd�D ]}� jt� q W G � fdd�d�}� j|� � d S )N�? c s e Zd Z� fdd�ZdS )zRBaseFutureDoneCallbackTests.test_remove_done_callbacks_list_mutation.<locals>.evilc s � j t� dS )NF)rH �id)r �other)rK r r �__eq__y s
zYBaseFutureDoneCallbackTests.test_remove_done_callbacks_list_mutation.<locals>.evil.__eq__N)r r r r� r )rK r r �evilx s r� )r0 rG �strrj r� rH )r �_r� r )rK r �(test_remove_done_callbacks_list_mutationo s
zDBaseFutureDoneCallbackTests.test_remove_done_callbacks_list_mutationc C sH dd� }| j � }|j|� |jt� |jt� |jd� tj| j� d S )Nc S s | j t� d S )N)rH r� )r r r r �mut� s zPBaseFutureDoneCallbackTests.test_schedule_callbacks_list_mutation_1.<locals>.mutr3 )r0 rG r� r r r r" )r r� rK r r r �'test_schedule_callbacks_list_mutation_1 s
zCBaseFutureDoneCallbackTests.test_schedule_callbacks_list_mutation_1c s\ | j � ��jt� xtd�D ]}�jt� q W d�d� G � ��fdd�d�}�j|� � d S )Nr� r� r c s e Zd Z� ��fdd�ZdS )zQBaseFutureDoneCallbackTests.test_schedule_callbacks_list_mutation_2.<locals>.evilc s � d7 � � �k r�j t� dS )Nr3 F)rG r� )r r� )� extra_cbsrK �
max_extra_cbsr r r� � s
zXBaseFutureDoneCallbackTests.test_schedule_callbacks_list_mutation_2.<locals>.evil.__eq__N)r r r r� r )r� rK r� r r r� � s r� )r0 rG r� rj r� rH )r r� r� r )r� rK r� r �'test_schedule_callbacks_list_mutation_2� s
zCBaseFutureDoneCallbackTests.test_schedule_callbacks_list_mutation_2)r r r r r r� r0 r� r� r� r� r� r� r+ r r )r% r r� s !
r� c @ s e Zd Zdd� ZdS )�CFutureDoneCallbackTestsc C s t j| jd�S )N)r" )r r� r" )r r r r r0 � s z$CFutureDoneCallbackTests._new_futureN)r r r r0 r r r r r� � s r� c @ s e Zd Zdd� ZdS )�PyFutureDoneCallbackTestsc C s t j| jd�S )N)r" )r r� r" )r r r r r0 � s z%PyFutureDoneCallbackTests._new_futureN)r r r r0 r r r r r� � s r� �__main__) �concurrent.futuresr� r� r^ r� r� Zunittestr r& r r rt r � ImportErrorr r
r r r
ZTestCaser r, Z
skipUnless�hasattrr� r� r� r� r� r �mainr r r r �<module> sH 3 4