Your IP : 216.73.216.213
�
f f�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 m Z e j d � Z
e j Z d d � Z Gd d � d e � 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 � Z Gd d � d e � Z Gd d � d e � Z e Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z d d d � Z e d! k r�e � n d S)"� N)�TestCase)�support� threadingc C se | j d � | j � z< y | j � \ } } | j � Wn t j k
rQ Yn XWd | j � Xd S)N� )Zlisten�setZaccept�close�socket�timeout)�evtZservZconnZaddr� r �8/opt/alt/python34/lib64/python3.4/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 r2 r r r �__init__W s zSocketStub.__init__c C s | j j | � d S)N)r3 �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 | � } n | S)N� r )r2 �len�pop�insert)r �size�outr r r �recv] s zSocketStub.recvN)r- r. r/ �__doc__r5 r8 r? r r r r r0 U s r0 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 r4 )r r r r �
sock_availj s zTelnetAlike.sock_availc G sD t j � � } t j j | | | � Wd QX| j | j � 7_ d S)N)r Zcaptured_stdoutr r �msg� _messages�getvalue)r rD r r> r r r rD l s zTelnetAlike.msgN)r- r. r/ r+ r rC rD r r r r rA f s rA 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 r5 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 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 sa d } x0 | j D]% } t | t � r | j j } Pq q W| rC g Sd d � | j j � D� Sd S)NFc S s g | ] } | | j f � q Sr )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)
r- r. r/ r5 �propertyrI rN rO rT rU r r r r rG r s rG c # s> � f d d � } z t j } | t _ d VWd | t _ Xd 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 s@ x | D] } q Wt | � � | d d � } d | _ Wd QX| S)za return a telnetlib.Telnet object that uses a SocketStub with
reads queued up to be read �dummyr � N)rX rE )r2 �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_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. 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 xxxmatchrD � 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 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.
rb i� rd � zr9 N)r r] �read_allr$ )r r2 rh r r7 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.
rb i� re r9 N)r] Z read_somer! r: r$ )r r r7 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.
rb �d Tr9 FN)r] �getattrr r4 r$ �EOFError)r Z func_namerg r �funcr7 r r r � _read_eager� s
zReadTests._read_eagerc C s | j d � | j d � d S)NZ
read_eagerZread_very_eager)rr )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)Nrb rn r9 )r] r$ �read_very_lazyr r2 � fill_rawqZ assertRaisesrp )r rg r r7 r r r rt � s
zReadTests.read_very_lazyc C s� d d } t | g � } | j d | j � � d } x] y- | j � } | | 7} | sd | j � n Wn t k
rz PYn X| j | j | � � q8 W| j | | � d S)Nrb rn r9 )r] r$ Z read_lazyru rp r! �
startswith)r rg r r7 Z read_datar r r �test_read_lazy s
zReadTests.test_read_lazyN)
r- r. r/ ri rl rm rr rs rt rw r r r r ra � s
ra c @ s+ e Z d Z d d d � Z d d � Z d S)�nego_collectorNc C s d | _ | | _ d | _ d S)Nr9 )�seen� sb_getter�sb_seen)r rz r r r r5 s znego_collector.__init__c C sM | j | | 7_ | t j k rI | j rI | j � } | j | 7_ n d S)N)ry �tl�SErz r{ )r r �cmdZoptZsb_datar r r �do_nego s znego_collector.do_nego)r- r. r/ r5 r r r r r rx s rx 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 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 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 r9 r Nre � )r] r: r rx � set_option_negotiation_callbackr rk ry r! �assertIn�cmdsr$ r| ZNOOPTrz )r r7 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) Nrb rn rd �
c S s g | ] } t j | � q Sr )r| r� )rP r~ r r r rQ 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 ddr9 s aabb)
r| r� ZSBr} r] rx Z read_sb_datar� r rk r$ r{ rz )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
re z : IAC DO 1
z
: IAC DONT 1
z
: IAC WILL 1
z
: IAC WONT 1
)r� r� ) r| r� �bytesZDOZDONTZWILLZWONTr] �set_debuglevelrk r� rE )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)Nre s xxxz send b'xxx'
)r] r� r� r� rE )r r Zexpectedr r r �test_debuglevel_writen s
z!OptionTests.test_debuglevel_writec
C s\ t g � � t d d � } d | _ Wd QX| j d � | j d � | j | j d � d S)NrY �0rZ re �testz0.*test)rX rA rE r� rD Z assertRegex)r r r r r �test_debug_accepts_str_portu s
z'OptionTests.test_debug_accepts_str_portN)r- r. r/ 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) z�
expect(expected, [timeout])
Read until the expected string has been seen, or a timeout is
hit (default is no timeout); may block.
rb r� s matchrd r9 Nre s
xxxxxxxxxxs
yyyyyyyyyyrf )r] rh r$ r )r rg r �_r7 r r r � test_expect� s zExpectTests.test_expectN)r- r. r/ r� r r r r r� s r� c C s t j t t t t t � d S)N)r Z run_unittestr ra r� r� r� )�verboser r r � test_main� s r� �__main__) r rJ r Ztime�
contextlibZunittestr r� r �
import_moduler r r
r �objectr0 r rA Z BaseSelectorrG �contextmanagerrX r] r^ ra rx r| r� r� r� r� r- r r r r �<module> s2 =
b P