j& dZddlZddlZddlZddlZddlZddlZddlZddlm Z e e  j Z e j Zeeejvr$ejdeeee ejvr$ejdee ddlmZmZmZmZmZmZmZmZmZmZmZmZmZm Z Gddej!Z"e#dkrej$dSdS)z Tier 4: Real-World Application Workloads Test Suite Contains >= 7 realistic, end-to-end workload test cases (Total: 10 test cases). N)Path) FILMMAKER_DIR PROMPTS_DIR ASSETS_DIR BIBLE_DIR PIPELINE_DIR EXPORTS_DIRWEB_REVIEW_DIRread_ep01_screenplayget_screenplay_scenesfind_male_crying_violationsload_muse_prompts_dataload_banana_prompts_datacreate_synthetic_wavcreate_synthetic_video_mp4cNeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d S) TestTier4Workloadsz^ Tier 4 tests execute realistic end-to-end application workloads across the pipeline. ct}|t|ddtjd|}|t|ddt |}|t|ddtjd|tj}|t|d d d S) z\Workload 1: Parse entire EP01 screenplay, extracting metadata, scene breakdown, and metrics.iz0EP01 screenplay must contain substantive contentz\d+\.\s*\*\*([^*]+)\*\*z$EP01 must list main character roster z$EP01 must contain at least 12 scenesu HỒI\s*(\d+)z Screenplay must delineate 3 actsN)r assertGreaterlenrefindallassertGreaterEqualr IGNORECASE)self screenplay char_rosterscenesactss 3C:\Projects\KieuStory\tests\test_tier4_workloads.py7test_workload_01_full_ep01_screenplay_parse_and_metricszJTestTier4Workloads.test_workload_01_full_ep01_screenplay_parse_and_metrics$s)++  3z??D2deeej!;ZHH  K 0 0!5[\\\'z22 F R1WXXXz*J FF D 1.PQQQQQcLt}t}||td||td|di}|t |ddt|ddD]k\}}| t|tot |dkt|tr| d|ldS) zKWorkload 2: Validate JSON registries for Gemini Banana and Muse.ai prompts.z4gemini_banana_prompts.json must be a JSON dictionaryz4muse_ai_video_prompts.json must be a JSON dictionarymotion_promptsrz)Motion prompts registry must not be emptyN motion_prompt) rrassertIsInstancedictgetrrlistitems assertTrue isinstancestrassertIn)r banana_data muse_datar'shot_id shot_infos r#Gtest_workload_02_prompt_registries_schema_and_cross_reference_integrityzZTestTier4Workloads.test_workload_02_prompt_registries_schema_and_cross_reference_integrity5s.00 *,,  k41ghhh i/efff"'7<< 3~..3^___#'~';';'='=">">ss"C : : GY OOJw44IW9I J J J)T** : oy999 : :r%c ddlm}m}|}tj5}t |}t |dz }t |dz }t|d}||d|} | ||d } || d |tj |d |d |d ddddg} tj| dd} || jd| j} |d| ddddS#1swxYwYdS)z\Workload 3: Generate synthetic audio and execute EBU R128 loudness normalization via FFmpeg.r)AudioContinuityEngine get_ffmpegztest_input.mp4ztest_normalized.mp4g@ duration_seczMust generate synthetic videog,) target_lufsz.normalize_loudness must succeed on valid videozOutput video must existz-iz-filter_complexzebur128=peak=truez-fnull-T)capture_outputtextzIntegrated loudness:N)audio_continuity_enginer9r:tempfileTemporaryDirectoryrr1rr/normalize_loudnessospathexists subprocessrun assertEqual returncodestderrr2)rr9r:ffmpegtmpdirtmp_path input_video output_videocreatedenginesuccess measure_cmdresrMs r#8test_workload_03_synthetic_audio_mastering_two_pass_lufszKTestTier4Workloads.test_workload_03_synthetic_audio_mastering_two_pass_lufsFsMMMMMMMM  ( * * :fF||Hh)99::Kx*??@@L13OOOG OOG%D E E E+*,,F// \W\/]]G OOG%U V V V OOBGNN<88:S T T Tl!#6fcK .TMMMC   S^Q / / /ZF MM0& 9 9 95 : : : : : : : : : : : : : : : : : :sDEEEcvddlm}tj5}t |}t |dz }t |dz }t |dz }t |dt |d|}|||g|d }||d |tj || tj |d d d d d S#1swxYwYd S) zVWorkload 4: Assemble multiple synthetic video clips using equal-power audio crossfade.r)r9z shot01.mp4z shot02.mp4zscene_crossfaded.mp4g@r;g?) crossfade_durz(stitch_with_audio_crossfade must succeediN)rBr9rCrDrr1rstitch_with_audio_crossfader/rFrGrHrgetsize) rr9rOrPv1v2 out_masterrTrUs r#>test_workload_04_synthetic_multi_shot_video_crossfade_assemblyzQTestTier4Workloads.test_workload_04_synthetic_multi_shot_video_crossfade_assemblygsAAAAAA  ( * * BfF||HX ,--BX ,--BX(>>??J 'r < < < < &r < < < <**,,F88"b:]`8aaG OOG%O P P P OOBGNN:66 7 7 7   rwz::D A A A B B B B B B B B B B B B B B B B B BsDD..D25D2cTddlm}ddlm}||}|d}||jd|d}||jd|}|d||d||d }||jd|d }||jd|d } || jd|d } || jd|d } || jddS)zKWorkload 5: Comprehensive HTTP walk across all Web Review Studio endpoints.r)app) TestClient/z /api/librarysummary raw_scenesz /api/videosz/api/charactersz /api/promptsz /api/episodesz /api/statusN) serverrbstarlette.testclientrcr,rK status_codejsonr2) rrbrcclientr_indexr_liblib_datar_vidsr_chars r_promptsr_epsr_statuss r#8test_workload_05_web_review_studio_api_full_catalog_walkzKTestTier4Workloads.test_workload_05_web_review_studio_api_full_catalog_walk{s333333C**S// ,c222 >** *C000::<< i*** lH---M** +S111**.// ,c222JJ~..  .444 ?++ *C000::m,, -s33333r%ct}t|}t|dkrIdd|D}|dt|d|dSdS)z[Workload 6: Exhaustive audit of all male dialogue lines and parentheticals across Acts 1-3.r c Hg|]}d|dd|dd|d S)zLine line_numz: characterz -> rA).0vs r# zfTestTier4Workloads.test_workload_06_complete_anti_male_tears_audit_across_all_acts..s>jjjZ[UQz]UUa nUU!F)UUjjjr%zAnti-Male-Tears audit failed (z violations): N)r r rjoinfail)rr violationsmsgs r#?test_workload_06_complete_anti_male_tears_audit_across_all_actszRTestTier4Workloads.test_workload_06_complete_anti_male_tears_audit_across_all_actss)++ 0<< z??Q  ))jj_ijjjkkC II\s:\\WZ\\ ] ] ] ] ]  r%c t}t|}d|D}|t|dkd|d}|t t jd|t jd|t t jd|t jd|t t jd|t jd |t t jd |t jd d S) z\Workload 7: Verify dramatic progression of Scene 05 from stroll to grave discovery to grief.c(g|]\}}d|vsd|v |S)u CẢNH 05uCẢNH 5r{)r|hcs r#r~zgTestTier4Workloads.test_workload_07_full_scene05_sub_beat_dramatic_arc_verification..s+PPP$!QK1,<,< aAr%rzScene 05 must existutiểu khê|suốizMust feature pastoral strollusè sè|nấm mồzMust feature discovery of mounduđau đớn thay|bạc mệnhzMust feature mourning for fateudấu hài|dấu giày|in rêuzMust feature footprints in mossN)r r r/rboolrsearchr)rrr!sc5_listsc5s r#@test_workload_07_full_scene05_sub_beat_dramatic_arc_verificationzSTestTier4Workloads.test_workload_07_full_scene05_sub_beat_dramatic_arc_verifications2)++ &z22PP&PPP H )+@AAAqk RY'dS)zZWorkload 8: Simulate orchestrator batch workflow (list shots, resolve frames, continuity).r)get_shots_for_sceneresolve_start_frame ep01_scene01z&Scene 01 must contain registered shotsN)production_orchestratorrrrrr/r0r1)rrrshotsr5 shot_data start_frames r#?test_workload_08_production_orchestrator_scene_batch_simulationzRTestTier4Workloads.test_workload_08_production_orchestrator_scene_batch_simulationsTTTTTTTT##N33 3u::q*RSSS"' Q Q GY--gyAAK OOK4/O:k33O3O P P P P Q Qr%cDtdz }|||d}|d||d||d||d|dS) zTWorkload 9: Verify feature assembly pipeline script and FFmpeg command construction.zassemble_ep01_feature.pyzutf-8)encodingfilter_complexz concat=n=libx264aacN)rr/rH read_textr2)r script_pathcodes r#;test_workload_09_master_feature_grand_assembly_verificationzNTestTier4Workloads.test_workload_09_master_feature_grand_assembly_verifications"%??   **,,---$$g$66 &--- k4((( i&&& eT"""""r%c >tdz tdz tdz tdz tdz tdz tdz tdz td z g }|D]F}||o|d |jGd S) zKWorkload 10: Health audit of core project directories, files, and licenses.00_Project_Bible01_Scripts_And_Episodes 02_AI_Prompts 04_Assets05_Production_Pipeline 06_Exports FilmMaker web_reviewlicensezCore directory missing: N)BASE_DIRr/rHis_dirname)r core_dirsds r#0test_workload_10_end_to_end_project_health_auditzCTestTier4Workloads.test_workload_10_end_to_end_project_health_audits ) ) 0 0  & { " / / | # { " | # y    \ \A OOAHHJJ5188::7ZRSRX7Z7Z [ [ [ [ \ \r%N)__name__ __module__ __qualname____doc__r$r7rXr`rurrrrrr{r%r#rrsRRR":::":::BBBB($4$4$4L^^^ C C C Q Q Q###\\\\\r%r__main__)%rrFsysrrkrCrIunittestpathlibr__file__resolveparent TESTS_DIRrr1rGinserttestsrrrrrr r r r r rrrrTestCaserrmainr{r%r#rs   DNN " " $ $ +  3x==  HOOAss8}}%%%3y>>!!HOOAss9~~&&&{\{\{\{\{\*{\{\{\| zHMOOOOOr%