← Back
Editing: 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 ;:
Save File
Cancel