File: //lib/python3/dist-packages/twisted/internet/test/__pycache__/test_win32serialport.cpython-38.pyc
U
W[� � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
zddlmZ ddlZW n ek
r� dZdZY nX edk r�G d d
� d
ej�ZG d d � d ej�ZG d
d� de�ZG dd� dej�ZdS )z+
Tests for L{twisted.internet.serialport}.
� N)�unittest)�Protocol)�Failure)�platform)� DoNothing)�
serialportc s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )�RegularFileSerialc s" t t| �j||� || _|| _d S �N)�superr �__init__�
captured_args�captured_kwargs)�self�args�kwargs�� __class__� �L/usr/lib/python3/dist-packages/twisted/internet/test/test_win32serialport.pyr s zRegularFileSerial.__init__c C s d S r r �r r r r �_reconfigurePort! s z"RegularFileSerial._reconfigurePortc C s d S r r r r r r �_reconfigure_port$ s z#RegularFileSerial._reconfigure_port)�__name__�
__module__� __qualname__r r r �
__classcell__r r r r r s r c s( e Zd ZeZ� fdd�Zdd� Z� ZS )�RegularFileSerialPortc s@ |� d�}d|kr|d= tjj| _|| j_tt| �j||� d S )N�cbInQue) �get�serialZwin32ZCOMSTAT�comstatr r
r r )r r r r r r r r + s
z RegularFileSerialPort.__init__c C s
d| j fS )NT)r r r r r �_clearCommError6 s z%RegularFileSerialPort._clearCommError)r r r r Z_serialFactoryr r! r r r r r r ( s r c @ s e Zd Zdd� Zdd� ZdS )�CollectReceivedProtocolc C s
g | _ d S r )�
received_datar r r r r ; s z CollectReceivedProtocol.__init__c C s | j �|� d S r )r# �append)r �datar r r � dataReceived> s z$CollectReceivedProtocol.dataReceivedN)r r r r r& r r r r r" : s r" c @ sv e Zd ZdZe�� sdZnes"dZdd� Zdd� Z dd � Z
d
d � Z d d
� Z dd� Z
dd� Zdd� Zdd� Zdd� ZdS )�Win32SerialPortTestszB
Minimal testing for Twisted's Win32 serial port support.
z This test must run on Windows.z-Windows serial port support is not available.c C sV t � | _t� | _t�� | _tj� | jd�| _d}t
| jd��}|� |� W 5 Q R X d S )NZ fake_serials 1234�wb) r �protocolr �reactor�tempfileZmkdtemp� directory�os�path�join�open�write)r r% �fr r r �setUpM s
zWin32SerialPortTests.setUpc C s t �| j� d S r )�shutilZrmtreer, r r r r �tearDownY s z Win32SerialPortTests.tearDownc C s� t | j| j| j�}| �| jf|jj� |jj}| �d|d � | �tj |d � | �tj
|d � | �tj |d � | �d|d � | �d|d � | �d |d
� |� t
td ��� d S ) zx
Test correct positional and keyword arguments have been
passed to the C{serial.Serial} object.
i�% ZbaudrateZbytesizeZparityZstopbitsr ZxonxoffZrtsctsNZtimeout�Cleanup)r r) r. r* � assertEqualZ_serialr r
r Z EIGHTBITSZ PARITY_NONEZ STOPBITS_ONE�connectionLostr � Exception)r �portr r r r �test_serialPortDefaultArgs\ s z/Win32SerialPortTests.test_serialPortDefaultArgsc C sf | � d| jj� t| j| j| j�}| � d|j� | � d| jj� | � || jj� |�tt d��� dS )z�
Test the port is connected at initialization time, and
C{Protocol.makeConnection} has been called on the desired protocol.
r � r6 N)
r7 r) Z connectedr r. r* Z transportr8 r r9 )r r: r r r �!test_serialPortInitiallyConnectedo s z6Win32SerialPortTests.test_serialPortInitiallyConnectedc C sb t | j| j| j|d�}|�� |�d� |�d� |�d� |�� |�� |�tt d��� d S )N�r) ZdeviceNameOrPortNumberr* r � s abcds ABCDr6 )
r r) r. r* �serialReadEventr1 ZserialWriteEventr8 r r9 )r r r: r r r �common_exerciseHandleAccess| s �
z0Win32SerialPortTests.common_exerciseHandleAccessc C s | j dd� d S �NF�r �rA r r r r �test_exerciseHandleAccess_1� s z0Win32SerialPortTests.test_exerciseHandleAccess_1c C s | j dd� d S �NTrC rD r r r r �test_exerciseHandleAccess_2� s z0Win32SerialPortTests.test_exerciseHandleAccess_2c C sR t � }t|| j| j|d�}|�� | �tdd� |jD ��� |�t t
d��� d S )Nr> c s s | ]}t |t�V qd S r )�
isinstance�bytes)�.0�dr r r � <genexpr>� s zEWin32SerialPortTests.common_serialPortReturnsBytes.<locals>.<genexpr>r6 ) r" r r. r* r@ Z
assertTrue�allr# r8 r r9 )r r r) r: r r r � common_serialPortReturnsBytes� s � �
z2Win32SerialPortTests.common_serialPortReturnsBytesc C s | j dd� d S rB �rN r r r r � test_serialPortReturnsBytes_1� s z2Win32SerialPortTests.test_serialPortReturnsBytes_1c C s | j dd� d S rF rO r r r r � test_serialPortReturnsBytes_2� s z2Win32SerialPortTests.test_serialPortReturnsBytes_2N)r r r �__doc__r Z isWindows�skipr r3 r5 r; r= rA rE rG rN rP rQ r r r r r' B s
r' )rR r- r4 r+ Z
twisted.trialr Ztwisted.internet.protocolr Ztwisted.python.failurer Ztwisted.python.runtimer Z%twisted.internet.test.test_serialportr Ztwisted.internetr r � ImportErrorZSerialr Z
SerialPortr r" ZTestCaser' r r r r �<module> s&