← Back
Editing: test_team.cpython-38.pyc
U W[>% � @ s� d Z ddlmZmZmZ ddlmZ ddlmZm Z ddl m Z ddl m Z ddlmZmZmZmZ G d d � d e ed ��ZG d d � d e�ZdS )z& Tests for L{twisted._threads._team}. � )�absolute_import�division�print_function)�SynchronousTestCase)�call�get)�proxyForInterface)�Failure� )�IWorker�Team�createMemoryWorker� AlreadyQuitc s( e Zd ZdZdd� Z� fdd�Z� ZS )�ContextualWorkerz: A worker implementation that supplies a context. c K s || _ || _dS )z: Create with a real worker and a context. N)� _realWorker�_context)�selfZ realWorkerZctx� r �A/usr/lib/python3/dist-packages/twisted/_threads/test/test_team.py�__init__ s zContextualWorker.__init__c s t t� ��� �fdd�� dS )z� Perform the given work with the context given to __init__. @param work: the work to pass on to the real worker. c s t � j��S �N)r r r �r Zworkr r �<lambda>% � z%ContextualWorker.do.<locals>.<lambda>N)�superr �dor �� __class__r r r s zContextualWorker.do)�__name__� __module__� __qualname__�__doc__r r � __classcell__r r r r r s r 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 )#� TeamTestsz Tests for L{Team} c sl t � \}� _t|dd�� _g � _g � _g � _g � _dd� � _� fdd�}g � _ � fdd�}t |||�� _ d S ) zl Set up a L{Team} with inspectable, synchronous workers that can be single-stepped. � coordinator��workerc S s dS )NFr r r r r r 9 r z!TeamTests.setUp.<locals>.<lambda>c sz �� � r d S t� \} ��j��� �j��� t| t�j�d�� �j�� � �j�� � � j �� ���fdd�}|� _ � S )Nr% c s" �� �j �� � �j��� d S r )�allUnquitWorkers�remove�activePerformersr )�cw� performer�realQuitr r r � quitAndRemoveE s z<TeamTests.setUp.<locals>.createWorker.<locals>.quitAndRemove) � noMoreWorkersr �workerPerformers�appendr) r �len�allWorkersEverr' �quit)r& r- �r )r* r+ r, r � createWorker; s z%TeamTests.setUp.<locals>.createWorkerc s � j �t� � d S r )�failuresr0 r r r4 r r � logExceptionM s z%TeamTests.setUp.<locals>.logExceptionN) r �coordinateOncer r$ r/ r2 r' r) r. r6 r �team)r r$ r5 r7 r r4 r �setUp. s zTeamTests.setUpc C s d}| � � rd}q|S )a Perform all work currently scheduled in the coordinator. @return: whether any coordination work was performed; if the coordinator was idle when this was called, return L{False} (otherwise L{True}). @rtype: L{bool} FT)r8 )r Zdidr r r � coordinateR s zTeamTests.coordinatec C s> d}|r:| � � }| jD ]}|| jkr|� q|p6| � � }qdS )zj Perform all work on the coordinator and worker performers that needs to be done. TN)r; r/ r) )r Z continuingr+ r r r �performAllOutstandingWorka s z#TeamTests.performAllOutstandingWorkc sb � fdd�� | j �� � | �� | �| j �� jd� | �� | �� jd� | �| j �� jd� dS )zd L{Team.do} does the work in a worker created by the createWorker callable. c s t d�� _d S )Nr& )r �whor �� somethingr r r? t s z4TeamTests.test_doDoesWorkInWorker.<locals>.something� r N)r9 r r; � assertEqual� statistics�busyWorkerCountr< r= r4 r r> r �test_doDoesWorkInWorkero s z!TeamTests.test_doDoesWorkInWorkerc C s8 | j �� }| �|jd� | �|jd� | �|jd� dS )z� L{Team.statistics} returns an object with idleWorkerCount, busyWorkerCount, and backloggedWorkCount integer attributes. r N)r9 rB rA �idleWorkerCountrC �backloggedWorkCount)r Zstatsr r r �test_initialStatistics~ s z TeamTests.test_initialStatisticsc C s* | j �d� | �� | �t| j�d� dS )zN L{Team.grow} increases the number of available idle workers. � N)r9 �growr< rA r1 r/ r4 r r r �test_growCreatesIdleWorkers� s z%TeamTests.test_growCreatesIdleWorkersc sL � fdd�� _ � j�d� � �� � �t� j�d� � �� j�� jd� dS )z� L{Team.grow} increases the number of available idle workers until the C{createWorker} callable starts returning None. c s t � j�dkS )N� )r1 r2 r r4 r r r � r z0TeamTests.test_growCreateLimit.<locals>.<lambda>rH rK N) r. r9 rI r< rA r1 r2 rB rE r4 r r4 r �test_growCreateLimit� s z TeamTests.test_growCreateLimitc C s> | j �d� | �� | j �d� | �� | �t| j�d� dS )zG L{Team.shrink} will quit the given number of workers. rH rK r N)r9 rI r< �shrinkrA r1 r' r4 r r r �test_shrinkQuitsWorkers� s z!TeamTests.test_shrinkQuitsWorkersc C s` | j �d� | �� | �t| j�d� | j �� | �t| j�d� | �� | �t| j�d� dS )zU L{Team.shrink} with no arguments will stop all outstanding workers. � r N)r9 rI r< rA r1 r' rM r4 r r r �test_shrinkToZero� s zTeamTests.test_shrinkToZeroc s� | j �d� | �� � fdd�� d� _| �| j �� jd� td�D ]}| j �� � qB| �� | �| j �� jd� dd� | _ | j �� � | �� | �| j �� jd� | �| j �� j d� | � � | �| j �� j d� | �� jd� d S ) z� When no additional workers are available, the given work is backlogged, and then performed later when the work was. rK c s � j d7 _ d S )Nr@ )�timesr r> r r r? � s z@TeamTests.test_moreWorkWhenNoWorkersAvailable.<locals>.somethingr c S s dS )NTr r r r r r � r z?TeamTests.test_moreWorkWhenNoWorkersAvailable.<locals>.<lambda>r@ � N) r9 rI r; rQ rA rB rE �ranger r. rF r< )r �ir r> r �#test_moreWorkWhenNoWorkersAvailable� s" z-TeamTests.test_moreWorkWhenNoWorkersAvailablec C sB | j �dd� � | �� | �t| j�d� | �| jd jt� dS )z� When an exception is raised in a task passed to L{Team.do}, the C{logException} given to the L{Team} at construction is invoked in the exception context. c S s dd S )Nr@ r r r r r r r � r z0TeamTests.test_exceptionInTask.<locals>.<lambda>r@ r N)r9 r r< rA r1 r6 �type�ZeroDivisionErrorr4 r r r �test_exceptionInTask� s z TeamTests.test_exceptionInTaskc C s0 | j �� | �t| j j� | �t| j jt� dS )zx L{Team.quit} causes future invocations of L{Team.do} and L{Team.quit} to raise L{AlreadyQuit}. N)r9 r3 � assertRaisesr r �listr4 r r r � test_quit� s zTeamTests.test_quitc C sZ t d�D ]}| j�t� q| �� | j�� | �� | �t| j�d� | � t | j j� dS )zk L{Team.quit} causes all idle workers, as well as the coordinator worker, to quit. rO r N) rS r9 r rZ r< r3 rA r1 r' rY r r$ �r �xr r r �test_quitQuits� s zTeamTests.test_quitQuitsc C s� | j �d� td�D ]}| j �t� q| �� | j �� | �� | �t| j �d� | � � | �t| j �d� | � t | j j� dS )z| L{Team.quit} causes all busy workers to be quit once they've finished the work they've been given. rO rH r N)r9 rI rS r rZ r; r3 rA r1 r' r< rY r r$ r\ r r r �test_quitQuitsLaterWhenBusy� s z%TeamTests.test_quitQuitsLaterWhenBusyc s^ �j �t� �j jj� � �fdd�}|�j j_�j �� ��t�j j� ��t�j jt� dS )z� If work happens after L{Team.quit} sets its C{Quit} flag, but before any other work takes place, the L{Team} should still exit gracefully. c s � � �� � d S r )r< r �Z originalSetr r r �performWorkConcurrently s zOTeamTests.test_quitConcurrentWithWorkHappening.<locals>.performWorkConcurrentlyN)r9 r rZ Z_quit�setr3 rY r )r ra r r` r �$test_quitConcurrentWithWorkHappening s z.TeamTests.test_quitConcurrentWithWorkHappeningc C s^ t d�D ]}| j�t� q| �� | �t| j�d� | j�d� | � � | �t| j�d� dS )zl L{Team.shrink} will wait for busy workers to finish being busy and then quit them. rO � rK N) rS r9 r rZ r; rA r1 r' rM r< r\ r r r �test_shrinkWhenBusy s z TeamTests.test_shrinkWhenBusyN)r r r r! r: r; r< rD rG rJ rL rN rP rU rX r[ r^ r_ rc re r r r r r# ) s" $ r# N)r! Z __future__r r r Ztwisted.trial.unittestr Ztwisted.python.contextr r Ztwisted.python.componentsr Ztwisted.python.failurer � r r r r r r# r r r r �<module> s
Save File
Cancel