3in(<dZddlZddlZddlZddlZddlmZddlZejjde ee jjddl mZmZGddZGddZGd d ZGd d ZGd dZGddZy)z#Unit tests for generate_carousel.pyN)Path)load_slide_data load_env_keysceZdZdZdZdZy)TestLoadSlideDataz%Test slide data loading functionalitychtd}|Jd|vsJt|ddk(sJ|ddk(sJy)z2Test that set_a.json loads correctly with 5 slides상시ANslidessetName)rlen)selfdatas F/home/jay/workspace/tools/carousel-gen/tests/test_generate_carousel.pytest_load_slide_dataz&TestLoadSlideData.test_load_slide_datasOy)44>"a'''I)+++cztjtd5tddddy#1swYyxYw)z'Test that unknown set raises ValueErrorz Unknown set)match unknown_setN)pytestraises ValueErrorr)rs r test_load_slide_data_unknown_setz2TestLoadSlideData.test_load_slide_data_unknown_sets- ]]:] ; + M * + + +s 1:N)__name__ __module__ __qualname____doc__rrrrrrs/,+rrceZdZdZdZy)TestSlideDataAccuracyz0Test accuracy of numeric and text data in slidesctd}|d}td|D}d}|jdgD] }d|vsd}n|sJdd}|jdgD] }d |vsd}n|sJd d}|jdgD] }d |vsd}n|sJd d}|jdgD] }d |vsd}n|sJdd} |jdgD] }d|vsd} n| sJdy)z3Test that numeric values match original source datar r c32K|]}|ddk(s |yw) slideNumberNr).0ss r zATestSlideDataAccuracy.test_slide_data_accuracy..,sBQAm,<,AqBs F bulletPointsu 1,000만원Tu+신입 최대 1,000만원 should be presentz50%u3경력직 최대 직전연봉 50% should be presentu4회/년u'해외여행 4회/년 should be presentu30개u(30개 지원 패키지 should be present잔여수수료u,잔여수수료 should be present in slide 3N)rnextget) rrr slide_3 found_10mbullet found_50pct found_4times found_30packsfound_remaining_fees rtest_slide_data_accuracyz.TestSlideDataAccuracy.test_slide_data_accuracy&sny)hB&BB kk."5 F&  GGGy kk."5 F"  QQQ{ kk."5 FV##  FFF| kk."5 F& $  HHH}$kk."5 F F*&*# #R$RR"rN)rrrrr3rrrr r #s :.Srr ceZdZdZdZy)TestNoForbiddenTermsz&Test that forbidden terms are not usedcttd}tj|d}d|vsJdd|vsJdy) u3Test that '잔존수당' term is not used in slidesr F) ensure_asciiu 잔존수당u0'잔존수당' forbidden term should not be usedr)u('잔여수수료' should be used insteadN)rjsondumps)rrdata_strs rtest_no_forbidden_termsz,TestNoForbiddenTerms.test_no_forbidden_termsZsJy)::d7X-a/aa-!H,X.XX,rN)rrrrr;rrrr5r5Ws 0 Yrr5ceZdZdZdZdZy)TestSlideApproachesz!Test slide approach (B/C) mappingctd}|d}dddddd}|D]&}|d}|d}||}||k(rJd|d |d |y ) z/Test that slide approaches are correctly mappedr r CBr$r r#approachSlide z: expected approach z, got Nr)rrr expected_approachesslide slide_numactual_approachexpected_approachs rtest_slide_approachesz)TestSlideApproaches.test_slide_approachesksy)h   cEm,I#J/O 3I > "&77 c #78I7J&Q`Pab c7  crctd}|d}|D]0}|ddk(s d|vsJd|dd|dr$Jd|dd y ) z)Test that Approach C slides have bgPromptr r rEr?bgPromptrFr#z$ has approach C but missing bgPromptz bgPrompt is emptyNrG)rrr rIs rtest_approach_c_has_bg_promptz1TestSlideApproaches.test_approach_c_has_bg_promptsy)h FEZ C'!U*XU=122VWX*Z(FU=122DEF(  FrN)rrrrrMrPrrrr=r=hs+c, Frr=ceZdZdZdZdZy)TestLoadEnvKeysz5Test environment variable loading from .env.keys filectjdddd5}|jd|jd|jd|jd |jd |jd |jd |jd |j}ddd gd}|D]"}tj j |d$d}ddl}|jj}d}|tj jddk(sJdtj jddk(sJdtj jddk(sJdtj jddk(sJdtj jddk(sJd tj jd!d"k(sJd# t|jy#1swYnxYw#tjwxYw)$zATest that export prefix parsing is correct (using temporary file)w.keysFutf-8modesuffixdeleteencodingz# This is a comment z&export OPENAI_API_KEY=sk-test-key-123 zPLAIN_VAR=plain_value z!export QUOTED_VAR='quoted_value' z$DOUBLE_QUOTED="double_quoted_value" z SPACED_VAR = spaced_value z EMPTY_VAR=  N)OPENAI_API_KEY PLAIN_VAR QUOTED_VAR DOUBLE_QUOTED SPACED_VAR EMPTY_VARrct|}|jr|jdjD]}|j }|s|j dr(d|vs-|j dr|dd}|j d\}}}|j j dj d }tjj|j |yy) zLoad API keys from filerVr[#=export N"' rexists read_text splitlinesstrip startswith partitionosenviron setdefault file_pathenv_pathlinekey_values rload_env_keys_from_filezCTestLoadEnvKeys.test_load_env_keys..load_env_keys_from_files ???$ ( 2 2G 2 D O O Q F#zz|(<#y9'+ABx,0NN3,?MCE$)KKM$7$7$<$B$B3$GEJJ11#))+uE F%rr]zsk-test-key-123z export prefix should be strippedr^ plain_valuez$plain var without export should workr_ quoted_valuezsingle quotes should be removedr`double_quoted_valuezdouble quotes should be removedra spaced_valuez*leading/trailing spaces should be strippedrbzempty values should be allowed) tempfileNamedTemporaryFilewritenamerrrspopgenerate_carouselr__code__r+runlink) rtmptmp_path test_varsvaroriginal_load_env_keysroriginal_env_pathr|s rtest_load_env_keysz"TestLoadEnvKeys.test_load_env_keyss  ( (    II- . II? @ II/ 0 II: ; II? @ II9 : IIn % IIdOxxH 5 $I! * sD) *&* " $ 1 ? ? H H  F $H -::>>"237HH 32 3H::>>+.-? 76 7?::>>,/>A 21 2A::>>/26KK 21 2K::>>,/>A =< =A::>>+."4 10 14 N ! ! #K  J N ! ! #sBH8EH(H%(Ictjdddd5}|jd|jd|j}ddd tj j d dd }|tj jd d k(sJd  t|jy#1swYyxYw#tjwxYw) z'Test that comments are properly ignoredrTrUFrVrWz# Full line comment zVALID_VAR=valid_value N VALID_VARct|}|jr|jdjD]}|j }|s|j dr(d|vs-|j dr|dd}|j d\}}}|j j dj d}tjj|j |yy) NrVrdrerfrgrhrirjrkrus rr|zTTestLoadEnvKeys.test_load_env_keys_handles_comments..load_env_keys_from_files ???$ ( 2 2G 2 D O O QF#zz|(<#y9'+ABx,0NN3,?MCE$)KKM$7$7$<$B$B3$GEJJ11#))+uEF%r valid_valuezvalid var should be loaded) rrrrrrrsrr+rr)rrrr|s r#test_load_env_keys_handles_commentsz3TestLoadEnvKeys.test_load_env_keys_handles_commentss  ( (    II- . II/ 0xxH  $ JJNN; - F $H -::>>+.-? -, -? N ! ! #?  > N ! ! #s/CAC C  C(N)rrrrrrrrrrRrRs?H$T!$rrRceZdZdZdZdZy)TestSlideStructurez*Test overall structure and required fieldsctd}|d}gd}|D])}|D]"}||vrJd|jdd|+y)z)Test that all slides have required fieldsr r )r#rEtitleheadlinerFr#z missing required field: N)rr+)rrr required_fieldsrIfields r$test_all_slides_have_required_fieldsz7TestSlideStructure.test_all_slides_have_required_fieldssmy)hJ XE( X~XUYY}566OPUwWX~ X Xrcttd}|d}|Dcgc]}|d }}|gdk(s Jd|ycc}w)z2Test that slide numbers are sequential from 1 to 5r r r#rAz-Slide numbers should be [1, 2, 3, 4, 5], got NrG)rrr r& slide_numberss rtest_slide_numbers_sequentialz0TestSlideStructure.test_slide_numbers_sequentialsSy)h39:a=): :/ L;M? K L/;s 5N)rrrrrrrrrrrs4 XLrr)rr8rrsysrpathlibrrpathinsertstr__file__parentrrrrr r5r=rRrrrrrs)  3tH~,,3345<++$1S1ShYY"#F#FLn$n$bLLr