File: //usr/lib/python3/dist-packages/twisted/test/__pycache__/test_threads.cpython-38.pyc
U
��W[�3 � @ sF d Z ddlmZmZ ddlZddlZddlZddlmZ ddl m
Z
ddl m Z m
Z
mZmZmZmZ ddlmZmZmZmZ G dd � d ej�ZG d
d � d �ZG d d
� d
ej�ZG dd� dej�ZdZ G dd� dej �Z G dd� dej�Z e� e d�dk�reee fD ] Z!de!_"�qnddl#Z#e�$e d�dk�rBe fD ] Z!de!_"�q4dS )zC
Test methods in twisted.internet.threads and reactor thread APIs.
� )�division�absolute_importN)�unittest)�range)�reactor�defer�
interfaces�threads�protocol�error)�failure�
threadable�log�
threadpoolc @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )�ReactorThreadsTestsz.
Tests for the reactor threading API.
c C s8 t �d� | �t jjd� t �d� | �t jjd� dS )z:
Try to change maximum number of threads.
�" � N)r �suggestThreadPoolSize� assertEqualr �max��self� r �;/usr/lib/python3/dist-packages/twisted/test/test_threads.py�test_suggestThreadPoolSize s
z.ReactorThreadsTests.test_suggestThreadPoolSizec C s t �tjd�S )z�
The reactor's threadpool is only available when the reactor is running,
so to have a sane behavior during the tests we make a dummy
L{threads.deferToThread} call.
r )r �
deferToThread�time�sleepr r r r �_waitForThread$ s z"ReactorThreadsTests._waitForThreadc s � fdd�}� � � �|�S )z�
Test callInThread functionality: set a C{threading.Event}, and check
that it's not in the main thread.
c sT t �� �g � � �fdd�}t�|� ��d� ��� sB��d� n��� dg� d S )Nc s � � t�� � ��� d S �N)�appendr
Z isInIOThread�setr ��result�waiterr r � threadedFunc5 s zGReactorThreadsTests.test_callInThread.<locals>.cb.<locals>.threadedFunc�x z Timed out waiting for event.F)� threading�Eventr � callInThread�wait�isSet�failr ��ignr% r r" r �cb2 s
z1ReactorThreadsTests.test_callInThread.<locals>.cb�r � addCallback�r r/ r r r �test_callInThread- s
z%ReactorThreadsTests.test_callInThreadc C s dd� }| � � �|�S )zk
Test callFromThread functionality: from the main thread, and from
another thread.
c sF t �� }t �� � � fdd�}t�|� t�|jd � t j|� gdd�S )Nc s t �� jd � d S r )r �callFromThread�callbackr �ZfiredByOtherThreadr r r% K s zIReactorThreadsTests.test_callFromThread.<locals>.cb.<locals>.threadedFuncT)ZfireOnOneErrback)r �Deferredr r) r4 r5 Z DeferredList)r. ZfiredByReactorThreadr% r r6 r r/ G s
�z3ReactorThreadsTests.test_callFromThread.<locals>.cbr0 r2 r r r �test_callFromThreadB s
z'ReactorThreadsTests.test_callFromThreadc s � fdd�}� � � �|�S )zT
Try to make an overflow on the reactor waker using callFromThread.
c s\ d �_ t�� � �� fdd�}t�|� � �d� � �� sB��d� �j d k rXt��j �S d S )Nc sJ t d�D ]4} zt�dd� � W q t�� � _Y q>Y qX q��� d S )Ni�� c S s d S r r r r r r �<lambda>c � z^ReactorThreadsTests.test_wakerOverflow.<locals>.cb.<locals>.threadedFunction.<locals>.<lambda>)r r r4 r ZFailurer! )�i)r r$ r r �threadedFunction^ s
zLReactorThreadsTests.test_wakerOverflow.<locals>.cb.<locals>.threadedFunctionr& �Timed out waiting for event) r r' r( r r) r* r+ r, r )r. r< r )r$ r r/ [ s
z2ReactorThreadsTests.test_wakerOverflow.<locals>.cbr0 r2 r r r �test_wakerOverflowW s z&ReactorThreadsTests.test_wakerOverflowc sJ t �� �g �g � � ����fdd�}� ���fdd�}��� �|��|�S )zK
Utility method to test L{threads.blockingCallFromThread}.
c s. � ���fdd�}t �|� t��j��� �S )Nc
sT zt �t��} W n, tk
r< } z� �|� W 5 d }~X Y n X ��| � ��� d S r )r ZblockingCallFromThreadr � Exceptionr r! )�r�e)�errors� reactorFunc�resultsr$ r r r% x s
zRReactorThreadsTests._testBlockingCallFromThread.<locals>.cb1.<locals>.threadedFunc)r r) r r r* Z
getTimeoutr- �rB rC rD r r$ r r �cb1w s
z<ReactorThreadsTests._testBlockingCallFromThread.<locals>.cb1c s �� � s��d� �� fS )Nr= )r+ r, �r. )rB rD r r$ r r �cb2� s
z<ReactorThreadsTests._testBlockingCallFromThread.<locals>.cb2)r' r( r r1 ZaddBoth)r rC rF rH r rE r �_testBlockingCallFromThreadp s
z/ReactorThreadsTests._testBlockingCallFromThreadc s$ dd� }� fdd�}� � |��|�S )z�
Test blockingCallFromThread facility: create a thread, call a function
in the reactor using L{threads.blockingCallFromThread}, and verify the
result returned.
c S s
t �d�S )N�foo)r Zsucceedr r r r rC � s zDReactorThreadsTests.test_blockingCallFromThread.<locals>.reactorFuncc s � � | d d d� d S )Nr rJ �r ��resr r r r/ � s z;ReactorThreadsTests.test_blockingCallFromThread.<locals>.cb�rI r1 �r rC r/ r r r �test_blockingCallFromThread� s z/ReactorThreadsTests.test_blockingCallFromThreadc s$ dd� }� fdd�}� � |��|�S )zx
Test blockingCallFromThread as above, but be sure the resulting
Deferred is not already fired.
c S s t �� } t�d| jd� | S )N皙�����?�egg)r r7 r � callLaterr5 ��dr r r rC � s zIReactorThreadsTests.test_asyncBlockingCallFromThread.<locals>.reactorFuncc s � � | d d d� d S )Nr rR rK rL r r r r/ � s z@ReactorThreadsTests.test_asyncBlockingCallFromThread.<locals>.cbrN rO r r r � test_asyncBlockingCallFromThread� s z4ReactorThreadsTests.test_asyncBlockingCallFromThreadc s$ dd� }� fdd�}� � |��|�S )z?
Test error report for blockingCallFromThread.
c S s t �td��S )N�bar)r r, � RuntimeErrorr r r r rC � s zIReactorThreadsTests.test_errorBlockingCallFromThread.<locals>.reactorFuncc s2 � � | d d t� � �| d d jd d� d S )N� r rW �ZassertIsInstancerX r �argsrL r r r r/ � s z@ReactorThreadsTests.test_errorBlockingCallFromThread.<locals>.cbrN rO r r r � test_errorBlockingCallFromThread� s z4ReactorThreadsTests.test_errorBlockingCallFromThreadc s$ dd� }� fdd�}� � |��|�S )z�
Test error report for blockingCallFromThread as above, but be sure the
resulting Deferred is not already fired.
c S s t �� } t�d| jtd�� | S )NrQ �spam)r r7 r rS ZerrbackrX rT r r r rC � s zNReactorThreadsTests.test_asyncErrorBlockingCallFromThread.<locals>.reactorFuncc s2 � � | d d t� � �| d d jd d� d S )NrY r r] rZ rL r r r r/ � s zEReactorThreadsTests.test_asyncErrorBlockingCallFromThread.<locals>.cbrN rO r r r �%test_asyncErrorBlockingCallFromThread� s z9ReactorThreadsTests.test_asyncErrorBlockingCallFromThreadN)�__name__�
__module__� __qualname__�__doc__r r r3 r8 r> rI rP rV r\ r^ r r r r r s
r c @ s e Zd ZdZdZdd� ZdS )�Counterr c C s, | j d }|| j d kr"d| _t�|| _ dS )zA non thread-safe method.rY N)�index�problem�
ValueError)r �nextr r r �add� s
z Counter.addN)r_ r` ra rd re rh r r r r rc � s rc c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�DeferredResultTestsz(
Test twisted.internet.threads.
c C s t �d� d S )N� �r r r r r r �setUp� s zDeferredResultTests.setUpc C s t �d� d S )Nr rk r r r r �tearDown� s z DeferredResultTests.tearDownc sR g � d�t �� �� ���fdd�}t�� fdd�t��D �tj|fi fg � �S )zW
L{threads.callMultipleInThread} calls multiple functions in a thread.
�
c s" �� � tt���� ��d � d S r )r �listr r5 r ��L�NrU r r r �finished� s z7DeferredResultTests.test_callMultiple.<locals>.finishedc s g | ]}� j |fi f�qS r )r )�.0r; )rq r r �
<listcomp>� s z9DeferredResultTests.test_callMultiple.<locals>.<listcomp>)r r7 r ZcallMultipleInThreadr r r4 �r rs r rp r �test_callMultiple� s ��z%DeferredResultTests.test_callMultiplec C s( t jddd�ddd�}|�| jd� |S ) z�
L{threads.deferToThread} executes the function passed, and correctly
handles the positional and keyword arguments given.
� c S s | | S r r ��x�yr r r r9 � r: z9DeferredResultTests.test_deferredResult.<locals>.<lambda>� r �r{ � )rx )r r r1 r �r rU r r r �test_deferredResult� s z'DeferredResultTests.test_deferredResultc s2 G dd� dt �� � fdd�}t�|�}| �|� �S )z�
Check that L{threads.deferToThread} return a failure object
with an appropriate exception instance when the called
function raises an exception.
c @ s e Zd ZdS )z:DeferredResultTests.test_deferredFailure.<locals>.NewErrorN�r_ r` ra r r r r �NewError s r� c s
� � �d S r r r �r� r r �
raiseError s z<DeferredResultTests.test_deferredFailure.<locals>.raiseError)r? r r �
assertFailure�r r� rU r r� r �test_deferredFailure� s
z(DeferredResultTests.test_deferredFailurec C s( t �dd� �}|�dd� � | �|t�S )z�
Check that a successful L{threads.deferToThread} followed by a one
that raises an exception correctly result as a failure.
c S s d S r r r r r r r9 r: zFDeferredResultTests.test_deferredFailureAfterSuccess.<locals>.<lambda>c S s t �dd� �S )Nc S s dd S )NrY r r r r r r r9 r: zXDeferredResultTests.test_deferredFailureAfterSuccess.<locals>.<lambda>.<locals>.<lambda>)r r rG r r r r9 r: )r r r1 r� �ZeroDivisionErrorr r r r � test_deferredFailureAfterSuccess s z4DeferredResultTests.test_deferredFailureAfterSuccessN)
r_ r` ra rb rl rm rw r� r� r� r r r r ri � s
ri c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �DeferToThreadPoolTestsz=
Test L{twisted.internet.threads.deferToThreadPool}.
c C s t �dd�| _| j�� d S )Nr rj )r Z
ThreadPool�tp�startr r r r rl & s z DeferToThreadPoolTests.setUpc C s | j �� d S r )r� �stopr r r r rm + s z DeferToThreadPoolTests.tearDownc C s. t jt| jddd�ddd�}|�| jd� |S ) z�
L{threads.deferToThreadPool} executes the function passed, and
correctly handles the positional and keyword arguments given.
rx c S s | | S r r ry r r r r9 5 r: z<DeferToThreadPoolTests.test_deferredResult.<locals>.<lambda>r| r r} r~ )rx )r �deferToThreadPoolr r� r1 r r r r r r� / s
�z*DeferToThreadPoolTests.test_deferredResultc s8 G dd� dt �� � fdd�}t�t| j|�}| �|� �S )z�
Check that L{threads.deferToThreadPool} return a failure object with an
appropriate exception instance when the called function raises an
exception.
c @ s e Zd ZdS )z=DeferToThreadPoolTests.test_deferredFailure.<locals>.NewErrorNr� r r r r r� @ s r� c s
� � �d S r r r r� r r r� B s z?DeferToThreadPoolTests.test_deferredFailure.<locals>.raiseError)r? r r� r r� r� r� r r� r r� : s z+DeferToThreadPoolTests.test_deferredFailureN)r_ r` ra rb rl rm r� r� r r r r r� ! s
r� a_
import time
import %(reactor)s
%(reactor)s.install()
from twisted.internet import reactor
def threadedCall():
print('threaded call')
reactor.callInThread(threadedCall)
# Spin very briefly to try to give the thread a chance to run, if it
# is going to. Is there a better way to achieve this behavior?
for i in range(100):
time.sleep(0.0)
c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� ThreadStartupProcessProtocolc C s || _ g | _g | _d S r )rs �out�errrv r r r �__init__] s z%ThreadStartupProcessProtocol.__init__c C s | j �|� d S r )r� r )r r� r r r � outReceivedb s z(ThreadStartupProcessProtocol.outReceivedc C s | j �|� d S r )r� r )r r� r r r � errReceivede s z(ThreadStartupProcessProtocol.errReceivedc C s | j �| j| j|f� d S r )rs r5 r� r� )r �reasonr r r � processEndedh s z)ThreadStartupProcessProtocol.processEndedN)r_ r` ra r� r� r� r� r r r r r� \ s r� c @ s e Zd ZdZdd� ZdS )�StartupBehaviorTestsa
Test cases for the behavior of the reactor threadpool near startup
boundary conditions.
In particular, this asserts that no threaded calls are attempted
until the reactor starts up, that calls attempted before it starts
are in fact executed once it has started, and that in both cases,
the reactor properly cleans itself up (which is tested for
somewhat implicitly, by requiring a child process be able to exit,
something it cannot do unless the threadpool has been properly
torn down).
c s� �� � }t|d��}|�tdtji � W 5 Q R X �fdd�}� fdd�}tj�� }tj �
t j �|d< t
�� �||�}t|�� t�� t jd|f|� |S ) N�wr c sR | \}}}|� tj�r(� �d||f � |r<t�d|f � � �|d|f � d S )Nz.Process did not exit cleanly (out: %s err: %s)z'Unexpected output on standard error: %sz(Expected no output, instead received:
%s)Zcheckr ZProcessTerminatedr, r �msgZ assertFalse)r# r� r� r� r r r �programFinished� s
�zMStartupBehaviorTests.testCallBeforeStartupUnexecuted.<locals>.programFinishedc s | � tj� � �d� | S )NZKILL)Ztrapr � TimeoutErrorZ
signalProcess)r� )�protor r �programTimeout� s
zLStartupBehaviorTests.testCallBeforeStartupUnexecuted.<locals>.programTimeout�
PYTHONPATH�python)Zmktemp�open�write�_callBeforeStartupProgramr r` �os�environ�copy�pathsep�join�sys�pathr r7 Z addCallbacksr� Z spawnProcess�
executable)r ZprognameZprogfiler� r� �envrU r )r� r r � testCallBeforeStartupUnexecuted| s
z4StartupBehaviorTests.testCallBeforeStartupUnexecutedN)r_ r` ra rb r� r r r r r� m s r� z(No thread support, nothing to test here.z7No process support, cannot run subprocess thread tests.)%rb Z
__future__r r r� r� r Z
twisted.trialr Ztwisted.python.compatr Ztwisted.internetr r r r r
r Ztwisted.pythonr r
r r ZTestCaser rc ri r� r� ZProcessProtocolr� r� ZIReactorThreads�cls�skipr' ZIReactorProcessr r r r �<module> s0 .L(-�