File: //usr/lib/python3/dist-packages/twisted/internet/test/__pycache__/test_win32events.cpython-38.pyc
U
W[d � @ s� d Z z ddlZW n ek
r( dZY nX ddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z ddl
mZ dd lmZ G d
d � d e�ZG d d
� d
e�Ze� �e�� � dS )z6
Tests for implementations of L{IReactorWin32Events}.
� N)� verifyObject)�Failure)� isInIOThread)�IReactorWin32Events)�Deferred)�ReactorBuilder)� getThreadIDc @ sP e Zd ZdZdZd Z ZZdd� Zdd� Z dd � Z
d
d � Z d d
� Z dd� Z
dS )�Listenera�
L{Listener} is an object that can be added to a L{IReactorWin32Events}
reactor to receive callback notification when a Windows event is set. It
records what thread its callback is invoked in and fires a Deferred.
@ivar success: A flag which is set to C{True} when the event callback is
called.
@ivar logThreadID: The id of the thread in which the C{logPrefix} method is
called.
@ivar eventThreadID: The id of the thread in which the event callback is
called.
@ivar connLostThreadID: The id of the thread in which the C{connectionLost}
method is called.
@ivar _finished: The L{Deferred} which will be fired when the event callback
is called.
FNc C s
|| _ d S �N)� _finished)�self�finished� r �H/usr/lib/python3/dist-packages/twisted/internet/test/test_win32events.py�__init__/ s zListener.__init__c C s t � | _dS )Nr )r � logThreadID�r r r r � logPrefix3 s zListener.logPrefixc C s d| _ t� | _| j�d � d S )NT)�successr �
eventThreadIDr �callbackr r r r �occurred8 s zListener.occurredc C s t d��d S )Nz Some problem)� RuntimeErrorr r r r �brokenOccurred> s zListener.brokenOccurredc C s t d�S )NzEntirely different problem)�EnvironmentErrorr r r r �returnValueOccurredB s z Listener.returnValueOccurredc C s t � | _| j�|� d S r
)r �connLostThreadIDr Zerrback)r �reasonr r r �connectionLostF s zListener.connectionLost)�__name__�
__module__� __qualname__�__doc__r r r r r r r r r r r r r r r s r c @ sF e Zd ZdZegZdd� Zdd� Zdd� Zdd � Z d
d � Z
d d
� Z dS )�Win32EventsTestsBuilderzD
Builder defining tests relating to L{IReactorWin32Events}.
c C s | � � }tt|� dS )zn
An instance of the reactor has all of the methods defined on
L{IReactorWin32Events}.
N)� buildReactorr r )r �reactorr r r �test_interfaceR s z&Win32EventsTestsBuilder.test_interfacec s� t � }| �� � t�dddd�}t� }|�� fdd�� t|�}� �||d� � �tj |� | �
� � | � |j � | �
||j� | �
||j� dS )z�
When an event which has been added to the reactor is set, the action
associated with the event is invoked in the reactor thread.
NFc s � � � S r
��stop�Zignored�r% r r �<lambda>d � z7Win32EventsTestsBuilder.test_addEvent.<locals>.<lambda>r )r r$ �
win32event� CreateEventr � addCallbackr �addEvent�callWhenRunning�SetEvent�
runReactor�
assertTruer � assertEqualr r )r �reactorThreadID�eventr
�listenerr r* r �
test_addEvent[ s
z%Win32EventsTestsBuilder.test_addEventc s� g �� �fdd�}| � � � t�dddd�}t� }t|�}|�|� � �||d� � �tj|� | � � � | �
|j � | � dg�� dS )z{
Using L{IReactorWin32Events.addEvent} does not change which thread is
reported as the I/O thread.
c s �� t� � � �� d S r
)�appendr r( r) �r% Zresultsr r �checkt s zAWin32EventsTestsBuilder.test_ioThreadDoesNotChange.<locals>.checkNFr T)
r$ r- r. r r r/ r0 r1 r2 r3 r4 r r5 )r r< r7 r
r8 r r; r �test_ioThreadDoesNotChangen s
z2Win32EventsTestsBuilder.test_ioThreadDoesNotChangec s� t � }| �� � t�dddd�}g }t� }|�|j� |�� fdd�� t|�}� �||d� � � tj
|� | � � � | � |d t
� |d �t� | �||j� | �dt| �t��� dS )z�
If the event handler raises an exception, the event is removed from the
reactor and the handler's C{connectionLost} method is called in the I/O
thread and the exception is logged.
NFc s � � � S r
r' r) r* r r r+ � r, zBWin32EventsTestsBuilder.test_disconnectedOnError.<locals>.<lambda>r r � )r r$ r- r. r �addBothr: r r0 r1 r2 r3 �assertIsInstancer �trapr r5 r �lenZflushLoggedErrors�r r6 r7 �resultr
r8 r r* r �test_disconnectedOnError� s
z0Win32EventsTestsBuilder.test_disconnectedOnErrorc s� t � }| �� � t�dddd�}g }t� }|�|j� |�� fdd�� t|�}� �||d� � � tj
|� | � � � | � |d t
� |d �t� | �||j� dS )z�
If the event handler returns a value, the event is removed from the
reactor and the handler's C{connectionLost} method is called in the I/O
thread.
NFc s � � � S r
r' r) r* r r r+ � r, zFWin32EventsTestsBuilder.test_disconnectOnReturnValue.<locals>.<lambda>r r )r r$ r- r. r r? r: r r0 r1 r2 r3 r@ r rA r r5 r rC r r* r � test_disconnectOnReturnValue� s
z4Win32EventsTestsBuilder.test_disconnectOnReturnValuec C sZ | � � }t�dddd�}t� }t|�}|�||d� |�|j� | �|� | � |j
� dS )z�
Event handlers added with L{IReactorWin32Events.addEvent} do not have
C{connectionLost} called on them if they are still active when the
reactor shuts down.
NFr ) r$ r- r. r r r0 r1 r( r3 Z assertIsNoner )r r% r7 r
r8 r r r � test_notDisconnectedOnShutdown� s
z6Win32EventsTestsBuilder.test_notDisconnectedOnShutdownN) r r r! r" r ZrequiredInterfacesr& r9 r= rE rF rG r r r r r# L s r# )r" r- � ImportErrorZzope.interface.verifyr Ztwisted.python.failurer Ztwisted.python.threadabler Ztwisted.internet.interfacesr Ztwisted.internet.deferr Z#twisted.internet.test.reactormixinsr r �objectr r# �globals�updateZmakeTestCaseClassesr r r r �<module> s
5{