File: //lib/python3/dist-packages/twisted/test/__pycache__/test_ident.cpython-38.pyc
U
W[g � @ s$ d Z ddlZddlmZ ddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z dd l
mZ z ddlZW n ek
r� ddlZY nX G d
d � d e j�ZG d d
� d
ej�ZG dd� dej�ZG dd� de�ZG dd� de j�Ze�dd�dd� dk�rdZdZ ndZdZ G dd � d e j�Z dS ) z0
Test cases for twisted.protocols.ident module.
� N)�ident)�failure)�error)�defer)�NativeStringIO)�unittest)�StringTransportc @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�ClassParserTestsz*
Test parsing of ident responses.
c C s t �� | _dS )z7
Create an ident client used in tests.
N)r Z IdentClient�client��self� r
�9/usr/lib/python3/dist-packages/twisted/test/test_ident.py�setUp! s zClassParserTests.setUpc C s6 t �� }| jj�|ddf� | j�d� | �|tj�S )zX
'UNKNOWN-ERROR' error should map to the L{ident.IdentError} exception.
�{ �� z 123, 456 : ERROR : UNKNOWN-ERROR) r �Deferredr
�queries�append� lineReceived�
assertFailurer �
IdentError�r �dr
r
r �test_indentError( s z!ClassParserTests.test_indentErrorc C s6 t �� }| jj�|ddf� | j�d� | �|tj�S )zN
'NO-USER' error should map to the L{ident.NoUser} exception.
�� r z234, 456 : ERROR : NO-USER) r r r
r r r r r �NoUserr r
r
r �test_noUSerError2 s z!ClassParserTests.test_noUSerErrorc C s6 t �� }| jj�|ddf� | j�d� | �|tj�S )zX
'INVALID-PORT' error should map to the L{ident.InvalidPort} exception.
iY �7 z 345, 567 : ERROR : INVALID-PORT) r r r
r r r r r � InvalidPortr r
r
r �test_invalidPortError< s z&ClassParserTests.test_invalidPortErrorc C s6 t �� }| jj�|ddf� | j�d� | �|tj�S )zV
'HIDDEN-USER' error should map to the L{ident.HiddenUser} exception.
r i z 567, 789 : ERROR : HIDDEN-USER) r r r
r r r r r �
HiddenUserr r
r
r �test_hiddenUserErrorF s z%ClassParserTests.test_hiddenUserErrorc C s@ t �� }| jj�|ddf� | j�t�t� � �� | �
|t j �S )zy
A pending query which failed because of a ConnectionLost should
receive an L{ident.IdentError}.
i� i� )
r r r
r r ZconnectionLostr ZFailurer ZConnectionLostr r r r r
r
r �test_lostConnectionP s z$ClassParserTests.test_lostConnectionN)
�__name__�
__module__� __qualname__�__doc__r r r r r" r# r
r
r
r r s
r c @ s e Zd Zdd� ZdS )�TestIdentServerc C s | j S �N)� resultValue�r Z
serverAddressZ
clientAddressr
r
r �lookup] s zTestIdentServer.lookupN�r$ r% r& r, r
r
r
r r( \ s r( c @ s e Zd Zdd� ZdS )�TestErrorIdentServerc C s | � � �d S r) )�
exceptionTyper+ r
r
r r, c s zTestErrorIdentServer.lookupNr- r
r
r
r r. b s r. c @ s e Zd ZdS )� NewExceptionN)r$ r% r& r
r
r
r r0 h s r0 c @ s e Zd Zdd� Zdd� ZdS )�ServerParserTestsc C s4 t � }|�t� � g }|j|_tj|_|�d� | � |d d� tj
|_|�d� | � |d d� tj |_|�d� | � |d d � tj |_|�d
� | � |d d � t
|_|�d
� | � |d d� | �t
�}| � t|�d� dD ]J}|d d �= |�d|f � |�d|f � | � |d|f d|f g� q�d S )Nz123, 345r z 123, 345 : ERROR : UNKNOWN-ERRORz432, 210� z432, 210 : ERROR : NO-USERz987, 654� z 987, 654 : ERROR : INVALID-PORTz756, 827� z 756, 827 : ERROR : HIDDEN-USERz987, 789� z 987, 789 : ERROR : UNKNOWN-ERROR)���r i i z%d, 5z5, %dz %d, 5 : ERROR : INVALID-PORTz 5, %d : ERROR : INVALID-PORT)r. �makeConnectionr r �sendLiner r r/ r � assertEqualr r r! r0 ZflushLoggedErrors�len)r �p�LZerrsZportr
r
r �
testErrorsn s>
��z ServerParserTests.testErrorsc C sB t � }|�t� � g }|j|_d|_|�d� | �|d d� d S )N)ZSYSZUSERz123, 456r z 123, 456 : USERID : SYS : USER)r( r7 r r r8 r* r r9 )r r; r< r
r
r � testSuccess� s
z ServerParserTests.testSuccessN)r$ r% r& r= r> r
r
r
r r1 m s %r1 z=Lr2 � Z0100007FZ04030201Z7F000001Z01020304c @ sP e Zd Zdeef Zde Zdd� Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dS )�ProcMixinTestszj4: %s:0019 %s:02FA 0A 00000000:00000000 00:00000000 00000000 0 0 10927 1 f72a5b80 3000 0 0 2 -1zd sl local_address rem_address st tx_queue rx_queue tr tm->when retrnsmt uid timeout inode
c C s t �� }| �|�t�d� d S )N� 127.0.0.1)r �ProcServerMixinr9 ZdottedQuadFromHexString�_addr1�r r; r
r
r �testDottedQuadFromHexString� s z*ProcMixinTests.testDottedQuadFromHexStringc C s" t �� }| �|�td �d� d S )Nz:0277)rA iw )r rB r9 Z
unpackAddressrC rD r
r
r �testUnpackAddress� s �z ProcMixinTests.testUnpackAddressc C s t �� }| �|�| j�d� d S )N)�rA � ��1.2.3.4� r )r rB r9 Z parseLine�linerD r
r
r �testLineParser� s
�z ProcMixinTests.testLineParserc sT g �t �� }� fdd�|_�fdd�|_� �|�dd�|jdf� � ��dg� d S )Nc s t � jg�S r) ��iterrL r
r r
r �<lambda>� � z4ProcMixinTests.testExistingAddress.<locals>.<lambda>c s � � | �dfd S )N�rootr2 )r )Zuid)�usernamer
r rP � rQ rG rI rR r )r rB �entriesZ getUsernamer9 r, Z SYSTEM_NAMErD r
)r rS r �testExistingAddress� s
�z"ProcMixinTests.testExistingAddressc sV t �� }� fdd�|_� �t j|jdd� � �t j|jdd� � �t j|jdd� d S )Nc s t � jg�S r) rN r
r r
r rP � rQ z7ProcMixinTests.testNonExistingAddress.<locals>.<lambda>�rA � rI rG )z1.2.3.5rK )rJ i� )r rB rT � assertRaisesr r, rD r
r r �testNonExistingAddress� s ���z%ProcMixinTests.testNonExistingAddressc sR g � � �fdd�}�� td|� t�� }��tj|jdd� ��di fg� � dS )zS
L{ident.ProcServerMixin.lookup} uses the Linux TCP process table.
c s � � | |f� t�j�S )z[
Mock for the open call to prevent actually opening /proc/net/tcp.
)r r �
sampleFile)�args�kwargs�Z
open_callsr r
r � mocked_open� s z8ProcMixinTests.testLookupProcNetTcp.<locals>.mocked_open�openrV rI )z
/proc/net/tcpN)Zpatch�builtinsr rB rX r r, r9 )r r^ r; r
r] r �testLookupProcNetTcp� s �z#ProcMixinTests.testLookupProcNetTcpN)
r$ r% r& rC �_addr2rL rZ rE rF rM rU rY ra r
r
r
r r@ � s �� r@ ) r' ZstructZtwisted.protocolsr Ztwisted.pythonr Ztwisted.internetr r Ztwisted.python.compatr Z
twisted.trialr Ztwisted.test.proto_helpersr r` � ImportErrorZ __builtin__ZTestCaser Z IdentServerr( r. � RuntimeErrorr0 r1 ZpackrC rb r@ r
r
r
r �<module> s. @1