Your IP : 216.73.216.213
�
f f�@ � @ s d Z 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 e j � Z Gd d � d e j � Z e d k re j � n d S)zTests for futures.py.� N)�mock)�
test_utils)�support)� test_supportc C s | S)N� )�fr r �C/opt/alt/python34/lib64/python3.4/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 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 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 d& d' � Z d( d) � Z e j d* � d+ d, � � Z d- d. � Z d/ d0 � Z d1 d2 � Z e j d � d3 d4 � � Z d5 d6 � Z d7 d8 � Z d9 d: � Z d; S)<� FutureTestsc C s& | j � | _ | j | j j � d S)N)�
new_test_loop�loopZ
addCleanup�close)�selfr r r �setUp s zFutureTests.setUpc C s\ t j d | j � } | j | j � � | j | j � � | j � | j | j � � d S)Nr )�asyncio�Futurer � assertFalse� cancelled�done�cancel�
assertTrue)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 r �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 r )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 r r r r r r r ZCancelledError�result� exception�InvalidStateError�
set_result�
set_exceptionr )r r r r r � test_cancel2 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 r )r r r r 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 | � | j | j � � | j
| j � � | j t | j � | j
| j � | � | j t j | j d � | j t j | j d � | j | j � � d S)Nr )� RuntimeErrorr r r r r# r" r% r r r r r! r' r$ r )r �excr r r r �test_exceptionJ s
zFutureTests.test_exceptionc C s< t j d | j � } | j t � | j | j � t � d S)Nr )r r r r% r) �assertIsInstancer" )r r r r r �test_exception_classX 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� Dd H} d | f V� Dd H} d | f Vd S)N�A�B�Cr )�x�y)r r r �fixture` s
z2FutureTests.test_yield_from_twice.<locals>.fixturer. r r/ r0 )r/ r )r0 r )r r r r' �nextr$ )r r3 �gr )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 � 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_source� __qualname__�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 r �_source_tracebackr' �reprr r$ r! r) r% r r" �add_done_callbackr � assertRegexr
r �range)r Zf_pending_debug�frameZ f_pending� f_cancelledZf_resultr* � f_exceptionr@ Zf_one_callbacksZ fake_reprZf_two_callbacksZ
first_reprZ last_reprZf_many_callbacks�iZcb_regexr r r �test_future_repro 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.futuresrN r r r r$ r r r' r! r% r) r r r )r rN r ZnewfrK Znewf_exceptionrJ 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
� Dd Hd S)Nr r )�futr r �coro� s z#FutureTests.test_iter.<locals>.coroc s � � \ } } d S)Nr )Zarg1Zarg2)rR r r �test� s z#FutureTests.test_iter.<locals>.test)r r r r �AssertionErrorr )r rS r )rR rQ r � test_iter� s
zFutureTests.test_iterzasyncio.base_events.loggerc C s/ t j d | j � } ~ | j | j j � d S)Nr )r r r r �error�called)r �m_logrQ r r r �test_tb_logger_abandoned� s z$FutureTests.test_tb_logger_abandonedc C s< t j d | j � } | j d � ~ | j | j j � d S)Nr r )r r r r$ r rV rW )r rX rQ r r r �!test_tb_logger_result_unretrieved� s
z-FutureTests.test_tb_logger_result_unretrievedc C sF t j d | j � } | j d � | j � ~ | j | j j � d S)Nr r )r r r r$ r! r rV rW )r rX rQ r r r � test_tb_logger_result_retrieved� s
z+FutureTests.test_tb_logger_result_retrievedc C sR t j d | j � } | j t d � � ~ t j | j � | j | j j � d S)Nr �boom)
r r r r% r) r � run_brieflyr rV rW )r rX rQ r r r �$test_tb_logger_exception_unretrieved� s
z0FutureTests.test_tb_logger_exception_unretrievedc C sL t j d | j � } | j t d � � | j � ~ | j | j j � d S)Nr r\ ) r r r r% r) r" r rV rW )r rX rQ 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 r\ )
r r r r% r) r r! r rV rW )r rX rQ r r r �)test_tb_logger_exception_result_retrieved� 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 �run s z)FutureTests.test_wrap_future.<locals>.runr7 �oir )�
concurrent�futures�ThreadPoolExecutor�submitr � wrap_futurer Zrun_until_completer, r r' ZassertNotEqualra rb )r rd �ex�f1�f2�resZidentr r r �test_wrap_future s z FutureTests.test_wrap_futurec C s8 t j d | j � } t j | � } | j | | � d S)Nr )r r r rj r )r rl rm r r r �test_wrap_future_future s z#FutureTests.test_wrap_future_futurezasyncio.futures.eventsc C s\ d d � } t j j d � } | j | d � } t j | � } | j | j j | j � d S)Nc S s | t j � f S)N)ra rb )rc r r r rd s z9FutureTests.test_wrap_future_use_global_loop.<locals>.runr7 re )
rf rg rh ri r rj r Zget_event_loopZ return_valuer )r Zm_eventsrd rk rl rm 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 ) rf rg r r rj r r r r] r r )r rl rm 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 r )rf rg r r rj r r$ r r r] r r r' r! r )r rl rm 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 r7 � � test_future_source_tracebackrB ) r rC r r �sys� _getframe�f_linenor, rD �listr' �__file__)r �futurer> r r r ru / s z(FutureTests.test_future_source_tracebackc C s� | j j | � d d � } | � } t j d | j � } | rL | j } n | 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[ | r:| 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)r* r r r � memory_error> s
zHFutureTests.check_future_exception_never_retrieved.<locals>.memory_errorr rt r8 r7 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\)$r= r r> 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)rt r8 rB rB )r rC r r rD r% r r] r Z
gc_collectrv � version_info�formatr: r; �type�
__traceback__rV Zassert_called_once_withr �ANY� call_argsrG �compile�DOTALL) r �debugrX r~ r* r{ Zsource_tracebackrI 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 )rg r � )r rg r r r Z _set_result_unless_cancelledr r )r rg rQ r r r � test_set_result_unless_cancelled� s
z,FutureTests.test_set_result_unless_cancelledN)!�__name__�
__module__r9 r r r r r& r( r+ r- r6 rM rP rU r ZpatchrY rZ r[ r^ r_ r` ro rp rq rr rs ru r� r� r� r� r r r r r s8
A
B r c @ sd e 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 S)�FutureDoneCallbackTestsc C s | j � | _ d S)N)r
r )r r r r r � s z FutureDoneCallbackTests.setUpc C s t j | j � d S)N)r r] r )r r r r r] � 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 r 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)Nr � �foo)r� rF r� r' r$ r] 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� rF r� r' r) r% r] r" )r r� r r* 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)Nr7 r� rt r � � r� ) r� r� rF r' Zremove_done_callbackrH r$ r] r! )r r� r Zcb1Zcb2Zcb3rL r r r �test_remove_done_callback� s$
z1FutureDoneCallbackTests.test_remove_done_callbackN)
r� r� r9 r r] r� r� r� r� r� r r r r r� � s r� �__main__)�__doc__Zconcurrent.futuresrf r: rv ra Zunittestr r r rS r � ImportErrorr r r
r ZTestCaser r� r� �mainr r r r �<module> s(
� oO