← Back
Editing: 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
Save File
Cancel