Your IP : 216.73.217.129
��Yf?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)Zlisten�setZaccept�close�socket�timeout)�evtZservZconnZaddr� r �8/opt/alt/python35/lib64/python3.5/test/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 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 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% r& r( r* r r r r
r s
r c @ s= e Z d Z d Z f d d � Z d d � Z d d � Z d S) �
SocketStubz* a socket proxy that re-defines sendall() c C s% t | � | _ g | _ d | _ d S)NF)�list�reads�writes�block)r r0 r r r
�__init__W s zSocketStub.__init__c C s | j j | � d S)N)r1 �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 )r0 �len�pop�insert)r �size�outr r r
�recv] s zSocketStub.recvN)r+ r, r- �__doc__r3 r6 r= r r r r
r. U s r. 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
r) 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 r2 )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 rB r r< r r r
rB l s zTelnetAlike.msgN)r+ r, r- r) r rA rB r r r r
r? f s r? 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
r3 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 )� selectorsZ SelectorKeyrF )r �fileobj�eventsr5 �keyr r r
�register{ s
zMockSelector.registerc C s | j j | � S)N)rF r9 )r rI 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 )rJ )�.0rK r r r
�
<listcomp>� s z'MockSelector.select.<locals>.<listcomp>)rF �
isinstancer? r r2 �values)r r r2 rI r r r
�select� s zMockSelector.selectc C s | j S)N)rF )r r r r
�get_map� s zMockSelector.get_map)
r+ r, r- r3 �propertyrG rL rM rR rS r r r r
rE r s rE c # s> � f d d � } z t j } | t _ d VWd | t _ Xd S)Nc s
t � � S)N)r. )Zignored)r0 r r
�new_conn� s z test_socket.<locals>.new_conn)r Zcreate_connection)r0 rU Zold_connr )r0 r
� test_socket� s
rV c
C sA x | D] } q Wt | � � | d d � } d | _ Wd QRX| S)za return a telnetlib.Telnet object that uses a SocketStub with
reads queued up to be read �dummyr � N)rV rC )r0 �cls�xr r r r
� test_telnet� s
r[ 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_selectorrE )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, r- r r r r r r
r\ � s r\ 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) zc
read_until(expected, timeout=None)
test the blocking version of read_util
s xxxmatchyyys matchs xxxmatchrB � x�2 � yr7 N� ���)r[ Z
read_untilr" ZcookedqZrawqr r0 r )r �wantr r5 r0 �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)zJ
read_all()
Read all data until EOF; may block.
r` i� rb � zr7 N)r r[ �read_allr" )r r0 rf r r5 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)zQ
read_some()
Read at least one byte or EOF; may block.
r` i� rc r7 N)r[ Z read_somer r8 r" )r r r5 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)z
read_*_eager()
Read all data available already queued or on the socket,
without blocking.
r` �d Tr7 FN)r[ �getattrr r2 r" �EOFError)r Z func_namere r �funcr5 r r r
� _read_eager� s
zReadTests._read_eagerc C s | j d � | j d � d S)NZ
read_eagerZread_very_eager)rp )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)Nr` rl r7 )r[ r" �read_very_lazyr r0 � fill_rawqZ assertRaisesrn )r re r r5 r r r
rr � 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)Nr` rl r7 )r[ r" Z read_lazyrs rn r �
startswith)r re r r5 Z read_datar r r
�test_read_lazy s
zReadTests.test_read_lazyN)
r+ r, r- rg rj rk rp rq rr ru r r r r
r_ � s
r_ c @ s+ e Z d Z d d d � Z d d � Z d S)�nego_collectorNc C s d | _ | | _ d | _ d S)Nr7 )�seen� sb_getter�sb_seen)r rx r r r
r3 s znego_collector.__init__c C sJ | j | | 7_ | t j k rF | j rF | j � } | j | 7_ d S)N)rw �tl�SErx ry )r r �cmdZoptZsb_datar r r
�do_nego s znego_collector.do_nego)r+ r, r- r3 r} r r r r
rv s rv c @ s" e Z d Z d Z d d � Z d S)�
WriteTestszKThe only thing that write does is replace each tl.IAC for
tl.IAC+tl.IACc 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 iacsr7 ) rz �IACr[ �writer r r1 r" �replace)r Z data_sampler5 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 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)z helper for testing IAC + cmd r7 r Nrc � )r[ r8 r rv � set_option_negotiation_callbackr} ri rw r �assertIn�cmdsr" rz ZNOOPTrx )r r5 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) Nr` rl rb �
c S s g | ] } t j | � q Sr )rz r )rN r| r r r
rO G s z1OptionTests.test_IAC_commands.<locals>.<listcomp>s
xxxxxxxxxxs
yyyyyyyyyy)r� r� rz 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 ddr7 s aabb)
rz r ZSBr{ r[ rv Z read_sb_datar� r} ri r" ry rx )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
rc z : IAC DO 1
z
: IAC DONT 1
z
: IAC WILL 1
z
: IAC WONT 1
)r� r� ) rz r �bytesZDOZDONTZWILLZWONTr[ �set_debuglevelri r� rC )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)Nrc s xxxz send b'xxx'
)r[ r� r� r� rC )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)NrW �0rX rc �testz0.*test)rV r? rC r� rB Z assertRegex)r r r r r
�test_debug_accepts_str_portu s
z'OptionTests.test_debug_accepts_str_portN)r+ r, r- rz 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) z�
expect(expected, [timeout])
Read until the expected string has been seen, or a timeout is
hit (default is no timeout); may block.
r` r� s matchrb r7 Nrc s
xxxxxxxxxxs
yyyyyyyyyyrd )r[ rf r" r )r re r �_r5 r r r
� test_expect� s zExpectTests.test_expectN)r+ r, r- r� r r r r
r� s r� �__main__) r rH r Ztime�
contextlibr� r Zunittest�
import_moduler r r ZTestCaser �objectr. r r? Z BaseSelectorrE �contextmanagerrV r[ r\ r_ rv rz r~ r� r� r+ �mainr r r r
�<module> s0 =
b P