File: //lib/python3/dist-packages/twisted/test/__pycache__/test_task.cpython-38.pyc
U
W[�� � @ s� d Z ddlmZmZ ddlmZ ddlmZmZm Z m
Z
m Z ddl m
Z
ddlmZ ejZddlmZ G dd � d ej�ZG d
d � d e�ZG d d
� d
ej�ZG dd� dej�ZG dd� dej�ZG dd� dej�ZG dd� de �Z G dd� dej �Z dS )z%
Tests for L{twisted.internet.task}.
� )�division�absolute_import)�unittest)�
interfaces�task�reactor�defer�error)�installReactor)� NoReactor)�failurec s e Zd Z� fdd�Z� ZS )�TestableLoopingCallc s t t| �j||� || _d S �N)�superr
�__init__�clock)�selfr �a�kw�� __class__� �8/usr/lib/python3/dist-packages/twisted/test/test_task.pyr s z TestableLoopingCall.__init__)�__name__�
__module__� __qualname__r �
__classcell__r r r r r
s r
c @ s e Zd ZdS )�
TestExceptionN�r r r r r r r r s r c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd S )!�
ClockTestsz<
Test the non-wallclock based clock implementation.
c C s t �� }| �|�� d� dS )zV
Test that the C{seconds} method of the fake clock returns fake time.
r N)r �Clock� assertEqual�seconds�r �cr r r � testSeconds& s zClockTests.testSecondsc C sR t �� }|jddd� ddd�}| �tj�|�� | �|�� d� | �|� � � dS )z{
Test that calls can be scheduled for later with the fake clock and
hands back an L{IDelayedCall}.
� c S s d S r r �r �br r r �<lambda>4 � z*ClockTests.testCallLater.<locals>.<lambda>� �r( N)
r r � callLater�
assertTruer Z IDelayedCall�
providedByr! �getTime�active�r r$ �callr r r �
testCallLater. s
zClockTests.testCallLaterc C s8 t �� }|jddd� ddd�}|�� | �|�� � dS )z3
Test that calls can be cancelled.
r& c S s d S r r r' r r r r) ? r* z3ClockTests.testCallLaterCancelled.<locals>.<lambda>r+ r, N)r r r- �cancel� assertFalser1 r2 r r r �testCallLaterCancelled: s z!ClockTests.testCallLaterCancelledc C sF t �� }|jddd� ddd�}|jddd� dd d�}| �||k� d
S ) z[
Test that the DelayedCall returned is not one previously
created.
�
c S s d S r r r' r r r r) J r* z3ClockTests.test_callLaterOrdering.<locals>.<lambda>r& r+ r, c S s d S r r r' r r r r) K r* � � N)r r r- r6 )r r$ Zcall1�call2r r r �test_callLaterOrderingD s z!ClockTests.test_callLaterOrderingc s` g � t �� }|�d� fdd��}|�d� | �� g � |�d� | �� dg� | �|�� � dS )zE
Test that advancing the clock will fire some calls.
r+ c s
� � d �S r ��appendr �Zeventsr r r) U r* z(ClockTests.testAdvance.<locals>.<lambda>r& N)r r r- �advancer! r6 r1 r2 r r? r � testAdvanceO s
zClockTests.testAdvancec s0 t �� }� �fdd�}|�d|�� |�d� dS )a
Test attempting to cancel the call in a callback.
AlreadyCalled should be raised, not for example a ValueError from
removing the call from Clock.calls. This requires call.called to be
set before the callback is called.
c s �� tj� j� d S r )� assertRaisesr Z
AlreadyCalledr5 r �r3 r r r �cbf s z(ClockTests.testAdvanceCancel.<locals>.cbr& N)r r r- r@ )r r$ rD r rC r �testAdvanceCancel] s z ClockTests.testAdvanceCancelc sr g � t �� }|jd� fdd�ddd�}|�d� | �|�� d� |�d� | �� g � |�d� | �� dg� d S )
z1
Test that calls can be delayed.
r& c s � � | |f�S r r= r' r? r r r) r r* z1ClockTests.testCallLaterDelayed.<locals>.<lambda>r+ r, g �?� �?�r& r+ N)r r r- �delayr! r0 r@ r2 r r? r �testCallLaterDelayedl s
z ClockTests.testCallLaterDelayedc s| g � t �� }|jd� fdd�ddd�}|�d� |�d� | �|�� d� |�d� | �� g � |�d� | �� dg� d S )
zL
Test that calls can have their time reset to a later time.
r+ c s � � | |f�S r r= r' r? r r r) � r* z4ClockTests.testCallLaterResetLater.<locals>.<lambda>r& r, r9 r: rG N)r r r- r@ �resetr! r0 r2 r r? r �testCallLaterResetLater{ s
z"ClockTests.testCallLaterResetLaterc s\ g � t �� }|jd� fdd�ddd�}|�d� | �|�� d� |�d� | �� dg� d S )
zO
Test that calls can have their time reset to an earlier time.
r: c s � � | |f�S r r= r' r? r r r) � r* z5ClockTests.testCallLaterResetSooner.<locals>.<lambda>r& r+ r, r9 rG N)r r r- rJ r! r0 r@ r2 r r? r �testCallLaterResetSooner� s
z#ClockTests.testCallLaterResetSoonerc C sL t �� }|�ddd� �}|�ddd� �}|�� }| �t||g�t|�� dS )zB
Test that we can get a list of all delayed calls
r& c S s d S r r ��xr r r r) � r* z1ClockTests.test_getDelayedCalls.<locals>.<lambda>r+ c S s d S r r rM r r r r) � r* N)r r r- �getDelayedCallsr! �set)r r$ r3 r; �callsr r r �test_getDelayedCalls� s
z ClockTests.test_getDelayedCallsc C s t �� }| �|�� g � dS )zk
Test that we get an empty list from getDelayedCalls on a newly
constructed Clock.
N)r r r! rO r# r r r �test_getDelayedCallsEmpty� s z$ClockTests.test_getDelayedCallsEmptyc C s t �� }| �tj�|�d� d S )Nz#Clock does not provide IReactorTime)r r r. r Z IReactorTimer/ r# r r r �test_providesIReactorTime� s �z$ClockTests.test_providesIReactorTimec sh g �ddg}t �� � � �fdd�}� �d|d�}|�d� � �d|d � � �d
gd � | ��|� d S )
a4
The order of calls scheduled by L{task.Clock.callLater} is honored when
adding a new call via calling L{task.Clock.callLater} again.
For example, if L{task.Clock.callLater} is invoked with a callable "A"
and a time t0, and then the L{IDelayedCall} which results from that is
C{reset} to a later time t2 which is greater than t0, and I{then}
L{task.Clock.callLater} is invoked again with a callable "B", and time
t1 which is less than t2 but greater than t0, "B" will be invoked before
"A".
�r( � @�r � @c s �� | � �� f�S r �r> r" ��n�r �resultr r r) � r* z<ClockTests.test_callLaterKeepsCallsOrdered.<locals>.<lambda>rF r rX rV r( r& r9 N�r r r- rJ �pumpr! �r �expected�logtimeZcall_ar r\ r � test_callLaterKeepsCallsOrdered� s
z*ClockTests.test_callLaterKeepsCallsOrderedc sh g �ddg}t �� � � �fdd�}� �d|d�}� �d|d� |�d � � �d
gd � | ��|� d S )
a:
The order of calls scheduled by L{task.Clock.callLater} is honored when
re-scheduling an existing call via L{IDelayedCall.reset} on the result
of a previous call to C{callLater}.
For example, if L{task.Clock.callLater} is invoked with a callable "A"
and a time t0, and then L{task.Clock.callLater} is invoked again with a
callable "B", and time t1 greater than t0, and finally the
L{IDelayedCall} for "A" is C{reset} to a later time, t2, which is
greater than t1, "B" will be invoked before "A".
rU rW c s �� | � �� f�S r rY rZ r\ r r r) � r* zAClockTests.test_callLaterResetKeepsCallsOrdered.<locals>.<lambda>rF r rV r( rX r& r9 Nr^ r` r r\ r �$test_callLaterResetKeepsCallsOrdered� s
z/ClockTests.test_callLaterResetKeepsCallsOrderedc sv g �ddg}t �� ���fdd�}��d|d�� � fdd�}��d |� ��d
|d � ��d gd
� | ��|� dS )a9
The order of calls scheduled by L{task.Clock.callLater} is honored when
re-scheduling an existing call via L{IDelayedCall.reset} on the result
of a previous call to C{callLater}, even when that call to C{reset}
occurs within the callable scheduled by C{callLater} itself.
)r$ rX )r( g @c s �� | � �� f�S r rY rZ r\ r r r) � r* zKClockTests.test_callLaterResetInsideCallKeepsCallsOrdered.<locals>.<lambda>rV r( c s � � d� d S )NrX )rJ r )�call_br r r � s zDClockTests.test_callLaterResetInsideCallKeepsCallsOrdered.<locals>.arF rX r$ � �?r8 N)r r r- r_ r! )r ra rb r r )re r r] r �.test_callLaterResetInsideCallKeepsCallsOrdered� s z9ClockTests.test_callLaterResetInsideCallKeepsCallsOrderedN)r r r �__doc__r% r4 r7 r< rA rE rI rK rL rR rS rT rc rd rg r r r r r " s
r c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(S ))� LoopTestsz[
Tests for L{task.LoopingCall} based on a fake L{IReactorTime}
implementation.
c C s t �dd� �}| �|jt� dS )zG
L{LoopingCall}'s default clock should be the reactor.
c S s d S r r r r r r r) r* z-LoopTests.test_defaultClock.<locals>.<lambda>N)r � LoopingCallr! r r )r r3 r r r �test_defaultClock s zLoopTests.test_defaultClockc s� g �d� t �� �� ��fdd�}t �|�}�|_d� |�d� | ��dg� | ���� � � ��d� | ��dg� d� ��d� | ��ddg� | ���� d � ��d� | ��ddg� d� ��d� | ��ddd
g� | ���� d
� dS ) z�
When more time than the defined interval passes during the execution
of a callback, L{LoopingCall} should schedule the next call for the
next interval which is still in the future.
Nc s �� ��� � ��� � d S r )r> r" r@ r �Z callDurationr �timesr r � aCallback s z3LoopTests.test_callbackTimeSkips.<locals>.aCallbackr+ rf r r& g @g @r: )r r rj r �startr! r" r@ �r rn r3 r rl r �test_callbackTimeSkips s,
z LoopTests.test_callbackTimeSkipsc s� g �t �� � � �fdd�}t �|�}� |_|�d� | ��dg� � �d� | ��ddg� � �d� | ��dddg� � �d� | ��dddg� dS ) a
When more time than the defined interval passes between when
L{LoopingCall} schedules itself to run again and when it actually
runs again, it should schedule the next call for the next interval
which is still in the future.
c s �� � �� � d S r rY r �r rm r r rn G s z2LoopTests.test_reactorTimeSkips.<locals>.aCallbackrf r r+ r& r9 N)r r rj r ro r! r@ rp r rr r �test_reactorTimeSkips> s
z LoopTests.test_reactorTimeSkipsc s� g �t �� � � �fdd�}t j�|�}� |_d}d}|�|� | ��dg� � �|| � | ��d|| dfg� � �d| | � | ��d|| dfd| d | dfg� � �d
� | ��d|| dfd| d | dfg� d S ) a�
When L{LoopingCall} schedules itself to run again, if more than the
specified interval has passed, it should schedule the next call for the
next interval which is still in the future. If it was created
using L{LoopingCall.withCount}, a positional argument will be
inserted at the beginning of the argument list, indicating the number
of calls that should have been made.
c s �� � �� | f� d S r rY )ZnumCallsrr r r rn k s z7LoopTests.test_reactorTimeCountSkips.<locals>.aCallbackrf �{�G�z�?�r r& r& r9 r: r+ r N)r r rj � withCountr ro r! r@ �r rn r3 ZINTERVALZREALISTIC_DELAYr rr r �test_reactorTimeCountSkips` s.
��
��z$LoopTests.test_reactorTimeCountSkipsc s� g �t �� � � �fdd�}t j�|�}� |_d}d}|�|� | ���� d� � �|| � | ���� || df� � �|d � | ���� d| | df� � �|d � | ���� d
| | df� d S ) a1
L{LoopingCall.withCount} counts only calls that were expected to be
made. So, if more than one, but less than two intervals pass between
invocations, it won't increase the count above 1. For example, a
L{LoopingCall} with interval T expects to be invoked at T, 2T, 3T, etc.
However, the reactor takes some time to get around to calling it, so in
practice it will be called at T+something, 2T+something, 3T+something;
and due to other things going on in the reactor, "something" is
variable. It won't increase the count unless "something" is greater
than T. So if the L{LoopingCall} is invoked at T, 2.75T, and 3T,
the count has not increased, even though the distance between
invocation 1 and invocation 2 is 1.75T.
c s �� � �� | f� d S r rY )�countrr r r rn � s z<LoopTests.test_countLengthyIntervalCounts.<locals>.aCallbackrf rt ru r& g �?g @g �?rX N) r r rj rv r ro r! �popr@ rw r rr r � test_countLengthyIntervalCounts� s
z)LoopTests.test_countLengthyIntervalCountsc
C s� t �� }g }t j�|j�}||_d}d}|| }|j|dd� t|�D ]}|�|� qF|t |g| � }|�|� |�
� } | � t |�dkd�
|�� | � | |kd�
| |�� | �t |�|� | �d|� d S )
a�
L{task.LoopingCall.withCount} should never invoke its callable with a
zero. Specifically, if a L{task.LoopingCall} created with C{withCount}
has its L{start <task.LoopingCall.start>} method invoked with a
floating-point number which introduces decimal inaccuracy when
multiplied or divided, such as "0.1", L{task.LoopingCall} will never
invoke its callable with 0. Also, the sum of all the values passed to
its callable as the "count" will be an integer, the number of intervals
that have elapsed.
This is a regression test for a particularly tricky case to implement.
r8 rF F�Znowg z {0} should be greater than zeroz*{0} should be greater than or equal to {1}r N)r r rj rv r> r ro �ranger@ �sumr" r. �abs�formatr! Z assertNotIn)
r r � accumulatorr3 ry Ztimespan�intervalrN �epsilonZ secondsValuer r r �#test_withCountFloatingPointBoundary� s4
��
��z-LoopTests.test_withCountFloatingPointBoundaryc sh t �� }g � � �fdd�}t j�|��|�_�jddd�}|�d� | �|� | �dddddg� � dS )zm
L{task.LoopingCall.withCount} with interval set to 0
will call the countCallable 1.
c s" � � | � t� �dkr ��� d S )Nr: �r> �len�stop�Zcnt�r� �loopr r �foo� s
z1LoopTests.test_withCountIntervalZero.<locals>.foor Fr| r& N) r r rj rv r ro r@ ZsuccessResultOfr! )r r r� �deferredr r� r �test_withCountIntervalZero� s
z$LoopTests.test_withCountIntervalZeroc s� t �� }t�� �g � � ��fdd�}t j�|��|�_�jddd� |�d� | � ddg� � |�d� | � ddg� � ��
d� |�d� | � dddddg� � dS ) z�
L{task.LoopingCall.withCount} with interval set to 0 and a delayed
call during the loop run will still call the countCallable 1 as if
no delay occurred.
c s2 � � | � t� �dkr�S t� �dkr.��� d S )Nr+ r: r� r� �r� r� r� r r r� s
z6LoopTests.test_withCountIntervalZeroDelay.<locals>.foor Fr| r& r+ N) r r r �Deferredrj rv r ro r@ r! �callback)r r r� r r� r � test_withCountIntervalZeroDelay s
z)LoopTests.test_withCountIntervalZeroDelayc s� t �� }t�� �g � � �fdd�}t j�|�}||_|jddd� |�d� | � ddg� � d|_
�� d � |�d
� | � ddd g� � |�d� | � ddd dg� � |�d � | � ddd ddg� � d S )
z�
L{task.LoopingCall.withCount} with interval set to 0 will still keep
the time when last called so when the interval is reset.
c s � � | � t� �dkr�S d S )Nr+ �r> r� r� �r� r� r r r� 7 s
zILoopTests.test_withCountIntervalZeroDelayThenNonZeroInterval.<locals>.foor Fr| r8 r& r+ N� r9 r: ) r r r r� rj rv r ro r@ r! r� r� )r r r� r� r r� r �2test_withCountIntervalZeroDelayThenNonZeroInterval. s"
z<LoopTests.test_withCountIntervalZeroDelayThenNonZeroIntervalc s� dddg}t �� }g � d� fdd� }t||dddd�}|�d�}g ��fd d
�}|�|� |�|� | �t� �d d t� �f � � D ]:\}}} }
| �|d� | �|d� | �| � | �|
d� q�|� � | �
�d
|� | � |j � d S )N皙�����?皙�����?c s � � | |||f� d S r r= )r r( r$ �d)�Lr r r� ^ s z(LoopTests.testBasicFunction.<locals>.foor r( r� �r� c s � � | � d S r r= �r] �� theResultr r �
saveResulte s z/LoopTests.testBasicFunction.<locals>.saveResultr9 zgot %d iterations, not 3r )NN)
r r r
ro � addCallbackr_ r! r� Z assertIsNoner� �assertIsr6 rQ ) r �timingsr r� �lc�Dr� r r( r$ r� r )r� r� r �testBasicFunctionT s*
�
zLoopTests.testBasicFunctionc s� dddg}t �� }g }t||jd �}|jddd�}g � � fdd�}|�|� |�|� | �t|�ddt|�f � |� � | �
� d |� | � |j � d S )
Nr� r� Fr| c s � � | � d S r r= r� r� r r r� � s z.LoopTests.testDelayedStart.<locals>.saveResultr+ zgot %d iterations, not 2r )
r r r
r> ro r� r_ r! r� r� r� r6 rQ )r r� r r� r� r� r� r r� r �testDelayedStart{ s
�zLoopTests.testDelayedStartc C s" t �dd� �}| �t|jd� d S )Nc S s d S r r r r r r r) � r* z(LoopTests.testBadDelay.<locals>.<lambda>���)r rj rB �
ValueErrorro )r r� r r r � testBadDelay� s zLoopTests.testBadDelayc sR g � � fdd�}t �� }t||�}|j|dd� |�� | �� � | �|j� d S )Nc s � � d � d S r r= r ��ranr r r� � s z$LoopTests._stoppingTest.<locals>.fooFr| )r r r
ro r� r6 rQ )r rH r� r r� r r� r �
_stoppingTest� s
zLoopTests._stoppingTestc C s
| � d�S )Nr �r� �r r r r �testStopAtOnce� s zLoopTests.testStopAtOncec C s
| � d�S )Nr8 r� r� r r r � testStoppingBeforeDelayedStart� s z(LoopTests.testStoppingBeforeDelayedStartc st g � � fdd�}t �� }t||�}|jddd� |�d� |�� |�d� | �� g � |�d� | �� dg� dS )z8
Test that L{LoopingCall} can be reset.
c s � � d � d S r r= r r� r r r� � s z!LoopTests.test_reset.<locals>.foor+ Fr| r& N)r r r
ro r@ rJ r! )r r� r$ r� r r� r �
test_reset� s
zLoopTests.test_resetc C s | � ttjtddd��d� dS )zO
L{LoopingCall.__repr__} includes the wrapped function's name.
r& r+ )�keyz6LoopingCall<None>(installReactor, *(1,), **{'key': 2})N)r! �reprr rj r
r� r r r �test_reprFunction� s �zLoopTests.test_reprFunctionc C s | � tt�tj��d� dS )zR
L{LoopingCall.__repr__} includes the wrapped method's full name.
z:LoopingCall<None>(TestableLoopingCall.__init__, *(), **{})N)r! r� r rj r
r r� r r r �test_reprMethod� s �zLoopTests.test_reprMethodc C sb t �dd� �}|j d}| �| jg�}| �dt|�� | �t|d d � | �||d d � dS ) z8
L{LoopingCall.deferred} is deprecated.
c S s d S r r r r r r r) � r* z4LoopTests.test_deferredDeprecation.<locals>.<lambda>z�twisted.internet.task.LoopingCall.deferred was deprecated in Twisted 16.0.0; please use the deferred returned by start() insteadr& r �category�messageN)r rj r� Z
flushWarnings�test_deferredDeprecationr! r� �DeprecationWarning)r r� r� �warningsr r r r� � s �z"LoopTests.test_deferredDeprecationN)r r r rh rk rq rs rx r{ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r ri � s( 3"0)8&&'
ri c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�ReactorLoopTestsc C s& dd� }t �|d�}| �|�d�t�S )Nc S s t | ��d S r �r rM r r r r� � s z)ReactorLoopTests.testFailure.<locals>.foo�barr� �r rj �
assertFailurero r )r r� r� r r r � testFailure� s z ReactorLoopTests.testFailurec s* � fdd�}t �|d�� | �� �d�t�S )Nc s � � � t| ��d S r )r� r rM �r� r r r� � s z-ReactorLoopTests.testFailAndStop.<locals>.foor� r� r� �r r� r r� r �testFailAndStop� s z ReactorLoopTests.testFailAndStopc s> g �� �fdd�}t �|�� � �d�}��fdd�}|�|�S )Nc s" �� d � t��dkr � �� d S )N� r� r )r� r� r r r� � s
z0ReactorLoopTests.testEveryIteration.<locals>.foor c s �� t� �d� d S )N� )r! r� )Zign)r� r r r �stopped s z4ReactorLoopTests.testEveryIteration.<locals>.stopped)r rj ro r� )r r� r� r� r )r� r� r r �testEveryIteration� s
z#ReactorLoopTests.testEveryIterationc sD t �� � � fdd�}t�|�| _| jjddd� t�d| j� � � S )Nc s � � t�d�� d S )Nz'This task also should never get called.)�errbackr ZDefaultExceptionr r� r r r�
s �z1ReactorLoopTests.testStopAtOnceLater.<locals>.foor& Fr| r ) r r� r rj �_lcro r r- �!_callback_for_testStopAtOnceLaterr� r r� r �testStopAtOnceLater s z$ReactorLoopTests.testStopAtOnceLaterc C s | j �� t�d|jd� d S )Nr Zsuccess)r� r� r r- r� )r r� r r r r� s
z2ReactorLoopTests._callback_for_testStopAtOnceLaterc sL ddg}t �� � � �fdd�}t� |����d� � �|� | �� j� d S )N皙�����?g�������?c s. t �� } | ��fdd�� � �d| jd � | S )Nc s � � � S r )r� )�_r� r r r) " r* z@ReactorLoopTests.testWaitDeferred.<locals>.foo.<locals>.<lambda>r& )r r� r� r- r� r� �r r� r r r� s z.ReactorLoopTests.testWaitDeferred.<locals>.foo)r r r
ro r_ r6 rQ )r r� r� r r� r �testWaitDeferred s
z!ReactorLoopTests.testWaitDeferredc sT dg}t �� � � fdd�}t� |�}|�d�}| �|t� � �|� | �� j� |S )N�333333�?c s t �� } � �d| jt� � | S )Nr� )r r� r- r� r r� �r r r r� 5 s z4ReactorLoopTests.testFailurePropagation.<locals>.foor& ) r r r
ro r� r r_ r6 rQ )r r� r� r� r� r r� r �testFailurePropagation+ s
z'ReactorLoopTests.testFailurePropagationc s� t �� }t�� � g �� �fdd�}t j�|�}||_|jddd�� | ��g � |� ddg� | �t
��d� � � d� |� dg� | �t
��d � | ��dd
g� dS ) a?
In the case that the function passed to L{LoopingCall.withCount}
returns a deferred, which does not fire before the next interval
elapses, the function should not be run again. And if a function call
is skipped in this fashion, the appropriate count should be
provided.
c s" �� | � t��dkr� S d S d S �Nr& r� )Z
possibleCount�r� ZdeferredCountsr r � countTrackerO s
z=ReactorLoopTests.test_deferredWithCount.<locals>.countTrackerr� Fr| g�������?r& Nr+ r9 ) r r r r� rj rv r ro r! r_ r� r� )r Z testClockr� r� r r� r �test_deferredWithCountC s
z'ReactorLoopTests.test_deferredWithCountN) r r r r� r� r� r� r� r� r� r� r r r r r� � s
r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �DeferLaterTestsz'
Tests for L{task.deferLater}.
c st g �t � � � �fdd�}t�� }tj|d|ddd�}|�| j� � |�d� | ��g � |�d� | ��d g� |S )
z�
The L{Deferred} returned by L{task.deferLater} is called back after
the specified delay with the result of the function passed in.
c s �� | |f� � S r r= �r� r� ��flagZresultsr r �callable| s z/DeferLaterTests.test_callback.<locals>.callabler9 r� r� )r� r+ r& r� )�objectr r �
deferLaterr� r� r@ r! �r r� r r� r r� r �
test_callbacku s
z DeferLaterTests.test_callbackc C s4 dd� }t �� }t �|d|�}|�d� | �|t�S )z�
The L{Deferred} returned by L{task.deferLater} is errbacked if the
supplied function raises an exception.
c S s
t � �d S r r� r r r r r� � s z.DeferLaterTests.test_errback.<locals>.callabler& )r r r� r@ r� r r� r r r � test_errback� s
z DeferLaterTests.test_errbackc sR g � t �� �t ��d� jd�}|�� � ��fdd�}��|tj� |�|� |S )z�
The L{Deferred} returned by L{task.deferLater} can be
cancelled to prevent the call from actually being performed.
r& Nc s �� g ��� � ��� � d S r )r! rO r6 )Zignored�Zcalledr r r r � cbCancelled� s z0DeferLaterTests.test_cancel.<locals>.cbCancelled) r r r� r> r5 r� r ZCancelledErrorr� )r r� r� r r� r � test_cancel� s
zDeferLaterTests.test_cancelN)r r r rh r� r� r� r r r r r� q s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) � _FakeReactorc C sD d| _ t�� | _| jj| _| jj| _| jj| _g | _g g d�| _d S )NF��before�during) �_runningr r �_clockr- r" rO � _whenRunning�_shutdownTriggersr� r r r r � s
z_FakeReactor.__init__c O s, | j d kr|||� n| j �|||f� d S r )r� r> )r r� �args�kwargsr r r �callWhenRunning� s
z _FakeReactor.callWhenRunningc G s0 |dks t �|dkst �| j| �||f� d S )Nr� �shutdown)�AssertionErrorr� r> )r ZphaseZeventr� r� r r r �addSystemEventTrigger� s z"_FakeReactor.addSystemEventTriggerc C s� d| _ | j}d| _|D ]\}}}|||� q| j rd| �� }|sFtd��| j�|d �� | �� � q,| j}d| _|d |d D ]\}}||� q�dS )z�
Call timed events until there are no more or the reactor is stopped.
@raise RuntimeError: When no timed events are left and the reactor is
still running.
TNzNo DelayedCalls leftr r� r� ) r� r� rO � RuntimeErrorr� r@ r0 r" r� )r Z whenRunningr� r� r� rQ ZshutdownTriggersZtriggerr r r �run� s z_FakeReactor.runc C s | j st�� �d| _ dS )z#
Stop the reactor.
FN)r� r ZReactorNotRunningr� r r r r� � s z_FakeReactor.stopN)r r r r r� r� r� r� r r r r r� � s
r� c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�
ReactTestsz3
Tests for L{twisted.internet.task.react}.
c sZ g � � fdd�}t � }| jttj||d�}| �d|j� | �� dg� | �|�� d� dS )z�
L{task.react} runs the reactor until the L{Deferred} returned by the
function it is passed is called back, then stops it.
c s, t �� }| �d� jd� | �d|jd � |S )Nr& Tr+ )r r� r- r> r� �r Zfinished�Z
timePassedr r �main� s z4ReactTests.test_runsUntilAsyncCallback.<locals>.main�Z_reactorr Tr+ N�r� rB �
SystemExitr �reactr! �coder" �r r� �r� exitErrorr r� r �test_runsUntilAsyncCallback� s �z&ReactTests.test_runsUntilAsyncCallbackc C sD dd� }t � }| jttj||d�}| �d|j� | �|�� d� dS )z�
L{task.react} returns quickly if the L{Deferred} returned by the
function it is passed has already been called back at the time it is
returned.
c S s
t �d �S r )r �succeed�r r r r r� s z3ReactTests.test_runsUntilSyncCallback.<locals>.mainr� r Nr� r� r r r �test_runsUntilSyncCallback� s �z%ReactTests.test_runsUntilSyncCallbackc sb G dd� dt �� � fdd�}t� }| jttj||d�}| �d|j� | �� �}| �t |�d� dS )z�
L{task.react} runs the reactor until the L{defer.Deferred} returned by
the function it is passed is errbacked, then it stops the reactor and
reports the error.
c @ s e Zd ZdS )z@ReactTests.test_runsUntilAsyncErrback.<locals>.ExpectedExceptionNr r r r r �ExpectedException s r c s t �� }| �d|j� � � |S r� )r r� r- r� r� �r r r r� s z3ReactTests.test_runsUntilAsyncErrback.<locals>.mainr� r& N)
� Exceptionr� rB r� r r� r! r� �flushLoggedErrorsr� �r r� r� r� �errorsr r r �test_runsUntilAsyncErrback
s �
z%ReactTests.test_runsUntilAsyncErrbackc sr G dd� dt �� � fdd�}t� }| jttj||d�}| �d|j� | �|�� d� | � � �}| �t
|�d� dS ) z�
L{task.react} returns quickly if the L{defer.Deferred} returned by the
function it is passed has already been errbacked at the time it is
returned.
c @ s e Zd ZdS )z?ReactTests.test_runsUntilSyncErrback.<locals>.ExpectedExceptionNr r r r r r * s r c s t �� � �S r )r �failr r r r r� - s z2ReactTests.test_runsUntilSyncErrback.<locals>.mainr� r& r N� r r� rB r� r r� r! r� r" r r� r r r r �test_runsUntilSyncErrback$ s �
z$ReactTests.test_runsUntilSyncErrbackc C sD dd� }t � }| jttj||d�}| �|�� d� | �d|j� dS )z�
L{task.react} doesn't try to stop the reactor if the L{defer.Deferred}
the function it is passed is called back after the reactor has already
been stopped.
c S s, | � d| j� t�� }| �dd|jd � |S �Nr& r� r� )r- r� r r� r� r� r� r r r r� > s �z0ReactTests.test_singleStopCallback.<locals>.mainr� r& r N)r� rB r� r r� r! r" r� r� r r r �test_singleStopCallback8 s �z"ReactTests.test_singleStopCallbackc sr G dd� dt �� � fdd�}t� }| jttj||d�}| �d|j� | �|�� d� | � � �}| �t
|�d� dS )z�
L{task.react} doesn't try to stop the reactor if the L{defer.Deferred}
the function it is passed is errbacked after the reactor has already
been stopped.
c @ s e Zd ZdS )z<ReactTests.test_singleStopErrback.<locals>.ExpectedExceptionNr r r r r r R s r c s. | � d| j� t�� }| �dd|j� � � |S r
)r- r� r r� r� r� r� r r r r� U s �z/ReactTests.test_singleStopErrback.<locals>.mainr� r& Nr r r r r �test_singleStopErrbackL s �
z!ReactTests.test_singleStopErrbackc sV g � � fdd�}t � }| jttj|dddg|d�}| �d|j� | �� dddg� dS ) z�
L{task.react} passes the elements of the list it is passed as
positional arguments to the function it is passed.
c s � � |||f� t�d �S r )�extendr r )r rN �y�z�r� r r r� l s z'ReactTests.test_arguments.<locals>.mainr& r+ r9 r� r N�r� rB r� r r� r! r� r� r r r �test_argumentsf s �zReactTests.test_argumentsc s^ � fdd�}t � }t� �. t|� � �ttj|g �}� �d|j� W 5 Q R X � � |� j
� dS )zp
L{twisted.internet.reactor} is used if no reactor argument is passed to
L{task.react}.
c s | � _ t�d �S r )�
passedReactorr r r r� r r r� { s z,ReactTests.test_defaultReactor.<locals>.mainr N) r� r r
rB r� r r� r! r� r� r )r r� r r� r r� r �test_defaultReactorv s z ReactTests.test_defaultReactorc C s6 dd� }t � }| jttj|g |d�}| �d|j� dS )z�
L{task.react} forwards the exit code specified by the C{SystemExit}
error returned by the passed function, if any.
c S s t �td��S )N� )r r
r� r r r r r� � s z1ReactTests.test_exitWithDefinedCode.<locals>.mainr� r Nr r� r r r �test_exitWithDefinedCode� s �z#ReactTests.test_exitWithDefinedCodec C s6 dd� }t � }| jttj|g |d�}| �d|j� dS )zw
L{task.react} handles when the reactor is stopped just before the
returned L{Deferred} fires.
c s$ t �� � � �fdd�}��|� � S )Nc s �� � � �d � d S r )r� r� r �r� r r r r� � s z;ReactTests.test_synchronousStop.<locals>.main.<locals>.stop)r r� r� )r r� r r r r� � s
z-ReactTests.test_synchronousStop.<locals>.mainr� r Nr r� r r r �test_synchronousStop� s �z ReactTests.test_synchronousStopc C s6 dd� }t � }| jttj|g |d�}| �d|j� dS )zv
L{task.react} handles when the reactor is stopped and the
returned L{Deferred} doesn't fire.
c S s | � d| j� t�� S r� )r- r� r r� r r r r r� � s z.ReactTests.test_asynchronousStop.<locals>.mainr� r Nr r� r r r �test_asynchronousStop� s �z ReactTests.test_asynchronousStopN)r r r rh r� r r r r r r r r r r r r r r r� � s
r� N) rh Z
__future__r r Z
twisted.trialr Ztwisted.internetr r r r r Ztwisted.internet.mainr
Z#twisted.internet.test.modulehelpersr r Ztwisted.pythonr rj r
r r ZTestCaser ri r� r� r� r� ZSynchronousTestCaser� r r r r �<module> s( ] k ;: