Xh d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm Z d dl
m Z d dl m
Z
mZ d dlmZ d dlmZmZ d dlmZ d dlmZ d d lmZ d d
lmZm Z d d l m Z m Z m Z m!Z! d d l"m#Z#m$Z$ d dl%Z
d d
l&m'Z' d dl%m*Z*m+Z+m,Z,m-Z- d dl
m.Z. G d de* Z/ G d de/ Z0 G d de/ Z1 G d de/ Z2 G d de/ Z3 G d de$ Z4 G d d e$ Z5 G d d e/ Z6 G d d!e/ Z7 G d" d#e/ Z8 G d$ d%e/ Z9 G d& d'e/ Z: G d( d)e/ Z; G d* d+e/ Z< G d, d-e/ Z= G d. d/e/ Z> G d0 d1e/ Z? G d2 d3e Z@ G d4 d5e@ ZA G d6 d7e/ ZB G d8 d9e@ ZCe G d: d;e@ ZDe G d< d=eD ZE G d> d?eE ZF G d@ dAeE ZG G dB dCeE ZH G dD dEeD ZIe G dF dGej ZK G dH dIeK ZL ej e.du dJ G dK dLeK ZN G dM dNe@ ZO G dO dPe@ ZP G dQ dRe@ ZQ G dS dTej ZR G dU dVe@ ZS G dW dXe@ ZTy# e($ r ejR w xY w# e($ r dZ.Y w xY w)Y N)Future)gen) HTTPError HTTPRequest)Event)gen_logapp_log)ResolverSimpleAsyncHTTPClient)
DictLoader)abstract_base_testignore_deprecation)AsyncHTTPTestCasegen_testbind_unused_port ExpectLog) ApplicationRequestHandler_websocket_mask_python)WebSocketHandlerwebsocket_connectWebSocketErrorWebSocketClosedError)speedupsc $ e Zd ZdZddZd Zd Zy)TestWebSocketHandlerzBase class for testing handlers that exposes the on_close event.
This allows for tests to see the close code and reason on the
server side.
Nc || _ || _ y N close_futurecompression_options)selfr" r# s O/var/www/html/myenv/lib/python3.12/site-packages/tornado/test/websocket_test.py
initializez TestWebSocketHandler.initialize6 s (#6 c | j S r r# r$ s r% get_compression_optionsz,TestWebSocketHandler.get_compression_options: s '''r' c | j 2| j j | j | j f y y r )r"
set_result
close_code close_reasonr* s r% on_closez TestWebSocketHandler.on_close= s6 ( ( ($//4;L;L)M N )r' NN)__name__
__module__ __qualname____doc__r& r+ r0 r' r% r r . s 7(Or' r c 0 e Zd Zej d Zy) EchoHandlerc # K | j |t |t y # t j $ r Y y t
$ r Y y w xY wwr )
write_message
isinstancebytesasyncioCancelledErrorr r$ messages r%
on_messagezEchoHandler.on_messageC sG $$Wj%.HI I%% # s/ A
"' A
AA
AA
AA
N)r2 r3 r4 r coroutinerA r6 r' r% r8 r8 B s ]] r' r8 c e Zd Zd Zy)ErrorInOnMessageHandlerc ddz y )N r r6 r? s r% rA z"ErrorInOnMessageHandler.on_messageN s Ar' Nr2 r3 r4 rA r6 r' r% rD rD M s r' rD c e Zd Zd Zy)
HeaderHandlerc ( t j | j d t j | j d t j | j dd t j | j
dd t j | j d | j | j g}|D ] } | t d | j | j j j dd y # t $ r Y Xw xY w) NzThis should not workzhttp://localhost/elsewhereX-Test ChocolateChipi z did not get expected exception) functoolspartialwriteredirect
set_header
set_cookie
set_statusflushfinish Exception RuntimeErrorr: requestheadersget)r$ methods_to_testmethods r% openzHeaderHandler.openS s djj*@ A dmm-I J doox < doo{F C doos 3 JJ KK
& F
@AA
4<<//33HbAB
s <D DDNr2 r3 r4 r_ r6 r' r% rI rI R s Cr' rI c e Zd Zd Zd Zy)HeaderEchoHandlerc ( | j dd y )NX-Extra-Response-HeaderExtra-Response-Value)rS r* s r% set_default_headersz%HeaderEchoHandler.set_default_headersi s 13IJr' c | j j j D ]7 \ }}|j j d s&| j || 9 y )Nzx-test)rZ r[ get_alllower
startswithrS )r$ kvs r% preparezHeaderEchoHandler.preparel sJ LL((002 &DAqwwy##H-1% &r' N)r2 r3 r4 rf rm r6 r' r% rb rb h s
K&r' rb c e Zd Zd Zy)NonWebSocketHandlerc & | j d y )Nok)rQ r* s r% r\ zNonWebSocketHandler.gets s
4r' Nr2 r3 r4 r\ r6 r' r% ro ro r s r' ro c e Zd Zd Zy)RedirectHandlerc & | j d y )N/echo)rR r* s r% r\ zRedirectHandler.getx s
g r' Nrr r6 r' r% rt rt w s r' rt c e Zd Zd Zy)CloseReasonHandlerc 6 d| _ | j dd y )NF goodbye)on_close_calledcloser* s r% r_ zCloseReasonHandler.open} s $
4#r' Nr` r6 r' r% rx rx | $r' rx c 6 e Zd Zej d Zd Zy)AsyncPrepareHandlerc # 0 K t j y wr )r momentr* s r% rm zAsyncPrepareHandler.prepare s jjs c & | j | y r r: r? s r% rA z AsyncPrepareHandler.on_message s 7#r' N)r2 r3 r4 r rB rm rA r6 r' r% r r s ]] $r' r c e Zd Zd Zy)PathArgsHandlerc & | j | y r r )r$ args r% r_ zPathArgsHandler.open s 3 r' Nr` r6 r' r% r r r' r c B e Zd Z fdZej
d Z xZS )CoroutineOnMessageHandlerc 2 t | di | d| _ y Nr r6 superr& sleepingr$ kwargs __class__s r% r& z$CoroutineOnMessageHandler.initialize
$V$
r' c # K | j dkD r| j d | xj dz
c_ t j d | xj dz c_ | j | y wNr z%another coroutine is already sleepingrF {Gz?r r: r sleepr? s r% rA z$CoroutineOnMessageHandler.on_message sY ==1 F G
iio
7#s A4A6)r2 r3 r4 r& r rB rA
__classcell__r s @r% r r s ]]$ $r' r c e Zd Zd Zy)RenderMessageHandlerc H | j | j d| y )N message.html)r@ )r:
render_stringr? s r% rA z RenderMessageHandler.on_message s 4--ng-NOr' NrG r6 r' r% r r s Pr' r c * e Zd Z fdZd Zd Z xZS )SubprotocolHandlerc 2 t | di | d| _ y NFr6 )r r& select_subprotocol_calledr s r% r& z SubprotocolHandler.initialize s
$V$).&r' c J | j r t d d| _ d|v ryy )Nz select_subprotocol called twiceT goodproto)r rX )r$ subprotocolss r% select_subprotocolz%SubprotocolHandler.select_subprotocol s- ) ) => >)-& , & r' c n | j s t d | j d| j z y )Nz select_subprotocol not calledzsubprotocol=%s)r rX r: selected_subprotocolr* s r% r_ zSubprotocolHandler.open s1 -- ;< < +d.G.GGHr' )r2 r3 r4 r& r r_ r r s @r% r r s /Ir' r c H e Zd Z fdZej
d Zd Z xZS )OpenCoroutineHandlerc @ t | di | || _ d| _ y r )r r& test
open_finished)r$ r r r s r% r& z OpenCoroutineHandler.initialize s"
$V$ "r' c # K | j j j t j d d| _ y w)Nr T)r message_sentwaitr r r r* s r% r_ zOpenCoroutineHandler.open s6 ii$$))++ii !s AAc T | j s t d | j d y )Nz&on_message called before open finishedrq )r rX r: r? s r% rA z OpenCoroutineHandler.on_message s% !! DE E 4 r' ) r2 r3 r4 r& r rB r_ rA r r s @r% r r s% #
]]" "
!r' r c e Zd Zd Zy)ErrorInOpenHandlerc t d )Nboom)rX r* s r% r_ zErrorInOpenHandler.open s
r' Nr` r6 r' r% r r r r' r c e Zd Zd Zy)ErrorInAsyncOpenHandlerc ^ K t j d d { t d 7 w)Nr r )r= r rX r* s r% r_ z ErrorInAsyncOpenHandler.open s( mmA s -+-Nr` r6 r' r% r r s r' r c e Zd Zd Zy)NoDelayHandlerc H | j d | j d y )NThello) set_nodelayr: r* s r% r_ zNoDelayHandler.open s 7#r' Nr` r6 r' r% r r r~ r' r c L e Zd Z fdZ fdZej d Z xZS )WebSocketBaseTestCasec 0 t | g | _ y r )r setUpconns_to_closer$ r s r% r zWebSocketBaseTestCase.setUp s
r' c d | j D ] }|j t |
y r )r r} r tearDown)r$ connr s r% r z WebSocketBaseTestCase.tearDown s, '' D JJL
r' c + K t d| j |fz fi |}| j j | t j
| w)Nzws://127.0.0.1:%d%s)r
get_http_portr appendr Return)r$ pathr wss r%
ws_connectz WebSocketBaseTestCase.ws_connect sW $ !T%7%7%94$@ @
DJ
""2&jjn s AA) r2 r3 r4 r r r rB r r r s @r% r r s% !
]] r' r c " e Zd Zd Zd Z fdZd Zd Zd Ze d Z
d Z e d Z e d
Z
e d Ze d Ze d
Ze d Ze d Ze d Ze d Ze d Ze d Ze d Ze d Ze d Ze d Ze d Ze d Z e d Z e d Z e d Z e d Z e d Z!e d Z"e d Z#e d! Z$e d" Z%e d# Z&e d$ Z'e d% Z( xZ)S )&
WebSocketTestc t | _ t dt t | j fdt
fdt fdt t | j fdt t | j fdt t | j fdt t | j fd t t | j fd
t t | j fd t t | j fd t t | j fd
t t | j fdt t | j | fdt" fdt$ fdt& fgt) ddi S )Nrv )r" /non_ws /redirectz/headerz /header_echo
/close_reason/error_in_on_message/async_preparez/path_args/(.*)
/coroutine/render /subprotocol/open_coroutine)r" r /error_in_open/error_in_async_open/nodelayr z{{ message }})template_loader)r r" r r8 dictro rt rI rb rx rD r r r r r r r r r r
r* s r% get_appzWebSocketTest.get_app sv "H+t9J9J'KL /0 o.M4T=N=N+OP"%d&7&78 $&d&7&78 ++d&7&78 %'d&7&78 &#d&7&78 !-d&7&78
0$DDUDU2VW"&d&7&78 &(d&7&7dC
"#56')@A^,a1
d '8N'O Pg4
4
r' c t S r r r* s r% get_http_clientz WebSocketTest.get_http_client& s $&&r' c ^ t | t j j y r )r r r _template_loadersclearr s r% r zWebSocketTest.tearDown* s
((..0r' c ^ | j d }| j |j d y )Nrv fetch assertEqualcoder$ responses r% test_http_requestz WebSocketTest.test_http_request. s$ ::g& ,r' c j | j ddddd }| j |j d y )Nrv Upgrade WebSocket13
Connectionr zSec-WebSocket-Versionr[ r r r s r% test_missing_websocket_keyz(WebSocketTest.test_missing_websocket_key3 < :: '&)-
,r' c j | j ddddd }| j |j d y )Nrv r r 12r r i r r s r% test_bad_websocket_versionz(WebSocketTest.test_bad_websocket_version> r r' c # K | j d }|j d |j }| j |d y w)Nrv r r r: read_messager r$ r r s r% test_websocket_genz WebSocketTest.test_websocket_genI sG ??7+
+ w'' ** 7+ A Ac t 5 t d j z j d d d j j }|j
d |j j j j } j |d j j fd |j j y # 1 sw Y xY w)Nws://127.0.0.1:%d/echo)callbackr c $ j S r )stop)fr$ s r% z8WebSocketTest.test_websocket_callbacks..Z s
diik r' ) r r r r r resultr: r r r" add_done_callbackr} r s ` r% test_websocket_callbacksz&WebSocketTest.test_websocket_callbacksP s
! (4+=+=+??$))
YY[
!
!
"99;%%' 7+ ++,AB
s )C66C?c # K | j d }|j dd |j }| j |d y w)Nrv s hello Tbinaryr r s r% test_binary_messagez!WebSocketTest.test_binary_message^ sI ??7+
+
t4 ** =1r c # K | j d }|j d |j }| j |d y wNrv u hello ér r s r% test_unicode_messagez"WebSocketTest.test_unicode_messagee sD ??7+
+
( ** >2 A
A c # K | j d }|j | j t 5 |j d d d d y # 1 sw Y y xY wwr )r r} assertRaisesr r: r$ r s r% )test_error_in_closed_client_write_messagez7WebSocketTest.test_error_in_closed_client_write_messagel sQ ??7+
+
3
4 - ^ , - - -s 9A"A
A"A A"c # K | j d }|j d |j }| j |d y w)Nr r z hellor r s r% test_render_messagez!WebSocketTest.test_render_messages sD ??9-
-
! ** >2r c # K | j d }|j d t t d 5 |j }d d d | j y # 1 sw Y xY ww)Nr r Uncaught exception)r r: r r r assertIsNoner s r% test_error_in_on_messagez&WebSocketTest.test_error_in_on_messagez sb ??#9:
:
!
w 4
5 / __ ..H / (# / /s 5A0A$
A0$A-)A0c # K | j t 5 }| j d d d d | j j j
d y # 1 sw Y 0xY ww)Nz /notfoundi )r r r r exceptionr )r$ cms r% test_websocket_http_failz&WebSocketTest.test_websocket_http_fail sS
y
) /R//+. . / **C0 / /s A'A/A'A$ A'c # K | j t 5 | j d d d d y # 1 sw Y y xY ww)Nr )r r r r* s r% test_websocket_http_successz)WebSocketTest.test_websocket_http_success s7
~
. -//), , - - - A5 A>Ac # K | j t 5 | j d d d d y # 1 sw Y y xY ww)Nr )r r r r* s r% test_websocket_http_redirectz*WebSocketTest.test_websocket_http_redirect s7
y
) ///+. . / / /r% c # K t \ }}|j | j t 5 t t
dd 5 t
d|z d d d d d d d y # 1 sw Y xY w# 1 sw Y y xY ww)Nz.*F)requiredzws://127.0.0.1:%d/i )connect_timeout)r r} r IOErrorr r r )r$ sockports r% test_websocket_network_failz)WebSocketTest.test_websocket_network_fail sz %'
d
w
' 7D59
'(4/
s4 3BA8A,A8# B,A5 1A88B=Bc # K t j t d| j z 5 }|j d |j d |j
j
d d d y # 1 sw Y y xY ww)Nr r world)
contextlibclosingr r r: streamr} r s r% "test_websocket_close_buffered_dataz0WebSocketTest.test_websocket_close_buffered_data so
$%=@R@R@T%TU
U
W % W % IIOO
s 2B=A:1 B:B?Bc # K t j t t d| j z ddi 5 }|j }| j
|d d d d y # 1 sw Y y xY ww)Nzws://127.0.0.1:%d/headerrK r r )r1 r2 r r r r r r s r% test_websocket_headersz$WebSocketTest.test_websocket_headers s|
' 2T5G5G5II!)7 3
0 __ ..H Xw / 0 0 0s ?A;%A/& A;/A84A;c # ^ K t j t t d| j z ddi 5 }| j |j j d d | j |j j d d d d d y # 1 sw Y y xY ww)Nz ws://127.0.0.1:%d/header_echoz X-Test-Hellor r rd re )r1 r2 r r r r r[ r\ r s r% test_websocket_header_echoz(WebSocketTest.test_websocket_header_echo s
' 7$:L:L:NN!/ 9
RZZ^^N ;W E
89;Q
s ?B-AB! B-!B*&B-c # 0 K | j d }|j }| j |d | j |j d | j |j
d | j \ }}| j |d y w)Nr rz r{ )r r assertIsr r. r/ r" )r$ r msgr reasons r% test_server_close_reasonz&WebSocketTest.test_server_close_reason s ???3
3OO%%
c4 - )4 ! . .. f
t$s BBc # K | j d }|j dd | j \ }}| j |d | j |d y w)Nrv rz r{ )r r} r" r )r$ r r r< s r% test_client_close_reasonz&WebSocketTest.test_client_close_reason sW ??7+
+
y! ! . .. f t$ + A A c # K | j d }|j }| j |d | j t 5 |j d d d d y # 1 sw Y y xY ww)Nr r )r r r: r r r: )r$ r r; s r% test_write_after_closez$WebSocketTest.test_write_after_close se ???3
3OO%%
c4
3
4 & W % & & &s A
A6A*! A6*A3/A6c # K | j d }|j d |j }| j |d y w)Nr r r r$ r ress r% test_async_preparez WebSocketTest.test_async_prepare sG ??#34
4
!OO%% g&r c # z K | j d }|j }| j |d y w)Nz/path_args/hellor r r r rD s r% test_path_argsz WebSocketTest.test_path_args s7 ??#56
6OO%% g& 9;c # K | j d }|j d |j d |j }| j |d |j }| j |d y w)Nr hello1hello2r rD s r% test_coroutinez WebSocketTest.test_coroutine sw ??<0
0 x(( x((OO%% h'OO%% h' BBc # . K | j }d|z }dd|z i}t j t t || 5 }|j d |j
}| j |d d d d y # 1 sw Y y xY ww)Nr Originzhttp://127.0.0.1:%dr r r r1 r2 r r r: r r r$ r- urlr[ r r s r% test_check_origin_valid_no_pathz-WebSocketTest.test_check_origin_valid_no_path s !!#&- 2T 9:
$[g%FG
G
0
W % __ ..H Xw / 0 0 0 AB
6B B BBc # . K | j }d|z }dd|z i}t j t t || 5 }|j d |j
}| j |d d d d y # 1 sw Y y xY ww)Nr rQ z http://127.0.0.1:%d/somethingr r rR rS s r% !test_check_origin_valid_with_pathz/WebSocketTest.test_check_origin_valid_with_path s !!#&- Bc # K | j }t j d| }|D ch c] }|d }}t j |vr| j d y d|z }ddi}| j
t 5 }t t || d d d | j j j d y c c}w # 1 sw Y 5xY ww) N localhostr z"localhost does not resolve to ipv4zws://localhost:%d/echorQ zhttp://subtenant.localhostr rZ )
r r
resolvesocketAF_INETskipTestr r r r r r r )r$ r- addrinfoaddrfamiliesrT r[ r! s r% $test_check_origin_invalid_subdomainsz2WebSocketTest.test_check_origin_invalid_subdomains8 s !!#
"++K>>(01DG11 >> ) MM> ? &- 9:
y
) GR#KW$EF F G
**C0 2 G Gs( 1C C AC C 4C CC c # K | j dddg }| j |j d |j }| j |d y w)Nr badprotor )r zsubprotocol=goodproto)r r r r rD s r% test_subprotocolsz WebSocketTest.test_subprotocolsO s` ?? *k)B #
00+>OO%% 56s AAc # K | j d }| j |j d |j }| j |d y w)Nr zsubprotocol=None)r r: r r r rD s r% test_subprotocols_not_offeredz+WebSocketTest.test_subprotocols_not_offeredX sK ??>2
2
b--t4OO%% 01s AAc # K t | _ | j d }|j d | j j |j }| j
|d y w)Nr r rq )r r r r: setr r rD s r% test_open_coroutinez!WebSocketTest.test_open_coroutine_ sd !G??#45
5 w'' OO%% d#s A5A7c # K t t d 5 | j d }|j }d d d | j y # 1 sw Y xY ww)Nr r r r r r r rD s r% test_error_in_openz WebSocketTest.test_error_in_openh sV
w 4
5 *'788B))C *
# * * A &AA A A c # K t t d 5 | j d }|j }d d d | j y # 1 sw Y xY ww)Nr r rs rD s r% test_error_in_async_openz&WebSocketTest.test_error_in_async_openo sV
w 4
5 *'=>>B))C *
# * *ru c # z K | j d }|j }| j |d y w)Nr r rH rD s r% test_nodelayzWebSocketTest.test_nodelayv s6 ??:.
.OO%% g&rJ )*r2 r3 r4 r r r r r r r r r r r r r r r" r$ r' r. r4 r6 r8 r= r? rB rF rI rN rU rX r] r_ ri rl rn rq rt rw ry r r s @r% r r s 6
p'1-
- - , , 2 2 3 3 - - 3 3 $ $ 1 1
- - / /
0
0 & % % , , & & ' ' ' '
( ( 0 0 0 0 1 1 1 1 1 1, 7 7 2 2 $ $ ' 'r' r c $ e Zd Z fdZd Z xZS ) NativeCoroutineOnMessageHandlerc 2 t | di | d| _ y r r r s r% r& z*NativeCoroutineOnMessageHandler.initialize~ r r' c K | j dkD r| j d | xj dz
c_ t j d d { | xj dz c_ | j | y 7 +wr r r? s r% rA z*NativeCoroutineOnMessageHandler.on_message sa ==1 F G
iio
7# s AA>A<,A>)r2 r3 r4 r& rA r r s @r% r{ r{ } s
$r' r{ c " e Zd Zd Zed Zy) WebSocketNativeCoroutineTestc & t dt fg S )N/native)r r{ r* s r% r z$WebSocketNativeCoroutineTest.get_app s Y(G HIJJr' c # K | j d }|j d |j d |j }| j |d |j }| j |d y w)Nr rL rM r rD s r% test_native_coroutinez2WebSocketNativeCoroutineTest.test_native_coroutine sw ??9-
- x(( x((OO%% h'OO%% h'rO N)r2 r3 r4 r r r r6 r' r% r r s K ( (r' r c V e Zd ZdZd Zd Zd Zdededdfd Ze d
Z
e d Z y) CompressionTestMixinz Hello world. Testing 123 123c G d dt }t dt t | j fd|t | j fg S )Nc " e Zd Zed Zd Zy)4CompressionTestMixin.get_app..LimitedHandlerc y)N r6 r* s r% max_message_sizezECompressionTestMixin.get_app..LimitedHandler.max_message_size s r' c J | j t t | y r )r: strlenr? s r% rA z?CompressionTestMixin.get_app..LimitedHandler.on_message s ""3s7|#45r' N)r2 r3 r4 propertyr rA r6 r' r% LimitedHandlerr s
6r' r rv r) /limited)r r r8 r get_server_compression_options)r$ r s r% r z CompressionTestMixin.get_app s_ 6 1 6 T-P-P-RS "T-P-P-RS
r' c y r r6 r* s r% r z3CompressionTestMixin.get_server_compression_options r' c y r r6 r* s r% get_client_compression_optionsz3CompressionTestMixin.get_client_compression_options r r' bytes_in bytes_outreturnNc t r NotImplementedErrorr$ r r s r% verify_wire_bytesz&CompressionTestMixin.verify_wire_bytes !##r' c # h K | j d| j }t d D ]K }|j | j |j }| j
|| j M | j
|j j t | j dz | j
|j j t | j dz | j |j j |j j y w)Nrv r) )r r ranger: MESSAGEr r protocol_message_bytes_outr _message_bytes_inr _wire_bytes_in_wire_bytes_outr$ r ir s r% test_message_sizesz'CompressionTestMixin.test_message_sizes s ?? )L)L)N #
q 5A T\\ * __ ..H Xt|| 4 5
77T\\9JQ9NO 66DLL8IA8MN r{{ 9 92;;;V;VWs D0D2c # $ K | j d| j }|j d |j }| j |d |j d |j }| j | y w)Nr r) ڀaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa128 aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa)r r r: r r r r s r% test_size_limitz$CompressionTestMixin.test_size_limit s ?? D,O,O,Q #
# ** 5) $ ** (#s BB) r2 r3 r4 r r r r intr r r r r6 r' r% r r sZ ,G
0$# $# $$ $ X X $ $r' r c e Zd ZdZd Zy)UncompressedTestMixinzESpecialization of CompressionTestMixin when we expect no compression.c | j |dt | j dz z | j |dt | j dz z y )Nr )r r r r s r% r z'UncompressedTestMixin.verify_wire_bytes sH AT\\):Q)>$?@ 1DLL(9A(=#>?r' N)r2 r3 r4 r5 r r6 r' r% r r s O@r' r c e Zd Zy)NoCompressionTestN)r2 r3 r4 r6 r' r% r r s r' r c e Zd Zd Zy)ServerOnlyCompressionTestc i S r r6 r* s r% r z8ServerOnlyCompressionTest.get_server_compression_options r' N)r2 r3 r4 r r6 r' r% r r r' r c e Zd Zd Zy)ClientOnlyCompressionTestc i S r r6 r* s r% r z8ClientOnlyCompressionTest.get_client_compression_options r r' N)r2 r3 r4 r r6 r' r% r r r r' r c e Zd Zd Zd Zd Zy)DefaultCompressionTestc i S r r6 r* s r% r z5DefaultCompressionTest.get_server_compression_options r r' c i S r r6 r* s r% r z5DefaultCompressionTest.get_client_compression_options r r' c | j |dt | j dz z | j |dt | j dz z | j ||dz y )Nr r r )
assertLessr r r r s r% r z(DefaultCompressionTest.verify_wire_bytes sX 1DLL(9A(=#>? !s4<<'81'<"=> HrM2r' N)r2 r3 r4 r r r r6 r' r% r r s 3r' r c @ e Zd ZdededefdZdej fdZy)MaskFunctionMixinmaskdatar c t r r r$ r r s r% r zMaskFunctionMixin.mask r r' r$ c | j | j dd d | j | j dd d | j | j dd d | j | j dd d | j | j d
d d | j | j d
d d y )Ns abcdr' b s 54321s TVPVPs ZXCVs 98765432s c`t`olpds s s s s s )r r r* s r% test_maskzMaskFunctionMixin.test_mask s 7C0#6 7D17; 7H5x@ 7K8+F
II)+F G '
II)+F G '
r' N)r2 r3 r4 r< r typingAnyr r6 r' r% r r s, $ $e $ $
r' r c e Zd Zd Zy)PythonMaskFunctionTestc t || S r r r s r% r zPythonMaskFunctionTest.mask s %dD11r' Nr2 r3 r4 r r6 r' r% r r s 2r' r z#tornado.speedups module not presentc e Zd Zd Zy)CythonMaskFunctionTestc . t j || S r )r websocket_maskr s r% r zCythonMaskFunctionTest.mask% s &&tT22r' Nr r6 r' r% r r # s 3r' r c " e Zd Zd Zed Zy)ServerPeriodicPingTestc B G d dt }t d|fgdd S )Nc e Zd Zd Zy)3ServerPeriodicPingTest.get_app..PingHandlerc & | j d y )Ngot pongr r$ r s r% on_pongz;ServerPeriodicPingTest.get_app..PingHandler.on_pong, "":.r' N)r2 r3 r4 r r6 r' r% PingHandlerr +
/r' r /r r )websocket_ping_intervalwebsocket_ping_timeoutr r r$ r s r% r z ServerPeriodicPingTest.get_app* s/ /. / ;
$(#$
r' c # K | j d }t d D ]& }|j }| j |d ( y w)Nr r r )r r r r r s r% test_server_pingz'ServerPeriodicPingTest.test_server_ping5 sI ??3'
'q 3A __ ..H Xz 2 3s A A N)r2 r3 r4 r r r r6 r' r% r r ) s
3 3r' r c " e Zd Zd Zed Zy)ClientPeriodicPingTestc < G d dt }t d|fg S )Nc e Zd Zd Zy)3ClientPeriodicPingTest.get_app..PingHandlerc & | j d y )Ngot pingr r s r% on_pingz;ClientPeriodicPingTest.get_app..PingHandler.on_pingA r r' Nr2 r3 r4 r r6 r' r% r r @ r r' r r r r s r% r z ClientPeriodicPingTest.get_app? s$ /. / S+ ./00r' c # K | j ddd }t d D ]& }|j }| j |d ( |j y w)Nr r r
ping_interval ping_timeoutr r )r r r r r} r s r% test_client_pingz'ClientPeriodicPingTest.test_client_pingF sZ ??3d?K
Kq 3A __ ..H Xz 2 3
r@ N)r2 r3 r4 r r r r6 r' r% r r > s 1 r' r c 2 e Zd Zd Zed Zed Zy)ServerPingTimeoutTestc X g | _ | G fddt }t d|fg }|S )Nc $ e Zd Zd fd Z xZS )2ServerPingTimeoutTest.get_app..PingHandlerc ~ j | _ | j j | t |
|| S )Nr! )handlersr r r& )r$ r" r# r r s r% r& z=ServerPingTimeoutTest.get_app..PingHandler.initializeU s= $
$$T* w)!-CV * r' r1 )r2 r3 r4 r& r )r r s @r% r r T s
r' r r )r r r )r$ r appr s @r% r z ServerPingTimeoutTest.get_appP s5 02
. C-./
r' c d _ d _ fd} | j j j _ y)z1Optionally suppress the client's "pong" response.Fr c 0 dt dt f fd }|S )Nopcoder c d | dk( r"xj dz
c_ j ry | | S )N
rF )pongs_received
drop_pongs)r r fcnr s r% _innerzCServerPingTimeoutTest.install_hook..wrapper.._innerh s2 S=%%*%}} 64((r' )r r< )r r
r s ` r% wrapperz3ServerPingTimeoutTest.install_hook..wrapperg s
)s
)%
) Mr' N)r r
r _handle_message)r r s ` r% install_hookz"ServerPingTimeoutTest.install_hook` s6
'.bkk.I.I&J #r' c # K d}| j d||dz }| j | | j d }t d D ]g }t j
| | j
|j | j
|j |j j gJ | j |j d d|_ t j
|dz | j |j d | j |j d
| j |j j d y w) Ng?r r r Tg ?i zping timed out)r r r r r r r r. r/ r assertGreaterEqualr
r r )r$ intervalr handler_s r% test_client_ping_timeoutz.ServerPingTimeoutTest.test_client_ping_timeoutu s5 ?? xhl #
" --"q 2A))H% %
g 0 0 1 g 2 2 3 ;;))1 11 2
1 115
ii3''
++T2 --/?@
//6s
B-E0B EN)r2 r3 r4 r staticmethodr r r r6 r' r% r r O s1 K K( %7 %7r' r c e Zd Zd Zy)PingCalculationTestc h ddl m} t j ddddddt j j }d}t j ddddd d
t j j }|j |j
||j
}| j |d y )
Nr )WebSocketProtocol13i rF r )tzinfor ; 6 )last_ping_timer nowr )tornado.websocketr datetimetimezoneutcping_sleep_time timestampr )r$ r r# r r"
sleep_times r% test_ping_sleep_timez(PingCalculationTest.test_ping_sleep_time s 9 aB1X=N=N=R=RS!** !QB8+<+<+@+@
)88)335
9
Q'r' N)r2 r3 r4 r+ r6 r' r% r r s
(r' r c " e Zd Zd Zed Zy)ManualPingTestc < G d dt }t d|fg S )Nc e Zd Zd Zy)+ManualPingTest.get_app..PingHandlerc F | j |t |t y )Nr )r: r; r< r s r% r z3ManualPingTest.get_app..PingHandler.on_ping s ""4
40G"Hr' Nr r6 r' r% r r0 s
Ir' r r r r s r% r zManualPingTest.get_app s& I. I S+ ./00r' c # H K | j d }| j t |j d |j d |j }| j |d |j d |j }| j |d y w)Nr ~aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaar s hellos binary hello)r r
ValueErrorpingr r )r$ r resps r% test_manual_pingz ManualPingTest.test_manual_ping s ??3'
' *bggy9
__&& x(
__&& /s B B"N)r2 r3 r4 r r r7 r6 r' r% r- r- s 1 0 0r' r- c " e Zd Zd Zed Zy)MaxMessageSizeTestc * t dt fgd S )Nr r )websocket_max_message_size)r r8 r* s r% r zMaxMessageSizeTest.get_app s S+ ./DQQr' c # K | j d }d}|j | |j }| j || |j |dz |j }| j |d | j |j
d | j |j d y w)Nr aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaabi zmessage too big)r r: r r r: r. r/ )r$ r r; r6 s r% test_large_messagez%MaxMessageSizeTest.test_large_message s ??3'
'
__&& s# s#__&&
dD! - *;N)r2 r3 r4 r r r? r6 r' r% r9 r9 s R = =r' r9 )Ur= r1 r% rO rc tracebackr unittesttornado.concurrentr tornador tornado.httpclientr r
tornado.locksr tornado.logr r tornado.netutilr
tornado.simple_httpclientr tornado.templater
tornado.test.utilr r tornado.testingr r r r tornado.webr r r$ tornado.utilr ImportError print_excr r r r r r r8 rD rI rb ro rt rx r r r r r r r r r r r r{ r r r r r r r TestCaser r skipIfr r r r r r- r9 r6 r' r% rR s
% 5 ( $ ; ' D T T 3
3
O+ O(& 2
C( C,&, &.
n
$- $ $. $ *
$ 4 $ P/ P
I- I&!/ !$ -
2 $) $ - &M') M'` $&: $
(#8
( @$0 @$ @$F @ 0 @ @ -
5
5
3 1 3
))
.2 . 2
T!#HI3 . 3 J3
3 2 3* 2 "L7 1 L7^((++ ("0* 00=. =]
I
" Hs 2
I I2 I/2I=<I=