o id @sdZddlZddlZddlZddlZdejd<ejdejejej e ddl m Z ddl mZmZmZddlmZddlmZdd lmZdd lmZejejd d ed ZddZedkroeedSdS)zj Standalone mock E2E test: seeds videos, runs worker, verifies results. Run: python -m tests.run_mock_e2e Ntrue MOCK_MODE) EnvConfig)MockDB VideoTask WorkerStats)R2Client)AIStudioProvider)OpenRouterProvider)Pipelinez1%(asctime)s [%(levelname)s] %(name)s: %(message)s)levelformatmock_e2ec s tddd}t}t|}tddd}tddd}t}tdddd td d dd td d dd g}||t dt |d| dddddiIdHt |||||d|d}d} | dIdH} | durin;t ddt d| jd| jdt d|| IdH} | r|d7}t d| jd| rdndq\t ddt dt dt d |d!t |t d"|jt d#|jt d$|jt d%|jt d&|jt d'|jt d(t |jt d)t |jtd*d+|jD} t d,| |jrNd-d.|jD} t| t | } t d/| d0t d1t|jd|t |ksaJd2t |d3|| shJd4|jdksrJd5t |jdks~Jd6t d7dS)8NTze2e-test-worker) mock_mode worker_idfake)api_keyrmock_telugu_001te)video_idlanguage segment_countmock_hindi_002himock_english_003enzSeeded z mock videosaistudiozmock-gputest)configdbr2primary_providerfallback_providerrstatsr z<============================================================zProcessing video: z (lang=)zVideo z: SUCCESSFAILEDz=== E2E TEST SUMMARY ===zVideos processed: /zTotal segments sent: zTotal segments completed: zTotal segments failed: z Total 429s: zTotal cache hits: zTotal batches: zDB results count: zDB flags count: css|]}|jdkVqdS)doneN)status).0vr/./home/ubuntu/transcripts/tests/run_mock_e2e.py Rszrun..zAll videos marked done: cSsg|]}|ddqS) quality_scorer)get)r-rr/r/r0 Wszrun..zAverage quality score: z.3fzSample result keys: z Expected z processed, got zNot all videos marked donezNo segments completedzNo results in DBz *** ALL E2E CHECKS PASSED ***)rrrr r rr seed_videosloggerinfolenregister_workerr claim_videorr process_video segments_sentsegments_completedsegments_failed segments_429 cache_hitsbatches_completedresultsflagsall video_queuesumlistkeys)rr r!primaryfallbackr$videospipeline processedtasksuccessall_donequality_scores avg_qualityr/r/r0runsp           &rT__main__)__doc__asynciologgingsysosenvironpathinsertdirnameabspath__file__ src.configrsrc.dbrrr src.r2_clientrsrc.providers.aistudior src.providers.openrouterr src.pipeliner basicConfigINFO getLoggerr7rT__name__r/r/r/r0s& &      L