Sindbad~EG File Manager
a
��g�2 � @ s8 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 k�r4e � � dS ) � N)�support)�
socket_helperc C sZ |� � | �� z:z|�� \}}|�� W n tjy> Y n0 W |�� n
|�� 0 d S �N)�listen�set�accept�close�socket�timeout)�evtZserv�conn�addr� r �//usr/local/lib/python3.9/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)� threading�Eventr r �AF_INET� SOCK_STREAM�sock�
settimeoutr Z bind_port�port�Threadr �thread� setDaemon�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 s20 Y | �|�� � d S r )r& r' r( r ZassertIsNotNone�
get_socketZ assertIsNone)r" �tnr r r �testContextManager- s ,z GeneralTests.testContextManagerc C sf | � t�� d u � t�d� z t�t| j�}W t�d � n t�d � 0 | �|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� z tjt| jd d�}W t�d � n t�d � 0 | � |j� � d u � |j�
� d S �Nr/ )r
) r0 r r1 r2 r& r' r( r r r4 r r) r r r �testTimeoutNone< s
z GeneralTests.testTimeoutNonec C s2 t jt| jdd�}| �|j�� d� |j�� d S r6 )r& r' r( r r3 r r4 r r) r r r �testTimeoutValueG s z GeneralTests.testTimeoutValuec C s: t �� }|jt| jdd� | �|j�� d� |j�� d S r6 ) r& r' �openr( r r3 r r4 r r) r r r �testTimeoutOpenL s z GeneralTests.testTimeoutOpenc C sJ t jt| jdd�}|j}| �|�� |� | �|�� |�� � |j�� d S r6 ) r& r' r( r r r3 r, �filenor )r" r* Zt_sockr r r � testGettersR s
zGeneralTests.testGettersN) �__name__�
__module__� __qualname__r# r% r+ r. r5 r7 r8 r: r< r r r r r s
r c @ s* e Zd ZdZd
dd�Zdd� Zdd� Zd S ) �
SocketStubz* a socket proxy that re-defines sendall() r c C s t |�| _g | _d| _d S )NF)�list�reads�writes�block)r" rB r r r �__init__\ s
zSocketStub.__init__c C s | j �|� d S r )rC �append)r" �datar r r �sendall` s zSocketStub.sendallc C sZ d}| j r(t|�|k r(|| j �d�7 }qt|�|krV| j �d||d � � |d |� }|S )N� r )rB �len�pop�insert)r" �size�outr r r �recvb s zSocketStub.recvN)r )r= r> r? �__doc__rE rH rO r r r r r@ Z s
r@ c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� TelnetAlikec C s
t � �d S r )�NotImplementedErrorr! r r r r; l s zTelnetAlike.filenoc C s d S r r r! r r r r n rI zTelnetAlike.closec C s
| j j S r )r rD r! r r r �
sock_availo s zTelnetAlike.sock_availc G sT t �� �&}tjj| |g|�R � W d � n1 s40 Y | j|�� 7 _d S r )r Zcaptured_stdoutr& r' �msg� _messages�getvalue)r" rT r rN r r r rT q s
4zTelnetAlike.msgN)r= r> r? r; r rS rT r r r r rQ k s rQ 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 rE y 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 )� selectors� SelectorKeyrY )r" �fileobj�eventsrG �keyr r r �register� s
zMockSelector.registerc C s | j �|�S r )rY rK )r" r] r r r �
unregister� s zMockSelector.unregisterc C sF d}| j D ]}t|t�r
|jj} q&q
|r.g S dd� | j �� D �S d S )NFc S s g | ]}||j f�qS r )r^ )�.0r_ r r r �
<listcomp>� rI z'MockSelector.select.<locals>.<listcomp>)rY �
isinstancerQ r rD �values)r" r
rD r] r r r �select� s
zMockSelector.selectc C s | j S r rX r! r r r �get_map� s zMockSelector.get_map)N)N)
r= r> r? rE �propertyrZ r` ra rf rg r r r r rW w s
rW c # s6 � fdd�}z t j}|t _d V W |t _n|t _0 d S )Nc s t � �S r )r@ )Zignored�rB r r �new_conn� s z test_socket.<locals>.new_conn)r �create_connection)rB rj Zold_connr ri r � test_socket� s rl r c C sZ | D ]}t |�tu sJ |��qt| �� |dd�}d|_W d � n1 sL0 Y |S )za return a telnetlib.Telnet object that uses a SocketStub with
reads queued up to be read �dummyr � N)�type�bytesrl rU )rB �cls�xr* r r r � test_telnet� s
$rs c @ s e Zd Zdd� Zdd� ZdS )�ExpectAndReadTestCasec C s t j| _tt _d S r )r& �_TelnetSelector� old_selectorrW r! r r r r# � s zExpectAndReadTestCase.setUpc C s | j t_d S r )rv r& ru r! r r r r% � s z ExpectAndReadTestCase.tearDownN)r= r> r? r# r% r r r r rt � s rt c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )� ReadTestsc C sr dg}t |�}|�d�}| j|d|j|j|jjfd� g d�}d�|dd� �}t |�}|�d�}| �||� dS ) zc
read_until(expected, timeout=None)
test the blocking version of read_util
s xxxmatchyyy� matchs xxxmatch)rT )s2 xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxrx s2 yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyrI N���)rs Z
read_untilr3 ZcookedqZrawqr rB r$ )r" �wantr* rG rB �expectr r r �test_read_until� s
zReadTests.test_read_untilc C s2 g d�}d� |�}t|�}|�� }| �||� dS )zJ
read_all()
Read all data until EOF; may block.
)�� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxs� yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyys� zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzrI N)r$ rs �read_allr3 )r" rB r{ r* rG r r r �
test_read_all� s
zReadTests.test_read_allc C sB t dg�}|�� }| �t|�dk� t � }|�� }| �d|� dS )zQ
read_some()
Read at least one byte or EOF; may block.
r} � rI N)rs Z read_somer0 rJ r3 )r" r* rG 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 q: ty^ Y qbY q:0 q:| �||� dS )z
read_*_eager()
Read all data available already queued or on the socket,
without blocking.
�d xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxTrI FN)rs �getattrr rD r3 �EOFError)r" � func_namerz r* �funcrG 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 sV d}t |g�}| �d|�� � |jjr0|�� q |�� }| �||� | �t|j� d S �Nr� rI )rs r3 �read_very_lazyr rB � fill_rawqZ assertRaisesr� )r" rz r* rG r r r r� � s
zReadTests.read_very_lazyc C s| d}t |g�}| �d|�� � d}z |�� }||7 }|s@|�� W n tyX Y qlY n0 | �|�|�� q"| �||� d S r� )rs r3 Z read_lazyr� r� r0 �
startswith)r" rz r* rG Z read_datar r r �test_read_lazy s
zReadTests.test_read_lazyN)
r= r> r? r| r r� r� r� r� r� r r r r rw � s
rw c @ s e Zd Zddd�Zdd� ZdS )�nego_collectorNc C s d| _ || _d| _d S )NrI )�seen� sb_getter�sb_seen)r" r� r r r rE s znego_collector.__init__c C s< | j || 7 _ |tjkr8| jr8| �� }| j|7 _d S r )r� �tl�SEr� r� )r" r �cmd�optZsb_datar r r �do_nego s znego_collector.do_nego)N)r= r> r? rE 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 sz ddt j d dt j t j d t j t jdg}|D ]@}t� }|�|� d�|jj�}| �|�t jt jt j �|� q4d S )Ns data sample without IACs data sample withs one IACs a fews iacsrI ) r� �IACrs �writer$ r rC r3 �replace)r" Z data_samplerG r* Zwrittenr r r �
test_write( s �
zWriteTests.test_writeN)r= r> r? rP 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� | � |dd� | j
� | � |dd� t j
� | � |t|| �� d|_dS )z helper for testing IAC + cmd rI r Nr� � )rs rJ r$ r� � set_option_negotiation_callbackr� r~ r� r0 �assertIn�cmdsr3 r� ZNOOPTr� )r" rG 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 )Nr� sd yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyy�
xxxxxxxxxx�
yyyyyyyyyyc S s g | ]}t j| �qS r )r� r� )rb r� r r r rc L rI 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 ddrI s aabb)
r� r� ZSBr� rs r� Z read_sb_datar� r� r~ r3 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 ]2\}}t|g�}|�d� |� � }| �
||j � q|d S ) N)� az : recv b''
�X z: IAC 88 not recognized
r� z : IAC DO 1
z
: IAC DONT 1
z
: IAC WILL 1
z
: IAC WONT 1
) r� r� rp ZDOZDONTZWILLZWONTrs �set_debuglevelr~ r� rU )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 )Nr� s xxxz send b'xxx'
)rs r� r� r� rU )r" r* Zexpectedr r r �test_debuglevel_writes s
z!OptionTests.test_debuglevel_writec C s^ t g �� tdd�}d|_W d � n1 s.0 Y |�d� |�d� | �|jd� d S )Nrm �0rn r� �testz0.*test)rl rQ rU r� rT Z assertRegexr) r r r �test_debug_accepts_str_portz s
$
z'OptionTests.test_debug_accepts_str_portN)r= r> r? r� ZAOZAYTZBRKZECZELZGAZIPZNOPr� r� r� r� r� r� r� r r r r r� 4 s $r� c @ s e Zd Zdd� ZdS )� ExpectTestsc C s@ g d�}t |�}|�dg�\}}}| �|d�|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.
)r� rx r� rx rI Nry )rs r{ r3 r$ )r" rz r* �_rG r r r � test_expect� s zExpectTests.test_expectN)r= r> r? r� r r r r r� � s r� �__main__) r r[ r& r �
contextlibr� r Z test.supportr Zunittestr( r ZTestCaser �objectr@ r' rQ � BaseSelectorrW �contextmanagerrl rs rt rw r� r� r� r� r� r= �mainr r r r �<module> s2 B
b P
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists