a j@svddlZddlZddlZddlmZmZddlmZddlm Z ddl m Z m Z Gdddej ZedkrredS) N)Mockpatch)recv)SSLWantReadError)WebSocketTimeoutException"WebSocketConnectionClosedExceptionc@s8eZdZdZddZddZddZdd Zd d Zd S) SocketBugsTestz(Test bugs found in socket handling logicc Cst}t|j_d|j_tdj}t}||_g|j_|t }t|dWdn1s`0Y| dt |j Wdn1s0YdS)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_selectorcmrFE:\soccer\ml_env\Lib\site-packages\websocket/tests/test_socket_bugs.py6test_bug_implicit_none_return_from_ssl_want_read_fixed$s   (zESocketBugsTest.test_bug_implicit_none_return_from_ssl_want_read_fixedc Cst}ttjd}||j_d|j_tdj}t}||_g|j _| t }t|dWdn1sj0Y| dt |jWdn1s0YdS)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 r Nr )rOSErrorerrnoEAGAINrrrrrrrrrrr)rrZ eagain_errorrrrrrr5test_bug_implicit_none_return_from_socket_error_fixed:s   (zDSocketBugsTest.test_bug_implicit_none_return_from_socket_error_fixedc Cst}t|j_d|j_tdj}t}||_g|j_|t }t|dWdn1s`0Y| dt |j Wdn1s0YdS)z BUG #6 FIX VERIFICATION: Test that selector timeout now raises correct exception type Bug was in _socket.py:115 returning None for timeout, treated as connection error Fixed: Now raises WebSocketTimeoutException directly r r r Nr r rrrr3test_bug_wrong_exception_for_selector_timeout_fixedUs   (zBSocketBugsTest.test_bug_wrong_exception_for_selector_timeout_fixedcCst}d|j_td|j_|t}t|dWdn1sF0Yddl }|j dkrv| dt |j n| dt |j dS)zJ Test that socket.timeout exceptions are properly handled r zOperation timed outr Nr) zConnection timed out)rrrsockettimeoutrrrrsys version_inforrr)rrrr(rrr&test_socket_timeout_exception_handlingps ( z5SocketBugsTest.test_socket_timeout_exception_handlingcCst}tdg|j_d|j_tdP}t}||_dg|j_t|d}||d|j |j Wdn1sz0YdS)zCTest the correct behavior when SSLWantReadError is properly handledsdata after retryr r Tr N) rrrrrrrrZ assertEqualregisterZ assert_called)rrrrresultrrr)test_correct_ssl_want_read_retry_behaviors     z8SocketBugsTest.test_correct_ssl_want_read_retry_behaviorN) __name__ __module__ __qualname____doc__rr"r#r*r-rrrrr!s r__main__)r r&ZunittestZ unittest.mockrrZwebsocket._socketrZwebsocket._ssl_compatrZwebsocket._exceptionsrrZTestCaserr.mainrrrrs  ~