Your IP : 216.73.216.85
3
� f�2 � @ s. d dl Z d dlZd dlZd dlZd dlmZ d dlZejd�Zej Z dd� Z
G dd� dej �Z G dd � d e
�ZG d
d � d ej�ZG d d
� d
ej�Zejdd� �Zf efdd�ZG dd� dej �ZG dd� de�ZG dd� de
�ZeZG dd� dej �ZG dd� dej �ZG d d � d e�Z e d k�r*ej � dS ) � N)�support� threadingc C sT |j � | j� z4y|j� \}}|j� W n tjk
r@ Y nX W d |j� X d S )N)Zlisten�setZaccept�close�socket�timeout)�evtZservZconnZaddr� r �8/opt/alt/python36/lib64/python3.6/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 sr t j� | _tjtjtj�| _| jjd� tj | j�| _
t j t | j| jfd�| _
| j
jd� | j
j� | jj� d S )N�< )�target�argsT)r ZEventr r ZAF_INETZ SOCK_STREAM�sockZ
settimeoutr Z bind_port�portZThreadr �threadZ setDaemon�start�wait)�selfr r r
�setUp s
zGeneralTests.setUpc C s | j j� | ` d S )N)r �join)r r r r
�tearDown# s
zGeneralTests.tearDownc C s t jt| j�}|jj� d S )N)� telnetlib�Telnet�HOSTr r r )r �telnetr r r
� testBasic' s zGeneralTests.testBasicc C s: t jt| j��}| j|j� � W d Q R X | j|j� � d S )N)r r r r ZassertIsNotNone�
get_socketZ assertIsNone)r Ztnr r r
�testContextManager, s z GeneralTests.testContextManagerc C s\ | j tj� d k� tjd� ztjt| j�}W d tjd � X | j|j j
� d� |j j � d S )N� ) �
assertTruer �getdefaulttimeout�setdefaulttimeoutr r r r � assertEqualr �
gettimeoutr )r r r r r
�testTimeoutDefault1 s
z GeneralTests.testTimeoutDefaultc C sb | j tj� d k� tjd� ztjt| jd d�}W d tjd � X | j |jj � d k� |jj
� d S )Nr )r ) r! 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|jj� d� |jj� d S )Nr )r )r r r r r$ r r% r )r r r r r
�testTimeoutValueF s z GeneralTests.testTimeoutValuec C s: t j� }|jt| jdd� | j|jj� d� |jj� d S )Nr )r ) r r �openr r r$ r r% r )r r r r r
�testTimeoutOpenK s z GeneralTests.testTimeoutOpenc C sJ t jt| jdd�}|j}| j|j� |� | j|j� |j� � |jj� d S )Nr )r ) r r r r r r$ r �filenor )r r Zt_sockr r r
� testGettersQ s
zGeneralTests.testGettersN) �__name__�
__module__� __qualname__r r r r r& r' r( r* r, r r r r
r s
r c @ s, e Zd ZdZf fdd�Zdd� Zdd� ZdS ) �
SocketStubz* a socket proxy that re-defines sendall() c C s t |�| _g | _d| _d S )NF)�list�reads�writes�block)r r2 r r r
�__init__[ s
zSocketStub.__init__c C s | j j|� d S )N)r3 �append)r �datar r r
�sendall_ s zSocketStub.sendallc C s^ d}x&| j r*t|�|k r*|| j jd�7 }qW t|�|krZ| j jd||d � � |d |� }|S )N� r )r2 �len�pop�insert)r �size�outr r r
�recva s zSocketStub.recvN)r- r. r/ �__doc__r5 r8 r? r r r r
r0 Y s r0 c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� TelnetAlikec C s
t � �d S )N)�NotImplementedError)r r r r
r+ k s zTelnetAlike.filenoc C s d S )Nr )r r r r
r m s zTelnetAlike.closec C s
| j j S )N)r r4 )r r r r
�
sock_availn s zTelnetAlike.sock_availc
G s> t j� �}tjj| |f|�� W d Q R X | j|j� 7 _d S )N)r Zcaptured_stdoutr r �msg� _messages�getvalue)r rD r r> r r r
rD p s
zTelnetAlike.msgN)r- r. r/ r+ r rC rD r r r r
rA j s rA c @ sD e Zd Zdd� Zedd� �Zddd�Zdd � Zdd
d �Zd d
� Z dS )� MockSelectorc C s
i | _ d S )N)�keys)r r r r
r5 x s zMockSelector.__init__c C s dS )Ng����MbP?r )r r r r
�
resolution{ s zMockSelector.resolutionNc C s t j|d||�}|| j|< |S )Nr )� selectorsZ SelectorKeyrH )r �fileobj�eventsr7 �keyr r r
�register s
zMockSelector.registerc C s | j j|�S )N)rH r; )r rK r r r
�
unregister� s zMockSelector.unregisterc C sH d}x"| j D ]}t|t�r |jj}P q W |r0g S dd� | j j� D �S d S )NFc S s g | ]}||j f�qS r )rL )�.0rM r r r
�
<listcomp>� s z'MockSelector.select.<locals>.<listcomp>)rH �
isinstancerA r r4 �values)r r r4 rK r r r
�select� s
zMockSelector.selectc C s | j S )N)rH )r r r r
�get_map� s zMockSelector.get_map)N)N)
r- r. r/ r5 �propertyrI rN rO rT rU r r r r
rG v s
rG c # s0 � fdd�}zt j}|t _d V W d |t _X d S )Nc s t � �S )N)r0 )Zignored)r2 r r
�new_conn� s z test_socket.<locals>.new_conn)r Zcreate_connection)r2 rW Zold_connr )r2 r
� test_socket� s
rX c
C sJ x | D ]}t |�tkst|��qW t| �� |dd�}d|_W dQ R X |S )za return a telnetlib.Telnet object that uses a SocketStub with
reads queued up to be read �dummyr � N)�type�bytes�AssertionErrorrX rE )r2 �cls�xr r r r
� test_telnet� s
r` c @ s e Zd Zdd� Zdd� ZdS )�ExpectAndReadTestCasec C s t j| _tt _d S )N)r �_TelnetSelector� old_selectorrG )r r r r
r � s zExpectAndReadTestCase.setUpc C s | j t_d S )N)rc r rb )r r r r
r � s z ExpectAndReadTestCase.tearDownN)r- r. r/ r r r r r r
ra � s ra c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )� ReadTestsc C s| dg}t |�}|jd�}| j|d|j|j|jjfd� dd ddd g}dj|d d � �}t |�}|jd�}| j||� d S ) zc
read_until(expected, timeout=None)
test the blocking version of read_util
s xxxmatchyyys matchs xxxmatch)rD � x�2 � yr9 N� ���)r` Z
read_untilr$ ZcookedqZrawqr r2 r )r �wantr r7 r2 �expectr r r
�test_read_until� s
zReadTests.test_read_untilc C s@ dd dd dd g}dj |�}t|�}|j� }| j||� dS )zJ
read_all()
Read all data until EOF; may block.
re i� rg � zr9 N)r r` �read_allr$ )r r2 rk r r7 r r r
�
test_read_all� s
zReadTests.test_read_allc C sF t dd g�}|j� }| jt|�dk� t � }|j� }| jd|� dS )zQ
read_some()
Read at least one byte or EOF; may block.
re i� rh r9 N)r` Z read_somer! r: r$ )r r r7 r r r
�test_read_some� s zReadTests.test_read_somec C sz dd }t |g�}t||�}d|j_| jd|� � d|j_d}x*y||� 7 }W q@ tk
rd P Y q@X q@W | j||� dS )z
read_*_eager()
Read all data available already queued or on the socket,
without blocking.
re �d Tr9 FN)r` �getattrr r4 r$ �EOFError)r Z func_namerj r �funcr7 r r r
� _read_eager� s
zReadTests._read_eagerc C s | j d� | j d� d S )NZ
read_eagerZread_very_eager)ru )r r r r
�test_read_eager� s
zReadTests.test_read_eagerc C s^ dd }t |g�}| jd|j� � x|jjr6|j� q$W |j� }| j||� | jt|j� d S )Nre rq r9 )r` r$ �read_very_lazyr r2 � fill_rawqZ assertRaisesrs )r rj r r7 r r r
rw � s
zReadTests.read_very_lazyc C s� dd }t |g�}| jd|j� � d}xLy |j� }||7 }|sF|j� W n tk
r^ P Y nX | j|j|�� q(W | j||� d S )Nre rq r9 )r` r$ Z read_lazyrx rs r! �
startswith)r rj r r7 Z read_datar r r
�test_read_lazy s
zReadTests.test_read_lazyN)
r- r. r/ rl ro rp ru rv rw rz r r r r
rd � s
rd c @ s e Zd Zddd�Zdd� ZdS )�nego_collectorNc C s d| _ || _d| _d S )Nr9 )�seen� sb_getter�sb_seen)r r} r r r
r5 s znego_collector.__init__c C s< | j || 7 _ |tjkr8| jr8| j� }| j|7 _d S )N)r| �tl�SEr} r~ )r r �cmdZoptZsb_datar r r
�do_nego s znego_collector.do_nego)N)r- r. r/ r5 r� r r r r
r{ s
r{ c @ s e Zd ZdZdd� ZdS )�
WriteTestszKThe only thing that write does is replace each tl.IAC for
tl.IAC+tl.IACc C s~ ddt j d dt j t j d t j t jdg}xH|D ]@}t� }|j|� dj|jj�}| j|jt jt jt j �|� q6W d S )Ns data sample without IACs data sample withs one IACs a fews iacsr9 ) r �IACr` �writer r r3 r$ �replace)r Z data_sampler7 r Zwrittenr r r
�
test_write' s
zWriteTests.test_writeN)r- r. r/ r@ 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 |�}tdj|��}t� }|j|j� |j� }|j}| jt|�dk� | j |dd� | j
� | j |dd� t j
� | j |t|| �� d|_dS )z helper for testing IAC + cmd r9 r Nrh � )r` r: r r{ � set_option_negotiation_callbackr� rn r| r! �assertIn�cmdsr$ r ZNOOPTr} )r r7 r Zdata_len�nego�txtr� r r r
�
_test_command7 s zOptionTests._test_commandc C sj xN| j D ]D}| jtj|g� | jdd tj|dd g� | jdtj|dg� qW | jdd� | j D �� d S ) Nre rq rg �
c S s g | ]}t j| �qS r )r r� )rP r� r r r
rQ K s z1OptionTests.test_IAC_commands.<locals>.<listcomp>s
xxxxxxxxxxs
yyyyyyyyyy)r� r� r r� )r r� r r r
�test_IAC_commandsE 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|j� |j � }| j
|d� t jt j d t j d t j d }| j
|j |� | j
d|j� � d |_ d S )Ns aas bbs ccs ddr9 s aabb)
r r� ZSBr� r` r{ Z read_sb_datar� r� rn r$ r~ r} )r �sendr r� r� Z want_sb_datar r r
�test_SB_commandsM s "&&.
$z OptionTests.test_SB_commandsc C s� d
t jtdg� dft jt j tdg� dft jt j tdg� dft jt j tdg� dft jt j tdg� d fg}x:|D ]2\}}t|g�}|jd� |j � }| j
||j � q~W d S ) N� a� : recv b''
�X z: IAC 88 not recognized
rh z : IAC DO 1
z
: IAC DONT 1
z
: IAC WILL 1
z
: IAC WONT 1
)r� r� ) r r� r\ ZDOZDONTZWILLZWONTr` �set_debuglevelrn r� rE )r Zgiven_a_expect_b�a�br r� r r r
�test_debuglevel_reads_ s
z!OptionTests.test_debuglevel_readsc C s0 t � }|jd� |jd� d}| j||j� d S )Nrh s xxxz send b'xxx'
)r` r� r� r� rE )r r Zexpectedr r r
�test_debuglevel_writer s
z!OptionTests.test_debuglevel_writec
C sJ t g �� tdd�}d|_W d Q R X |jd� |jd� | j|jd� d S )NrY �0rZ rh �testz0.*test)rX rA rE r� rD Z assertRegex)r r r r r
�test_debug_accepts_str_porty s
z'OptionTests.test_debug_accepts_str_portN)r- r. r/ r ZAOZAYTZBRKZECZELZGAZIPZNOPr� r� r� r� r� r� r� r r r r
r� 3 s $r� c @ s e Zd Zdd� ZdS )� ExpectTestsc C sB ddd g}t |�}|jdg�\}}}| j|dj|dd
� �� dS ) z�
expect(expected, [timeout])
Read until the expected string has been seen, or a timeout is
hit (default is no timeout); may block.
re r� s matchrg r9 Nrh s
xxxxxxxxxxs
yyyyyyyyyyri )r` rk r$ r )r rj r �_r7 r r r
� test_expect� s
zExpectTests.test_expectN)r- r. r/ r� r r r r
r� � s r� �__main__) r rJ r �
contextlibr� r Zunittest�
import_moduler r r ZTestCaser �objectr0 r rA Z BaseSelectorrG �contextmanagerrX r` ra rd r{ r r� r� r� r- �mainr r r r
�<module> s.
B
b P