File: //usr/local/lib/python3.7/test/__pycache__/test_telnetlib.cpython-37.opt-2.pyc
B
��g�2 � @ s, d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlZejZdd� Z G dd� dej
�Z G dd� de �Z
G d d
� d
ej�ZG d d � d ej�Zejd
d� �Zdefdd�ZG dd� dej
�ZG dd� de�ZG dd� de �ZeZG dd� dej
�ZG dd� dej
�ZG d d � d e�Ze d k�r(e� � dS ) � N)�supportc C sT |� � | �� z4y|�� \}}|�� W n tjk
r@ Y nX W d |�� X d S )N)Zlisten�setZaccept�close�socket�timeout)�evtZservZconnZaddr� r �//usr/local/lib/python3.7/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 �� | _t�tjtj�| _| j�d� t� | j�| _
t j t | j| jfd�| _
| j
�d� | j
�� | j�� d S )N�< )�target�argsT)� threadingZEventr 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 �� | ` d S )N)r �join)r r r r �tearDown# s
zGeneralTests.tearDownc C s t �t| j�}|j�� d S )N)� telnetlib�Telnet�HOSTr r r )r �telnetr r r � testBasic' s zGeneralTests.testBasicc C s: t �t| j��}| �|�� � W d Q R X | �|�� � d S )N)r r r r ZassertIsNotNone�
get_socketZ assertIsNone)r �tnr r r �testContextManager, s z GeneralTests.testContextManagerc C s\ | � t�� d k� t�d� zt�t| j�}W d t�d � X | �|j �
� d� |j � � d S )N� ) �
assertTruer �getdefaulttimeout�setdefaulttimeoutr r r r � assertEqualr �
gettimeoutr )r r r r r �testTimeoutDefault1 s
z GeneralTests.testTimeoutDefaultc C sb | � t�� d k� t�d� ztjt| jd d�}W d t�d � X | � |j� � d k� |j�
� 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�� d� |j�� 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 �� }|jt| jdd� | �|j�� d� |j�� 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�� 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 Zd dd�Zdd� Zdd� ZdS )
�
SocketStubr c C s t |�| _g | _d| _d S )NF)�list�reads�writes�block)r r3 r r r �__init__[ s
zSocketStub.__init__c C s | j �|� d S )N)r4 �append)r �datar r r �sendall_ s zSocketStub.sendallc C s^ d}x&| j r*t|�|k r*|| j �d�7 }qW t|�|krZ| j �d||d � � |d |� }|S )N� r )r3 �len�pop�insert)r �size�outr r r �recva s zSocketStub.recvN)r )r. r/ r0 r6 r9 r@ r r r r r1 Y s
r1 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 r: zTelnetAlike.closec C s
| j j S )N)r r5 )r r r r �
sock_availn s zTelnetAlike.sock_availc G s> t �� �}tjj| |f|�� W d Q R X | 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/ r0 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 r6 x s 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 SelectorKeyrH )r �fileobj�eventsr8 �keyr r r �register s
zMockSelector.registerc C s | 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 �� 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 r5 �values)r r r5 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/ r0 r6 �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)r1 )Zignored)r3 r r �new_conn� s z test_socket.<locals>.new_conn)r Zcreate_connection)r3 rW Zold_connr )r3 r � test_socket� s
rX r c C s6 x | D ]}qW t | �� |dd�}d|_W d Q R X |S )N�dummyr � )rX rE )r3 �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)r` r r_ )r r r r r � s z ExpectAndReadTestCase.tearDownN)r. r/ r0 r r r r r r r^ � s r^ 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� dddg}d�|d d� �}t |�}|�d�}| �||� d S ) Ns xxxmatchyyys matchs xxxmatch)rD s2 xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxs2 yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyr: ���)r] Z
read_untilr% ZcookedqZrawqr r3 r )r �wantr r8 r3 �expectr r r �test_read_until� s
zReadTests.test_read_untilc C s4 dddg}d� |�}t|�}|�� }| �||� d S )Ns� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxs� yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyys� zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzr: )r r] �read_allr% )r r3 rd r r8 r r r �
test_read_all� s
zReadTests.test_read_allc C sB t dg�}|�� }| �t|�dk� t � }|�� }| �d|� d S )Ns� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx� r: )r] Z read_somer" r; r% )r r r8 r r r �test_read_some� s
zReadTests.test_read_somec C sv d}t |g�}t||�}d|j_| �d|� � d|j_d}x*y||� 7 }W q< tk
r` P Y q<X q<W | �||� d S )Nsd xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxTr: F)r] �getattrr r5 r% �EOFError)r Z func_namerc r �funcr8 r r r � _read_eager� s
zReadTests._read_eagerc C s | � d� | � d� d S )NZ
read_eagerZread_very_eager)rm )r r r r �test_read_eager� s
zReadTests.test_read_eagerc C sZ d}t |g�}| �d|�� � x|jjr2|�� q W |�� }| �||� | �t|j� d S )Nsd xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxr: )r] r% �read_very_lazyr r3 � fill_rawqZ assertRaisesrk )r rc r r8 r r r ro � s
zReadTests.read_very_lazyc C s� d}t |g�}| �d|�� � d}xLy |�� }||7 }|sB|�� W n tk
rZ P Y nX | �|�|�� q$W | �||� d S )Nsd xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxr: )r] r% Z read_lazyrp rk r" �
startswith)r rc r r8 Z read_datar r r �test_read_lazy s
zReadTests.test_read_lazyN)
r. r/ r0 re rg ri rm rn ro rr r r r r ra � s
ra c @ s e Zd Zddd�Zdd� ZdS )�nego_collectorNc C s d| _ || _d| _d S )Nr: )�seen� sb_getter�sb_seen)r ru r r r r6 s znego_collector.__init__c C s< | j || 7 _ |tjkr8| jr8| �� }| j|7 _d S )N)rt �tl�SEru rv )r r �cmd�optZsb_datar r r �do_nego s znego_collector.do_nego)N)r. r/ r0 r6 r{ r r r r rs s
rs c @ s e Zd Zdd� ZdS )�
WriteTestsc C s~ ddt j d dt j t j d t j t jdg}xH|D ]@}t� }|�|� d�|jj�}| �|�t jt jt j �|� q6W d S )Ns data sample without IACs data sample withs one IACs a fews iacsr: ) rw �IACr] �writer r r4 r% �replace)r Z data_sampler8 r Zwrittenr r r �
test_write' s
zWriteTests.test_writeN)r. r/ r0 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 rh � )r] r; r rs � set_option_negotiation_callbackr{ rf rt r" �assertIn�cmdsr% rw ZNOOPTru )r r8 r Zdata_len�nego�txtry r r r �
_test_command7 s zOptionTests._test_commandc C sb xF| j D ]<}| �tj|g� | �dtj|dg� | �dtj|dg� qW | �dd� | j D �� d S )Nsd xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxsd yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyys
xxxxxxxxxxs
yyyyyyyyyyc S s g | ]}t j| �qS r )rw r} )rP ry r r r rQ K s z1OptionTests.test_IAC_commands.<locals>.<listcomp>)r� r� rw r} )r ry 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� |� � }| �
|d� t jt j d t j d t j d }| �
|j |� | �
d|�� � d |_ d S )Ns aas bbs ccs ddr: s aabb)
rw r} ZSBrx r] rs Z read_sb_datar� r{ rf r% rv ru )r �sendr r� r� Z want_sb_datar r r �test_SB_commandsM 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}x:|D ]2\}}t|g�}|�d� |� � }| �
||j � q~W d S ) N)� az : 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
) rw r} �bytesZDOZDONTZWILLZWONTr] �set_debuglevelrf 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 � }|�d� |�d� d}| �||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 |�d� |�d� | �|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/ r0 rw ZAOZAYTZBRKZECZELZGAZIPZNOPr� r� r� r� r� r� r� r r r r r� 3 s $r� c @ s e Zd Zdd� ZdS )� ExpectTestsc C sB dddg}t |�}|�dg�\}}}| �|d�|d d� �� d S )Ns
xxxxxxxxxxs matchs
yyyyyyyyyyr: rb )r] rd r% r )r rc r �_r8 r r r � test_expect� s
zExpectTests.test_expectN)r. r/ r0 r� r r r r r� � s r� �__main__) r rJ r r �
contextlibr� r Zunittestr r
ZTestCaser �objectr1 r rA Z BaseSelectorrG �contextmanagerrX r] r^ ra rs rw r| r� r� r. �mainr r r r �<module> s. B
b P