Your IP : 216.73.216.213
���]?2 � @ s� d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l Z e j d � Z e j
Z
d d � Z Gd d � d e j � Z
Gd d � d e � Z Gd
d � d e j � Z Gd d
� d
e j � Z e j d d � � Z f e d d � Z Gd d � d e j � Z Gd d � d e � Z Gd d � d e � Z e Z Gd d � d e j � Z Gd d � d e j � Z Gd d � d e � Z e d k r�e j � d S) � N)�support� threadingc C sb | j � | j � z< y | j � \ } } | j � Wn t j k
rN Yn XWd | j � Xd S)N)�listen�set�accept�close�socket�timeout)�evtZserv�conn�addr� r
�3/opt/alt/python35/lib64/python3.5/test_telnetlib.py�server
s
r c @ sp e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d S)� GeneralTestsc C s� t j � | _ t j t j t j � | _ | j j d � t j | j � | _
t j d t d | j | j f � | _
| j
j d � | j
j � | j j � d S)N�< �target�argsT)r �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 j � | ` d S)N)r �join)r r
r
r �tearDown$ s
zGeneralTests.tearDownc C s& t j t | j � } | j j � d S)N)� telnetlib�Telnet�HOSTr r r )r �telnetr
r
r � testBasic( s zGeneralTests.testBasicc C sz | j t j � d k � t j d � z t j t | j � } Wd t j d � X| j | j j
� d � | j j � d S)N� ) �
assertTruer �getdefaulttimeout�setdefaulttimeoutr# r$ r% r � assertEqualr �
gettimeoutr )r r& r
r
r �testTimeoutDefault- s
z GeneralTests.testTimeoutDefaultc C s� | j t j � d k � t j d � z t j t | j d d �} Wd t j d � X| j | j j � d k � | j j
� d S)Nr( r ) r) r r* r+ r# r$ r% r r r- r )r r& r
r
r �testTimeoutNone7 s
z GeneralTests.testTimeoutNonec C sE t j t | j d d �} | j | j j � d � | j j � d S)Nr r( )r# r$ r% r r, r r- r )r r& r
r
r �testTimeoutValueB s z GeneralTests.testTimeoutValuec C sO t j � } | j t | j d d �| j | j j � d � | j j � d S)Nr r( ) r# r$ �openr% r r, r r- r )r r& r
r
r �testTimeoutOpenG s z GeneralTests.testTimeoutOpenc C sg t j t | j d d �} | j } | j | j � | � | j | j � | j � � | j j � d S)Nr r( ) r# r$ r% r r r, Z
get_socket�filenor )r r& Zt_sockr
r
r � testGettersM s
zGeneralTests.testGettersN) �__name__�
__module__� __qualname__r r" r' r. r/ r0 r2 r4 r
r
r
r r s
r c @ s7 e Z d Z f d d � Z d d � Z d d � Z d S)�
SocketStubc C s% t | � | _ g | _ d | _ d S)NF)�list�reads�writes�block)r r: r
r
r �__init__W s zSocketStub.__init__c C s | j j | � d S)N)r; �append)r �datar
r
r �sendall[ s zSocketStub.sendallc C s� d } x5 | j r= t | � | k r= | | j j d � 7} q Wt | � | k r} | j j d | | d � � | d | � } | S)N� r )r: �len�pop�insert)r �size�outr
r
r �recv] s zSocketStub.recvN)r5 r6 r7 r= r@ rG r
r
r
r r8 U s r8 c @ s@ e Z d Z d d � Z d d � Z d d � Z d d � Z d S)
� TelnetAlikec C s
t � � d S)N)�NotImplementedError)r r
r
r r3 g s zTelnetAlike.filenoc C s d S)Nr
)r r
r
r r i s zTelnetAlike.closec C s | j j S)N)r r< )r r
r
r �
sock_availj s zTelnetAlike.sock_availc G sE t j � � } t j j | | | � Wd QRX| j | j � 7_ d S)N)r Zcaptured_stdoutr# r$ �msg� _messages�getvalue)r rK r rF r
r
r rK l s zTelnetAlike.msgN)r5 r6 r7 r3 r rJ rK r
r
r
r rH f s rH c @ sd e Z d Z d d � Z e d d � � Z d d d � Z d d � Z d d
d � Z d d
� Z d S)� MockSelectorc C s
i | _ d S)N)�keys)r r
r
r r= t s zMockSelector.__init__c C s d S)Ng����MbP?r
)r r
r
r �
resolutionw s zMockSelector.resolutionNc C s) t j | d | | � } | | j | <| S)Nr )� selectors� SelectorKeyrO )r �fileobj�eventsr? �keyr
r
r �register{ s
zMockSelector.registerc C s | j j | � S)N)rO rC )r rS r
r
r �
unregister� s zMockSelector.unregisterc C s^ d } x- | j D]" } t | t � r | j j } Pq W| r@ g Sd d � | j j � D� Sd S)NFc S s g | ] } | | j f � q Sr
)rT )�.0rU r
r
r �
<listcomp>� s z'MockSelector.select.<locals>.<listcomp>)rO �
isinstancerH r r< �values)r r r< rS r
r
r �select� s zMockSelector.selectc C s | j S)N)rO )r r
r
r �get_map� s zMockSelector.get_map)
r5 r6 r7 r= �propertyrP rV rW r\ r] r
r
r
r rN r s rN c # s> � f d d � } z t j } | t _ d VWd | t _ Xd S)Nc s
t � � S)N)r8 )Zignored)r: r
r �new_conn� s z test_socket.<locals>.new_conn)r �create_connection)r: r_ Zold_connr
)r: r � test_socket� s
ra c
C sA x | D] } q Wt | � � | d d � } d | _ Wd QRX| S)N�dummyr � )ra rL )r: �cls�xr& r
r
r � test_telnet� s
rf c @ s( e Z d Z d d � Z d d � Z d S)�ExpectAndReadTestCasec C s t j | _ t t _ d S)N)r# �_TelnetSelector� old_selectorrN )r r
r
r r � s zExpectAndReadTestCase.setUpc C s | j t _ d S)N)ri r# rh )r r
r
r r" � s z ExpectAndReadTestCase.tearDownN)r5 r6 r7 r r" r
r
r
r rg � s rg c @ sd e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d S)� ReadTestsc C s� d g } t | � } | j d � } | j | d d | j | j | j j f �d d d d d g } d j | d d
� � } t | � } | j d � } | j | | � d S) Ns xxxmatchyyys matchs xxxmatchrK � x�2 � yrA � ���)rf Z
read_untilr, ZcookedqZrawqr r: r! )r �wantr& r? r: �expectr
r
r �test_read_until� s + zReadTests.test_read_untilc C sV d d d d d d g } d j | � } t | � } | j � } | j | | � d S)Nrk i� rm � zrA )r! rf �read_allr, )r r: rq r& r? r
r
r �
test_read_all� s zReadTests.test_read_allc C sa t d d g � } | j � } | j t | � d k � t � } | j � } | j d | � d S)Nrk i� rn rA )rf Z read_somer) rB r, )r r& r? r
r
r �test_read_some� s zReadTests.test_read_somec C s� d d } t | g � } t | | � } d | j _ | j d | � � d | j _ d } x+ y | | � 7} Wq\ t k
r� PYq\ Xq\ W| j | | � d S)Nrk �d TrA F)rf �getattrr r< r, �EOFError)r � func_namerp r& �funcr? r
r
r � _read_eager� s
zReadTests._read_eagerc C s | j d � | j d � d S)NZ
read_eagerZread_very_eager)r| )r r
r
r �test_read_eager� s
zReadTests.test_read_eagerc C s d d } t | g � } | j d | j � � x | j j rK | j � q2 W| j � } | j | | � | j t | j � d S)Nrk rw rA )rf r, �read_very_lazyr r: � fill_rawqZ assertRaisesry )r rp r& r? r
r
r r~ � s
zReadTests.read_very_lazyc C s� d d } t | g � } | j d | j � � d } xZ y* | j � } | | 7} | sa | j � Wn t k
rw PYn X| j | j | � � q8 W| j | | � d S)Nrk rw rA )rf r, Z read_lazyr ry r) �
startswith)r rp r& r? Z read_datar
r
r �test_read_lazy s
zReadTests.test_read_lazyN)
r5 r6 r7 rr ru rv r| r} r~ r� r
r
r
r rj � s
rj c @ s+ e Z d Z d d d � Z d d � Z d S)�nego_collectorNc C s d | _ | | _ d | _ d S)NrA )�seen� sb_getter�sb_seen)r r� r
r
r r= s znego_collector.__init__c C sJ | j | | 7_ | t j k rF | j rF | j � } | j | 7_ d S)N)r� �tl�SEr� r� )r r �cmd�optZsb_datar
r
r �do_nego s znego_collector.do_nego)r5 r6 r7 r= r� r
r
r
r r� s r� c @ s e Z d Z d d � Z d S)�
WriteTestsc C s� d d t j d d t j t j d t j t j d g } xb | D]Z } t � } | j | � d j | j j � } | j | j t j t j t j � | � qC Wd S)Ns data sample without IACs data sample withs one IACs a fews iacsrA ) r� �IACrf �writer! r r; r, �replace)r Z data_sampler? r& Zwrittenr
r
r �
test_write# s
zWriteTests.test_writeN)r5 r6 r7 r� r
r
r
r r� s r� c @ s� e Z d Z e j e j e j e j e j e j e j
e j g Z d d � Z
d d � Z d d � Z d d � Z d d
� Z d d � Z d
S)� OptionTestsc C s� t | � } t d j | � � } t � } | j | j � | j � } | j } | j t | � d k � | j | d d � | j
� | j | d d � t j
� | j | t | | � � d | _ d S)NrA r rn � )rf rB r! r� � set_option_negotiation_callbackr� rt r� r) �assertIn�cmdsr, r� ZNOOPTr� )r r? r& Zdata_len�nego�txtr� r
r
r �
_test_command3 s zOptionTests._test_commandc C s� xg | j D]\ } | j t j | g � | j d d t j | d d g � | j d t j | d g � q
W| j d d � | j D� � d S) Nrk rw rm �
c S s g | ] } t j | � q Sr
)r� r� )rX r� r
r
r rY G s z1OptionTests.test_IAC_commands.<locals>.<listcomp>s
xxxxxxxxxxs
yyyyyyyyyy)r� r� r� r� )r r� r
r
r �test_IAC_commandsA s
$ z OptionTests.test_IAC_commandsc C s| t j t j t j t j t j t j t j t j t j t j t j t j t j t j d t j t j t j t j d t j t j t j t j t j t j d t j t j d t j t j g } t | � } t | j � } | j | j � | j � } | j
| d � t j t j d t j d t j d } | j
| j | � | j
d | j � � d | _ d S)Ns aas bbs ccs ddrA s aabb)
r� r� ZSBr� rf r� Z read_sb_datar� r� rt r, r� r� )r �sendr& r� r� Z want_sb_datar
r
r �test_SB_commandsI s )--7 *z OptionTests.test_SB_commandsc C s� d
t j t d g � d f t j t j t d g � d f t j t j t d g � d f t j t j t d g � d f t j t j t d g � d f g } xO | D]G \ } } t | g � } | j d � | j � } | j
| | j � q� Wd S) N� a� : recv b''
�X z: IAC 88 not recognized
rn z : IAC DO 1
z
: IAC DONT 1
z
: IAC WILL 1
z
: IAC WONT 1
)r� r� ) r� r� �bytesZDOZDONTZWILLZWONTrf �set_debuglevelrt r� rL )r Zgiven_a_expect_b�a�br& r� r
r
r �test_debuglevel_reads[ s &
z!OptionTests.test_debuglevel_readsc C s@ t � } | j d � | j d � d } | j | | j � d S)Nrn s xxxz send b'xxx'
)rf r� r� r� rL )r r& Zexpectedr
r
r �test_debuglevel_writen s
z!OptionTests.test_debuglevel_writec
C s] t g � � t d d � } d | _ Wd QRX| j d � | j d � | j | j d � d S)Nrb �0rc rn �testz0.*test)ra rH rL r� rK Z assertRegex)r r& r
r
r �test_debug_accepts_str_portu s
z'OptionTests.test_debug_accepts_str_portN)r5 r6 r7 r� ZAOZAYTZBRKZECZELZGAZIPZNOPr� r� r� r� r� r� r� r
r
r
r r� / s 6 r� c @ s e Z d Z d d � Z d S)� ExpectTestsc C s] d d d g } t | � } | j d g � \ } } } | j | d j | d d � � � d S)
Nrk r� s matchrm rA rn s
xxxxxxxxxxs
yyyyyyyyyyro )rf rq r, r! )r rp r& �_r? r
r
r � test_expect� s zExpectTests.test_expectN)r5 r6 r7 r� r
r
r
r r� s r� �__main__) r rQ r# �time�
contextlibr� r Zunittest�
import_moduler r% r ZTestCaser �objectr8 r$ rH � BaseSelectorrN �contextmanagerra rf rg rj r� r� r� r� r� r5 �mainr
r
r
r �<module> s0 =
b P