o [jD@sdZddlZddlmmZddlZddlZddl m Z ddl m Z ddl mZddlmZddlmZmZmZmZmZddlmZmZd d ZdS) zMDiagnostic test: trace active_sessions through Start New Chat -> Accept flow.N)status) get_database)create_session)get_or_create_conversation)active_sessionsadd_active_sessionremove_active_sessionnotify_visitor_socketupdate_active_session_status) AsyncMockpatchc Csdtjddd}|jd|ddd}|d }d d |i}|jd d |i|d }|d}|d|}tdt|}td|tvtd|rY|dndtd|ri|ddundtd|ryt |dndd} || u} | st d| fd|| fdt vst |rt |ndt | d} dd| i} tt | d} } |d} d}| |k} | st d| fd | |ft | t |d!}d"d#|i}tt |d} } }|d} d}| |u} | st d| fd$| |ft | t |d!}d"d#|i}tt |d} } }|jd%d|id}|j} tj}| |k}|st d|fd&| |fd't vsRt |rWt |nd't | dt vsit trnt tndt |d(}d)d*|i}tt |d} }}td+t|}td|r|dndtd|r|ddund|d} d,}| |k} | st d| fd | |ft | t |d!}d"d#|i}tt |d} } }|d} d}| |u} | st d| fd$| |ft | t |d!}d"d#|i}tt |d} } }|}td-||d.} d/}| |k} | sZt d| fd | |ft | t |d!}d"d#|i}tt |d} } }|d} d,}| |k} | st d| fd | |ft | t |d!}d"d#|i}tt |d} } }|jd0|d1d2d}|j} tj}| |k}|st d|fd&| |fd3t vst |rt |nd3t | dt vst trt tndt |d(}d)d*|i}tt |d} }}td4t|}td|tvtd|r|dndtd|r)|ddundd} || u} | snt d| fd|| fdt vsNt |rSt |ndt | d} t d5d6d| i} tt | d} } |d} d7}| |k} | st d| fd | |ft | t |d!}d"d#|i}tt |d} } }|d} d}| |u} | st d| fd$| |ft | t |d!}t d8d9d#|i}tt |d} } }|}td-||d.} d/}| |k} | st d| fd | |ft | t |d!}d"d#|i}tt |d} } }|d} d7}| |k} | sTt d| fd | |ft | t |d!}d"d#|i}tt |d} } }|d:|d;}td<td=|jtd>||j} tj}| |k}|st d|fd&| |fd?t vst |rt |nd?t | dt vst trt tndt |d(}d)d*|i}tt |d} }}|}|d@}t|}td|tvtd|r|dndtd|r|ddundtd|rt |dndd} || u} | s[t d| fd|| fdt vs;t |r@t |ndt | d} t dAd6d| i} tt | d} } |d} d}| |k} | st d| fd | |ft | t |d!}d"d#|i}tt |d} } }|d}tdB|d} || u} | st d| fd|| fdCt vst |rt |ndCt | d} t dDd6d| i} tt | d} } |}tdE||d.} d/}| |k} | s t d| fd | |ft | t |d!}d"d#|i}tt |d} } }|d} d}| |k} | sVt d| fd | |ft | t |d!}d"d#|i}tt |d} } }tdFt|}tdG|ro|dndtdH|r|ddund|jd%d|id}tdI|jtdJ||jtjkrGt|}tdK|r|dndtdL|r|ddund|}td-||d.} d/}| |k} | st d| fd | |ft | t |d!}d"d#|i}tt |d} } }|d} d,}| |k} | s ws connect -> accept -> end -> start-new-chat -> accept again. Traces whether the websocket reference is valid at every step. zdiag-Nz.comz/api/v1/tenant/registerzDiagnostic Test Store)domainname)jsonsiteIdOriginzhttp://z/api/v1/session/init)rheaders sessionIdz /ws/session/z [DIAG-1] After WS connect:z in active_sessions: z status: rzN/Az websocket is None: websocketz websocket type: )is not)z%(py0)s is not %(py3)smem)py0py3zassert %(py5)spy5RINGING)==)z%(py1)s == %(py4)s)py1py4zassert %(py6)spy6)z%(py1)s is not %(py4)sz/api/v1/handoff/accept)zT%(py2)s {%(py2)s = %(py0)s.status_code } == %(py6)s {%(py6)s = %(py4)s.HTTP_200_OK }accept1)rpy2rrzassert %(py8)spy8z [DIAG-2] After first accept:IN_CHATz visitor received event: typestatus_changedz/api/v1/handoff/endagent)rendedByend_respz [DIAG-3] After end chat:z7active_sessions entry should still exist after end chatz >assert %(py5)sENDEDz,websocket should be preserved after end chatz >assert %(py6)sz/api/v1/session/z/new-conversationz [DIAG-4] After Start New Chat:z response status: z response body: new_conv_respconversationIdz9active_sessions entry should exist after new-conversationz websocket ref: ws_after_new_convz3CRITICAL: websocket is None after new-conversation!z" visitor received RINGING event: z- [DIAG-5] About to accept new conversation...z pre-accept status: z pre-accept websocket is None: z accept response status: z accept response body: z post-accept status: z! post-accept websocket is None: zP [DIAG-RESULT] SUCCESS - visitor received IN_CHAT after new-conversation accept!z7 [DIAG-RESULT] FAILURE - accept returned 409 CONFLICT: z- This means claim_session_atomically failed!Fz5Accept returned 409 - claim_session_atomically failedz >assert %(py1)srz, [DIAG-RESULT] UNEXPECTED - accept returned z: zUnexpected accept status: )uuiduuid4hexpostrwebsocket_connectprintrgetr$ @pytest_ar_call_reprcompare @py_builtinslocals_should_repr_global_name _safereprAssertionError_format_explanation status_coder HTTP_200_OK receive_json_format_assertmsgHTTP_409_CONFLICT) clientrreg_respsite_idr init_resp session_idwsr @py_assert2 @py_assert1 @py_format4 @py_format6 @py_assert0 @py_assert3 @py_format5 @py_format7r @py_assert5 @py_format9ev1r(ev2r* new_conv_data new_conv_idr,ev3mem_before_accept2accept2ev4 @py_format2rZ9/web/html/dev/chatbot/backend/tests/test_diag_new_conv.py,test_new_conversation_accept_websocket_traces     xjl "llll "lvll ""l ll " "ll :B\`$r\)__doc__builtinsr6_pytest.assertion.rewrite assertionrewriter4r-pytestfastapir app.databaser#app.repositories.session_repositoryr(app.repositories.conversation_repositoryrapp.services.presencerrrr r unittest.mockr r r\rZrZrZr[s"