ZiddlZddlZddlZddlmZmZddlmZddlm Z ddl m Z m Z  Gddej ZedkrejdSdS) N)Mockpatch)recv)SSLWantReadError)WebSocketTimeoutException"WebSocketConnectionClosedExceptionc0eZdZdZdZdZdZdZdZdS)SocketBugsTestz(Test bugs found in socket handling logicct}t|j_d|j_t d5}t}||_g|j_|t5}t|ddddn #1swxYwY| dt|j ddddS#1swxYwYdS)a BUG #5 FIX VERIFICATION: Test SSLWantReadError timeout now raises correct exception Bug was in _socket.py:100-101 - SSLWantReadError except block returned None implicitly Fixed: Now properly handles timeout with WebSocketTimeoutException ?selectors.DefaultSelectordN%Connection timed out waiting for data rrr side_effect gettimeout return_valuerselect assertRaisesrassertInstr exceptionself mock_sockmock_selector_class mock_selectorcms [/root/projects/butler/venv/lib/python3.11/site-packages/websocket/tests/test_socket_bugs.py6test_bug_implicit_none_return_from_ssl_want_read_fixedzESocketBugsTest.test_bug_implicit_none_return_from_ssl_want_read_fixed$aFF %5%7%7 ",/ ) . / / V3F FFM/<  ,02M -""#<== %Y$$$ % % % % % % % % % % % % % % % MMA3r|CTCT U U U V V V V V V V V V V V V V V V V V V6B CB C"B #+CC"Cct}ttjd}||j_d|j_td5}t}||_g|j _| t5}t |ddddn #1swxYwY| dt|jddddS#1swxYwYdS)a BUG #5 FIX VERIFICATION: Test that socket.error with EAGAIN now handles timeout correctly Bug was in _socket.py:102-105 - socket.error except block returned None implicitly Fixed: Now properly handles timeout with WebSocketTimeoutException z Resource temporarily unavailabler r rNr)rOSErrorerrnoEAGAINrrrrrrrrrrr)rr eagain_errorrrrs r5test_bug_implicit_none_return_from_socket_error_fixedzDSocketBugsTest.test_bug_implicit_none_return_from_socket_error_fixed:sqFF u|-OPP &2 ",/ ) . / / V3F FFM/<  ,02M -""#<== %Y$$$ % % % % % % % % % % % % % % % MMA3r|CTCT U U U V V V V V V V V V V V V V V V V V Vs6r __main__)r%r.unittest unittest.mockrrwebsocket._socketrwebsocket._ssl_compatrwebsocket._exceptionsrrTestCaser r9mainr=r>rrGs %%%%%%%%""""""222222 &{1{1{1{1{1X&{1{1{1| zHMOOOOOr>