Bj ddlZddlZddlZddlmZmZddlmZddlm Z ddl m Z m Z  GddejZedk(rej yy) N)Mockpatch)recv)SSLWantReadError)WebSocketTimeoutException"WebSocketConnectionClosedExceptionc.eZdZdZdZdZdZdZdZy)SocketBugsTestz(Test bugs found in socket handling logicct}t|j_d|j_t d5}t}||_g|j_|jt5}t|dddd|jdtjdddy#1swY7xYw#1swYyxYw)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/workspace/ytshorts/venv/lib/python3.12/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%F %5%7 ",/ ) . / V3F FM/<  ,02M -""#<= %Y$ % MMA3r||CT U V V  % % V V$8C8 B;-C;C CCct}ttjd}||j_d|j _td5}t}||_g|j_|jt5}t |dddd|jdtjdddy#1swY7xYw#1swYyxYw)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;sF u||-OP &2 ",/ ) . / V3F FM/<  ,02M -""#<= %Y$ % MMA3r||CT U V V  % % V Vs$8C C -C C CC"ct}t|j_d|j_t d5}t}||_g|j_|jt5}t|dddd|jdtjdddy#1swY7xYw#1swYyxYw)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 rNrrrs r3test_bug_wrong_exception_for_selector_timeout_fixedzBSocketBugsTest.test_bug_wrong_exception_for_selector_timeout_fixedVr!r"ct}d|j_tjd|j _|jt5}t |ddddddl }|jdk\r&|jdtjy|jdtjy#1swYhxYw)zJ Test that socket.timeout exceptions are properly handled r zOperation timed outrNr) zConnection timed out)rrrsockettimeoutrrrrsys version_inforrr)rrrr0s r&test_socket_timeout_exception_handlingz5SocketBugsTest.test_socket_timeout_exception_handlingqsF ,/ )&,^^4I%J "  8 9 !R C  !    w & MM0#bll2C D MM/R\\1B C ! !s  C  Cct}tdg|j_d|j_t d5}t}||_dg|j_t|d}|j|d|jj|jjdddy#1swYyxYw)zCTest the correct behavior when SSLWantReadError is properly handledsdata after retryr r TrN) rrrrrrrr assertEqualregister assert_called)rrrrresults r)test_correct_ssl_want_read_retry_behaviorz8SocketBugsTest.test_correct_ssl_want_read_retry_behaviorsF '7&8:M%N ",/ ) . / 13F FM/<  ,15M -)S)F   V%8 9  " " 0 0 2  . . 0 1 1 1s A6CC N) __name__ __module__ __qualname____doc__r r(r*r2r8rr r "s"2V,V6V6D21r>r __main__)r%r.unittest unittest.mockrrwebsocket._socketrwebsocket._ssl_compatrwebsocket._exceptionsrrTestCaser r9mainr=r>rrGsP %"2 ({1X&&{1| zHMMOr>