← Back
Editing: test_internet.cpython-38.pyc
U ��W[V� � @ s d Z ddlmZmZ ddlZddlZddlZddlmZ ddl m Z ddl m Z m Z mZmZmZ ddl mZmZ zddl mZ W n ek r� dZY nX er�ejs�dZdd lmZmZ es�dd lmZ G d d � d e j�Z G d d� de j�Z G dd� de j�Z G dd� de j�Z G dd� de j�Z dZ!G dd� de j"�Z#G dd� de j�Z$e�%e d��s^de$_&G dd � d e j�Z'G d d � d e j(�Z)G d d � d e j*�Z+G d!d"� d"e j�Z,G d#d$� d$e-�Z.G d%d&� d&ej/�Z0G d'd(� d(e.�Z1G d)d*� d*e j�Z2G d+d,� d,e j�Z3dS )-zB Tests for lots of functionality provided by L{twisted.internet}. � )�division�absolute_importN)�_PY3)�unittest)�reactor�protocol�error�abstract�defer)� interfaces�base)�ssl)�Deferred�passthru)�utilc @ 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(d)� Zd*d+� Zd,S )-�ThreePhaseEventTestszQ Tests for the private implementation helpers for system event triggers. c C s dd� | _ t� | _t�� | _dS )zR Create a trigger, an argument, and an event to be used by tests. c S s d S �N� ��xr r �</usr/lib/python3/dist-packages/twisted/test/test_internet.py�<lambda>' � z,ThreePhaseEventTests.setUp.<locals>.<lambda>N)�trigger�object�argr Z_ThreePhaseEvent�event��selfr r r �setUp# s zThreePhaseEventTests.setUpc C s | � t| jjd| j| j� dS )zt L{_ThreePhaseEvent.addTrigger} should raise L{KeyError} when called with an invalid phase. �xxxN)� assertRaises�KeyErrorr � addTriggerr r r r r r �test_addInvalidPhase, s �z)ThreePhaseEventTests.test_addInvalidPhasec C s6 | j �d| j| j� | �| j j| j| jfi fg� dS )z� L{_ThreePhaseEvent.addTrigger} should accept C{'before'} as a phase, a callable, and some arguments and add the callable with the arguments to the before list. �beforeN)r r# r r � assertEqualr% r r r r �test_addBeforeTrigger6 s �z*ThreePhaseEventTests.test_addBeforeTriggerc C s6 | j �d| j| j� | �| j j| j| jfi fg� dS )z� L{_ThreePhaseEvent.addTrigger} should accept C{'during'} as a phase, a callable, and some arguments and add the callable with the arguments to the during list. �duringN)r r# r r r& r( r r r r �test_addDuringTriggerB s �z*ThreePhaseEventTests.test_addDuringTriggerc C s6 | j �d| j| j� | �| j j| j| jfi fg� dS )z� L{_ThreePhaseEvent.addTrigger} should accept C{'after'} as a phase, a callable, and some arguments and add the callable with the arguments to the after list. �afterN)r r# r r r& r* r r r r �test_addAfterTriggerN s �z)ThreePhaseEventTests.test_addAfterTriggerc C s4 | j �d| j| j�}| j �|� | �| j jg � dS )z� L{_ThreePhaseEvent.removeTrigger} should accept an opaque object previously returned by L{_ThreePhaseEvent.addTrigger} and remove the associated trigger. r% N)r r# r r � removeTriggerr& r% �r Zhandler r r �test_removeTriggerZ s z'ThreePhaseEventTests.test_removeTriggerc C s | � t| jjt� � dS )z� L{_ThreePhaseEvent.removeTrigger} should raise L{ValueError} when given an object not previously returned by L{_ThreePhaseEvent.addTrigger}. N)r! � ValueErrorr r, r r r r r � test_removeNonexistentTriggere s z2ThreePhaseEventTests.test_removeNonexistentTriggerc C s6 | j �d| j| j�}| j �|� | �t| j j|� dS )z� L{_ThreePhaseEvent.removeTrigger} should raise L{ValueError} the second time it is called with an object returned by L{_ThreePhaseEvent.addTrigger}. r% N)r r# r r r, r! r/ r- r r r �test_removeRemovedTriggerm s z.ThreePhaseEventTests.test_removeRemovedTriggerc C s$ | � t| jjd| j| jfi f� dS )z� L{_ThreePhaseEvent.removeTrigger} should raise L{ValueError} if it is given a trigger handle which resembles a valid trigger handle aside from its phase being incorrect. r N)r! r" r r, r r r r r r � test_removeAlmostValidTriggerx s �z2ThreePhaseEventTests.test_removeAlmostValidTriggerc C s� g }| j �d|jd� | j �d|jd� | j �d|jd� | j �d|jd� | j �d|jd� | j �d|jd � | �|g � | j �� | �|dddddd g� d S ) z� L{_ThreePhaseEvent.fireEvent} should call I{before}, I{during}, and I{after} phase triggers in that order. r* )�firstr* r( �r3 r( r% �r3 r% ��secondr% �r7 r( )r7 r* N)r r# �appendr& � fireEvent�r �eventsr r r �test_fireEvent� s" ��z#ThreePhaseEventTests.test_fireEventc s� g }t � � | j�d� fdd�� | j�d|jd� | j�d|jd� | �|g � | j�� | �|g � � �d� | �|ddg� dS )z� L{_ThreePhaseEvent.fireEvent} should wait for any L{Deferred} returned by a I{before} phase trigger before proceeding to I{during} events. r% c s � S r r r �Z beforeResultr r r � r z>ThreePhaseEventTests.test_asynchronousBefore.<locals>.<lambda>r( r* N)r r r# r9 r& r: �callbackr; r r> r �test_asynchronousBefore� s z,ThreePhaseEventTests.test_asynchronousBeforec s� g }G dd� dt �� � fdd�}| j�d|� | j�d|jd� | j�d|jd� | j�� | �|ddg� | �� �}| �t|�d� dS ) z� If a before-phase trigger raises a synchronous exception, it should be logged and the remaining triggers should be run. c @ s e Zd ZdS )zHThreePhaseEventTests.test_beforeTriggerException.<locals>.DummyExceptionN��__name__� __module__� __qualname__r r r r �DummyException� s rE c s � � �d S r r r �rE r r �raisingTrigger� s zHThreePhaseEventTests.test_beforeTriggerException.<locals>.raisingTriggerr% r( � N�� Exceptionr r# r9 r: r& ZflushLoggedErrors�len�r r< rG �errorsr rF r �test_beforeTriggerException� s z0ThreePhaseEventTests.test_beforeTriggerExceptionc s� g }G dd� dt �� � fdd�}| j�d|� | j�d|jd� | j�d|jd� | j�� | �|ddg� | �� �}| �t|�d� dS ) z� If a during-phase trigger raises a synchronous exception, it should be logged and the remaining triggers should be run. c @ s e Zd ZdS )zHThreePhaseEventTests.test_duringTriggerException.<locals>.DummyExceptionNrA r r r r rE � s rE c s � � �d S r r r rF r r rG � s zHThreePhaseEventTests.test_duringTriggerException.<locals>.raisingTriggerr( r* rH NrI rL r rF r �test_duringTriggerException� s z0ThreePhaseEventTests.test_duringTriggerExceptionc sl g }� �fdd�}�j �d|jd�� �j �d|� �j �d|jd� ��tdt�j j� ��|ddg� dS )z� If a before-phase trigger tries to remove another before-phase trigger which has already run, a warning should be emitted. c s �j �� � d S r �r r, r �Z beforeHandler r r r, � s zWThreePhaseEventTests.test_synchronousRemoveAlreadyExecutedBefore.<locals>.removeTriggerr% r5 r6 zdRemoving already-fired system event triggers will raise an exception in a future version of Twisted.N)r r# r9 Z assertWarns�DeprecationWarning�__file__r: r& )r r< r, r rQ r �+test_synchronousRemoveAlreadyExecutedBefore� s �z@ThreePhaseEventTests.test_synchronousRemoveAlreadyExecutedBeforec s\ g }�j �d� �fdd�� �j �d|jd�� �j �d|jd� �j �� ��|dg� dS )z� If a before-phase trigger removes another before-phase trigger which has not yet run, the removed trigger should not be run. r% c s �j �� �S r rP r rQ r r r � r zJThreePhaseEventTests.test_synchronousRemovePendingBefore.<locals>.<lambda>r5 r6 N�r r# r9 r: r& r; r rQ r �#test_synchronousRemovePendingBefore� s � � z8ThreePhaseEventTests.test_synchronousRemovePendingBeforec s\ g }�j �d� �fdd�� �j �d|jd�� �j �d|jd� �j �� ��|dg� dS )z If a before-phase trigger removes a during-phase trigger, the during-phase trigger should not be run. r% c s �j �� �S r rP r �� duringHandler r r r r zJThreePhaseEventTests.test_synchronousBeforeRemovesDuring.<locals>.<lambda>r( r* NrU r; r rW r �#test_synchronousBeforeRemovesDuring� s � z8ThreePhaseEventTests.test_synchronousBeforeRemovesDuringc sv g }t � � | j�d� fdd�� | j�d|jd�}| j�d|jd� | j�� | j�|� � �d� | �|dg� dS )z� If a before-phase trigger returns a L{Deferred} and later removes a during-phase trigger before the L{Deferred} fires, the during-phase trigger should not be run. r% c s � S r r r r> r r r r zKThreePhaseEventTests.test_asynchronousBeforeRemovesDuring.<locals>.<lambda>r( r* N)r r r# r9 r: r, r? r& )r r< rX r r> r �$test_asynchronousBeforeRemovesDuring s z9ThreePhaseEventTests.test_asynchronousBeforeRemovesDuringc s` g ��fdd�}�j �d|� �j �d� �fdd�� �j �d|�� �j �� ���dg� dS ) a If a before-phase trigger removes a during-phase trigger which is identical to an already-executed before-phase trigger aside from their phases, no warning should be emitted and the during-phase trigger should not be run. c s � � d� d S )Nr �r9 r �r< r r r " s z]ThreePhaseEventTests.test_synchronousBeforeRemovesConspicuouslySimilarDuring.<locals>.triggerr% c s �j �� �S r rP r )� duringTriggerr r r r & r z^ThreePhaseEventTests.test_synchronousBeforeRemovesConspicuouslySimilarDuring.<locals>.<lambda>r( r N)r r# r: r& �r r r )r] r< r r �7test_synchronousBeforeRemovesConspicuouslySimilarDuring s � zLThreePhaseEventTests.test_synchronousBeforeRemovesConspicuouslySimilarDuringc s\ g }�j �d� �fdd�� �j �d|jd�� �j �d|jd� �j �� ��|dg� dS )z� If a during-phase trigger removes another during-phase trigger which has not yet run, the removed trigger should not be run. r( c s �j �� �S r rP r rW r r r 3 r zJThreePhaseEventTests.test_synchronousRemovePendingDuring.<locals>.<lambda>r4 r8 NrU r; r rW r �#test_synchronousRemovePendingDuring, s" � � � z8ThreePhaseEventTests.test_synchronousRemovePendingDuringc C sd g }| j �d|jd� | j �d|jd� | j �d|jd� | j �� | j �� | �|dddg� dS )zm A trigger should only be called on the first call to L{_ThreePhaseEvent.fireEvent}. r% r( r* NrU r; r r r �test_triggersRunOnce< s z)ThreePhaseEventTests.test_triggersRunOncec sP g � � �fdd�}�j �d� jd� �j �d|� �j �� ��� ddg� dS )z� The temporary list L{_ThreePhaseEvent.finishedBefore} should be emptied and the state reset to C{'BASE'} before the first during-phase trigger executes. c s. � � d� ���jjg � ���jjd� d S )Nr( ZBASE)r9 r& r ZfinishedBefore�stater �r< r r r r] Q s zNThreePhaseEventTests.test_finishedBeforeTriggersCleared.<locals>.duringTriggerr% r( NrU )r r] r rc r �"test_finishedBeforeTriggersClearedJ s z7ThreePhaseEventTests.test_finishedBeforeTriggersClearedN)rB rC rD �__doc__r r$ r' r) r+ r. r0 r1 r2 r= r@ rN rO rT rV rY rZ r_ r` ra rd 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$S )%�SystemEventTestsa Tests for the reactor's implementation of the C{fireSystemEvent}, C{addSystemEventTrigger}, and C{removeSystemEventTrigger} methods of the L{IReactorCore} interface. @ivar triggers: A list of the handles to triggers which have been added to the reactor. c C s g | _ dS )zI Create an empty list in which to store trigger handles. N)�triggersr r r r r e s zSystemEventTests.setUpc C s> | j r:| j �� }zt�|� W q ttfk r6 Y q X q dS )zA Remove all remaining triggers from the reactor. N)rg �popr �removeSystemEventTriggerr/ r" r^ r r r �tearDownl s zSystemEventTests.tearDownc C s t �|||�}| j�|� |S )zS Add a trigger to the reactor and remember it in C{self.triggers}. )r ZaddSystemEventTriggerrg r9 )r r �phase�func�tr r r r# x s zSystemEventTests.addTriggerc C s t �|� | j�|� dS )zd Remove a trigger by its handle from the reactor and from C{self.triggers}. N)r ri rg �remover^ r r r r, � s z SystemEventTests.removeTriggerc sJ d}g � � fdd�}| � |||� | �� g � t�|� | �� d g� d S )N�testc s � � d � d S r r[ r r\ r r r � s z<SystemEventTests._addSystemEventTriggerTest.<locals>.trigger�r# r& r �fireSystemEvent)r rk � eventTyper r r\ r �_addSystemEventTriggerTest� s z+SystemEventTests._addSystemEventTriggerTestc C s | � d� dS )z� L{IReactorCore.addSystemEventTrigger} should accept the C{'before'} phase and not call the given object until the right event is fired. r% N�rs r r r r �test_beforePhase� s z!SystemEventTests.test_beforePhasec C s | � d� dS )z� L{IReactorCore.addSystemEventTrigger} should accept the C{'during'} phase and not call the given object until the right event is fired. r( Nrt r r r r �test_duringPhase� s z!SystemEventTests.test_duringPhasec C s | � d� dS )z� L{IReactorCore.addSystemEventTrigger} should accept the C{'after'} phase and not call the given object until the right event is fired. r* Nrt r r r r �test_afterPhase� s z SystemEventTests.test_afterPhasec C s d}| � t| jd|dd� � dS )z� L{IReactorCore.addSystemEventTrigger} should reject phases other than C{'before'}, C{'during'}, or C{'after'}. ro r c S s d S r r r r r r r � r z4SystemEventTests.test_unknownPhase.<locals>.<lambda>N)r! r" r# )r rr r r r �test_unknownPhase� s �z"SystemEventTests.test_unknownPhasec sf d}g � � fdd�}� fdd�}| � d||� | � d||� | �� g � t�|� | �� ddg� dS ) z� L{IReactorCore.addSystemEventTrigger} should call triggers added to the C{'before'} phase before it calls triggers added to the C{'during'} phase. ro c s � � d� d S �Nr% r[ r r\ r r � beforeTrigger� s zASystemEventTests.test_beforePreceedsDuring.<locals>.beforeTriggerc s � � d� d S �Nr( r[ r r\ r r r] � s zASystemEventTests.test_beforePreceedsDuring.<locals>.duringTriggerr% r( Nrp �r rr rz r] r r\ r �test_beforePreceedsDuring� s z*SystemEventTests.test_beforePreceedsDuringc sf d}g � � fdd�}� fdd�}| � d||� | � d||� | �� g � t�|� | �� ddg� dS ) z� L{IReactorCore.addSystemEventTrigger} should call triggers added to the C{'during'} phase before it calls triggers added to the C{'after'} phase. ro c s � � d� d S r{ r[ r r\ r r r] � s z@SystemEventTests.test_duringPreceedsAfter.<locals>.duringTriggerc s � � d� d S )Nr* r[ r r\ r r � afterTrigger� s z?SystemEventTests.test_duringPreceedsAfter.<locals>.afterTriggerr( r* Nrp )r rr r] r~ r r\ r �test_duringPreceedsAfter� s z)SystemEventTests.test_duringPreceedsAfterc s� t � �d}g � �fdd�}� fdd�}| �d||� | �d||� | �� g � t�|� | �� g � ��d� | �� dg� dS ) z� If a trigger added to the C{'before'} phase of an event returns a L{Deferred}, the C{'during'} phase should be delayed until it is called back. ro c s � S r r r )�triggerDeferredr r rz � s zBSystemEventTests.test_beforeReturnsDeferred.<locals>.beforeTriggerc s � � d� d S r{ r[ r r\ r r r] � s zBSystemEventTests.test_beforeReturnsDeferred.<locals>.duringTriggerr% r( N�r r# r& r rq r? r| r )r< r� r �test_beforeReturnsDeferred� s z+SystemEventTests.test_beforeReturnsDeferredc s� t � �t � �d}g � �fdd�}�fdd�}� fdd�}| �d||� | �d||� | �d ||� | �� g � t�|� | �� g � ��d � | �� g � ��d � | �� d g� d S ) z� If more than one trigger added to the C{'before'} phase of an event return L{Deferred}s, the C{'during'} phase should be delayed until they are all called back. ro c s � S r r r �� firstDeferredr r �firstBeforeTrigger� s zNSystemEventTests.test_multipleBeforeReturnDeferred.<locals>.firstBeforeTriggerc s � S r r r )�secondDeferredr r �secondBeforeTrigger� s zOSystemEventTests.test_multipleBeforeReturnDeferred.<locals>.secondBeforeTriggerc s � � d� d S r{ r[ r r\ r r r] s zISystemEventTests.test_multipleBeforeReturnDeferred.<locals>.duringTriggerr% r( Nr� )r rr r� r� r] r �r< r� r� r �!test_multipleBeforeReturnDeferred� s" z2SystemEventTests.test_multipleBeforeReturnDeferredc s� d}g � t � �t � ��fdd�}�fdd�}� �fdd�}� fdd �}| �d ||� | �d ||� | �d ||� | �d ||� | �� g � t�|� | �� d g� ��d � | �� d d g� d S ) aa If a trigger added to the C{'before'} phase of an event calls back a L{Deferred} returned by an earlier trigger in the C{'before'} phase of the same event, the remaining C{'before'} triggers for that event should be run and any further L{Deferred}s waited on before proceeding to the C{'during'} events. ro c s � S r r r r� r r r� s zaSystemEventTests.test_subsequentBeforeTriggerFiresPriorBeforeDeferred.<locals>.firstBeforeTriggerc s � � d � d S r )r? r r� r r r� s zbSystemEventTests.test_subsequentBeforeTriggerFiresPriorBeforeDeferred.<locals>.secondBeforeTriggerc s � � d� �S ry r[ r �r< r� r r �thirdBeforeTrigger s zaSystemEventTests.test_subsequentBeforeTriggerFiresPriorBeforeDeferred.<locals>.thirdBeforeTriggerc s � � d� d S r{ r[ r r\ r r r] " s z\SystemEventTests.test_subsequentBeforeTriggerFiresPriorBeforeDeferred.<locals>.duringTriggerr% r( Nr� )r rr r� r� r� r] r r� r �4test_subsequentBeforeTriggerFiresPriorBeforeDeferred s" zESystemEventTests.test_subsequentBeforeTriggerFiresPriorBeforeDeferredc sj d}g � � fdd�}� fdd�}| � d||� | �| � d||�� | �� g � t�|� | �� dg� dS ) z� A trigger removed with L{IReactorCore.removeSystemEventTrigger} should not be called when the event fires. ro c s � � d� d S )Nr3 r[ r r\ r r r� 6 s zJSystemEventTests.test_removeSystemEventTrigger.<locals>.firstBeforeTriggerc s � � d� d S )Nr7 r[ r r\ r r r� 8 s zKSystemEventTests.test_removeSystemEventTrigger.<locals>.secondBeforeTriggerr% r3 N)r# r, r& r rq )r rr r� r� r r\ r � test_removeSystemEventTrigger/ s � z.SystemEventTests.test_removeSystemEventTriggerc C sh | � dddd� �}| �|� | �ttjd� | �ttj|� | �ttj|d d|d dd� f� dS ) aM Passing an object to L{IReactorCore.removeSystemEventTrigger} which was not returned by a previous call to L{IReactorCore.addSystemEventTrigger} or which has already been passed to C{removeSystemEventTrigger} should result in L{TypeError}, L{KeyError}, or L{ValueError} being raised. r( ro c S s d S r r r r r r r J r zKSystemEventTests.test_removeNonExistentSystemEventTrigger.<locals>.<lambda>Nr )r rH )r# r, r! � TypeErrorr ri r/ r" )r �br r r �(test_removeNonExistentSystemEventTriggerB s" � ��z9SystemEventTests.test_removeNonExistentSystemEventTriggerc s g � d}t � �� �fdd�}� fdd�}d}t � �� �fdd�}� fd d �}| �d ||� | �d ||� | �d ||� | �d ||� | �� g � t�|� | �� d g� t�|� | �� d dg� ��d� | �� d ddg� ��d� | �� d dddg� dS )z� L{IReactorCore.fireSystemEvent} should behave the same way for a particular system event regardless of whether Deferreds are being waited on for a different system event. z first-eventc s � � d� �S )N�r% r3 r[ r )r< r� r r �beforeFirstEvent` s zQSystemEventTests.test_interactionBetweenDifferentEvents.<locals>.beforeFirstEventc s � � d� d S )N�r* r3 r[ r r\ r r �afterFirstEventc s zPSystemEventTests.test_interactionBetweenDifferentEvents.<locals>.afterFirstEventz second-eventc s � � d� �S )N�r% r7 r[ r r� r r �beforeSecondEventh s zRSystemEventTests.test_interactionBetweenDifferentEvents.<locals>.beforeSecondEventc s � � d� d S )N�r* r7 r[ r r\ r r �afterSecondEventk s zQSystemEventTests.test_interactionBetweenDifferentEvents.<locals>.afterSecondEventr% r* r� r� Nr� r� r� )r Z firstEventr� r� Z secondEventr� r� r r� r �&test_interactionBetweenDifferentEventsV s6 � �z7SystemEventTests.test_interactionBetweenDifferentEventsN)rB rC rD re r rj r# r, rs ru rv rw rx r} r r� r� r� r� r� r� r r r r rf \ s$ rf c @ sp e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� ZdS )� TimeTestsz9 Tests for the IReactorTime part of the reactor. c C s t �� }| �|| | |� dS )aN L{twisted.internet.reactor.seconds} should return something like a number. 1. This test specifically does not assert any relation to the "system time" as returned by L{time.time} or L{twisted.python.runtime.seconds}, because at some point we may find a better option for scheduling calls than wallclock-time. 2. This test *also* does not assert anything about the type of the result, because operations may not return ints or floats: For example, datetime-datetime == timedelta(0). N)r �secondsr& )r Znowr r r � test_seconds� s zTimeTests.test_secondsc C sJ t j}dd� t _z$t �ddd� �}| �|�� d� W 5 |t _X |�� dS )z� L{reactor.callLater<twisted.internet.interfaces.IReactorTime.callLater>} should use the reactor's seconds factory to produce the time at which the DelayedCall will be called. c S s dS �N�d r r r r r r � r zITimeTests.test_callLaterUsesReactorSecondsInDelayedCall.<locals>.<lambda>� c S s d S r r r r r r r � r �i N)r r� � callLaterr& �getTime�cancel�r Zoseconds�callr r r �-test_callLaterUsesReactorSecondsInDelayedCall� s z7TimeTests.test_callLaterUsesReactorSecondsInDelayedCallc C sJ t j}dd� t _z$t �ddd� �}| �|�� d� W 5 |t _X |�� dS )z� L{reactor.callLater<twisted.internet.interfaces.IReactorTime.callLater>} should propagate its own seconds factory to the DelayedCall to use as its own seconds factory. c S s dS r� r r r r r r � r zWTimeTests.test_callLaterUsesReactorSecondsAsDelayedCallSecondsFactory.<locals>.<lambda>r� c S s d S r r r r r r r � r r� N)r r� r� r& r� r� r r r �;test_callLaterUsesReactorSecondsAsDelayedCallSecondsFactory� s zETimeTests.test_callLaterUsesReactorSecondsAsDelayedCallSecondsFactoryc C s( t � }t�d|jd� |�| jd� |S )zc Test that a DelayedCall really calls the function it is supposed to call. r N)r r r� r? � addCallbackr& �r �dr r r �test_callLater� s zTimeTests.test_callLaterc C s< t �dtt�}| �|j� |j}|�d� | �|j|� dS )zS A L{DelayedCall} that is reset will be scheduled at the new time. � rH N)r r� r Z addCleanupr� �time�reset�assertNotEqual)r r� ZorigTimer r r �test_callLaterReset� s z TimeTests.test_callLaterResetc sP g � � fdd�}t �d|�}|�� t� �� ��fdd�}t �dt jd|� �S )zL Test that when a DelayedCall is cancelled it does not run. c s � � d � d S r r[ r )�calledr r �function� s z2TimeTests.test_cancelDelayedCall.<locals>.functionr c s4 z�� � g � W n ��� Y n X ��d � d S r )r& �errbackr? r �r� r� r r r �check� s z/TimeTests.test_cancelDelayedCall.<locals>.check)r r� r� r )r r� r� r� r r� r �test_cancelDelayedCall� s z TimeTests.test_cancelDelayedCallc C s, t �ddd� �}|�� | �tj|j� dS )z Test that cancelling a DelayedCall which has already been cancelled raises the appropriate exception. r c S s d S r r r r r r r � r z;TimeTests.test_cancelCancelledDelayedCall.<locals>.<lambda>N)r r� r� r! r ZAlreadyCancelled)r r� r r r � test_cancelCancelledDelayedCall� s z)TimeTests.test_cancelCancelledDelayedCallc s& t � �� ��fdd�}t�d|�� �S )z� Test that cancelling a DelayedCall in the DelayedCall's function as that function is being invoked by the DelayedCall raises the appropriate exception. c s8 z�� tj� j� W n ��� Y n X ��d � d S r �r! r Z AlreadyCalledr� r� r? r �r� r� r r r �later s z@TimeTests.test_cancelCalledDelayedCallSynchronous.<locals>.laterr �r r r� �r r� r r� r �'test_cancelCalledDelayedCallSynchronous� s z1TimeTests.test_cancelCalledDelayedCallSynchronousc s2 t � �� ��fdd���fdd�}t�d|�� �S )z| Test that cancelling a DelayedCall after it has run its function raises the appropriate exception. c s8 z�� tj� j� W n ��� Y n X ��d � d S r r� r r� r r r� s zATimeTests.test_cancelCalledDelayedCallAsynchronous.<locals>.checkc s t �d� � d S �Nr )r r� r )r� r r r� s zATimeTests.test_cancelCalledDelayedCallAsynchronous.<locals>.laterr r� r� r )r� r� r� r r �(test_cancelCalledDelayedCallAsynchronous s z2TimeTests.test_cancelCalledDelayedCallAsynchronousc C sB t �ddd� �}z"| �|�� t�� d dk � W 5 |�� X d S )N� c S s d S r r r r r r r " r z-TimeTests.testCallLaterTime.<locals>.<lambda>rH )r r� r� � assertTruer� r� r� r r r �testCallLaterTime! s "zTimeTests.testCallLaterTimec s� t jddd� ddd�� t� � � �d� t� � � �� t� � t jddd� ftd��dd d id ft gi�� t� � � �� t� � � fd d �}t� �|�}t �d|jd �� t� � |S )Nr c S s d S r r )r �yr r r r , r z:TimeTests.testDelayedCallStringification.<locals>.<lambda>r r� )r� r� c S s d S r r r r r r r 3 r ZhelloZworldy $@c s t � � d S r )�str)Zignored��dcr r � calledBack8 s z<TimeTests.testDelayedCallStringification.<locals>.calledBack) r r� r� r� r� �ranger r� r? )r r� r� r r� r � testDelayedCallStringification) s , z(TimeTests.testDelayedCallStringificationc C sX dd� }t �ddd� di dd� dd� |�}| �|�� d� |�d � | �|�� d � d S ) z� Test that the C{seconds} argument to DelayedCall gets used instead of the default timing function, if it is not None. c S s dS )Nr� r r r r r r� E s z9TimeTests.testDelayedCallSecondsOverride.<locals>.secondsr� c S s d S r r r r r r r G r z:TimeTests.testDelayedCallSecondsOverride.<locals>.<lambda>r c S s d S r r r� r r r r G r c S s d S r r r� r r r r H r � � N)r Z DelayedCallr& r� r� )r r� r� r r r � testDelayedCallSecondsOverride@ s � z(TimeTests.testDelayedCallSecondsOverrideN)rB rC rD re r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s r� c @ sJ e Zd Zdd� Ze�ed�dkr&de_dd� Zdd� Z d d � Z d d � Z dS ) �!CallFromThreadStopsAndWakeUpTestsc s t � � � fdd�}t�|� � S )Nc s t �d� t�� jd � d S )Ng�������?)r� �sleepr �callFromThreadr? r �r� r r �wakeS s z:CallFromThreadStopsAndWakeUpTests.testWakeUp.<locals>.wake)r r Z callInThread)r r� r r� r � testWakeUpP s z,CallFromThreadStopsAndWakeUpTests.testWakeUpNz-Nothing to wake up for without thread supportc C s d| _ d S )NT)�stoppedr r r r �_stopCallFromThreadCallback] s z=CallFromThreadStopsAndWakeUpTests._stopCallFromThreadCallbackc C s t �| j|� t �d| j� d S r� )r r� �_callFromThreadCallback2r� r� r� r r r �_callFromThreadCallback` s z9CallFromThreadStopsAndWakeUpTests._callFromThreadCallbackc C s4 z| � | j� W n |�� Y n X |�d � d S r )r� r� r� r? r� r r r r� d s z:CallFromThreadStopsAndWakeUpTests._callFromThreadCallback2c C s d| _ t�� }t�| j|� |S )z� Ensure that callFromThread from inside a callFromThread callback doesn't sit in an infinite loop and lets other things happen too. F)r� r r r r� r� r� r r r �testCallFromThreadStopsm s z9CallFromThreadStopsAndWakeUpTests.testCallFromThreadStops) rB rC rD r� r �IReactorThreadsr �skipr� r� r� r� r r r r r� O s r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )� DelayedTestsc C s d| _ d| _i | _t�� | _d S r� )�finished�counter�timersr r �deferredr r r r r z s zDelayedTests.setUpc C s | j �� D ] }|�� q d S r )r� �valuesr� )r rm r r r rj � s zDelayedTests.tearDownc C sX | j �� }tt�� �}g }|D ]}||kr |�|� q |r@d| _| �|dt|� � d S )NrH z7Should have been missing no calls, instead was missing ) r� r� �listr �getDelayedCallsr9 r� � assertFalse�repr)r �l1�l2Zmissingr� r r r � checkTimers� s �zDelayedTests.checkTimersc C s | j |= | �� d S r )r� r� �r �tagr r r r? � s zDelayedTests.callbackc C s | � |� | �d| j � d S )N� )r? �addTimerr� r r r r� � s zDelayedTests.addCallbackc C s d| _ | �|� | j�d � d S )NrH )r� r? r� r� r r r �done� s zDelayedTests.donec C s6 t �|d || j�| j| j< | jd7 _| �� d S )N�{�G�z�?rH )r r� r� r� r� )r Zwhenr? r r r r� � s � zDelayedTests.addTimerc s� t td�sd S � �� � �d� j� � �d� j� � �d� j� � j}� �d� j� � �d� j� � �d� j� � j| � � � j|= � �� � j �� fdd �� � j S ) Nr� �# � � � � � c s � � � S r )r� r r r r r � r z2DelayedTests.testGetDelayedCalls.<locals>.<lambda>) �hasattrr r� r� r� r? r� r� r� r� r� )r Zwhichr r r �testGetDelayedCalls� s z DelayedTests.testGetDelayedCallsc s@ t �d�jjd�� ��� �� � � �fdd�}�j�|� �jS )zM L{IDelayedCall.active} returns False once the call has run. r� Tc s �� � �� � | S r )r� �active)Zsuccess�Zdcallr r r �checkDeferredCall� s z3DelayedTests.test_active.<locals>.checkDeferredCall)r r� r� r? r� r� r� )r r� r r� r � test_active� s zDelayedTests.test_activeN) rB rC rD r rj r� r? r� r� r� r� r� r r r r r� y s r� a from __future__ import print_function import %(reactor)s %(reactor)s.install() from twisted.internet import reactor class Foo: def __init__(self): reactor.callWhenRunning(self.start) self.timer = reactor.callLater(3, self.failed) def start(self): reactor.resolve('localhost').addBoth(self.done) def done(self, res): print('done', res) reactor.stop() def failed(self): print('failed') self.timer = None reactor.stop() f = Foo() reactor.run() c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �ChildResolveProtocolc C s || _ d S r )� onCompletion)r r� r r r �__init__� s z ChildResolveProtocol.__init__c C s g | _ g | _d S r )�outputr r r r r �connectionMade� s z#ChildResolveProtocol.connectionMadec C s | j �|� d S r )r� r9 )r �outr r r � outReceived� s z ChildResolveProtocol.outReceivedc C s | j �|� d S r )r r9 )r �errr r r � errReceived� s z ChildResolveProtocol.errReceivedc C s | j �|| j| jf� d | _ d S r )r� r? r� r )r �reasonr r r � processEnded� s z!ChildResolveProtocol.processEndedN)rB rC rD r� r� r� r r r r r r r� � s r� c @ s e Zd Zdd� ZdS )� ResolveTestsc s� t j�� �� �}t|d�� }tj}|�td|i � W 5 Q R X t j � � }t j � t j�|d< t� }t|�}t�|t jdd|f|� � fdd�}|�|� |S )N�wr � PYTHONPATH�pythonz-uc sX | \}}}d� |�}tr*dtj�d� }nd}||krT� �d|�� |d� |�f � d S )Nr s done 127.0.0.1�asciis done 127.0.0.1 z�The child process failed to produce the desired results: Reason for termination was: %r Output stream was: %r Error stream was: %r )�joinr �os�linesep�encodeZfailZgetErrorMessage)�resultr r� r Zexpected_outputr r r � cbFinished s ��z1ResolveTests.testChildResolve.<locals>.cbFinished)r �path�abspathZmktemp�openr rC �write�resolve_helper�environ�copy�pathsepr �sysr r� Z spawnProcess� executabler� )r Z helperPathZ helperFileZ reactorName�envZhelperDeferredZ helperProtor r r r �testChildResolve� s z ResolveTests.testChildResolveN)rB rC rD r r r r r r � s r z8cannot run test: reactor doesn't support IReactorProcessc @ sL e Zd ZdZe�ed�dkr dZdd� Zdd� Z dd � Z d d � Z d d � Z dS )�CallFromThreadTestsz- Task scheduling from threads tests. Nz&Nothing to test without thread supportc C s d| _ t� | _d S r� )r� r r� r r r r r 0 s zCallFromThreadTests.setUpc O s t j||� dS )z) Override in subclasses. N)r r� )r �args�kwargsr r r �schedule5 s z CallFromThreadTests.schedulec s* � fdd�}t d�D ]}� �|� q� jS )z� L{IReactorThreads.callFromThread} can be used to schedule a large number of calls in the reactor thread. c s( � j d7 _ � j dkr$� j�d� d S )NrH r� T�r� r� r? r r r r �addAndMaybeFinishA s zVCallFromThreadTests.test_lotsOfThreadsAreScheduledCorrectly.<locals>.addAndMaybeFinishr� )r� r r� )r r �ir r r �'test_lotsOfThreadsAreScheduledCorrectly< s z;CallFromThreadTests.test_lotsOfThreadsAreScheduledCorrectlyc sb g � � �fdd�}�j �|� ��� jd� ��� jd� ��� jd� ��tj�j jd� �j S )zO Callbacks should be invoked in the order they were scheduled. c s �� � dddg� d S )NrH r� r� )r& )�_��orderr r r r� R s zECallFromThreadTests.test_threadsAreRunInScheduledOrder.<locals>.checkrH r� r� N)r� r� r r9 r r� r? )r r� r r$ r �"test_threadsAreRunInScheduledOrderL s z6CallFromThreadTests.test_threadsAreRunInScheduledOrderc s* � fdd�}� � |� � �� jd� � jS )zU Scheduled tasks should not be run until the reactor starts running. c s d� _ � j�d� d S )NrH Tr r r r r � incAndFinishb s zUCallFromThreadTests.test_scheduledThreadsNotRunUntilReactorRuns.<locals>.incAndFinishr )r r& r� r� )r r' r r r �+test_scheduledThreadsNotRunUntilReactorRuns^ s z?CallFromThreadTests.test_scheduledThreadsNotRunUntilReactorRuns) rB rC rD re r r� r r� r r r" r& r( r r r r r ) s r c @ s e Zd ZdZdS )� MyProtocolz Sample protocol. N)rB rC rD re r r r r r) n s r) c @ s e Zd ZdZeZdS )� MyFactoryz Sample factory. N)rB rC rD re r) r r r r r r* s s r* c @ s e Zd Zdd� ZdS )� ProtocolTestsc C s0 t � }|�d �}| �|j|� | �||j� d S r )r* Z buildProtocolr& �factoryZassertIsInstancer )r r, r r r r � testFactory} s zProtocolTests.testFactoryN)rB rC rD r- r r r r r+ { s r+ c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � DummyProducera� Very uninteresting producer implementation used by tests to ensure the right methods are called by the consumer with which it is registered. @type events: C{list} of C{str} @ivar events: The producer/consumer related events which have happened to this producer. Strings in this list may be C{'resume'}, C{'stop'}, or C{'pause'}. Elements are added as they occur. c C s g | _ d S r r\ r r r r r� � s zDummyProducer.__init__c C s | j �d� d S )N�resume�r< r9 r r r r �resumeProducing� s z DummyProducer.resumeProducingc C s | j �d� d S )N�stopr0 r r r r � stopProducing� s zDummyProducer.stopProducingc C s | j �d� d S )N�pauser0 r r r r �pauseProducing� s z DummyProducer.pauseProducingN)rB rC rD re r� r1 r3 r5 r r r r r. � s r. c @ s, e Zd ZdZdZdZdd� Zdd� ZeZdS ) �SillyDescriptora A descriptor whose data buffer gets filled very fast. Useful for testing FileDescriptor's IConsumer interface, since the data buffer fills as soon as at least four characters are written to it, and gets emptied in a single doWrite() cycle. r� Tc C s t |�S )z( Always write all data. )rK )r �datar r r � writeSomeData� s z SillyDescriptor.writeSomeDatac C s dS )z1 Do nothing: bypass the reactor. Nr r r r r � startWriting� s z SillyDescriptor.startWritingN) rB rC rD re Z bufferSize� connectedr8 r9 Z stopWritingr r r r r6 � s r6 c s, e Zd ZdZ� fdd�Z� fdd�Z� ZS )�ReentrantProducera� Similar to L{DummyProducer}, but with a resumeProducing method which calls back into an L{IConsumer} method of the consumer against which it is registered. @ivar consumer: The consumer with which this producer has been or will be registered. @ivar methodName: The name of the method to call on the consumer inside C{resumeProducing}. @ivar methodArgs: The arguments to pass to the consumer method invoked in C{resumeProducing}. c s$ t t| ��� || _|| _|| _d S r )�superr; r� �consumer� methodName� methodArgs)r r= r> r? �� __class__r r r� � s zReentrantProducer.__init__c s& t t| ��� t| j| j�| j� d S r )r< r; r1 �getattrr= r> r? r r@ r r r1 � s z!ReentrantProducer.resumeProducing)rB rC rD re r� r1 � __classcell__r r r@ r r; � s r; c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dS )� ProducerTestsz< Test abstract.FileDescriptor's consumer interface. c C sH t �� }d|_t� }|�|d� | �|jdg� | �t|jt� d� dS )z� Verify that registering a non-streaming producer invokes its resumeProducing() method and that you can only register one producer at a time. rH r r/ N) r �FileDescriptorr: r. �registerProducerr& r< r! � RuntimeError�r �fdZdpr r r �test_doubleProducer� s z!ProducerTests.test_doubleProducerc C s4 t �� }d|_t� }|�|d� | �|jdg� dS )z� Verify that registering a producer when the connection has already been closed invokes its stopProducing() method. rH r r2 N)r rE Z disconnectedr. rF r& r< rH r r r � test_unconnectedFileDescriptor� s z,ProducerTests.test_unconnectedFileDescriptorc C sn t � }t� }|j|dd� | �|jdg� |jdd�= |dkrR|�dddd g� n |�d � | �|jg � dS ) z� Pull consumers don't get their C{pauseProducing} method called if the descriptor buffer fills up. @param _methodName: Either 'write', or 'writeSequence', indicating which transport method to write data to. F�Z streamingr/ N� writeSequence� 1� 2� 3� 4s 1234)r6 r. rF r& r< rM r )r r> � descriptor�producerr r r �_dontPausePullConsumerTest� s z(ProducerTests._dontPausePullConsumerTestc C s | � d�S )z� Verify that FileDescriptor does not call producer.pauseProducing() on a non-streaming pull producer in response to a L{IConsumer.write} call which results in a full write buffer. Issue #2286. r �rT r r r r �!test_dontPausePullConsumerOnWrite s z/ProducerTests.test_dontPausePullConsumerOnWritec C s | � d�S )a@ Like L{test_dontPausePullConsumerOnWrite}, but for a call to C{writeSequence} rather than L{IConsumer.write}. C{writeSequence} is not part of L{IConsumer}, but L{abstract.FileDescriptor} has supported consumery behavior in response to calls to L{writeSequence} forever. rM rU r r r r �)test_dontPausePullConsumerOnWriteSequence s z7ProducerTests.test_dontPausePullConsumerOnWriteSequencec C s� t � }|dkrdddg}nd}t|||�}|j|dd� t||�|� | �|jdg� |jd d �= |�� | �|jd dg� |jd d �= |�� | �|jd dg� d S ) NrM � s� ps ams spamTrL r4 r/ )r6 r; rF rB r& r< ZdoWrite)r r> rR r7 rS r r r � _reentrantStreamingProducerTest! s z-ProducerTests._reentrantStreamingProducerTestc C s | � d�S )z� Verify that FileDescriptor tracks producer's paused state correctly. Issue #811, fixed in revision r12857. r �rZ r r r r �)test_reentrantStreamingProducerUsingWriteB s z7ProducerTests.test_reentrantStreamingProducerUsingWritec C s | � d�S )a/ Like L{test_reentrantStreamingProducerUsingWrite}, but for calls to C{writeSequence}. C{writeSequence} is B{not} part of L{IConsumer}, however C{abstract.FileDescriptor} has supported consumery behavior in response to calls to C{writeSequence} forever. rM r[ r r r r �1test_reentrantStreamingProducerUsingWriteSequenceJ s z?ProducerTests.test_reentrantStreamingProducerUsingWriteSequenceN) rB rC rD re rJ rK rT rV rW rZ r\ r] r r r r rD � s !rD c @ sf e Zd Ze�ed�dk r dd� Ze�ed�dk r8dd� Ze� ed�dk rbe rbe fdd�Z e rbde _ dS ) �PortStringificationTestsNc C sF t �dt�� �}|�� j}| �t|��t|��dd||f � |� � S �Nr ����%d not found in %s) r Z listenTCPr � ServerFactory�getHost�portr� r� �find� stopListening�r �p�portNor r r �testTCPY s �z PortStringificationTests.testTCPc C sF t �dt�� �}|�� j}| �t|��t|��dd||f � |� � S r_ ) r Z listenUDPr ZDatagramProtocolrc rd r� r� re rf rg r r r �testUDPa s �z PortStringificationTests.testUDPc C s\ t �td�}t�dt�� |�||��}|�� j }| � t |�� t |��dd||f � |� � S )Nz server.pemr r` ra )r ZsibpathrS r Z listenSSLr rb Z DefaultOpenSSLContextFactoryrc rd r� r� re rf )r r Zpemrh ri r r r �testSSLi s �z PortStringificationTests.testSSLz-Re-enable once the Python 3 SSL port is done.)rB rC rD r Z IReactorTCPr rj Z IReactorUDPrk Z IReactorSSLr rl r r� r r r r r^ W s r^ )4re Z __future__r r r r r� Ztwisted.python.compatr Z twisted.trialr Ztwisted.internetr r r r r r r r � ImportErrorZ supportedZtwisted.internet.deferr r Ztwisted.pythonr ZTestCaser rf r� r� r� r ZProcessProtocolr� r ZIReactorProcessr� r ZProtocolr) ZFactoryr* r+ r r. rE r6 r; rD r^ r r r r �<module> sV ? 2 D*W)�E
Save File
Cancel