File: //usr/lib/python3/dist-packages/twisted/test/__pycache__/test_ident.cpython-310.pyc
o
�b> � @ s� d Z ddlZddlZddlmZ ddlmZmZ ddlm Z ddl
m Z ddl m
Z
ddlmZ G d d
� d
ej�ZG d d � d e j�ZG d
d� de j�ZG dd� de�ZG dd� dej�Ze�dd�dd� dkrmdZdZndZdZG dd� dej�ZdS ) z0
Test cases for twisted.protocols.ident module.
� N)�StringIO)�defer�error)�ident)�failure)�StringTransport)�unittestc @ 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 � IdentClient�client��self� r �9/usr/lib/python3/dist-packages/twisted/test/test_ident.py�setUp s zClassParserTests.setUpc C �6 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 � z!ClassParserTests.test_indentErrorc C r )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_noUSerError( r z!ClassParserTests.test_noUSerErrorc C r )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_invalidPortError1 r z&ClassParserTests.test_invalidPortErrorc C r )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_hiddenUserError: r 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 �connectionLostr �Failurer �ConnectionLostr r r r r r r �test_lostConnectionC s z$ClassParserTests.test_lostConnectionN)
�__name__�
__module__� __qualname__�__doc__r r r r# r% r) r r r r r s r c @ � e Zd Zdd� ZdS )�TestIdentServerc C s | j S �N)� resultValue�r
�
serverAddress�
clientAddressr r r �lookupO s zTestIdentServer.lookupN�r* r+ r, r5 r r r r r/ N � r/ c @ r. )�TestErrorIdentServerc C s | � � �r0 )�
exceptionTyper2 r r r r5 T s zTestErrorIdentServer.lookupNr6 r r r r r8 S r7 r8 c @ s e Zd ZdS )� NewExceptionN)r* r+ r, r r r r r: X s r: 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 ]%}|d d �= |�d|f � |�d|f � | � |d|f d|f g� qrd 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)r8 �makeConnectionr r �sendLiner r r9 r � assertEqualr r" r$ r: �flushLoggedErrors�len)r
�p�L�errs�portr r r �
testErrors] s@
���z ServerParserTests.testErrorsc C sB t � }|�t� � g }|j|_d|_|�d� | �|d d� d S )N)�SYS�USERz123, 456r z 123, 456 : USERID : SYS : USER)r/ rA r r rB r1 r rC )r
rF rG r r r � testSuccess� s
z ServerParserTests.testSuccessN)r* r+ r, rJ rM r r r r r; \ s (r; z=Lr< � �0100007F�04030201�7F000001�01020304c @ 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 �ProcServerMixinrC �dottedQuadFromHexString�_addr1�r
rF r r r �testDottedQuadFromHexString� s z*ProcMixinTests.testDottedQuadFromHexStringc C s" t �� }| �|�td �d� d S )Nz:0277)rT iw )r rU rC �
unpackAddressrW rX r r r �testUnpackAddress� s z ProcMixinTests.testUnpackAddressc C s t �� }| �|�| j�d� d S )N)�rT � ��1.2.3.4� r )r rU rC � parseLine�linerX r r r �testLineParser� s �z ProcMixinTests.testLineParserc sT g �t �� }� fdd�|_�fdd�|_� �|�dd�|jdf� � ��dg� d S )Nc � t � jg�S r0 ��iterrb r r r r �<lambda>� � z4ProcMixinTests.testExistingAddress.<locals>.<lambda>c s � � | �dfd S )N�rootr< )r )�uid)�usernamer r rg � s r\ r^ ri r )r rU �entries� getUsernamerC r5 � SYSTEM_NAMErX r )r
rk r �testExistingAddress� s �z"ProcMixinTests.testExistingAddressc sV t �� }� fdd�|_� �t j|jdd� � �t j|jdd� � �t j|jdd� d S )Nc rd r0 re r r r r rg � rh z7ProcMixinTests.testNonExistingAddress.<locals>.<lambda>�rT � r^ r\ )z1.2.3.5r` )r_ i� )r rU rl � assertRaisesr r5 rX 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�openrp r^ )z
/proc/net/tcpN)�patch�builtinsr rU rr r r5 rC )r
ry rF r rw r �testLookupProcNetTcp� s z#ProcMixinTests.testLookupProcNetTcpN)
r* r+ r, rW �_addr2rb rt rY r[ rc ro rs r} r r r r rS � s ���
rS ) r- r| �struct�ior �twisted.internetr r �twisted.protocolsr �twisted.pythonr �twisted.test.proto_helpersr �
twisted.trialr �TestCaser � IdentServerr/ r8 � RuntimeErrorr: r; �packrW r~ rS r r r r �<module> s( :4