o i @sdZddlZddlZddlZddlZddlZddlmZeddZeddZ edd Z d d Z d d Z ddZ ddZddZddZedkrZeeZeerUdnddSdS)z] Test script for ElevenLabs ConvAI Server Tests WebSocket connection, latency, STT, LLM, TTS N)datetime SERVER_URLzhttp://localhost:8080WS_URLzws://localhost:8080ELEVENLABS_API_KEYc shtdtdtdzt4IdHv}|td4IdHR}|IdH}td|jtd||dd kr[td  WdIdHWdIdHWd Std  WdIdHWdIdHWd S1IdHs{wYWdIdHWdS1IdHswYWdSty}ztd|WYd}~d Sd}~ww)zTest health check endpoint= ============================================================zTEST 1: Health Check<============================================================Nz/healthStatus: Response: statusoku✓ Health check PASSEDTu✗ Health check FAILEDFu✗ Health check FAILED: printaiohttp ClientSessiongetrjsonr ExceptionsessionresponsedataerM/var/www/html/vikas/2025-Nov-21-Voicebot/BackEnd/agent-starter-python/test.pytest_health_checks4 2 rc shtdtdtdz t4IdH}|td4IdH}|IdH}td|jtd|dtd |d td |d td |dd|di}|rtd|di}td|dddtd|dddtd|dddtd|ddd|di}tdtd|dddtd|dddtd|dddtd|dddtd WdIdHWdIdHWdS1IdHswYWdIdHWdS1IdHswYWdSty3}ztd |WYd}~d!Sd}~ww)"zTest metrics endpointrzTEST 2: Metrics EndpointrNz/metricsr zActive Sessions: active_sessionszTotal Sessions: total_sessionsz Total Turns: total_turnsz Avg Latency: average_end_to_end_latency_msmscost_breakdownz Cost Breakdown:totalz STT: $sttrz.6fz TTS: $ttsz LLM: $llmz Total: $per_turn_averagez Per Turn Average:u ✓ Metrics check PASSEDTu✗ Metrics check FAILED: Fr )rrrr!r"per_turnrrrr test_metrics+sL   4r(c s0tdtdtdzpt4IdHZ}|td4IdH6}|IdH}td|jtdtj|dd td  WdIdHWdIdHWd S1IdHs_wYWdIdHWdS1IdHsvwYWdSty}ztd |WYd}~d Sd}~ww)zTest LLM list endpointrzTEST 3: LLM ListrNz /llm/listr r indentu ✓ LLM list check PASSEDTu✗ LLM list check FAILED: F) rrrrrrr dumpsrrrrr test_llm_listRs*2r-c shtdtdtdggggd}zt4IdH}|tdIdH}d}td|t}|td |d IdHd }d }|2zp3dHW}|j tj j krt |j }|d d} t|d} td| d| ddd| vsd| vr|d| nd| vsd| vr|d| nd| vsd| vr|d| |d7}||krn qK|j tj jkrnqK6|IdHtd|D],\} } | rt| t| } td| d| ddt| dqtd| d qtd! WdIdHWd"S1IdHswYWdSty3}ztd#|WYd}~d$Sd}~ww)%z4Test WebSocket conversation with latency measurementrzTEST 4: WebSocket Conversationr)r#r%r$r"Nz/ws/zHello, how are you?z Sending: conversation_message)typemessager r/unknownz Received: z (z.0fzms) transcriptr#r%modelaudior$z Latency Results:z : z ms (avg of z responses)z : No datau" ✓ WebSocket conversation PASSEDTu#✗ WebSocket conversation FAILED: F)rrr ws_connectrtimesend_strrr,r/ WSMsgTypeTEXTloadsrrappendERRORcloseitemssumlenupperr) latenciesrws test_message start_timeresponse_count max_responsesmsgr response_typeelapsedkeyvaluesavgrrrrtest_websocket_conversationesp  (69rRc s@tdtdtdzxt4IdHb}dddd}|jtd |d 4IdH6}|IdH}td |jtd tj|d dtd WdIdHWdIdHWdS1IdHsgwYWdIdHWdS1IdHs~wYWdSty}ztd|WYd}~dSd}~ww)z#Test LLM usage calculation endpointrzTEST 5: LLM Usage CalculaterNr3id)character_counttts_character_countllm_token_countz/llm-usage/calculate)rr r r)r*u! ✓ LLM usage calculation PASSEDTu"✗ LLM usage calculation FAILED: F) rrrpostrrr r,r)r usage_datarrrrrrtest_llm_usage_calculates:2rYcs>tdtdtdtdttdttdtg}|dtIdHf|d tIdHf|d t IdHf|d t IdHf|d t IdHftdtd tdd}d}|D]\}}|rydnd}t|d||r|d7}qq|d7}qqtd|d|d|dkS)z Run all testsrz#ELEVENLABS CONVAI SERVER TEST SUITErz Server URL: zWebSocket URL: zTime: z Health CheckNMetricszLLM ListzWebSocket ConversationzLLM Usage Calculatez TEST SUMMARYru ✓ PASSEDu ✗ FAILEDr8r7z Total: z passed, z failed) rrrrnow isoformatr?rr(r-rRrY)resultspassedfailednameresultr rrr run_all_testss4    rb__main__r7)__doc__asynciorr:rosrgetenvrrrrr(r-rRrYrb__name__runsuccessexitrrrrs(    'L'