Your IP : 216.73.216.213
3
� f�W � @ sp d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl Z ddl m
Z
ddl m Z yddl m
Z
W n ek
r� ddl mZ
Y nX dd � Zd
d � Zd d
� ZG dd� 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 d d � d ee
j��Z G d d � d ee
j�Z e d k�rlej � dS )!zTests for futures.py.� N)�mock)�
test_utils)�futures)�support)� test_supportc C s | S )N� )�fr r �C/opt/alt/python36/lib64/python3.6/test/test_asyncio/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 � st�| jd k r | j�| jS )N)r �AssertionErrorr r )r r r r �result7 s
zDuckFuture.resultc C s | j � st�| jS )N)r r r )r r r r � exception= s zDuckFuture.exceptionc C s$ | j � st�|d k st�|| _d S )N)r r r )r r r r r �
set_resultA s zDuckFuture.set_resultc C s$ | j � st�|d k st�|| _d S )N)r r r )r r r r r �
set_exceptionF s zDuckFuture.set_exceptionc c s( | j � sd| _| V | j � s t�| j� S )NT)r �_asyncio_future_blockingr r )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|�}||ks t�d S )N)r
�asyncio� wrap_futurer )r r �gr r r �test_wrap_futureZ s
zDuckTests.test_wrap_futurec C s t � }tj|�}||ks t�d S )N)r
r' Z
ensure_futurer )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 r2 r r r r �MyFuturep s r3 � )r# )
� assertFalser' �isfuture�
assertTruer ZMockr1 r# �typer )r r3 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# )r1 r# r5 r r r r7 )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# r1 �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� TypeErrorr1 )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 rB � 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# )r1 r# r7 r r r r? r' ZCancelledErrorr r rD r r r5 )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> )r1 r# r? r' rD r r r5 r r7 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)rE r1 r# r? r' rD r ZassertRaisesRegexr@ r �
StopIterationr5 r r7 r r rO 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# )r1 r# r rE �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>.fixturerV r> rW rX )rW r> )rX r> )r1 r# rO �nextr )r r[ 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# r4 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_debugr1 �_source_tracebackrO rJ r r r rE r r; r rH r
� assertRegexr r �range)r Zf_pending_debug�frameZ f_pending� f_cancelledZf_resultrR � f_exceptionre 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.futuresrq r1 r# r r7 r rO r r rE r? r r )r rq r Znewfrn Znewf_exceptionrm 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 )rL r r �coroV s z'BaseFutureTests.test_iter.<locals>.coroc s � � \} }d S )Nr )Zarg1Zarg2)rt r r �testY s z'BaseFutureTests.test_iter.<locals>.test)r1 r# r? r r )r ru r )rt rL r � test_iterS s
zBaseFutureTests.test_iterzasyncio.base_events.loggerc C s" | j | jd�}~| j|jj� d S )N)r# )r1 r# r5 �error�called)r �m_logrL 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# )r1 r# r rG r r5 rw rx )r ry rL 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> )r1 r# r r5 rw rx )r ry rL 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> )r1 r# r r r5 rw rx )r ry rL 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) r1 r# r rE r � run_brieflyr �
gc_collectr7 rw rx )r ry rL 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~ )r1 r# r rE r r5 rw rx )r ry rL 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~ ) r1 r# r rE r? r r5 rw rx )r ry rL 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>.runr4 �oi)r# T)�wait)�
concurrentr �ThreadPoolExecutor�submitr' r( r# Zrun_until_completer7 r6 rO 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# )r1 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 rB � 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>.runr4 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 r7 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 r5 r rO r r7 )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# r4 � � � test_future_source_traceback���) r# rh r1 �sys� _getframe�f_linenorT ri �listrO �__file__)r �futurerc 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)rR 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# rh r1 r r r r r� r� � version_infor8 �
__traceback__rw Zassert_called_once_withr �ANY� call_argsrj r_ �compile�DOTALL) r �debugry r� rR 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� )r1 r# r r Z _set_result_unless_cancelledr7 r )r rL 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# r4 r� r zStopIteration was expected)r4 r� )r4 r� ) r1 r# r r �sendrQ r/ ZfailrO )r rL �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� )r1 r# rK r? r@ �throwrG r� )r rL 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# )rk r1 r# r )r r� ro rL r r r �test_future_del_collect s z'BaseFutureTests.test_future_del_collect))r r r r1 r! r9 r: r= rA rM rN rP rS rU r] rp rs 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 r1 / 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)r1 rH r� rO 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� )r1 rH r� rO rE r r r )r r� r rR 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 )Nr4 r� r� r � � r� ) r1 r� rH rO rI rk r r r )r r� r Zcb1Zcb2Zcb3ro 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)rI �id)r �other)rL r r �__eq__y s
zYBaseFutureDoneCallbackTests.test_remove_done_callbacks_list_mutation.<locals>.evil.__eq__N)r r r r� r )rL r r �evilx s r� )r1 rH �strrk r� rI )r �_r� r )rL 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)rI r� )r r r r �mut� s zPBaseFutureDoneCallbackTests.test_schedule_callbacks_list_mutation_1.<locals>.mutr4 )r1 rH r� r r r r# )r r� rL 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 )Nr4 F)rH r� )r r� )� extra_cbsrL �
max_extra_cbsr r r� � s
zXBaseFutureDoneCallbackTests.test_schedule_callbacks_list_mutation_2.<locals>.evil.__eq__N)r r r r� r )r� rL r� r r r� � s r� )r1 rH r� rk r� rI )r r� r� r )r� rL r� r �'test_schedule_callbacks_list_mutation_2� s
zCBaseFutureDoneCallbackTests.test_schedule_callbacks_list_mutation_2)r r r r! r r� r1 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 r1 � s z$CFutureDoneCallbackTests._new_futureN)r r r r1 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 r1 � s z%PyFutureDoneCallbackTests._new_futureN)r r r r1 r r r r r� � s r� �__main__) �__doc__Zconcurrent.futuresr� r� r_ r� r� Zunittestr r' r r ru r � ImportErrorr r
r r r
ZTestCaser r- Z
skipUnless�hasattrr� r� r� r� r� r �mainr r r r �<module> sJ 3 4