Your IP : 216.73.216.213
���]�L � @ s= d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l Z d d l m Z y d d l m
Z
Wn" e k
r� d d l m Z
Yn Xd d � Z
d d � Z d
d � Z Gd d
� d
� Z Gd d � d e j � Z Gd d � d e j � Z Gd d � d e j � Z e d k r9e j � d S)� N)�mock)�
test_utils)�support)� test_supportc C s | S)N� )�fr r �1/opt/alt/python35/lib64/python3.5/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 Z d Z d Z d Z d Z d Z d d � Z d d � Z d d � Z d d
� Z
d d � Z d
d � Z d d � Z
d d � Z d S)�
DuckFutureFNc C s | j � r d Sd | _ d S)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$ | j d k p$ | j d k S)N)r �_DuckFuture__result�_DuckFuture__exception)r r r r r
. s zDuckFuture.donec C s | j d k r | j � | j S)N)r r )r r r r �result3 s zDuckFuture.resultc C s | j S)N)r )r r r r � exception9 s zDuckFuture.exceptionc C s
| | _ d S)N)r )r r r r r �
set_result= s zDuckFuture.set_resultc C s
| | _ d S)N)r )r r r r r �
set_exceptionB s zDuckFuture.set_exceptionc c s$ | j � s d | _ | V| j � S)NT)r
�_asyncio_future_blockingr )r r r r �__iter__G 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 Z d Z � f d d � Z d d � Z d d � Z � S)� DuckTestsc s3 t � j � | j � | _ | j | j j � d S)N)�super�setUp�
new_test_loop�loop�
addCleanup�close)r )� __class__r r r Q s
zDuckTests.setUpc C s t � } t j | � } d S)N)r �asyncio� wrap_future)r r �gr r r �test_wrap_futureV s zDuckTests.test_wrap_futurec C s t � } t j | � } 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) r r )r$ r r O s r c s� e Z d Z � f d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z e j d � d d � � Z e j d � d d � � Z e j d � d d! � � Z e j d � d" d# � � Z e j d � d$ d% � � Z e j d � d& d' � � Z e j d � d( d) � � Z d* d+ � Z d, d- � Z d. d/ � Z d0 d1 � Z d2 d3 � Z d4 d5 � Z e j d � d6 d7 � � Z d8 d9 � Z d: d; � Z! d< d= � Z" � S)>� FutureTestsc s3 t � j � | j � | _ | j | j j � d S)N)r r r r! r" r# )r )r$ r r r c s
zFutureTests.setUpc C s� Gd d � d � } | j t j | � � | j t j | � � � | j t j d � � | j t j t j � � | j t j t j � � � | j t j t j t j � � � t j d | j � } | j t j | � � | j � d S)Nc @ s" e Z d Z d Z d d � Z d S)z+FutureTests.test_isfuture.<locals>.MyFutureNc S s
d | _ d S)NF)r )r r r r �__init__l s z4FutureTests.test_isfuture.<locals>.MyFuture.__init__)r r r r r+ r r r r �MyFuturei s r, � r! ) � assertFalser% Zisfuture�
assertTrue�Futurer ZMockr! r )r r, r r r r �
test_isfutureh s "zFutureTests.test_isfuturec C s\ t j d | j � } | j | j � � | j | j � � | j � | j | j � � d S)Nr! )r% r0 r! r. r r
r r/ )r r r r r �test_initial_state s
z FutureTests.test_initial_statec C s6 t j | j � t 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 z.FutureTests.test_init_constructor_default_loopc C s | j t t j d � d S)N�* )� assertRaises� TypeErrorr% r0 )r r r r �test_constructor_positional� s z'FutureTests.test_constructor_positionalc C s� t j d | j � } | j | j � � | j | j � � | j | j � � | j t j | j � | j t j | j
� | j t j | j d � | j t j | j
d � | j | j � � d S)Nr! )r% r0 r! r/ r r r
r7 �CancelledErrorr r �InvalidStateErrorr r r. )r r r r r � test_cancel� s zFutureTests.test_cancelc C s� t j d | j � } | j t j | j � | j d � | j | j � � | j | j
� � | j | j � d � | j | j � d � | j t j | j d � | j t j | j
d � | j | j � � d S)Nr! r6 )r% r0 r! r7 r; r r r. r r/ r
� assertEqualr r r )r r r r r � test_result� s
zFutureTests.test_resultc C s� t � } t j d | j � } | j t j | j � | j t d | j t
� | j | � | j | j � � | j
| j � � | j t | j � | j | j � | � | j t j | j d � | j t j | j d � | j | j � � d S)Nr! z!StopIteration .* cannot be raised)� RuntimeErrorr% r0 r! r7 r; r ZassertRaisesRegexr8 r �
StopIterationr. r r/ r
r r= r r )r �excr r r r �test_exception� s
zFutureTests.test_exceptionc C s< t j d | j � } | j t � | j | j � t � d S)Nr! )r% r0 r! r r? �assertIsInstancer )r r r r r �test_exception_class� s
z FutureTests.test_exception_classc s� t j d | j � � � f d d � } | � } | j t | � d � | j t | � � � � j d � | j t | � d � | j t | � d � d S)
Nr! c 3 s5 d V� Ed H} d | f V� Ed H} d | f Vd S)N�A�B�Cr )�x�y)r r r �fixture� s
z2FutureTests.test_yield_from_twice.<locals>.fixturerE r6 rF rG )rF r6 )rG r6 )r% r0 r! r= �nextr )r rJ r' r )r r �test_yield_from_twice� s
z!FutureTests.test_yield_from_twicec C s� | j j d � t j d | j � } | j d } | j t | � d | d | d f � | j � | j j d � t j d | j � } | j t | � d � | j � t j d | j � } | j � | j t | � d � t j d | j � } | j d � | j t | � d
� | j | j � d � t
� } t j d | j � } | j | � | j t | � d � | j | j
� | � d d
� } t j d | j � } | j t � | t � }
| j t | � d |
� | j � | j t | � d � t j d | j � } | j t � | j t � | t � } | t � }
| j t | � d | |
f � t j d | j � } | j t � x! t d � D] } | j t � q�W| j t � d | |
f } | j t | � d | � | j � | j t | � d � d S)NTr! r- 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 s8 t j | � \ } } d | j | | f } t j | � S)Nz
%s() at %s:%s)r Zget_function_sourcer �re�escape)�func�filename�lineno�textr r r � func_repr� s z/FutureTests.test_future_repr.<locals>.func_reprz<Future pending cb=\[%s\]>z <Future pending cb=\[%s, %s\]>� z%s, <8 more>, %s���)r! � set_debugr% r0 �_source_tracebackr= �reprr r r r? r r3 r �add_done_callbackr � assertRegexr
r �range)r Zf_pending_debug�frameZ f_pending� f_cancelledZf_resultrA � f_exceptionrT Zf_one_callbacksZ fake_reprZf_two_callbacksZ
first_reprZ last_reprZf_many_callbacks�iZcb_regexr r r �test_future_repr� sh
z FutureTests.test_future_reprc C sB d d l m } t j d | j � } | j d � t j d | j � } | | | � | j | j � � | j | j � d � t j d | j � } | j
t � � t j d | j � } | | | � | j | j � � | j t | j � t j d | j � } | j
� t j d | j � } | | | � | j | j � � d S)Nr )�_copy_future_stater! �
)Zasyncio.futuresrb r% r0 r! r r/ r
r= r r r? r7 r r )r rb r Znewfr_ Znewf_exceptionr^ Znewf_cancelledr r r �test_copy_state s$
zFutureTests.test_copy_statec sW t j d | j � � � f d d � � � f d d � } | j t | � � j � d S)Nr! c 3 s
� Ed Hd S)Nr r )�futr r �coro. s z#FutureTests.test_iter.<locals>.coroc s � � \ } } d S)Nr )Zarg1Zarg2)rf r r �test1 s z#FutureTests.test_iter.<locals>.test)r% r0 r! r7 �AssertionErrorr )r rg r )rf re r � test_iter+ s
zFutureTests.test_iterzasyncio.base_events.loggerc C s/ t j d | j � } ~ | j | j j � d S)Nr! )r% r0 r! r. �error�called)r �m_logre r r r �test_tb_logger_abandoned7 s z$FutureTests.test_tb_logger_abandonedc C sI t j d | j � } | j t � � | j � ~ | j | j j � d S)Nr! ) r% r0 r! r � Exceptionr r. rj rk )r rl re r r r �&test_tb_logger_not_called_after_cancel= s
z2FutureTests.test_tb_logger_not_called_after_cancelc C s< t j d | j � } | j d � ~ | j | j j � d S)Nr! r6 )r% r0 r! r r. rj rk )r rl re r r r �!test_tb_logger_result_unretrievedE s
z-FutureTests.test_tb_logger_result_unretrievedc C sF t j d | j � } | j d � | j � ~ | j | j j � d S)Nr! r6 )r% r0 r! r r r. rj rk )r rl re r r r � test_tb_logger_result_retrievedL s
z+FutureTests.test_tb_logger_result_retrievedc C s\ t j d | j � } | j t d � � ~ t j | j � t j � | j | j
j � d S)Nr! �boom) r% r0 r! r r? r � run_brieflyr �
gc_collectr/ rj rk )r rl re r r r �$test_tb_logger_exception_unretrievedT s
z0FutureTests.test_tb_logger_exception_unretrievedc C sL t j d | j � } | j t d � � | j � ~ | j | j j � d S)Nr! rr ) r% r0 r! r r? r r. rj rk )r rl re r r r �"test_tb_logger_exception_retrieved] s
z.FutureTests.test_tb_logger_exception_retrievedc C sU t j d | j � } | j t d � � | j t | j � ~ | j | j j � d S)Nr! rr )
r% r0 r! r r? r7 r r. rj rk )r rl re r r r �)test_tb_logger_exception_result_retrievede s
z5FutureTests.test_tb_logger_exception_result_retrievedc C s� d d � } t j j d � } | j | d � } t j | d | j �} | j j | � \ } } | j | t j � | j
| d � | j | t j
� � d S)Nc S s | t j � f S)N)� threading� get_ident)�argr r r �runo s z)FutureTests.test_wrap_future.<locals>.runr- �oir! )�
concurrent�futures�ThreadPoolExecutor�submitr% r& r! Zrun_until_completerC r0 r= ZassertNotEqualrx ry )r r{ �ex�f1�f2�res�identr r r r( m s z FutureTests.test_wrap_futurec C s8 t j d | j � } t j | � } | j | | � d S)Nr! )r% r0 r! r& r3 )r r� r� r r r �test_wrap_future_futurey s z#FutureTests.test_wrap_future_futurec
s� t j d � �q } � f d d � | _ d d � } t j j d � } | j | d � } t j | � } � j � j
| j � Wd QRXd S)Nzasyncio.futures.eventsc s � j S)N)r! r )r r r �<lambda>� s z>FutureTests.test_wrap_future_use_global_loop.<locals>.<lambda>c S s | t j � f S)N)rx ry )rz r r r r{ � s z9FutureTests.test_wrap_future_use_global_loop.<locals>.runr- r| ) r �patchZget_event_loopr} r~ r r� r% r& r3 r! r4 )r �eventsr{ r� r� r� r )r r � test_wrap_future_use_global_loop~ s z,FutureTests.test_wrap_future_use_global_loopc C sk t j j � } t j | d | j �} | j � t j | j � | j | j
� � | j | j
� � d S)Nr! ) r} r~ r0 r% r& r! r r rs r/ r )r r� r� r r r �test_wrap_future_cancel� s
z#FutureTests.test_wrap_future_cancelc C s� t j j � } t j | d | j �} | j d � | j � t j | j � | j
| j � � | j | j
� d � | j | j � � d S)Nr! r6 )r} r~ r0 r% r& r! r r r rs r. r r= r r/ )r r� r� r r r �test_wrap_future_cancel2� s
z$FutureTests.test_wrap_future_cancel2c C sy | j j d � t j d | j � } t j � j d } | j | j t � | j
| j d d d � t | d f � d S)NTr! r- � � test_future_source_tracebackrV ) r! rW r% r0 �sys� _getframe�f_linenorC rX �listr= �__file__)r �futurerR r r r r� � s z(FutureTests.test_future_source_tracebackc C s� | j j | � d d � } | � } t j d | j � } | rI | j } | j | � d } t j | j � t j � t
j d k r� | r� | d } d j d t
j | d � d
| d � } n d } t | � | | j f } | j j t j d | �n[ | r7| d } d
j d t
j | d � d
| d � } n d } | j j t j d d �| j j d d }
| j |
t
j | t
j � � d S)Nc S s= y
t � � Wn) t k
r8 } z | SWYd d } ~ Xn Xd S)N)� MemoryError�
BaseException)rA r r r � memory_error� s
zHFutureTests.check_future_exception_never_retrieved.<locals>.memory_errorr! r� rM r- aD ^Future exception was never retrieved\nfuture: <Future finished exception=MemoryError\(\) created at {filename}:{lineno}>\nsource_traceback: Object created at \(most recent call last\):\n File.*\n File "{filename}", line {lineno}, in check_future_exception_never_retrieved\n future = asyncio\.Future\(loop=self\.loop\)$rQ r rR z[^Future exception was never retrieved\nfuture: <Future finished exception=MemoryError\(\)>$�exc_infoa ^Future/Task exception was never retrieved\nFuture/Task created at \(most recent call last\):\n File.*\n File "{filename}", line {lineno}, in check_future_exception_never_retrieved\n future = asyncio\.Future\(loop=self\.loop\)\nTraceback \(most recent call last\):\n.*\nMemoryError$zb^Future/Task exception was never retrieved\nTraceback \(most recent call last\):\n.*\nMemoryError$F)r� rM rV rV )r! rW r% r0 rX r r rs r rt r� � version_info�formatrN rO �type�
__traceback__rj Zassert_called_once_withr �ANY� call_argsr[ �compile�DOTALL) r �debugrl r� rA r� Zsource_tracebackr] Zregexr� �messager r r �&check_future_exception_never_retrieved� s8
z2FutureTests.check_future_exception_never_retrievedc C s | j d � d S)NF)r� )r r r r �%test_future_exception_never_retrieved� s z1FutureTests.test_future_exception_never_retrievedc C s | j d � d S)NT)r� )r r r r �+test_future_exception_never_retrieved_debug� s z7FutureTests.test_future_exception_never_retrieved_debugc C sV d d l m } t j d | j � } | j � | j | d � | j | j � � d S)Nr )r~ r! � )r% r~ r0 r! r Z _set_result_unless_cancelledr/ r )r r~ re r r r � test_set_result_unless_cancelled� s
z,FutureTests.test_set_result_unless_cancelled)#r r r r r1 r2 r5 r9 r< r> rB rD rL ra rd ri r r� rm ro rp rq ru rv rw r( r� r� r� r� r� r� r� r� r� r r )r$ r r* a s<
A
B r* c sj e Z d Z � f d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z � S)�FutureDoneCallbackTestsc s t � j � | j � | _ d S)N)r r r r! )r )r$ r r r � s
z FutureDoneCallbackTests.setUpc C s t j | j � d S)N)r rs r! )r r r r rs � s z#FutureDoneCallbackTests.run_brieflyc s � � f d d � } | S)Nc s � j � � d S)N)�append)r� )�bag�thingr r � bag_appender s z<FutureDoneCallbackTests._make_callback.<locals>.bag_appenderr )r r� r� r� r )r� r� r �_make_callback� s z&FutureDoneCallbackTests._make_callbackc C s t j d | j � S)Nr! )r% r0 r! )r r r r � _new_future s z#FutureDoneCallbackTests._new_futurec C s� g } | j � } | j | j | d � � | j | j | d � � | j | g � | j d � | j � | j | d d g � | j | j � d � d S)Nr6 � �foo)r� rZ r� r= r rs r )r r� r r r r �$test_callbacks_invoked_on_set_result s
z<FutureDoneCallbackTests.test_callbacks_invoked_on_set_resultc C s� g } | j � } | j | j | d � � | j | g � t � } | j | � | j � | j | d g � | j | j � | � d S)N�d )r� rZ r� r= r? r rs r )r r� r rA r r r �'test_callbacks_invoked_on_set_exception s
z?FutureDoneCallbackTests.test_callbacks_invoked_on_set_exceptionc C s2 g } | j � } | j | d � } | j | d � } | j | d � } | j | � | j | � | j | j | � d � | j | j | � d � | j | � x! t d � D] } | j | � q� W| j | j | � d � | j | g � | j d � | j � | j | d g � | j | j � d � d S)Nr- r� r� r � � r� ) r� r� rZ r= Zremove_done_callbackr\ r rs r )r r� r Zcb1Zcb2Zcb3r` r r r �test_remove_done_callback# s$
z1FutureDoneCallbackTests.test_remove_done_callback)
r r r r rs r� r� r� r� r� r r )r$ r r� � s r� �__main__)�concurrent.futuresr} rN r� rx Zunittestr r% r rg r � ImportErrorr r r
r r ZTestCaser r* r� r �mainr r r r �<module> s*
3� �P