File: //usr/lib/python3/dist-packages/twisted/internet/test/__pycache__/test_threads.cpython-38.pyc
U
W[�! � @ s� d Z ddlmZmZ eZddlmZ ddlZddl Z ddl
m Z ddl m
Z
ddlmZ ddlmZ G d d
� d
e
�Ze� �e�� � dS ) z2
Tests for implementations of L{IReactorThreads}.
� )�division�absolute_import)�refN)� isInIOThread)�ReactorBuilder)�
ThreadPool)�IReactorThreadsc @ s^ e Zd ZdZefZdd� Zdd� Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dd� ZdS )�ThreadTestsBuilderzD
Builder for defining tests relating to L{IReactorThreads}.
c s� g �| � � ���� � | �� t� | �� jd� � ��fdd�}��|� | ��d� | ��d d� | ��d d� | �� j d � d
S ) z�
C{reactor.getThreadPool()} returns an instance of L{ThreadPool} which
starts when C{reactor.run()} is called and stops before it returns.
z(Pool should not start before reactor.runc s$ �� � j� �� � j� ��� d S �N)�append�started�joined�stop� ��pool�reactor�stater �D/usr/lib/python3/dist-packages/twisted/internet/test/test_threads.py�f( s z0ThreadTestsBuilder.test_getThreadPool.<locals>.f� r z#Pool should start after reactor.run� z-Pool should not be joined before reactor.stopz0Pool should be stopped after reactor.run returnsN)
� buildReactor�
getThreadPoolZassertIsInstancer Z assertFalser �callWhenRunning�
runReactor�
assertTruer
)�selfr r r r �test_getThreadPool s. �
� ��z%ThreadTestsBuilder.test_getThreadPoolc C s, | � � }|�d� |�� }| �|jd� dS )zm
C{reactor.suggestThreadPoolSize()} sets the maximum size of the reactor
threadpool.
� N)r ZsuggestThreadPoolSizer � assertEqual�max)r r r r r r �test_suggestThreadPoolSize: s
z-ThreadTestsBuilder.test_suggestThreadPoolSizec sV | � � � � fdd�}� �d� j|� � �� }| �� d� � �� }| �|| dk � dS )aR
A function scheduled with L{IReactorThreads.callFromThread} invoked
from a delayed call is run immediately in the next reactor iteration.
When invoked from the reactor thread, previous implementations of
L{IReactorThreads.callFromThread} would skip the pipe/socket based wake
up step, assuming the reactor would wake up on its own. However, this
resulted in the reactor not noticing an insert into the thread queue at
the right time (in this case, after the thread queue has been processed
for that reactor iteration).
c s � � � d S r
)r r �r r r �
threadCallS s zAThreadTestsBuilder.test_delayedCallFromThread.<locals>.threadCallr �< � N)r � callLater�callFromThreadZsecondsr r )r r$ ZbeforeZafterr r# r �test_delayedCallFromThreadE s z-ThreadTestsBuilder.test_delayedCallFromThreadc sP | � � � g �� �fdd�}� �d� j� j|� | �� d� | ��t�� g� dS )z�
A function scheduled with L{IReactorThreads.callFromThread} invoked
from another thread is run in the reactor thread.
c s �� t�� � � �� d S r
)r � threading�
currentThreadr r �r �resultr r r$ x s z:ThreadTestsBuilder.test_callFromThread.<locals>.threadCallr � N)r r' � callInThreadr( r r r* r+ )r r$ r r, r �test_callFromThreadp s
� z&ThreadTestsBuilder.test_callFromThreadc C sB | � � }t|�� �}|�|j� | �|� t�� | �|� � dS )a0
When the reactor stops, L{ReactorBase._stopThreadPool} drops the
reactor's direct reference to its internal threadpool and removes
the associated startup and shutdown triggers.
This is the case of the thread pool being created before the reactor
is run.
N) r r r r r r �gc�collect� assertIsNone)r r �
threadpoolr r r �test_stopThreadPool� s
z&ThreadTestsBuilder.test_stopThreadPoolc sJ | � � � g �� �fdd�}� �|� | �� � t�� | ��d � � dS )a�
We must handle the case of shutting down the thread pool when it was
started after the reactor was run in a special way.
Some implementation background: The thread pool is started with
callWhenRunning, which only returns a system trigger ID when it is
invoked before the reactor is started.
This is the case of the thread pool being created after the reactor
is started.
c s �� t� �� �� � �� d S r
)r r r r r �r ZthreadPoolRefsr r �acquireThreadPool� s z[ThreadTestsBuilder.test_stopThreadPoolWhenStartedAfterReactorRan.<locals>.acquireThreadPoolr N)r r r r1 r2 r3 )r r7 r r6 r �-test_stopThreadPoolWhenStartedAfterReactorRan� s
z@ThreadTestsBuilder.test_stopThreadPoolWhenStartedAfterReactorRanc C sZ | � � }t|�� �}|�d� |jjdkrB| �|j� | �|� nt �
� | �|� � dS )a[
When the reactor has its shutdown event fired before it is run, the
thread pool is completely destroyed.
For what it's worth, the reason we support this behavior at all is
because Trial does this.
This is the case of the thread pool being created without the reactor
being started at al.
ZshutdownZAsyncioSelectorReactorN) r r r ZfireSystemEvent� __class__�__name__r3 r4 r r1 r2 )r r Z
threadPoolRefr r r �,test_cleanUpThreadPoolEvenBeforeReactorIsRun� s
z?ThreadTestsBuilder.test_cleanUpThreadPoolEvenBeforeReactorIsRunc s@ g �| � � � � �fdd�}� �|� | �� � | �dg�� dS )z�
The reactor registers itself as the I/O thread when it runs so that
L{twisted.python.threadable.isInIOThread} returns C{True} if it is
called in the thread the reactor is running in.
c s �� t� � � �� d S r
)r r r r �r Zresultsr r �check� s z3ThreadTestsBuilder.test_isInIOThread.<locals>.checkTN)r r r r �r r= r r< r �test_isInIOThread� s
z$ThreadTestsBuilder.test_isInIOThreadc s@ g �| � � � � �fdd�}� �|� | �� � | �dg�� dS )z�
The reactor registers itself as the I/O thread when it runs so that
L{twisted.python.threadable.isInIOThread} returns C{False} if it is
called in a different thread than the reactor is running in.
c s �� t� � � �� j� d S r
)r r r( r r r< r r r= � s z6ThreadTestsBuilder.test_isNotInIOThread.<locals>.checkFN)r r/ r r r> r r< r �test_isNotInIOThread� s
z'ThreadTestsBuilder.test_isNotInIOThreadN)r: �
__module__� __qualname__�__doc__r ZrequiredInterfacesr r" r) r0 r5 r8 r; r? r@ r r r r r s + r )rC Z
__future__r r �typeZ
__metaclass__�weakrefr r1 r* Ztwisted.python.threadabler Z#twisted.internet.test.reactormixinsr Ztwisted.python.threadpoolr Ztwisted.internet.interfacesr r �globals�updateZmakeTestCaseClassesr r r r �<module> s T