File: //usr/lib/python3/dist-packages/twisted/test/__pycache__/test_internet.cpython-310.pyc
o
�b�� � @ s d Z ddlZddlZddlZddlmZ ddlmZmZm Z m
Z
m Z m Z m
Z
ddlmZmZ ddlmZ ddlmZ ddlmZ zdd lmZ W n eyS dZY nw eZer]ejs]dZG d
d � d e�ZG d d
� d
e�Z G dd� de�Z G dd� de�Z G dd� de�Z dZ G dd� de j!�Z"ee �#e
d� d�G dd� de��Z$ee �%e
d� d�G dd � d e��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$�Z,G d%d&� d&ej-�Z.G d'd(� d(e,�Z/G d)d*� d*e�Z0G d+d,� d,e�Z1G d-d.� d.e�Z2dS )/zB
Tests for lots of functionality provided by L{twisted.internet}.
� N)�skipIf)�abstract�base�defer�error�
interfaces�protocol�reactor)�Deferred�passthru)� Connector)�util)�TestCase)�sslc @ 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 � 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 �_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 �6 | 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_addBeforeTrigger5 � "z*ThreePhaseEventTests.test_addBeforeTriggerc C r& )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_addDuringTrigger> r* z*ThreePhaseEventTests.test_addDuringTriggerc C r& )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_addAfterTriggerG r* 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 �handler r r �test_removeTriggerP 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_removeNonexistentTriggerZ 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" r3 r0 r r r �test_removeRemovedTriggera 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_removeAlmostValidTriggerk 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 �� | �|g d
�� d S ) z�
L{_ThreePhaseEvent.fireEvent} should call I{before}, I{during}, and
I{after} phase triggers in that order.
r- ��firstr- r+ �r8 r+ r' �r8 r' ��secondr' �r< r+ �r< r- )r: r; r9 r= r7 r> N)r r$ �appendr( � fireEvent�r �eventsr r r �test_fireEventu 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 r r r �� beforeResultr r r � r z>ThreePhaseEventTests.test_asynchronousBefore.<locals>.<lambda>r+ r- N)r
r r$ r? r( r@ �callbackrA r rE r �test_asynchronousBefore� s
z,ThreePhaseEventTests.test_asynchronousBeforec � 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 @ � e Zd ZdS )zHThreePhaseEventTests.test_beforeTriggerException.<locals>.DummyExceptionN��__name__�
__module__� __qualname__r r r r �DummyException� � rO c � � � �r r r �rO r r �raisingTrigger� � zHThreePhaseEventTests.test_beforeTriggerException.<locals>.raisingTriggerr' r+ � N�� Exceptionr r$ r? r@ r( �flushLoggedErrors�len�r rB rS �errorsr rR r �test_beforeTriggerException� �
z0ThreePhaseEventTests.test_beforeTriggerExceptionc rI ) z�
If a during-phase trigger raises a synchronous exception, it should be
logged and the remaining triggers should be run.
c @ rJ )zHThreePhaseEventTests.test_duringTriggerException.<locals>.DummyExceptionNrK r r r r rO � rP rO c rQ r r r rR r r rS � rT zHThreePhaseEventTests.test_duringTriggerException.<locals>.raisingTriggerr+ r- rU NrV rZ r rR r �test_duringTriggerException� r] 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 �� beforeHandler r r r/ � � zWThreePhaseEventTests.test_synchronousRemoveAlreadyExecutedBefore.<locals>.removeTriggerr' r: r; zdRemoving already-fired system event triggers will raise an exception in a future version of Twisted.N)r r$ r? � assertWarns�DeprecationWarning�__file__r@ r( )r rB r/ r r` r �+test_synchronousRemoveAlreadyExecutedBefore� s ��z@ThreePhaseEventTests.test_synchronousRemoveAlreadyExecutedBeforec �\ 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 � �j �� �S r r_ r r` r r r � � zJThreePhaseEventTests.test_synchronousRemovePendingBefore.<locals>.<lambda>r: r; N�r r$ r? r@ r( rA r r` r �#test_synchronousRemovePendingBefore� � �
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 rh r r_ r �� duringHandler r r r � ri zJThreePhaseEventTests.test_synchronousBeforeRemovesDuring.<locals>.<lambda>r+ r- Nrj rA r rm 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 rD r r r rE r r r r zKThreePhaseEventTests.test_asynchronousBeforeRemovesDuring.<locals>.<lambda>r+ r- N)r
r r$ r? r@ r/ rG r( )r rB rn r rE 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 � � � d� d S )Nr �r? r �rB r r r � z]ThreePhaseEventTests.test_synchronousBeforeRemovesConspicuouslySimilarDuring.<locals>.triggerr' c rh r r_ r )�
duringTriggerr r r r ri z^ThreePhaseEventTests.test_synchronousBeforeRemovesConspicuouslySimilarDuring.<locals>.<lambda>r+ r N)r r$ r@ r( �r r r )ru rB r r �7test_synchronousBeforeRemovesConspicuouslySimilarDuring
s
zLThreePhaseEventTests.test_synchronousBeforeRemovesConspicuouslySimilarDuringc rg )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 rh r r_ r rm r r r % ri zJThreePhaseEventTests.test_synchronousRemovePendingDuring.<locals>.<lambda>r9 r= Nrj rA r rm r �#test_synchronousRemovePendingDuring rl z8ThreePhaseEventTests.test_synchronousRemovePendingDuringc C sb g }| j �d|jd� | j �d|jd� | j �d|jd� | j �� | j �� | �|g d�� dS )zm
A trigger should only be called on the first call to
L{_ThreePhaseEvent.fireEvent}.
r' r+ r- )r' r+ r- Nrj rA 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+ �BASE)r? r( r �finishedBefore�stater �rB r r r ru B s
zNThreePhaseEventTests.test_finishedBeforeTriggersCleared.<locals>.duringTriggerr' r+ Nrj )r ru r r} r �"test_finishedBeforeTriggersCleared: s
z7ThreePhaseEventTests.test_finishedBeforeTriggersClearedN)rL rM rN �__doc__r r% r) r, r. r2 r4 r5 r6 rC rH r\ r^ rf rk ro rp rw rx ry 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 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 W s
zSystemEventTests.setUpc C sD | j r | j �� }zt�|� W n ttfy Y nw | j sdS dS )zA
Remove all remaining triggers from the reactor.
N)r� �popr �removeSystemEventTriggerr3 r# rv r r r �tearDown] s
��zSystemEventTests.tearDownc C s t �|||�}| j�|� |S )zS
Add a trigger to the reactor and remember it in C{self.triggers}.
)r �addSystemEventTriggerr� r? )r r �phase�func�tr r r r$ h 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 r� r� �removerv r r r r/ p s
z SystemEventTests.removeTriggerc sJ d}g � � fdd�}| � |||� | �� g � t�|� | �� d g� d S )N�testc � � � d � d S r rr r rs r r r | rt z<SystemEventTests._addSystemEventTriggerTest.<locals>.trigger�r$ r( r �fireSystemEvent)r r� � eventTyper r rs r �_addSystemEventTriggerTestx s
z+SystemEventTests._addSystemEventTriggerTestc C � | � 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�r� r r r r �test_beforePhase� � z!SystemEventTests.test_beforePhasec C r� )z�
L{IReactorCore.addSystemEventTrigger} should accept the C{'during'}
phase and not call the given object until the right event is fired.
r+ Nr� r r r r �test_duringPhase� r� z!SystemEventTests.test_duringPhasec C r� )z�
L{IReactorCore.addSystemEventTrigger} should accept the C{'after'}
phase and not call the given object until the right event is fired.
r- Nr� r r r r �test_afterPhase� r� 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'}.
r� r! c S r r r r r r r r � r z4SystemEventTests.test_unknownPhase.<locals>.<lambda>N)r" r# r$ )r r� r r r �test_unknownPhase� s z"SystemEventTests.test_unknownPhasec �f 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.
r� c rq �Nr' rr r rs r r �
beforeTrigger� rt zASystemEventTests.test_beforePreceedsDuring.<locals>.beforeTriggerc rq �Nr+ rr r rs r r ru � rt zASystemEventTests.test_beforePreceedsDuring.<locals>.duringTriggerr' r+ Nr� �r r� r� ru r rs r �test_beforePreceedsDuring� �
z*SystemEventTests.test_beforePreceedsDuringc r� ) z�
L{IReactorCore.addSystemEventTrigger} should call triggers added to the
C{'during'} phase before it calls triggers added to the C{'after'}
phase.
r� c rq r� rr r rs r r ru � rt z@SystemEventTests.test_duringPreceedsAfter.<locals>.duringTriggerc rq )Nr- rr r rs r r � afterTrigger� rt z?SystemEventTests.test_duringPreceedsAfter.<locals>.afterTriggerr+ r- Nr� )r r� ru r� r rs r �test_duringPreceedsAfter� r� 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.
r� c rD r r r )�triggerDeferredr r r� � � zBSystemEventTests.test_beforeReturnsDeferred.<locals>.beforeTriggerc rq r� rr r rs r r ru � rt zBSystemEventTests.test_beforeReturnsDeferred.<locals>.duringTriggerr' r+ N�r
r$ r( r r� rG r� r )rB 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.
r� c rD r r r ��
firstDeferredr r �firstBeforeTrigger� r� zNSystemEventTests.test_multipleBeforeReturnDeferred.<locals>.firstBeforeTriggerc rD r r r )�secondDeferredr r �secondBeforeTrigger� r� zOSystemEventTests.test_multipleBeforeReturnDeferred.<locals>.secondBeforeTriggerc rq r� rr r rs r r ru � rt zISystemEventTests.test_multipleBeforeReturnDeferred.<locals>.duringTriggerr' r+ Nr� )r r� r� r� ru r �rB 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.
r� c rD r r r r� r r r� r� zaSystemEventTests.test_subsequentBeforeTriggerFiresPriorBeforeDeferred.<locals>.firstBeforeTriggerc r� r )rG r r� r r r� rt zbSystemEventTests.test_subsequentBeforeTriggerFiresPriorBeforeDeferred.<locals>.secondBeforeTriggerc � � � d� �S r� rr r �rB r� r r �thirdBeforeTrigger �
zaSystemEventTests.test_subsequentBeforeTriggerFiresPriorBeforeDeferred.<locals>.thirdBeforeTriggerc rq r� rr r rs r r ru rt z\SystemEventTests.test_subsequentBeforeTriggerFiresPriorBeforeDeferred.<locals>.duringTriggerr' r+ Nr� )r r� r� r� r� ru 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.
r� c rq )Nr8 rr r rs r r r� . rt zJSystemEventTests.test_removeSystemEventTrigger.<locals>.firstBeforeTriggerc rq )Nr< rr r rs r r r� 1 rt zKSystemEventTests.test_removeSystemEventTrigger.<locals>.secondBeforeTriggerr' r8 N)r$ r/ r( r r� )r r� r� r� r rs 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+ r� c S r r r r r r r r B r zKSystemEventTests.test_removeNonExistentSystemEventTrigger.<locals>.<lambda>Nr )r! rU )r$ r/ r" � TypeErrorr r� r3 r# )r �br r r �(test_removeNonExistentSystemEventTrigger: 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� | �� g d�� ��d� | �� g d�� 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 r� )N�r' r8 rr r )rB r� r r �beforeFirstEventU r� zQSystemEventTests.test_interactionBetweenDifferentEvents.<locals>.beforeFirstEventc rq )N�r- r8 rr r rs r r �afterFirstEventY rt zPSystemEventTests.test_interactionBetweenDifferentEvents.<locals>.afterFirstEventz second-eventc r� )N�r' r< rr r r� r r �beforeSecondEvent_ r� zRSystemEventTests.test_interactionBetweenDifferentEvents.<locals>.beforeSecondEventc rq )N�r- r< rr r rs r r �afterSecondEventc rt zQSystemEventTests.test_interactionBetweenDifferentEvents.<locals>.afterSecondEventr' r- r� r� N)r� r� r� )r� r� r� r� r� )r �
firstEventr� r� � secondEventr� r� r r� r �&test_interactionBetweenDifferentEventsJ s6
�
�z7SystemEventTests.test_interactionBetweenDifferentEventsN)rL rM rN r r r� r$ r/ r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� M s& $ r� 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 �nowr r r � test_seconds� s zTimeTests.test_secondsc C sP t j}dd� t _zt �ddd� �}| �|�� d� W |t _n|t _w |�� 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 � dS �N�d r r r r r r � r zITimeTests.test_callLaterUsesReactorSecondsInDelayedCall.<locals>.<lambda>� c S r r r r r r r r � r �i N)r r� � callLaterr( �getTime�cancel�r �oseconds�callr r r �-test_callLaterUsesReactorSecondsInDelayedCall� �
z7TimeTests.test_callLaterUsesReactorSecondsInDelayedCallc C sP t j}dd� t _zt �ddd� �}| �|�� d� W |t _n|t _w |�� 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 r� r� r r r r r r � r zWTimeTests.test_callLaterUsesReactorSecondsAsDelayedCallSecondsFactory.<locals>.<lambda>r� c S r r r r r r r r � r r� N)r r� r� r( r� r� r r r �;test_callLaterUsesReactorSecondsAsDelayedCallSecondsFactory� r� 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� rG � 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.
� rU N)r r� r �
addCleanupr� �time�reset�assertNotEqual)r r� �origTimer 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 r� r rr r )�calledr r �function� rt z2TimeTests.test_cancelDelayedCall.<locals>.functionr c s< z�� � g � W n ty ��� Y d S w ��d � d S r )r( �
BaseException�errbackrG 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 r r r r r r r r � r z;TimeTests.test_cancelCancelledDelayedCall.<locals>.<lambda>N)r r� r� r" r �AlreadyCancelled)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 �@ z
�� tj� j� W n ty ��� Y d S w ��d � d S r �r" r �
AlreadyCalledr� r� r� rG r �r� r� r r r �later� � �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 r� r r� r r� r r r� r� zATimeTests.test_cancelCalledDelayedCallAsynchronous.<locals>.checkc s t �d� � d S �Nr )r r� r )r� r r r� rb zATimeTests.test_cancelCalledDelayedCallAsynchronous.<locals>.laterr r� r� r )r� r� r� r r �(test_cancelCalledDelayedCallAsynchronous
s
z2TimeTests.test_cancelCalledDelayedCallAsynchronousc C sH t �ddd� �}z| �|�� t�� d dk � W |�� d S |�� w )N�
c S r r r r r r r r r z-TimeTests.testCallLaterTime.<locals>.<lambda>rU )r r� �
assertTruer� r� r� r� r r r �testCallLaterTime s zTimeTests.testCallLaterTimec s� t jddd� ddd�� t� � � �d� t� � � �� t� � t jddd� gtd��R dd d
id ft gi�� t� � � �� t� � � fd d
�}t� �|�}t �d|jd �� t� � |S )Nr c S r r r )r �yr r r r ) r z:TimeTests.testDelayedCallStringification.<locals>.<lambda>r r� )r� r� c S r r r r r r r r 1 r �hello�worldy $@c s t � � d S r )�str)�ignored��dcr r �
calledBack7 s z<TimeTests.testDelayedCallStringification.<locals>.calledBack) r r� r� r� r� �ranger
r� rG )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 r� )Nr� r r r r r r� E r� z9TimeTests.testDelayedCallSecondsOverride.<locals>.secondsr� c S r r r r r r r r I r z:TimeTests.testDelayedCallSecondsOverride.<locals>.<lambda>r c S r r r r� r r r r I r � �
N)r � DelayedCallr( r� r� )r r� r� r r r � testDelayedCallSecondsOverride? s �
z(TimeTests.testDelayedCallSecondsOverrideN)rL rM rN r r� r� r� r� r� r� r� r� r� r� r� r r r r r r� � s
r� c @ sH e Zd Zee�ed� d�dd� �Zdd� Zdd� Z d d
� Z
d d � Z dS )
�!CallFromThreadStopsAndWakeUpTestsNz-Nothing to wake up for without thread supportc s t � � � fdd�}t�|� � S )Nc s t �d� t�� jd � d S )Ng�������?)r� �sleepr �callFromThreadrG r �r� r r �wakeY s
z:CallFromThreadStopsAndWakeUpTests.testWakeUp.<locals>.wake)r
r � callInThread)r r r r r �
testWakeUpQ s
z,CallFromThreadStopsAndWakeUpTests.testWakeUpc C s
d| _ d S )NT)�stoppedr r r r �_stopCallFromThreadCallbacka �
z=CallFromThreadStopsAndWakeUpTests._stopCallFromThreadCallbackc C s t �| j|� t �d| j� d S r� )r r �_callFromThreadCallback2r� r r� r r r �_callFromThreadCallbackd s z9CallFromThreadStopsAndWakeUpTests._callFromThreadCallbackc C s< z| � | j� W n ty |�� Y d S w |�d � d S r )r� r r� r� rG r� r r r r h 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 �testCallFromThreadStopsq s z9CallFromThreadStopsAndWakeUpTests.testCallFromThreadStops) rL rM rN r r �IReactorThreadsr r r r r r
r r r r r P 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 ~ s zDelayedTests.setUpc C s | j �� D ]}|�� qd S r )r �valuesr� )r r� r r r r� � s
�zDelayedTests.tearDownc C sX | j �� }tt�� �}g }|D ] }||vr|�|� q|r d| _| �|dt|� � d S )NrU z7Should have been missing no calls, instead was missing ) r r �listr �getDelayedCallsr? r � assertFalse�repr)r �l1�l2�missingr� r r r � checkTimers� s
���zDelayedTests.checkTimersc C s | j |= | �� d S r )r r �r �tagr r r rG � s zDelayedTests.callbackc C s | � |� | �d| j � d S )N� )rG �addTimerr r r r r� � s
zDelayedTests.addCallbackc C s d| _ | �|� | j�d � d S )NrU )r rG r r r r r �done� s
zDelayedTests.donec C s6 t �|d || j�| j| j< | jd7 _| �� d S )N�{�G�z�?rU )r r� r r r )r �whenrG 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 � s z2DelayedTests.testGetDelayedCalls.<locals>.<lambda>) �hasattrr r r r! rG r r� r r� r )r �whichr 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)�success��dcallr r r �checkDeferredCall� s z3DelayedTests.test_active.<locals>.checkDeferredCall)r r� r rG r� r- r� )r r1 r r/ r � test_active� s
zDelayedTests.test_activeN) rL rM rN r r� r rG r� r! r r, r2 r r r r r } s r a�
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 r4 r r r �__init__� r
z ChildResolveProtocol.__init__c C s g | _ g | _d S r )�outputr r r r r �connectionMade� s
z#ChildResolveProtocol.connectionMadec C � | j �|� d S r )r6 r? )r �outr r r � outReceived� rb z ChildResolveProtocol.outReceivedc C r8 r )r r? )r �errr r r � errReceived� rb z ChildResolveProtocol.errReceivedc C s | j �|| j| jf� d | _ d S r )r4 rG r6 r )r �reasonr r r � processEnded� s
z!ChildResolveProtocol.processEndedN)rL rM rN r5 r7 r: r< r> r r r r r3 � s r3 z8cannot run test: reactor doesn't support IReactorProcessc @ � e Zd Zdd� ZdS )� ResolveTestsc s� t j�� �� �}t|d��}tj}|�td|i � W d � n1 s$w Y t j �
� }t j � t
j�|d< t� }t|�}t�|t
jdd|f|� � fdd�}|�|� |S )N�wr �
PYTHONPATH�pythonz-uc sR | \}}}d� |�}dtj�d� }||kr'� �d�|�� |d� |��� d S d S )N� s done 127.0.0.1�asciiz�The child process failed to produce the desired results:
Reason for termination was: {!r}
Output stream was: {!r}
Error stream was: {!r}
)�join�os�linesep�encode�fail�format�getErrorMessage)�resultr= r6 r �expected_outputr r r �
cbFinished s
��z1ResolveTests.testChildResolve.<locals>.cbFinished)rG �path�abspath�mktemp�openr rM �write�resolve_helper�environ�copy�pathseprF �sysr
r3 � spawnProcess�
executabler� )r �
helperPath�
helperFile� reactorName�env�helperDeferred� helperProtorO r r r �testChildResolve s �
�
z ResolveTests.testChildResolveN)rL rM rN rb r r r r r@ s r@ z&Nothing to test without thread supportc @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�CallFromThreadTestsz-
Task scheduling from threads tests.
c C s d| _ t� | _d S r� )r r
r r r r r r : s zCallFromThreadTests.setUpc O s t j|i |�� dS )z)
Override in subclasses.
N)r r )r �args�kwargsr r r �schedule> 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 d S )NrU r� T�r r rG r r r r �addAndMaybeFinishJ s
�zVCallFromThreadTests.test_lotsOfThreadsAreScheduledCorrectly.<locals>.addAndMaybeFinishr� )r� rf r )r rh �ir r r �'test_lotsOfThreadsAreScheduledCorrectlyD 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 �� � g d�� d S )N)rU r� r� )r( )�_��orderr r r r� Z s zECallFromThreadTests.test_threadsAreRunInScheduledOrder.<locals>.checkrU r� r� N)r r� rf r? r r rG )r r� r rl r �"test_threadsAreRunInScheduledOrderT 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 )NrU Trg r r r r � incAndFinishj s zUCallFromThreadTests.test_scheduledThreadsNotRunUntilReactorRuns.<locals>.incAndFinishr )rf r( r r )r ro r r r �+test_scheduledThreadsNotRunUntilReactorRunse s
z?CallFromThreadTests.test_scheduledThreadsNotRunUntilReactorRunsN) rL rM rN r r rf rj rn rp r r r r rc 1 s rc c @ s e Zd ZdZdS )�
MyProtocolz
Sample protocol.
N)rL rM rN r r r r r rq v s rq c @ s e Zd ZdZeZdS )� MyFactoryz
Sample factory.
N)rL rM rN r rq r r r r r rr | s rr c @ r? )�
ProtocolTestsc C s0 t � }|�d �}| �|j|� | �||j� d S r )rr �
buildProtocolr( �factory�assertIsInstancer )r ru r r r r � testFactory� s
zProtocolTests.testFactoryN)rL rM rN rw r r r r rs � � rs 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 rs r r r r r5 � r
zDummyProducer.__init__c C � | j �d� d S )N�resume�rB r? r r r r �resumeProducing� rb z DummyProducer.resumeProducingc C rz )N�stopr| r r r r �
stopProducing� rb zDummyProducer.stopProducingc C rz )N�pauser| r r r r �pauseProducing� rb z DummyProducer.pauseProducingN)rL rM rN r r5 r} r r� r r r r ry � s
ry 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.
)rY )r �datar r r �
writeSomeData� s z SillyDescriptor.writeSomeDatac C r� )z1
Do nothing: bypass the reactor.
Nr r r r r � startWriting� r z SillyDescriptor.startWritingN) rL rM rN r �
bufferSize� connectedr� r� � stopWritingr r r r r� � s r� 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 � �� || _|| _|| _d S r )�superr5 �consumer�
methodName�
methodArgs)r r� r� r� �� __class__r r r5 � s
zReentrantProducer.__init__c s" t � �� t| j| j�| j� d S r )r� r} �getattrr� r� r� r r� r r r} � s
z!ReentrantProducer.resumeProducing)rL rM rN r r5 r} �
__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.
rU r r{ N) r �FileDescriptorr� ry �registerProducerr( rB r" � RuntimeError�r �fd�dpr 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.
rU r r~ N)r r� � disconnectedry r� r( rB r� r r r � test_unconnectedFileDescriptor� s
z,ProducerTests.test_unconnectedFileDescriptorc C sj t � }t� }|j|dd� | �|jdg� |jdd�= |dkr'|�g d�� 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�� streamingr{ N�
writeSequence)� 1� 2� 3� 4s 1234)r� ry r� r( rB r� rT )r r� �
descriptor�producerr r r �_dontPausePullConsumerTest� s
z(ProducerTests._dontPausePullConsumerTestc C �
| � 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.
rT �r� r r r r �!test_dontPausePullConsumerOnWrite
s
z/ProducerTests.test_dontPausePullConsumerOnWritec C r� )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.
r� r� r r r r �)test_dontPausePullConsumerOnWriteSequence �
z7ProducerTests.test_dontPausePullConsumerOnWriteSequencec C s� t � }|dkr g d�}nd}t|||�}|j|dd� t||�|� | �|jdg� |jd d �= |�� | �|jddg� |jd d �= |�� | �|jddg� d S )Nr� )� s� ps ams spamTr� r� r{ )r� r� r� r� r( rB �doWrite)r r� r� r� r� r r r � _reentrantStreamingProducerTest s
z-ProducerTests._reentrantStreamingProducerTestc C r� )z�
Verify that FileDescriptor tracks producer's paused state correctly.
Issue #811, fixed in revision r12857.
rT �r� r r r r �)test_reentrantStreamingProducerUsingWrite@ s
z7ProducerTests.test_reentrantStreamingProducerUsingWritec C r� )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.
r� r� r r r r �1test_reentrantStreamingProducerUsingWriteSequenceG r� z?ProducerTests.test_reentrantStreamingProducerUsingWriteSequenceN) rL rM rN r r� r� r� r� r� r� r� r� r r r r r� � s
r� c @ sp e Zd Zee�ed� d�dd� �Zee�ed� d�dd� �Z ee�
ed� d�ee d �e fd
d ���Z dS ) �PortStringificationTestsNzIReactorTCP is neededc C �F t �dt�� �}|�� j}| �t|��t|��dd||f � |� � S �Nr ����%d not found in %s)
r � listenTCPr �
ServerFactory�getHost�portr� r� �find�
stopListening�r �p�portNor r r �testTCPT �
�z PortStringificationTests.testTCPzIReactorUDP is neededc C r� r� )
r � listenUDPr �DatagramProtocolr� r� r� r� r� r� r� r r r �testUDP] r� z PortStringificationTests.testUDPzIReactorSSL is neededzSSL support is missingc C s\ t �td�}t�dt�� |�||��}|�� j }| �
t |�� t |��dd||f � |�
� S )Nz
server.pemr r� r� )r
�sibpathre r � listenSSLr r� � DefaultOpenSSLContextFactoryr� r� r� r� r� r� )r r �pemr� r� r r r �testSSLf s �
�z PortStringificationTests.testSSL)
rL rM rN r r � IReactorTCPr r� � IReactorUDPr� � IReactorSSLr r� r r r r r� S s
r� c @ r? )�ConnectorReprTestsc C s@ t ddt� dt� �}d}|t|�|�� f }| �t|�|� d S )N� localhosti� r zA<twisted.internet.tcp.Connector instance at 0x%x disconnected %s>)r r �id�getDestinationr( r )r �c�expectr r r �
test_tcp_repru s z ConnectorReprTests.test_tcp_reprN)rL rM rN r� r r r r r� t rx r� )3r rG rY r� �unittestr �twisted.internetr r r r r r r �twisted.internet.deferr
r �twisted.internet.tcpr �twisted.pythonr
�twisted.trial.unittestr r �_ssl� ImportError� supportedr r� r� r r rU �ProcessProtocolr3 �IReactorProcessr@ r rc �Protocolrq �Factoryrr rs ry r� r� r� r� r� r� r r r r �<module> sb $ �
0 @ F-Z �, �Az!