File: //lib/python3/dist-packages/twisted/test/__pycache__/test_cooperator.cpython-310.pyc
o
�b[S � @ s� d Z ddlmZmZmZ ddlmZ G dd� d�ZG dd� d�ZG dd � d ej �Z
G d
d � d e �Z G d d
� d
ej �Z
G dd� dej �ZdS )z_
This module contains tests for L{twisted.internet.task.Cooperator} and
related functionality.
� )�defer�reactor�task)�unittestc @ s e Zd ZdZdd� Zdd� ZdS )�FakeDelayedCallzA
Fake delayed call which lets us simulate the scheduler.
c C s || _ d| _dS )z+
A function to run, later.
FN)�func� cancelled��selfr � r �>/usr/lib/python3/dist-packages/twisted/test/test_cooperator.py�__init__ s
zFakeDelayedCall.__init__c C �
d| _ dS )z.
Don't run my function later.
TN)r �r
r r r �cancel �
zFakeDelayedCall.cancelN)�__name__�
__module__� __qualname__�__doc__r
r r r r r r s r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �
FakeSchedulerz/
A fake scheduler for testing against.
c C s
g | _ dS )zD
Create a fake scheduler with a list of work to do.
N)�workr r r r r
& r zFakeScheduler.__init__c C s t |�}| j�|� |S )z;
Schedule a unit of work to be done later.
)r r �append)r
�thunk�unitr r r �__call__, s zFakeScheduler.__call__c C s* | j g }| _ |D ] }|js|�� q dS )zL
Do all of the work that is currently available to be done.
N)r r r )r
r r r r r �pump4 s ��zFakeScheduler.pumpN)r r r r r
r r r r r r r ! s
r c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"S )#�CooperatorTests�donec C s |� tj� | jS �N)�trapr �SchedulerStopped�RESULT)r
�errr r r �ebIterA s zCooperatorTests.ebIterc C s | � � d S r )�fail)r
�ignr r r �cbIterE s zCooperatorTests.cbIterc s" � fdd���d�� �fdd��S )zU
Test that Cooperators refuse new tasks when they have been stopped.
c sJ t �� }|�� |�td�| �}|�� j� |�� j� |�� fdd��S )Nr c � � � | � j�S r �� assertEqualr" ��resultr r r �<lambda>S � zNCooperatorTests.testStoppedRejectsNewTasks.<locals>.testwith.<locals>.<lambda>) r �
Cooperator�stop� coiterate�iter� addCallbackr' �
addErrbackr$ )�stuff�c�dr r r �testwithM s z<CooperatorTests.testStoppedRejectsNewTasks.<locals>.testwithNc s � t �� �S r )r �Deferred�r& )r8 r r r- U s z<CooperatorTests.testStoppedRejectsNewTasks.<locals>.<lambda>)r3 r r )r
r8 r �testStoppedRejectsNewTasksH s z*CooperatorTests.testStoppedRejectsNewTasksc s^ t �� }dd� � d� _|�� � �}|��j� |��j� |�� � �fdd�}|�|� |S )zq
Test that a running iterator will not run to completion when the
cooperator is stopped.
c s s � t d�E d H d S )N� )�ranger r r r �myiter^ � �z/CooperatorTests.testStopRunning.<locals>.myiter���c s �� | �j� �� � jd� d S �Nr@ )r* r" �valuer+ �r> r
r r � doassertsg s z2CooperatorTests.testStopRunning.<locals>.doasserts) r r/ rB r1 r3 r' r4 r$ r0 )r
r6 r7 rD r rC r �testStopRunningW s
z CooperatorTests.testStopRunningc sv t �� �t �� ����fdd�}t�� � � �|� �}� �fdd�}��|� |��j� |��j� |��fdd��S )z�
An iterator run with L{Cooperator.coiterate} paused on a L{Deferred}
yielded by that iterator will fire its own L{Deferred} (the one
returned by C{coiterate}) when L{Cooperator.stop} is called.
c 3 s$ � t �d�jd � � V ��� d S �Nr )r � callLater�callbackr% r )� outstandingDr
� testControlDr r r> w s � z3CooperatorTests.testStopOutstanding.<locals>.myiterc s � � � ��d� d S )N� arglebargle)r0 rH r: )r6 rI r r � stopAndGo s z6CooperatorTests.testStopOutstanding.<locals>.stopAndGoc r( r r) r+ r r r r- � r. z5CooperatorTests.testStopOutstanding.<locals>.<lambda>) r r9 r r/ r1 r3 r' r4 r$ )r
r> r7 rL r )r6 rI r
rJ r �testStopOutstandingn s
z#CooperatorTests.testStopOutstandingc C s( t �� }dd� }|�|� �}| �|t�S )Nc s s
� t � �r )� RuntimeErrorr r r r r> � s �z3CooperatorTests.testUnexpectedError.<locals>.myiter�r r/ r1 �
assertFailurerN )r
r6 r> r7 r r r �testUnexpectedError� s z#CooperatorTests.testUnexpectedErrorc C �( dd� }t �� }|�|� �}| �|t�S )Nc s s& � t �� } t�d| jt� � | V d S rF )r r9 r rG �errbackrN )�Dr r r r> � s �
z@CooperatorTests.testUnexpectedErrorActuallyLater.<locals>.myiterrO �r
r> r6 r7 r r r � testUnexpectedErrorActuallyLater� s z0CooperatorTests.testUnexpectedErrorActuallyLaterc C rR )Nc s s � t �t� �V d S r )r r% rN r r r r r> � r? zCCooperatorTests.testUnexpectedErrorNotActuallyLater.<locals>.myiterrO rU r r r �#testUnexpectedErrorNotActuallyLater� s z3CooperatorTests.testUnexpectedErrorNotActuallyLaterc s^ g � � fdd�}g d��t �� }g }�D ] }|�|�||��� qt�|��� ��fdd��S )Nc 3 s � | D ]
}� � |� d V qd S r �r )�things�th)�Lr r r> � s
�
�z/CooperatorTests.testCooperation.<locals>.myiter)�abc�� � r< �def)� � � c s �� t� �tt�� d��S �Nr )r* �tuple�sum�zipr: �r[ �groupsOfThingsr
r r r- � s z1CooperatorTests.testCooperation.<locals>.<lambda>)r r/ r r1 r � DeferredListr3 )r
r> r6 �tasksr5 r rh r �testCooperation� s
�z CooperatorTests.testCooperationc s~ g �� �fdd�}G dd� d�� t j� d�}|�|� ��| j� |j�� |�� |�� | � � j
� | � �t t
d��� d S )Nc 3 s2 � t d�D ]} ��| � | dkrd� _| V qd S )N�d � T)r= r �stopped)�i��_TPF�outputr r r> � s �
�z6CooperatorTests.testResourceExhaustion.<locals>.myiterc @ � e Zd ZdZdd� ZdS )z4CooperatorTests.testResourceExhaustion.<locals>._TPFFc S s | j S r )ro r r r r r � s z=CooperatorTests.testResourceExhaustion.<locals>._TPF.__call__N)r r r ro r r r r r rr � s rr )�terminationPredicateFactory�
)r r/ r1 r4 r$ � _delayedCallr �_tickr0 �
assertTruero r* �listr= )r
r> r6 r rq r �testResourceExhaustion� s
z&CooperatorTests.testResourceExhaustionc s� g �G dd� d�� � ��fdd�}t j|dd� d����td��}g ���fd d
�}|�|� d }�sS|d 7 }�rF��d ��� |d 7 }�s9|d
krO��d� �r3dS dS )a*
If a callback to a deferred returned by coiterate calls coiterate on
the same Cooperator, we should make sure to only do the minimal amount
of scheduling work. (This test was added to demonstrate a specific bug
that was found while writing the scheduler.)
c @ s" e Zd Zdd� Zdefdd�ZdS )z9CooperatorTests.testCallbackReCoiterate.<locals>.FakeCallc S s
|| _ d S r �r r r r r r
� s
zBCooperatorTests.testCallbackReCoiterate.<locals>.FakeCall.__init__�returnc S s d| j �d�S )Nz
<FakeCall �>r| r r r r �__repr__� s zBCooperatorTests.testCallbackReCoiterate.<locals>.FakeCall.__repr__N)r r r r
�strr r r r r �FakeCall� s r� c s& �� �t��� ��� | �� �d S rA )� assertFalse�reprr ��f)r� �callsr
r r �sched� s z6CooperatorTests.testCallbackReCoiterate.<locals>.schedc S � dd� S )Nc S � dS �NTr r r r r r- � � zKCooperatorTests.testCallbackReCoiterate.<locals>.<lambda>.<locals>.<lambda>r r r r r r- � � z9CooperatorTests.testCallbackReCoiterate.<locals>.<lambda>�� schedulerru r c s � � td����j� d S rd )r1 r2 �addBothr r: )r6 r r r � anotherTask� s z<CooperatorTests.testCallbackReCoiterate.<locals>.anotherTaskr r^ �2 zCooperator took too longN)r r/ r1 r2 r3 �popr r% )r
r� r7 r� r r )r� r6 r� r r
r �testCallbackReCoiterate� s( �
�
�z'CooperatorTests.testCallbackReCoiteratec s� dg� � fdd�}t j|d�}|�tddg��}|�tddg��}| �� d j|j� |�� | �� d j � | �|j
� d � |�� | � � d j � | � |j
� |�tddg�� | �� d j � | �|j
� d � dS )a
If the last task in a Cooperator is removed, the scheduled call for
the next tick is cancelled, since it is no longer necessary.
This behavior is useful for tests that want to assert they have left
no reactor state behind when they're done.
Nc s t | �� d< � d S rF )r r� �r� r r r� s zFCooperatorTests.test_removingLastTaskStopsScheduledCall.<locals>.sched)r� r^ r_ r )
r r/ � cooperater2 r* r rx r0 r� r rw ry � assertIsNone)r
r� �coop�task1�task2r r� r �'test_removingLastTaskStopsScheduledCall� s z7CooperatorTests.test_removingLastTaskStopsScheduledCallc C s t �� }| �|j� dS )zm
L{Cooperator.running} reports C{True} if the L{Cooperator}
was started on creation.
N)r r/ ry �running�r
r6 r r r �test_runningWhenStarted" s z'CooperatorTests.test_runningWhenStartedc C s t jdd�}| �|j� dS )zk
L{Cooperator.running} reports C{False} if the L{Cooperator}
has not been started.
F��startedN)r r/ r� r� r� r r r �test_runningWhenNotStarted* s z*CooperatorTests.test_runningWhenNotStartedc C s0 t jdd�}|�� | �|j� | �|j� dS )zb
L{Cooperator.running} reports C{True} when the L{Cooperator}
is running.
Fr� N)r r/ �start�
addCleanupr0 ry r� r� r r r �test_runningWhenRunning2 s z'CooperatorTests.test_runningWhenRunningc C s, t jdd�}|�� |�� | �|j� dS )zj
L{Cooperator.running} reports C{False} after the L{Cooperator}
has been stopped.
Fr� N)r r/ r� r0 r� r� r� r r r �test_runningWhenStopped< s z'CooperatorTests.test_runningWhenStoppedN)r r r r" r$ r' r; rE rM rQ rV rW rl r{ r� r� r� r� r� r� r r r r r > s$
*$
r c @ s e Zd ZdZdS )�UnhandledExceptionz0
An exception that should go unhandled.
N)r r r r r r r r r� G s r� c @ rt )�
AliasTestszg
Integration test to verify that the global singleton aliases do what
they're supposed to.
c s8 t �� � � fdd�}|� }t�|�}| �|tjj� � S )zZ
L{twisted.internet.task.cooperate} ought to run the generator that it is
c 3 s" � dV dV dV � � d� d S )Nr^ r_ r< �yay)rH r �r7 r r �doitY s
�z'AliasTests.test_cooperate.<locals>.doit)r r9 r r� �assertIn�_theCooperator�_tasks)r
r� �it�theTaskr r� r �test_cooperateS s
zAliasTests.test_cooperateN)r r r r r� r r r r r� M s r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*S )+�
RunStateTestsz�
Tests to verify the behavior of L{CooperativeTask.pause},
L{CooperativeTask.resume}, L{CooperativeTask.stop}, exhausting the
underlying iterator, and their interactions with each other.
c C sV d| _ d| _d| _g | _t� | _tj| jdd� d�| _| j� | �
� �| _| j� � dS )z�
Create a cooperator with a fake scheduler and a termination predicate
that ensures only one unit of work will take place per tick.
Fc S r� )Nc S r� r� r r r r r r- z r� z7RunStateTests.setUp.<locals>.<lambda>.<locals>.<lambda>r r r r r r- z r� z%RunStateTests.setUp.<locals>.<lambda>r� N) � _doDeferNext� _doStopNext�
_doDieNextr r r� r r/ �
cooperatorr� �workerr� r r r r �setUpl s �zRunStateTests.setUpc c sb � d} |d7 }| j r d| _ t�� }| j�|� |V n| jr!dS | jr't� �| j�|� |V q)z�
This is a sample generator which yields Deferreds when we are testing
deferral and an ascending integer count otherwise.
r Tr^ FN)r� r r9 r r r� r� r� )r
rp r7 r r r r� s � �zRunStateTests.workerc C s | ` | `dS )z�
Drop references to interesting parts of the fixture to allow Deferred
errors to be noticed when things start failing.
N)r r� r r r r �tearDown� s zRunStateTests.tearDownc C r )z@
Defer the next result from my worker iterator.
TN)r� r r r r � deferNext� r zRunStateTests.deferNextc C r )zm
Make the next result from my worker iterator be completion (raising
StopIteration).
TN)r� r r r r �stopNext� s
zRunStateTests.stopNextc C s dd� }d| _ dS )zk
Make the next result from my worker iterator be raising an
L{UnhandledException}.
c S s | � t� d S r )r r� )�failurer r r �ignoreUnhandled� �
z.RunStateTests.dieNext.<locals>.ignoreUnhandledTN)r� )r
r� r r r �dieNext� s
zRunStateTests.dieNextc C s� | j �� | �| jdg� | j �� | �| jddg� | j�� | j �� | �| jddg� | j�� | �| jddg� | j �� | �| jg d�� dS )z�
Cooperators should stop running their tasks when they're paused, and
start again when they're resumed.
r^ r_ r] N)r� r r* r r �pause�resumer r r r �test_pauseResume� s
z RunStateTests.test_pauseResumec C s< | � tj| jj� | j�� | j�� | � tj| jj� dS )z�
L{CooperativeTask.resume} should raise a L{TaskNotPaused} exception if
it was not paused; e.g. if L{CooperativeTask.pause} was not invoked
more times than L{CooperativeTask.resume} on that object.
N)� assertRaisesr � NotPausedr� r� r r r r �test_resumeNotPaused� s
z"RunStateTests.test_resumeNotPausedc C s� | j �� | j�� | �| jg � | j �� | j�� | �| jg � | j �� | j�� | �| jg � | j �� | j�� | �| jdg� dS )z~
Pauses on tasks should behave like a stack. If a task is paused twice,
it needs to be resumed twice.
r^ N)r r� r� r r* r r� r r r r �test_pauseTwice� s
z RunStateTests.test_pauseTwicec C s� | � � | j�� | �t| j�d� | �| jd tj� | j�� | �t| j�d� | j �
� | j�� | �t| j�d� | j � � | j�� | �t| j�d� | jd � d� | j�� | �t| j�d� | �| jd d� dS )a
C{pause()}ing a task while it is waiting on an outstanding
L{defer.Deferred} should put the task into a state where the
outstanding L{defer.Deferred} must be called back I{and} the task is
C{resume}d before it will continue processing.
r^ r zSTUFF!r_ N)
r� r� r r* �lenr �assertIsInstancer r9 r r� r� rH r r r r �test_pauseWhileDeferred� s
z%RunStateTests.test_pauseWhileDeferredc s� | j �� }| j �� }g � g �g }g }� fdd�}�fdd�}|�|� |�|� |�|j� |�|j� | �� | j�� | �t� �d� | �t��d� | � � d | j j
� | � �d | j j
� | �|dg� | �|dg� dS ) a/
L{CooperativeTask.whenDone} returns a Deferred which fires when the
Cooperator's iterator is exhausted. It returns a new Deferred each
time it is called; callbacks added to other invocations will not modify
the value that subsequent invocations will fire with.
c � � � | � dS )Nr^ rX r+ )�results1r r � callbackOne r� z0RunStateTests.test_whenDone.<locals>.callbackOnec r� )Nr_ rX r+ )�results2r r � callbackTwo r� z0RunStateTests.test_whenDone.<locals>.callbackTwor^ r r_ N) r �whenDoner3 r r� r� r r* r� �assertIs� _iterator)r
� deferred1� deferred2�final1�final2r� r� r )r� r� r �
test_whenDone s(
zRunStateTests.test_whenDonec C sV | j �� }g }|�|j� | �� | j�� | �t|�d� | �|d � t
�t
� dS )z�
L{CooperativeTask.whenDone} returns a L{defer.Deferred} that will fail
when the iterable's C{next} method raises an exception, with that
exception.
r^ r N) r r� r4 r r� r� r r* r� �checkr� )r
r� �resultsr r r �test_whenDoneError/ s
z RunStateTests.test_whenDoneErrorc C sR | j �� }g }|�|j� | j �� | �t|�d� | �|d �t j�t j� dS )z�
L{CooperativeTask.whenDone} returns a L{defer.Deferred} that fails with
L{TaskStopped} when the C{stop} method is called on that
L{CooperativeTask}.
r^ r N) r r� r4 r r0 r* r� r� � TaskStopped)r
r� �errorsr r r �test_whenDoneStop= s
z RunStateTests.test_whenDoneStopc C s> | � � | j�� g }| j�� �|j� | �|| jjg� dS )z�
L{CooperativeTask.whenDone} will return a L{defer.Deferred} that will
succeed immediately if its iterator has already completed.
N) r� r� r r r� r3 r r* r� )r
r� r r r �test_whenDoneAlreadyDoneJ s
z&RunStateTests.test_whenDoneAlreadyDonec C sf | j �� | j�� | �t| j�d� | �t j| j j� | �t j| j j � | j�� | �| jg � dS )z�
C{stop()}ping a task should cause it to be removed from the run just as
C{pause()}ing, with the distinction that C{resume()} will raise a
L{TaskStopped} exception.
r N)
r r0 r� r r* r� r r� r� r� r r r r �test_stopStopsU s
z RunStateTests.test_stopStopsc C s: | j �� | j �� | j �� | j�� | �| jg � dS )z�
C{resume()}ing a paused, stopped task should be a no-op; it should not
raise an exception, because it's paused, but neither should it actually
do more work from the task.
N)r r� r0 r� r� r r* r r r r r �test_pauseStopResumed s
z"RunStateTests.test_pauseStopResumec C s� | � � | j�� | j�� }| �| jjd� g }|�|j � | j�� | j�
� | j�� |� d� | j�� | �|dg� | �| jg � dS )a
As a corrolary of the interaction of C{pause()} and C{unpause()},
C{stop()}ping a task which is waiting on a L{Deferred} should cause the
task to gracefully shut down, meaning that it should not be unpaused
when the deferred fires.
r^ � N) r� r� r r r� r* r � _pauseCountr� r r0 rH )r
r7 r� r r r �test_stopDeferredp s
z RunStateTests.test_stopDeferredc C �( | � � | j�� | �tj| jj� dS )zx
C{stop()}ping a L{CooperativeTask} whose iterator has been exhausted
should raise L{TaskDone}.
N)r� r� r r� r �TaskDoner0 r r r r �test_stopExhausted� �
z RunStateTests.test_stopExhaustedc C r� )z�
C{stop()}ping a L{CooperativeTask} whose iterator has encountered an
error should raise L{TaskFailed}.
N)r� r� r r� r �
TaskFailedr0 r r r r �test_stopErrored� r� z RunStateTests.test_stopErroredc sL g � � �fdd�}�j �� �|� ��� �j�� ��� �j jdg� dS )a�
If a callback of a L{Deferred} from L{CooperativeTask.whenDone} calls
C{Cooperator.stop} on its L{CooperativeTask._cooperator}, the
L{Cooperator} will stop, but the L{CooperativeTask} whose callback is
calling C{stop} should already be considered 'stopped' by the time the
callback is running, and therefore removed from the
L{CoooperativeTask}.
c s" � � | � �j�� � � d� d S )Nr )r r� r0 r+ ��callbackPhasesr
r r �stopit� s
z;RunStateTests.test_stopCooperatorReentrancy.<locals>.stopitr N)r r� r3 r� r� r r* r� )r
r� r r� r � test_stopCooperatorReentrancy� s
z+RunStateTests.test_stopCooperatorReentrancyN)r r r r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� e s, +
r� N)r �twisted.internetr r r �
twisted.trialr r r �TestCaser � Exceptionr� r� r� r r r r �<module> s