File: //lib/python3/dist-packages/twisted/test/__pycache__/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