Your IP : 216.73.216.213
B
;��]g � @ s$ d Z ddlZddlmZ ddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z dd l
mZ y 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 �J/opt/alt/python37/lib64/python3.7/site-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.
�{ i� 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.
�� i� z234, 456 : ERROR : NO-USER) r r r
r r r r r �NoUser)r r 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 i7 z 345, 567 : ERROR : INVALID-PORT) r r r
r r r r r � InvalidPort)r r 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.
i7 i z 567, 789 : ERROR : HIDDEN-USER) r r r
r r r r r �
HiddenUser)r r 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 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 �
serverAddress�
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 )N)�
exceptionType)r r& r' r r r
r( c s zTestErrorIdentServer.lookupN)r r! r" r( r r r r
r) b s r) c @ s e Zd ZdS )� NewExceptionN)r r! r" r r r r
r+ h s r+ c @ s e Zd Zdd� Zdd� ZdS )�ServerParserTestsc C s8 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� xRdD ]J}|d d �= |�d|f � |�d|f � | � |d|f d|f g� q�W 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 r+ ZflushLoggedErrors�len)r �p�LZerrs�portr r r
�
testErrorsn s8
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$ r2 r r r3 r% r r4 )r r6 r7 r r r
� testSuccess� s
z ServerParserTests.testSuccessN)r r! r" r9 r: r r r r
r, m s %r, z=Lr- � 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 )Nz 127.0.0.1)r �ProcServerMixinr4 ZdottedQuadFromHexString�_addr1)r r6 r r r
�testDottedQuadFromHexString� s z*ProcMixinTests.testDottedQuadFromHexStringc C s" t �� }| �|�td �d� d S )Nz:0277)z 127.0.0.1iw )r r= r4 Z
unpackAddressr> )r r6 r r r
�testUnpackAddress� s z ProcMixinTests.testUnpackAddressc C s t �� }| �|�| j�d� d S )N))z 127.0.0.1� )z1.2.3.4i� r )r r= r4 Z parseLine�line)r r6 r r r
�testLineParser� s
z ProcMixinTests.testLineParserc sT g �t �� }� fdd�|_�fdd�|_� �|�dd�|jdf� � ��dg� d S )Nc s t � jg�S )N)�iterrB r )r r r
�<lambda>� � z4ProcMixinTests.testExistingAddress.<locals>.<lambda>c s � � | �dfd S )N�rootr- )r )�uid)�usernamer r
rE � rF )z 127.0.0.1rA )z1.2.3.4i� rG r )r r= �entriesZ getUsernamer4 r( Z SYSTEM_NAME)r r6 r )r rI 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 )N)rD rB r )r r r
rE � rF z7ProcMixinTests.testNonExistingAddress.<locals>.<lambda>)z 127.0.0.1� )z1.2.3.4i� )z 127.0.0.1rA )z1.2.3.5i� )z1.2.3.4i� )r r= rJ � assertRaisesr r( )r r6 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)�
open_callsr r r
� mocked_open� s z8ProcMixinTests.testLookupProcNetTcp.<locals>.mocked_open�open)z 127.0.0.1rL )z1.2.3.4i� )z
/proc/net/tcpN)�patch�builtinsr r= rM r r( r4 )r rS r6 r )rR r r
�testLookupProcNetTcp� s z#ProcMixinTests.testLookupProcNetTcpN)
r r! r" r> �_addr2rB rO r? r@ rC rK rN rW r r r r
r< � s
r< ) r# �structZtwisted.protocolsr Ztwisted.pythonr Ztwisted.internetr r Ztwisted.python.compatr Z
twisted.trialr Ztwisted.test.proto_helpersr rV � ImportErrorZ __builtin__ZTestCaser Z IdentServerr$ r) � RuntimeErrorr+ r, �packr> rX r< r r r r
�<module> s. @1