P1iddlmZddlmZddlmZddlmZddlm Z m Z ddl Z ddl m Z ddlmZmZmZmZmZmZGd d eZy) ) annotations)TestCase)get_user_model)reverse)datetimeN)Person) REL_LIUHE REL_BANSANHE REL_TIANGANHE REL_CHONGREL_HAIREL_XINGcveZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZy)APIResponseStructureTestszJTest API response structures for proper t=r/s mapping and index alignment.c ht}|jjddd|_|jj |jt jjddtdddtd d |jd |_ |jj|jj|jjjd i}|r0td|jdd|jdytdy)N 13900000000xztest@example.com)phonepasswordemailOwnerM rT)namegender birth_date birth_time created_byownerdayzOwner day pillar: God=god_idxz, Earth= earth_idxz1Warning: Owner day pillar not calculated properly)robjects create_useruserclient force_loginr createrrr"calculate_bazisave bazi_resultgetprint)selfUser day_pillars G/home/cursorai/projects/iching/api/tests/test_api_response_structure.pysetUpzAPIResponseStructureTests.setUpsLL,,#5G-    *^^**dAq1AB{tyy+  !!# ZZ++//r:  *:>>)+D*EXjnn]hNiMjk l E Fctd}|jj|}|j|jd|j }d|vrE|j |dgd|ddk(r%|j d||j d|yy|j d||j d|y) z4Test that the with-relations endpoint is accessible.api:bazi-with-relationsstatus) processingreadyerrorr<resultscountN)rr)r/ assertEqual status_codejsonassertInr1urlrespdatas r4#test_with_relations_endpoint_existsz=APIResponseStructureTests.test_with_relations_endpoint_exists)s/0{{s# ))3/yy{ t  MM$x.*J KH~( i. gt,) MM)T * MM'4 (r6c tjjddtdddt dd|j }|j |jddlm }|d |j jd td }|jj|}|j|jd |j!}|j#d||j#d||ddkDrbd}|dD]}|d|jk(s|}n|r#|j#d||d}|j%|t&|D cgc]} | jddk(s| } } t)| dkDr| D]} |j| ddd|j#d| d|j%| dt*d|j#| dt,t.t0t2t4t6gd| d|j9d| d|j#d| dyt;d|jyt;d|jd yt;d!ycc} w)"zBTest that relation type 'r' uses 'c' field for relationship codes. TestPerson1Fr rrrrr r! call_commandrecalc_bazi_relationsTr(forcer8r9r>r?Nid relation_goodtrzRelation type must be 'r'cz!Relation reason missing 'c' fieldzRelation code must be integerzInvalid relation code: iz)Relation reason should not have 'i' fieldby"Relation reason missing 'by' fieldz*No relation type reasons found for person Person $ not found in with-relations results+No results found in with-relations endpoint)r r&r+rrr(r,r-django.core.managementrPrTrr)r/r@rArBrCassertIsInstancelistlenintr r r r rr assertNotInr0) r1personrPrErFrG person_dataresultrUrWrelation_reasonsreasons r4!test_relation_type_r_uses_c_fieldz;APIResponseStructureTests.test_relation_type_r_uses_c_field<sM&&stD!R7HB{tyy'    8,499<vc{m%LN ((f6ab dF4XY##3(FvyykRS {*NOP ? @;$Ss I68I6c tjjddtdddt dd|j }|j |jddlm }|d |j jd td }|jj|}|j|jd |j!}|j#d||j#d||ddkDrd}|dD]}|d|jk(s|}n|r|d}|D cgc]} | jddk(s| } } t%| dkDrq| D]k} |j| ddd|j#d| d|j'| dt(d|j+d| d|j#d| dmyt-d|jyt-d|jdyycc} w) z>Test that sha god type 's' uses 'i' field for sha god indices. TestPerson2rrLr rrNrOrQTrRr8r9r>r?NrTrUrVszSha god type must be 's'rYz Sha god reason missing 'i' fieldzSha god index must be integerrXz(Sha god reason should not have 'c' fieldrZz!Sha god reason missing 'by' fieldz)No sha god type reasons found for person r\r])r r&r+rrr(r,r-r_rPrTrr)r/r@rArBrCrbr`rcrdr0) r1rerPrErFrGrfrgrUrWsha_god_reasonsris r4 test_sha_god_type_s_uses_i_fieldz:APIResponseStructureTests.test_sha_god_type_s_uses_i_fields &&stD!R7HB{tyy'    8,499<r?NrTrUrVrWrX5eu#天干合 reason missing '5e' fieldzFive element must be integerzInvalid five element: z Five element should be 0-4, got z&No tianganhe reasons found for person r\r]r^)r r&r+rrr(r,r-r_rPrTrr)r/r@rArBrCr rbr`rcranger0) r1rerPrErFrGrfrgrWtianganhe_reasonsris r4test_tianganhe_has_5e_fieldz5APIResponseStructureTests.test_tianganhe_has_5e_fields"&&stD!R7HB{tyy'    8,499<ef fTlE!H@`aghlam`n>op#4B699+NO {*NOP ? @/%Ys35H6c tjjddtdddt dd|j }|j |jddlm }|d |j jd td }|jj|}|j|jd |j!}|j#d||j#d||ddkDrd}|dD]}|d|jk(s|}n|r|dDcgc]0}|jddk(r|jdt$k7r|2} }t'| dkDr| D]} |j)d| d| yt+d|jyt+d|jdyt+dycc}w)u8Test that non-天干合 relations don't have '5e' field. TestPerson4rrLrrrNrOrQTrRr8r9r>r?NrTrUrVrWrXrtz-Non-tianganhe relation should not have '5e': z,No non-tianganhe relations found for person r\r]r^)r r&r+rrr(r,r-r_rPrTrr)r/r@rArBrCr rbrdr0) r1rerPrErFrGrfrgrW non_tianganheris r4(test_non_tianganhe_relations_no_5e_fieldzBAPIResponseStructureTests.test_non_tianganhe_relations_no_5e_field s&&stD!R7HB{tyy'    8,499<   ]NEm_U W.Dr6c ddlm}dddd}|jD]<\}}|jj |}|j ||d|d|d |>y ) z>Test that sha god indices align with bzshagod.py gShagodNames.r)bzshagodu天乙u文昌u桃花)rruzSha god index z mismatch: expected z, got N) iching.utilsrr gShagodNamesr/r@)r1sgexpected_indicesindex expected_name actual_names r4(test_sha_god_indices_align_with_bzshagodzBAPIResponseStructureTests.test_sha_god_indices_align_with_bzshagodRss/  %5$:$:$< E=//--e4K   [-+E72F}oU[\g[hi k%=r6ctd}|jj|}|j|jd|j }|j d||j d|d|vr+|j|dduxst|dtd|vr+|j|dduxst|dt|j|dty)z5Test that paginated responses have correct structure.r8r9r?r>nextNprevious) rr)r/r@rArBrC assertTrue isinstancestrr`rarDs r4&test_api_response_pagination_structurez@APIResponseStructureTests.test_api_response_pagination_structurebs/0{{s# ))3/yy{ gt$ i& T> OODLD0QJtF|S4Q R   OOD,4Y 4 CSUX8Y Z d9ot4r6c tjjddtdddt dd|j }|j |jddlm }|d |j jd td }|jj|}|j}t|d dkDr|d d}gd}|D]}|j!||d||j!d|ddddg} | D]}|j!||d||j#|dt$|j#|dt$|j'd|d|j'd|dyy)zBTest that person records in API responses have complete structure. StructureTestrKirrNrOrQTrRr8r>)rTrrrrgzMissing required field: r.rgz#Missing bazi_result in result fieldrU relation_badzMissing relation field: relation_good_countz0Count fields should be omitted from API responserelation_bad_countN)r r&r+rrr(r,r-r_rPrTrr)r/rBrbrCr`rard) r1rerPrErFrG person_recordrequired_fieldsfieldrelation_fieldss r4%test_complete_person_record_structurez?APIResponseStructureTests.test_complete_person_record_structurevs&& dAq9IB{tyy'    8,499<rTrUrrZr[z'by' field must be a listz.'by' field should have 2 elements for pairwisezBreakdown must be a listrzBreakdown must have 4 elementsopzInvalid 'who' value: )ymdhzInvalid pillar: gezInvalid bazi type: zValue must be integerrmrzInvalid value for z: )r r&r+rrr(_skip_bazi_calculationr.r-r_rPrTrr)r/rBrCr`rar@rbrcrv)r1rerPrErFrGrfrg all_relationsriby_field breakdownwhopillar bazi_typevalues r4test_by_field_structurez1APIResponseStructureTests.test_by_field_structures:&&stD!Q7GB{tyy' )-%*+q)*    =/ 2 8,499<BRSYRZ@[\MM)c3Z;Nyk9Z[))%6MNMM%Y#5Er2)N"4YKr% IK"*( r6c tjjddtdddt dd|j }d|_d d d d d d ddd ddd d|_|jdgddl m }|d|j jdtd}|jj|}|j}|dDcgc]}|d }}|j!|j|dycc}w)z4Test API structure when person has no relationships. NoRelationsrKirrNTrrMrrr rrrr.rrOrQrRrr>rTzGPerson with no relations should not appear in person-relations endpointN)r r&r+rrr(rr.r-r_rPrTrr)r/rBrd)r1rerPrErFrGrg person_idss r4test_empty_relations_structurez8APIResponseStructureTests.test_empty_relations_structures&&stD!Q7GB{tyy' )-%+,q)*    =/ 2 8,499<d?\|jd@|dA|d@}|j!|dB|r|j#|t,dC|D]}gdD}|D]}|j||dE|dF|j#|dGt&dH|j#|dt$dI|j#|dJt,dK|j#|dLt,dM|j#|dNt,dO|j#|dPt&dQgdR}|D]}|j||dS||dT}|j!|dU|r{|D]v} | |vs|| } |jd| dV| dW|jd!| dV| dX|j#| dt&dV| dY|j#| d!t&dV| dZx|jd[|d\|d[}|j!|d]|r!gd^}|D]}|j||d_||jd`|da|jdb|dc|j#|d`t,dd|j#|dbt,deyf)gzJTest that the individual BaZi GET API returns complete response structure. BaZiAPITestrrLrrMrNapi:bazi-detailpkkwargsr9zAPI should return 200, got ) rTrrrr twin_type father_dob mother_dobnotes created_at updated_atr!zMissing basic field: rgzMissing 'result' fieldrzMissing pillar: Pillar z should not be null) rr earth_elementten_god hidden_godsemptyr$r% god_element star_luck_idx self_luck_idxzMissing field 'z' in  pillarrrr$r% CRITICAL: .5 must not be null - this was causing the recent issuerzGod in z should be stringrz Earth in r$z God index in z should be intr%zEarth index in rz Empty in z should be boolz should not be empty stringrzHidden god missing 'god' fieldrz"Hidden god missing 'ten_god' fieldsha_godszMissing 'sha_gods' fieldzsha_gods should not be nullgodszMissing 'gods' in sha_gods[]zgods in sha_gods[z] should be listrzSha god missing 'name' fieldkindnessz Sha god missing 'kindness' fieldzSha god kindness should be int)rrrz%Sha god kindness should be 0, 1, or 2sha_god_locationsz!Missing 'sha_god_locations' fieldz$sha_god_locations should not be nullz sha_god_locations should be list)gidrlocgerupczSha god location missing 'z' fieldrzgid should be intzgod should be stringrzloc should be listrzge should be listrzpillar should be listrzupc should be int) life_number life_palace ten_year_fate year_fate current_yearcurrent_10yearr. nominal_ageelement_strengthszMissing field: r.zbazi_result should not be null bazi_result[z] missing 'god' fieldz] missing 'earth' fieldz].god should be intz].earth should be intnumberzMissing 'number' fieldznumber should not be null)headno_nosrzodiacformulaliunianmingpanpairs81year_5ehave_nosyear_5ecyear_5elyear_5es number_5e number_5ec number_5el number_5es year_number zodiac_boostzMissing number field: rUzMissing 'relation_good' fieldrzMissing 'relation_bad' fieldzrelation_good should be listzrelation_bad should be listN)r r&r+rrr(r,r-rrTr)r/r@rArBrCassertIsNotNoner`rrcboolassertNotEqualra)r1rerErFrG basic_fieldsrrgpillarsr pillar_data pillar_fieldscritical_fields hidden_godr pillar_sharr sha_locationslocationr other_fieldsr.r number_fieldss r4$test_bazi_get_api_response_structurez>APIResponseStructureTests.test_bazi_get_api_response_structures`&&stD!R7HB| '   'vyy0AB{{s# ))32MdN^N^M_0`ayy{f !E MM%)>ug'F G" h&>?h3F MM&&,W]V^^lEmn%%k'&:DIfXUdBef##K$6gfXMhmn}-&1-&@  eZ9YZ i=ab'ACL j&*DE*% X'DE !X%!)&!1JMM&*8STZS[[\6]^%f-D))$8I&Qa6bc# fc3QR j#7YZ--c*osDde c*oyBij $" )63VW23  ],RS   ! !-7Y Z)"N,EMM%5OPUwV]3^_-%%huos|TZS[[pEqr " h&>?h V%@A qM' eV/EeW-MN' ot-LM nd,JK d?3T;YZ d>2D:WXr6c tjjddtdddt dd|j }|j |jtd d |ji }|jj|}|j|jd |j}td dd|_|j|jj|}|j|jd |j}|d}dD]}|j!||d|d||s"|j#d||d|d|j#d||d|dgd} | D]!} |j!||| d|d| d#|j%||ddd|d|j%||ddd|d|j!|d d!|j!|d"d#|j!|d$d%|d$rdD]}|j#||d$d&|d'|d$|} |j!| d&|d(| s>|j#d| d&|d)|j#d| d&|d*|j!| dd&|d+|j!| dd&|d,|j%|d-|d-d.y/)0zETest that BaZi GET API returns complete response after record update. UpdateTestrKrrurr{rrNrrrr9rrgrrz$ should be recalculated after updaterzUpdated z pillar missing 'god' fieldrz pillar missing 'earth' fieldrzCRITICAL: Updated rrrz.god should not be empty stringz!.earth should not be empty stringrz,sha_gods should be recalculated after updaterz5sha_god_locations should be recalculated after updater.z)bazi_result must not be null after updater] missing after updatez] must not be null after update].god missing after update].earth missing after updatez#].god must not be null after updatez%].earth must not be null after updaterz%Birth date should change after updateN)r r&r+rrr(r,r-rrTr)r/r@rArBrr rCr ) r1rerE initial_resp initial_data updated_resp updated_datargrrrrs r4test_bazi_get_api_after_updatez8APIResponseStructureTests.test_bazi_get_api_after_updates3&&cd4B6GB| '   'vyy0AB{{s+  1137#((* !q"- {{s+  1137#((* h'7F  76(Bf1g hf~ eVF^xxGb5cd gvf~&If7gh#K,E((vu-,VHAeW !  VM24\] - MM% !68c d   !6u!=?m n !r6c tjjddtdddt dd|j }|j |jtdd |ji }|jj|}|j|jd |j}|d }gd }|D]p}|j||d|d||s"gd}|D]!} |j||| d|d| d# dvsQ|j!||| dd|d| dr|j|dd|j|dd|j|dd|dr|D]}|j#||dd|d|d|} |j| d|d | s>|j#d!| d|d"|j#d#| d|d$|j| d!d|d%|j| d#d|d&gd'} | D]U} | |vs|| t%|| t&t(fr(t%|| t*s<|j!|| dd)| d*Wy()+zfTest that BaZi GET API returns NO NULL values for critical fields that were previously causing issues. NoNullTestrirr-rNrrrr9rgrzCRITICAL: Pillar z must not be nullrrrrrrz must not be empty stringrz#CRITICAL: sha_gods must not be nullrz,CRITICAL: sha_god_locations must not be nullr.z&CRITICAL: bazi_result must not be nullz CRITICAL: bazi_result must have rzCRITICAL: bazi_result[z] must not be nullrz] must have 'god' fieldrz] must have 'earth' fieldz].god must not be nullz].earth must not be null)rrrNzField z& should not be empty string if present)r r&r+rrr(r,r-rrTr)r/r@rArBr r rCrrcfloatr) r1rerErFrGrgcritical_pillarsrcritical_pillar_fieldsrrimportant_fieldss r40test_bazi_get_api_no_null_values_critical_fieldszJAPIResponseStructureTests.test_bazi_get_api_no_null_values_critical_fieldss&&cd4B6GB| '   'vyy0AB{{s# ))3/yy{h<&F  3DVHL]1^ _f~)Q&3E((vu-$VHAeW4ij4,,''vu-$VHAeW4MN'* VJ/1VW V$78:hi VM24\] - * ff]&;?_`f_ggn=op$]3F; $$[4J6(Rd2efMM%8NvhVm6noMM';:PQWPXXq8rs((U);?UV\U]]s=tu((W)=AWX^W__w?xy+I%E%=,!&-#u>#F5M37++F5M2wNt?uv&r6c Jtjjddtdddt dd|j }|j |jtd d |ji }|jj|}|j|jd |j}i}|d }dD]3}||s i||<||jD]\}} |  | |||<5tddd|_|j|jj|} |j| jd | j} | d } |jD]\}} |j#| |d|d| |s%| jD]\}}| |j|}|j#|d|d|dt%|t&t(fsJt%|t&t(fsa|dk\r|j+|d|d|d|dks|j-|d|d|dgd}|D]}|j#| |d|d| drdD]}|j/|| dd|d| d|}|j#|d|d |s>|j/d!|d|d"|j/d#|d|d$|j#|d!d|d%|j#|d#d|d&yy)'zRTest that updating a BaZi record preserves all values and doesn't introduce nulls.UpdatePreserveTestrKrrrrmrrNrrrr9rgrNrrrrz became null after updaterrz5 became null after update - this was the recent issuerz should not become negativerMz should not exceed valid range)rrr.r.rrz] became null after updaterrrr z].god became null after updatez ].earth became null after update)r r&r+rrr(r,r-rrTr)r/r@rArBrrr rrcr.assertGreaterEqualassertLessEqualrC)r1rerEr!r"initial_non_null_valuesrgrrrr#r$updated_resultfields initial_value updated_valuerrs r4-test_bazi_get_api_update_preserves_all_valueszGAPIResponseStructureTests.test_bazi_get_api_update_preserves_all_valuesFsk&&%cd4B>OB| '   'vyy0AB{{s+  1137#((* #%h'7Ff~24'/$*6N$8$8$:LE5(AF/7>%;7!q"- {{s+  1137#((* &h/5;;=NFF  !776(Jc9d ef%,2LLN(E=$26$:$>$>u$EM((%$VHAeW4ij "-#u>:m^ach]iCj(A- 33M1PQRWQXXsFtu(B. 00vhaPUwVtDuv-; >(K$E  !6*UGKd8e f% - (: fn]&C|TZS[[qErs,];FC $$[LHb2cdMM% VHLf6ghMM';,vhNj8kl((U);|F8Sq=rs((W)=fXUu?vw; )r6N)__name__ __module__ __qualname____doc__r5rHrjrprxr}rrrrrrrr%r*r2r<r6r4rrsxTG.)&HAT?QD=AB6At Wk 5(*OX7Krc>NY`GDR%oNJwXMxr6r) __future__r django.testrdjango.contrib.authr django.urlsrdatetimerrrB bazi.modelsr iching.utils.bazi_relationsr r r r rrrrAr6r4rIs4" .  DxDxr6