Your IP : 216.73.216.196
o
0�i�2 � @ s: d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl Z ej
Z
dd� Z G dd� de j �Z
G dd � d e�ZG d
d � d ej�ZG d d
� d
ej�Zejdd� �Zdefdd�ZG dd� de j �ZG dd� de�ZG dd� de�ZeZG dd� de j �ZG dd � d e j �Z G d d � d e�Z e d kr�e � � dS dS ) � N)�support)�
socket_helperc C sd |� � | �� z$z |�� \}}|�� W n ty Y nw W |�� d S W |�� d S |�� w �N)Zlisten�setZaccept�close� TimeoutError)�evtZservZconnZaddr� r �:/opt/alt/python310/lib64/python3.10/test/test_telnetlib.py�server
s �� �r c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )� GeneralTestsc C sn t �� | _t�tjtj�| _| j�d� t� | j�| _
t j t | j| jfd�| _
d| j
_| j
�� | j�� d S )N�< )�target�argsT)� threadingZEventr �socketZAF_INETZ SOCK_STREAM�sockZ
settimeoutr Z bind_port�portZThreadr �threadZdaemon�start�wait��selfr r r
�setUp s
zGeneralTests.setUpc C s | j �� | ` d S r )r �joinr r r r
�tearDown$ s
zGeneralTests.tearDownc C s t �t| j�}|j�� d S r )� telnetlib�Telnet�HOSTr r r �r �telnetr r r
� testBasic( s zGeneralTests.testBasicc C sN t �t| j��}| �|�� � W d � n1 sw Y | �|�� � d S r )r r r r ZassertIsNotNone�
get_socketZ assertIsNone)r Ztnr r r
�testContextManager- s �z GeneralTests.testContextManagerc C sf | � t�� d u � t�d� zt�t| j�}W t�d � nt�d � w | �|j �
� d� |j � � d S )N� ) �
assertTruer �getdefaulttimeout�setdefaulttimeoutr r r r � assertEqualr �
gettimeoutr r r r r
�testTimeoutDefault2 s
z GeneralTests.testTimeoutDefaultc C sl | � t�� d u � t�d� ztjt| jd d�}W t�d � nt�d � w | � |j� � d u � |j�
� d S �Nr$ )�timeout) r% r r& r' r r r r r r) r r r r r
�testTimeoutNone< s
z GeneralTests.testTimeoutNonec C s2 t jt| jdd�}| �|j�� d� |j�� d S r+ )r r r r r( r r) r r r r r
�testTimeoutValueG s z GeneralTests.testTimeoutValuec C s: t �� }|jt| jdd� | �|j�� d� |j�� d S r+ ) r r �openr r r( r r) r r r r r
�testTimeoutOpenL s z GeneralTests.testTimeoutOpenc C sJ t jt| jdd�}|j}| �|�� |� | �|�� |�� � |j�� d S r+ ) r r r r r r( r" �filenor )r r Zt_sockr r r
� testGettersR s
zGeneralTests.testGettersN) �__name__�
__module__� __qualname__r r r! r# r* r- r. r0 r2 r r r r
r s
r c @ s( e Zd Z d dd�Zdd� Zdd� ZdS )
�
SocketStubr c C s t |�| _g | _d| _d S )NF)�list�reads�writes�block)r r8 r r r
�__init__\ s
zSocketStub.__init__c C s | j �|� d S r )r9 �append)r �datar r r
�sendall` s zSocketStub.sendallc C sj d}| j r t|�|k r || j �d�7 }| j r t|�|k s t|�|kr3| j �d||d � � |d |� }|S )N� r )r8 �len�pop�insert)r �size�outr r r
�recvb s � zSocketStub.recvN)r )r3 r4 r5 r; r> rE r r r r
r6 Z s
r6 c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� TelnetAlikec C s t � �r )�NotImplementedErrorr r r r
r1 l � zTelnetAlike.filenoc C s d S r r r r r r
r n s zTelnetAlike.closec C s
| j j S r )r r: r r r r
�
sock_availo �
zTelnetAlike.sock_availc G sT t �� �}tjj| |g|�R � W d � n1 sw Y | j|�� 7 _d S r )r Zcaptured_stdoutr r �msg� _messages�getvalue)r rK r rD r r r
rK q s
�zTelnetAlike.msgN)r3 r4 r5 r1 r rI rK r r r r
rF k s
rF c @ sD e Zd Zdd� Zedd� �Zddd�Zdd � Zdd
d �Zd d
� Z dS )� MockSelectorc C s
i | _ d S r ��keysr r r r
r; y rJ zMockSelector.__init__c C s dS )Ng����MbP?r r r r r
�
resolution| s zMockSelector.resolutionNc C s t �|d||�}|| j|< |S )Nr )� selectorsZ SelectorKeyrP )r �fileobj�eventsr= �keyr r r
�register� s
zMockSelector.registerc C s | j �|�S r )rP rA )r rS r r r
�
unregister� � zMockSelector.unregisterc C sB d}| j D ]
}t|t�r|jj} nq|rg S dd� | j �� D �S )NFc S s g | ]}||j f�qS r )rT )�.0rU r r r
�
<listcomp>� � z'MockSelector.select.<locals>.<listcomp>)rP �
isinstancerF r r: �values)r r, r: rS r r r
�select� s
�zMockSelector.selectc C s | j S r rO r r r r
�get_map� rH zMockSelector.get_mapr )
r3 r4 r5 r; �propertyrQ rV rW r^ r_ r r r r
rN w s
rN c # s6 � � fdd�}zt j}|t _d V W |t _d S |t _w )Nc s t � �S r )r6 )Zignored�r8 r r
�new_conn� s z test_socket.<locals>.new_conn)r Zcreate_connection)r8 rb Zold_connr ra r
� test_socket� s � �rc r c C sJ | D ]}qt | �� |dd�}d|_W d � |S 1 s w Y |S )N�dummyr � )rc rL )r8 �cls�xr r r r
� test_telnet� s
��rh c @ s e Zd Zdd� Zdd� ZdS )�ExpectAndReadTestCasec C s t j| _tt _d S r )r �_TelnetSelector� old_selectorrN r r r r
r � s
zExpectAndReadTestCase.setUpc C s | j t_d S r )rk r rj r r r r
r � rX z ExpectAndReadTestCase.tearDownN)r3 r4 r5 r r r r r r
ri � s ri c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )� ReadTestsc C st dg}t |�}|�d�}| j|d|j|j|jjfd� g d�}d�|d d� �}t |�}|�d�}| �||� d S )Ns xxxmatchyyy� matchs xxxmatch)rK )s2 xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxrm s2 yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyr? ���)rh Z
read_untilr( ZcookedqZrawqr r8 r )r �wantr r= r8 �expectr r r
�test_read_until� s
zReadTests.test_read_untilc C s4 g d�}d� |�}t|�}|�� }| �||� d S )N)�� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxs� yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyys� zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzr? )r rh �read_allr( )r r8 rp r r= r r r
�
test_read_all� s
zReadTests.test_read_allc C sD t dg�}|�� }| �t|�dk� t � }|�� }| �d|� d S )Nrr � r? )rh Z read_somer% r@ r( )r r r= r r r
�test_read_some� s
zReadTests.test_read_somec C sr d}t |g�}t||�}d|j_| �d|� � d|j_d} z||� 7 }W n ty/ Y nw q | �||� d S )N�d xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxTr? F)rh �getattrr r: r( �EOFError)r Z func_namero r �funcr= r r r
� _read_eager� s
��zReadTests._read_eagerc C s | � d� | � d� d S )NZ
read_eagerZread_very_eager)r{ r r r r
�test_read_eager� s
zReadTests.test_read_eagerc C s\ d}t |g�}| �d|�� � |jjr|�� |jjs|�� }| �||� | �t|j� d S �Nrw r? )rh r( �read_very_lazyr r8 � fill_rawqZ assertRaisesry )r ro r r= r r r
r~ � s
� zReadTests.read_very_lazyc C sz d}t |g�}| �d|�� � d} z|�� }||7 }|s!|�� W n ty+ Y n
w | �|�|�� q| �||� d S r} )rh r( Z read_lazyr ry r% �
startswith)r ro r r= Z read_datar r r
�test_read_lazy s"
� �� zReadTests.test_read_lazyN)
r3 r4 r5 rq rt rv r{ r| r~ r� r r r r
rl � s
rl c @ s e Zd Zddd�Zdd� ZdS )�nego_collectorNc C s d| _ || _d| _d S )Nr? )�seen� sb_getter�sb_seen)r r� r r r
r; s
znego_collector.__init__c C sD | j || 7 _ |tjkr | jr | �� }| j|7 _d S d S d S r )r� �tl�SEr� r� )r r �cmd�optZsb_datar r r
�do_nego s
�znego_collector.do_negor )r3 r4 r5 r; r� r r r r
r� s
r� c @ s e Zd Z dd� ZdS )�
WriteTestsc C sz ddt j d dt j t j d t j t jdg}|D ] }t� }|�|� d�|jj�}| �|�t jt jt j �|� qd S )Ns data sample without IACs data sample withs one IACs a fews iacsr? ) r� �IACrh �writer r r9 r( �replace)r Z data_sampler= r Zwrittenr r r
�
test_write( s �
�zWriteTests.test_writeN)r3 r4 r5 r� r r r r
r� $ s r� c @ s` e Zd Zejejejejejej ej
ej gZ dd� Z
dd� Zdd� Zdd� Zd d
� Zd d � Zd
S )� OptionTestsc C s� t |�}td�|��}t� }|�|j� |�� }|j}| �t|�dk� | � |d d� | j
� | � |dd� t j
� | � |t|| �� d |_d S )Nr? r ru � )rh r@ r r� � set_option_negotiation_callbackr� rs r� r% �assertIn�cmdsr( r� ZNOOPTr� )r r= r Zdata_len�nego�txtr� r r r
�
_test_command8 s
zOptionTests._test_commandc C s^ | j D ] }| �tj|g� | �dtj|dg� | �dtj|dg� q| �dd� | j D �� d S )Nrw sd yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyy�
xxxxxxxxxx�
yyyyyyyyyyc S s g | ]}t j| �qS r )r� r� )rY r� r r r
rZ L r[ z1OptionTests.test_IAC_commands.<locals>.<listcomp>)r� r� r� r� )r r� r r r
�test_IAC_commandsF s
z OptionTests.test_IAC_commandsc C s0 t jt j t j t j t jt j t j t j t j t j t jt j t j t j d t j t j t jt j d t j t j t j t j t jt j d t j t j d t j t j g}t|�}t|j�}|�|j� |� � }| �
|d� t jt j d t j d t j d }| �
|j |� | �
d|�� � d |_ d S )Ns aas bbs ccs ddr? s aabb)
r� r� ZSBr� rh r� Z read_sb_datar� r� rs r( r� r� )r �sendr r� r� Z want_sb_datar r r
�test_SB_commandsN s "&&*�
$
z OptionTests.test_SB_commandsc C s� dt jtdg� dft jt j tdg� dft jt j tdg� dft jt j tdg� dft jt j tdg� dfg}|D ]\}}t|g�}|�d� |� � }| �
||j � q>d S ) N)� az : recv b''
�X z: IAC 88 not recognized
ru z : IAC DO 1
z
: IAC DONT 1
z
: IAC WILL 1
z
: IAC WONT 1
) r� r� �bytesZDOZDONTZWILLZWONTrh �set_debuglevelrs r� rL )r Zgiven_a_expect_b�a�br r� r r r
�test_debuglevel_reads` s �
z!OptionTests.test_debuglevel_readsc C s0 t � }|�d� |�d� d}| �||j� d S )Nru s xxxz send b'xxx'
)rh r� r� r� rL )r r Zexpectedr r r
�test_debuglevel_writes s
z!OptionTests.test_debuglevel_writec C s^ t g �� tdd�}d|_W d � n1 sw Y |�d� |�d� | �|jd� d S )Nrd �0re ru �testz0.*test)rc rF rL r� rK Z assertRegexr r r r
�test_debug_accepts_str_portz s
�
z'OptionTests.test_debug_accepts_str_portN)r3 r4 r5 r� ZAOZAYTZBRKZECZELZGAZIPZNOPr� r� r� r� r� r� r� r r r r
r� 4 s $ r� c @ s e Zd Zdd� ZdS )� ExpectTestsc C sB g d�}t |�}|�dg�\}}}| �|d�|d d� �� d S )N)r� rm r� rm r? rn )rh rp r( r )r ro r �_r= r r r
� test_expect� s
zExpectTests.test_expectN)r3 r4 r5 r� r r r r
r� � s r� �__main__) r rR r r �
contextlibr� r Z test.supportr Zunittestr r ZTestCaser �objectr6 r rF Z BaseSelectorrN �contextmanagerrc rh ri rl r� r� r� r� r� r3 �mainr r r r
�<module> s6 B
b P
�