python (3.12.0)
Αed3 f d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZ d dl Z ej d ej d Z ej Z
d Z G d d e j Z G d
d e Z G d d
e j& Z G d dej* Zej. d ZdefdZ G d de j Z G d de Z G d de Z e Z G d de j Z G d d e j Z G d d e Z e!d k( r e jD yy) N)support)
socket_helperwarnings_helperT)module telnetlibc |j | j |j \ }}|j |j y # t $ r Y w xY w# |j w xY wN)listensetacceptclose TimeoutError)evtservconnaddrs G/BuggyBox/python/3.12.0/bootstrap/lib/python3.12/test/test_telnetlib.pyserverr s\ KKMGGI[[]
d
s# #A A" A% !A""A% %A7c B e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z y
) GeneralTestsc t j | _ t j t j t j
| _ | j j d t j | j | _
t j t | j | j f | _
d| j _ | j j | j j! y )N< )targetargsT) threadingEventr socketAF_INET SOCK_STREAMsock
settimeoutr bind_portportThreadr threaddaemonstartwaitselfs r setUpzGeneralTests.setUp s ??$MM&..&2D2DE R !++DII6 &&fDHHTYY;OP !
c < | j j | ` y r )r% joinr) s r tearDownzGeneralTests.tearDown' s Kr, c t j t | j }|j j y r )r TelnetHOSTr# r r
r* telnets r testBasiczGeneralTests.testBasic+ s( !!$ 2 r, c t j t | j 5 }| j |j d d d | j
j y # 1 sw Y )xY wr )r r1 r2 r# assertIsNotNone
get_socket assertIsNone)r* tns r testContextManagerz GeneralTests.testContextManager0 sU
dDII
. 2" 1 2 "--/* 2 2s A--A6c | j t j d u t j d t j t | j }t j d | j |j j d |j j y # t j d w xY w)N )
assertTruer getdefaulttimeoutsetdefaulttimeoutr r1 r2 r# assertEqualr
gettimeoutr
r3 s r testTimeoutDefaultz GeneralTests.testTimeoutDefault5 s 002d:; $ + %%dDII6F $ $T * //126
$ $T *s $B: :Cc | j t j d u t j d t j t | j d }t j d | j |j j d u |j j y # t j d w xY wNr= )timeout) r> r r? r@ r r1 r2 r# r rB r
r3 s r testTimeoutNonez GeneralTests.testTimeoutNone? s 002d:; $ + %%dDIIt%DF $ $T * ..0D89
$ $T *s &B= =Cc t j t | j d }| j |j
j
d |j
j y rE )r r1 r2 r# rA r rB r
r3 s r testTimeoutValuez GeneralTests.testTimeoutValueJ sJ !!$ 2!> //126 r, c t j }|j t | j d | j |j j d |j j y rE ) r r1 openr2 r# rA r rB r
r3 s r testTimeoutOpenz GeneralTests.testTimeoutOpenO sS !!# D$))R 0 //126 r, c 8 t j t | j d }|j }| j |j
| | j |j |j |j j y rE ) r r1 r2 r# r rA r8 filenor
)r* r4 t_socks r testGetterszGeneralTests.testGettersU sk !!$ 2!> **,f5 &--/: r, N) __name__
__module__ __qualname__r+ r/ r5 r; rC rG rI rL rP r, r r r s/
+
r, r c $ e Zd ZdZddZd Zd Zy)
SocketStubz* a socket proxy that re-defines sendall() c @ t | | _ g | _ d| _ y NF)listreadswritesblock)r* rZ s r __init__zSocketStub.__init___ s %[
r, c : | j j | y r )r[ append)r* datas r sendallzSocketStub.sendallc s 4 r, c d}| j rGt | |k r9|| j j d z
}| j rt | |k r9t | |kD r$| j j d||d |d | }|S )Nr, r )rZ lenpopinsert)r* sizeouts r recvzSocketStub.recve sw jjSX_ 4::>>!$ $C jjSX_ s8d? JJ aTU ,et*C
r, N)rT )rQ rR rS __doc__r] ra rh rT r, r rV rV ] s 4!r, rV c $ e Zd Zd Zd Zd Zd Zy) TelnetAlikec t r )NotImplementedErrorr) s r rN zTelnetAlike.filenoo s !##r, c y r rT r) s r r
zTelnetAlike.closeq s r, c 0 | j j S r )r r\ r) s r
sock_availzTelnetAlike.sock_availr s IIOO#$r, c t j 5 }t j j | |g| d d d | xj
j
z
c_ y # 1 sw Y -xY wr )r captured_stdoutr r1 msg _messagesgetvalue)r* rs r rg s r rs zTelnetAlike.msgt sW
$
$
& 3# s 2T 2 3 #,,.( 3 3s "A##A,N)rQ rR rS rN r
rp rs rT r, r rk rk n s $%r, rk c > e Zd Zd Zed ZddZd ZddZd Z y) MockSelectorc i | _ y r keysr) s r r] zMockSelector.__init__| s r, c y)NgMbP?rT r) s r
resolutionzMockSelector.resolution s r, Nc T t j |d|| }|| j |< |S )Nr ) selectors SelectorKeyrz )r* fileobjeventsr` keys r registerzMockSelector.register s+ ##GQ= '
r, c 8 | j j | S r )rz rd )r* r s r
unregisterzMockSelector.unregister s yy}}W%%r, c d}| j D ]* }t |t s|j j } n |rg S | j j D cg c] }||j f c}S c c}w rX )rz
isinstancerk r r\ valuesr )r* rF r\ r r s r selectzMockSelector.select sj yy G';/ ** I151A1A1CD#S#**%D DDs A4c | j S r ry r) s r get_mapzMockSelector.get_map s yyr, r )
rQ rR rS r] propertyr| r r r r rT r, r rw rw z s0
& Er, rw c # K fd} t j }|t _ d |t _ y # t _ w xY ww)Nc t S r )rV )ignoredrZ s r new_connz test_socket.<locals>.new_conn s % r, )r create_connection)rZ r old_conns ` r test_socketr s? !,++#+
#+
$, s A 5 A
AArT c v | D ] } t | 5 |dd }d|_ ddd |S # 1 sw Y S xY w)za return a telnetlib.Telnet object that uses a SocketStub with
reads queued up to be read dummyr N)r rt )rZ clsxr4 s r test_telnetr sL # # U Wa M Ms .8c e Zd Zd Zd Zy)ExpectAndReadTestCasec L t j | _ t t _ y r )r _TelnetSelector old_selectorrw r) s r r+ zExpectAndReadTestCase.setUp s % 5 5$0 !r, c . | j t _ y r )r r r r) s r r/ z ExpectAndReadTestCase.tearDown s $($5$5 !r, N)rQ rR rS r+ r/ rT r, r r r s 16r, r c 6 e Zd Zd Zd Zd Zd Zd Zd Zd Z y) ReadTestsc L dg}t | }|j d }| j |d|j |j |j
j f g d}dj |dd }t | }|j d }| j || y) zc
read_until(expected, timeout=None)
test the blocking version of read_util
s xxxmatchyyy matchs xxxmatch)rs )s2 xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxr s2 yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyr, N)r
read_untilrA cookedqrawqr rZ r. )r* wantr4 r` rZ expects r test_read_untilzReadTests.test_read_until s
T" * { fkkN_N_0`a0%*% U# * v&r, c g d}dj | }t | }|j }| j || y)zJ
read_all()
Read all data until EOF; may block.
) xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxs yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyys zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzr, N)r. r read_allrA )r* rZ r r4 r` s r
test_read_allzReadTests.test_read_all s?
5% U# v&r, c t dg }|j }| j t | dk\ t }|j }| j d| y)zQ
read_some()
Read at least one byte or EOF; may block.
r r, N)r read_somer> rc rA )r* r4 r` s r test_read_somezReadTests.test_read_some sW j\* ! D Q' ! d#r, c
d}t |g }t || }d|j _ | j d | d|j _ d} | | z
} # t
$ r Y nw xY w| j || y)z
read_*_eager()
Read all data available already queued or on the socket,
without blocking.
d xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxTr, FN)r getattrr r\ rA EOFError)r* func_namer r4 funcr` s r _read_eagerzReadTests._read_eager s dV$vy) df% !
t$s
A# # A/.A/c H | j d | j d y )N
read_eagerread_very_eager)r r) s r test_read_eagerzReadTests.test_read_eager s!
& *+r, c ^ d}t |g }| j d|j |j j r'|j |j j r'|j }| j || | j
t |j y Nr r, )r rA read_very_lazyr rZ fill_rawq assertRaisesr )r* r r4 r` s r r zReadTests.read_very_lazy s dV$ f 3 3 56kk kk $$& t$ (F$9$9:r, c : d}t |g }| j d|j d} |j }||z
}|s|j | j |j
| I# t $ r Y nw xY w| j || y r )r rA read_lazyr r r>
startswith)r* r r4 r` read_datas r test_read_lazyzReadTests.test_read_lazy s dV$ f . . 01
" , , . ! $$&
OODOOD 1 2
t$s 'A; ; BBN)
rQ rR rS r r r r r r r rT r, r r r s% '"
$ %(,;%r, r c e Zd ZddZd Zy)nego_collectorNc . d| _ || _ d| _ y )Nr, )seen sb_gettersb_seen)r* r s r r] znego_collector.__init__ s " r, c | xj ||z z
c_ |t j k( r3| j r&| j }| xj |z
c_ y y y r )r tlSEr r )r* r cmdoptsb_datas r do_negoznego_collector.do_nego sF S3Y "%%<DNNnn&G LLG #L +<r, r )rQ rR rS r] r rT r, r r r s
$r, r c e Zd ZdZd Zy)
WriteTestszKThe only thing that write does is replace each tl.IAC for
tl.IAC+tl.IACc ddt j z dz dt j z t j z dz t j z t j dg}|D ] }t }|j | dj |j
j }| j |j t j t j t j z | y )Ns data sample without IACs data sample withs one IACs a fews iacsr, ) r IACr writer. r r[ rA replace)r* data_sampler` r4 writtens r
test_writezWriteTests.test_write+ s 1*RVV3kA "&&(2661H<rvvEvv
JD ]F LL hhv{{ 1 12G T\\"&& ? I Jr, N)rQ rR rS ri r rT r, r r r ' s
Jr, r c e Zd Zej ej
ej ej ej ej ej ej gZ d Z
d Zd Zd Zd Zd Zy) OptionTestsc t | }t dj | }t }|j |j
|j
}|j }| j t | dkD | j |dd | j | j |dd t j | j |t ||z d|_ y)z helper for testing IAC + cmd r, r Nr )r rc r. r set_option_negotiation_callbackr r r r> assertIncmdsrA r NOOPTr )r* r` r4 data_lennegotxtr s r
_test_commandzOptionTests._test_command; s T"sxx~& ..t||<oo ii C1 %
c"1gtyy) Qq288, 3sSy>2 r, c r | j D ]i }| j t j |g | j dt j |dg | j dt j |dg k | j | j D cg c] }t j |z c} y c c}w )Nr sd yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyy
xxxxxxxxxx
yyyyyyyyyy)r r r r )r* r s r test_IAC_commandsz OptionTests.test_IAC_commandsI s 99 BC } -
BFFC B C 2663 @ A B
dii@cBFFSL@A@s B4c t j t j z t j z t j z t j t j z t j z t j z t j z t j z t j t j z t j z t j z dz t j z t j z t j t j z dz t j z t j z t j z t j z t j t j z dz t j z t j z dz t j z t j z g}t | }t |j }|j |j |j }| j |d t j t j z dz t j z dz t j z dz }| j |j | | j d|j
d |_ y )Ns aas bbs ccs ddr, s aabb)
r r SBr r r read_sb_datar r r rA r r )r* sendr4 r r want_sb_datas r test_SB_commandsz OptionTests.test_SB_commandsQ s '"%%/'"&&02669BEEA'"&&058266ABEEI&/"&&8266ABEEI&/"&&85@266IBEEQ T" f 1 12..t||<oo c"vv02669EABFFJUR |4 f 1 1 34 r, c d dt j t dg z dft j t j z t dg z dft j t j z t dg z dft j t j
z t dg z dft j t j z t dg z dfg}|D ]N \ }}t |g }|j d |j }| j ||j P y ) 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 bytesDODONTWILLWONTr set_debuglevelr r rt )r* given_a_expect_babr4 r s r test_debuglevel_readsz!OptionTests.test_debuglevel_readsc s
#
VVeRDk
!#> ?
VVbee^eQCj
(. 9
VVbgg
qc
*,< =
VVbgg
qc
*,< =
VVbgg
qc
*,< =
% /DAq !%F ! !! $//#C MM!V - - . /
r, c t }|j d |j d d}| j ||j y )Nr s xxxz send b'xxx'
)r r r r rt )r* r4 expecteds r test_debuglevel_writez!OptionTests.test_debuglevel_writev s< a V "
h 0 01r, c t g 5 t dd }d|_ d d d j d |j d | j |j d y # 1 sw Y HxY w)Nr 0r r testz0.*test)r rk rt r rs assertRegexr3 s r test_debug_accepts_str_portz'OptionTests.test_debug_accepts_str_port} sd
_ " #.F !F " a
6 )):6 " "s A''A0N)rQ rR rS r AOAYTBRKECELGAIPNOPr r r r r r r rT r, r r r 7 sW EE266266255"%%rvv FD B $&27r, r c e Zd Zd Zy) ExpectTestsc g d}t | }|j dg \ }}}| j |dj |dd y)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 r r r, Nr )r r rA r. )r* r r4 _r` s r test_expectzExpectTests.test_expect sH 0 T"]]H:.
1T sxxSb 23r, N)rQ rR rS r rT r, r r r s 4r, r __main__)#r r~ r
contextlibr r test.supportr r unittestrequires_working_socketimport_deprecatedr r2 r TestCaser objectrV r1 rk BaseSelectorrw contextmanagerr r r r r r r r r rQ mainrT r, r <module>r! s=
7 t , -O - -k : @ 8$$ @ D "
)""
9)) @ k 6H - - 6`%% `%D
$V
$ J"" J M7(## M7`
4'
4 zHMMO r, 
Browse
Usage
Versions