z8SocketTest.test_default_socket_option...s$ -0CFf(( ( s "N)assertIsInstancer list assertGreaterlenany assertTrue)selftcp_nodelay_founds rtest_default_socket_optionz%SocketTest.test_default_socket_option(sR 3T: 345q9  4I   )*ct}d|j_t|d}|j|d|jj dy)zTest normal recv operation test data N)rr return_value assertEqualassert_called_once_withr! mock_sockresults rtest_recv_normalzSocketTest.test_recv_normal3sCF &2 #i# ...q1r$ct}td|j_|j t 5}t|dddd|j tjdy#1swY/xYw)zTest recv with TimeoutErrorConnection timed outr'N) r TimeoutErrorr side_effect assertRaisesrr)str exceptionr!r,cms rtest_recv_timeout_errorz"SocketTest.test_recv_timeout_error=sgF %12H%I "   8 9 R A   R\\*,BC  s A55A>ct}tjd}d|_||j_d|j _|jt5}t |ddddddl }|jdk\r&|jtjdy|jtjdy#1swYhxYw) zTest recv with socket.timeoutSocket timed out)r:>@r'Nr) r0)rrtimeoutargsrr2 gettimeoutr(r3rsys version_infor)r4r5)r!r, timeout_excr7rAs rtest_recv_socket_timeoutz#SocketTest.test_recv_socket_timeoutGsF nn%78 0 %0 ",0 )   8 9 R A      w &   S.0F G   S.0B C  s  CCct}td}d|_||j_|j t 5}t|dddd|jtjdy#1swY/xYw)z Test recv with SSL timeout errorThe operation timed out)rFr'N) rr r?rr2r3rr)r4r5)r!r,ssl_excr7s rtest_recv_ssl_timeoutz SocketTest.test_recv_ssl_timeout]ssF 453 %, "   8 9 R A   R\\*,EF  s  A>>Bct}td}d|_||j_|j t5t|ddddy#1swYyxYw)z$Test recv with SSL non-timeout errorSSL certificate error)rJr'N)rr r?rr2r3)r!r,rGs rtest_recv_ssl_non_timeout_errorz*SocketTest.test_recv_ssl_non_timeout_errorisVF 231 %, "  x (  A    s  AA"ct}d|j_|jt5}t|dddd|j t jdy#1swY/xYw)z1Test recv with empty response (connection closed)r$r'Nz#Connection to remote host was lost.)rrr(r3rr)r4r5r6s rtest_recv_empty_responsez#SocketTest.test_recv_empty_responsetsaF &) #   A B b A   R\\*,QR  s A,,A5cxt}tdg|j_t d5}t}||_dg|j _t|d}|j|d|jj|jjdddy#1swYyxYw)z.Test recv with SSLWantReadError (should retry)sdata after retryselectors.DefaultSelectorTdN) rr rr2rr(selectr)register assert_calledcloser!r,mock_selector_class mock_selectorr-s rtest_recv_ssl_want_read_errorz(SocketTest.test_recv_ssl_want_read_error~sF '7&8:M%N " . / 03F FM/<  ,15M -)S)F   V%8 9  " " 0 0 2    - - / 0 0 0s A6B00B9c\t}t|j_d|j_t d5}t}||_g|j_|jt5t|dddddddy#1swYxYw#1swYyxYw)z.Test recv with SSLWantReadError that times outg?rOrPN) rr rr2r@r(rrQr3r)r!r,rVrWs rtest_recv_ssl_want_read_timeoutz*SocketTest.test_recv_ssl_want_read_timeoutsF %5%7 ",/ ) . / %3F FM/<  ,02M -""#<= %Y$ %  % %  % %  % %s$8B"8 BB"B B""B+ct}gd}td|5}t|}|j|d|j|jddddy#1swYyxYw)zTest recv_line functionality)Helr^o zwebsocket._socket.recv)r2sHello N)rrrr) call_count)r!r, recv_calls mock_recvr-s rtest_recv_linezSocketTest.test_recv_linesbF ; + D 6 y)F   VZ 0   Y111 5  6 6 6s :AA(ct}d|j_d|j_t|d}|j |d|jj dy)zTest normal send operationr'r;r&Nrrr(r@r)assert_called_withr+s rtest_send_normalzSocketTest.test_send_normalsQF &' #,0 )i. #)),7r$ct}d|j_d|j_t|d}|j |d|jj dy)z*Test send with zero timeout (non-blocking)r'rr&N)rrr(r@r)r*r+s rtest_send_zero_timeoutz!SocketTest.test_send_zero_timeoutsQF &' #,- )i. #..| !,, !,,.HI %0 "   v|| , * L ) * * *s B##B,ct}d|j_tjd}d|_||j _|jtj5t |ddddy#1swYyxYw)z-Test send with socket error that has no errnor;Generic socket error)rr&N) rr@r(rrsr?rr2r3)r!r,no_errno_errors rtest_send_socket_error_no_errnoz*SocketTest.test_send_socket_error_no_errnosmF ,0 ) &<=7%3 "   v|| , * L ) * * *s ( A>>Bct}d|j_tjd}t j |_t j df|_||j_ td5}t}||_g|j_t|d}|j|ddddy#1swYyxYw)z!Test send write operation timeoutr;rqrOr&rNrrrvs rtest_send_write_timeoutz"SocketTest.test_send_write_timeout(sF ,0 )||$FG "\\ "\\+MN %1 " . / (3F FM/<  ,02M -)\2F   VQ ' ( ( (s 9ACC ct}d|j_d|j_t|d}|j |d|jj dy)z.Test send with string data (should be encoded)r'r;z test datar&Nrgr+s rtest_send_string_dataz SocketTest.test_send_string_data>sQF &' #,0 )i- #)),7r$c@t}d|j_tjd}t j |_t j df|_|dg|j_ td5}t}||_dg|j_t|d}|j|d|jj|jj|jjdddy#1swYyxYw)zTest send retry mechanismr;rqr'rOTr&N)rr@r(rrsrtrur?rr2rrQr)rRrSrTrvs rtest_send_partial_send_retryz'SocketTest.test_send_partial_send_retryIsF ,0 )||$FG "\\ "\\+MN '3A%6 " . / 03F FM/<  ,15M -)\2F   VQ '  " " 0 0 2  . . 0    - - / 0 0 0s ;BDDN)__name__ __module__ __qualname__r#r.r8rDrHrKrMrXrZrerirkrmrorxr}rrrrrr$rrr'sp +2DD, G S0$ % 6 8 = I0&(,(, * *(, 80r$r__main__)rtrunittest unittest.mockrrrtimewebsocket._socketrrrr websocket._ssl_compatr r r r websocket._exceptionsrrTestCaserrmainrr$rrs` 00 JJ  &z0""z0z  zHMMOr$