i<dZddlmZddlZddlmcmZddl Z e jjddddl Z ddl Z ddlZddlmZmZmZmZddlmZddlmZmZdZd5d Zd Zd6d Zd6d Zd7d ZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ*dZ+dZ,dZ-e.dk(rddl/Z/defde fd e!fd!e"fd"e#fd#e$fd$e%fd%e&fd&e'fd'e(fd(e)fd)e*fd*e+fd+e,fd,e-fgZ0dZ1e2e0Z3e0D]\Z4Z5 e5e6d-e4e1d.z Z1e6d1e1d2e3d3e1e3k(r e6d4ye jtd.yy#e7$r)Z8e6d/e4d0e8e/jrYdZ8[8zdZ8[8wwxYw)8u Tests for Story 6.02 (Track B): RedisMasterState — Versioned OCC State Store Black Box tests (BB): verify the public contract from the outside — get_snapshot return types, commit_patch success/conflict, initialize_state. White Box tests (WB): verify internal mechanics — new_version increment, JSON storage format, WatchError → conflict, patch application, JSON decode. Story: 6.02 File under test: core/coherence/redis_master_state.py ALL tests use mocks — NO real Redis connection is made. ) annotationsNz/mnt/e/genesis-system) AsyncMock MagicMockpatch PropertyMock) WatchError)RedisMasterState CommitResultcHtjj|S)z>!T!>T !!!!>T!!!!!!6!!!6!!!>!!!T!!!!!!!   ""  """" """"""6"""6""" """""""""" ??#e#?e ####?e######6###6###?###e#######rcVt}t|}t|jdddd}t |t }|s ddt jvstjtrtjtnddt jvstj|rtj|nddt jvstjt rtjt ndtj|d z}ttj|d }|j}d}||u}|stjd |fd ||fdt jvstj|rtj|ndtj|tj|d z}dd|iz}ttj|d x}x}}|j}d}||k(}|stjd|fd||fdt jvstj|rtj|ndtj|tj|d z}dd|iz}ttj|d x}x}}|j }d}||u}|stjd |fd||fdt jvstj|rtj|ndtj|tj|d z}dd|iz}ttj|d x}x}}y )uTBB3: initialize_state → CommitResult(success=True, new_version=1, conflict=False).zsess-newgenesisT)agentreadyr[r\r]r r^Nrbrdrerfrgrhr;riFrj)r.r rinitialize_stater\r rFrGrDrHrIrJrKrlrErmrn) r+rLr]rprqrNrrrPrss r*test_bb3_initialize_state_sets_version_onerzs"$J : &C %%jIPT2UV WF fl ++ ++++++:+++:++++++f+++f++++++l+++l+++ +++++++ >>!T!>T !!!!>T!!!!!!6!!!6!!!>!!!T!!!!!!!   ""  """" """"""6"""6""" """""""""" ??#e#?e ####?e######6###6###?###e#######rctdddi}t|}t|jddg}|j}d}||u}|st j d |fd ||fd tjvst j|rt j|nd t j|t j|d z}d d|iz}tt j|dx}x}}|j}d}||u}|st j d |fd||fd tjvst j|rt j|nd t j|t j|d z}d d|iz}tt j|dx}x}}y)uTBB4: commit_patch with wrong version → CommitResult(success=False, conflict=True).xrh)r8rrYrZFrbrdr]rerfrgNTrj)r9r rrkrlrDrErFrGrHrIrJrKrnr+rLr]rNrrrprPrss r4test_bb4_commit_patch_wrong_version_returns_conflictrs0qQxPJ : &C !!*ar!B CF >>"U">U """">U""""""6"""6""">"""U""""""" ??"d"?d """"?d""""""6"""6"""?"""d"""""""rcbddd}td|}t|}t|jd\}}d}||k(}|st j d|fd||fd t jvst j|rt j|nd t j|d z}d d |iz}tt j|d x}}||k(}|st j d|fd||fdt jvst j|rt j|nddt jvst j|rt j|nddz} dd| iz} tt j| d }y )zFBB5: get_snapshot on existing session returns correct (version, data).rv)rwtasksrz sess-existingr;r=rr>rArBN)z%(py0)s == %(py2)sr existing_datar?r`assert %(py4)sra) r3r rrCrDrErFrGrHrIrJrK) rr+rLrrrMrNrOrP @py_format3rqs r=test_bb5_get_snapshot_existing_state_returns_version_and_datars'!4M' FJ : &C((9:MGT7a<7a77a =    4=      4   4      =   =       rc@tdddi}td|_t|}t |j ddd i}|j }d }||u}|stjd |fd ||fd tjvstj|rtj|nd tj|tj|dz}dd|iz}ttj|dx}x}}|j}d}||u}|stjd |fd||fd tjvstj|rtj|nd tj|tj|dz}dd|iz}ttj|dx}x}}|j }d}||k(}|stjd|fd||fd tjvstj|rtj|nd tj|tj|dz}dd|iz}ttj|dx}x}}y)u]BB6: initialize_state when key already exists → CommitResult(success=False, conflict=True).rhexistingTrNrz sess-existsnewrFrbrdr]rerfrgrjrr;ri)r3rr"r rryrlrDrErFrGrHrIrJrKrnrmrs r5test_bb6_initialize_state_key_exists_returns_conflictrs'T8JKJD1JN : &C %%meV_E FF >>"U">U """">U""""""6"""6""">"""U""""""" ??"d"?d """"?d""""""6"""6"""?"""d"""""""   ""  """" """"""6"""6""" """"""""""rctdddi}t|}t|jddg}|j}d}||k(}|st j d|fd ||fd tjvst j|rt j|nd t j|t j|d z}d d |iz}tt j|dx}x}}y)z8WB1: new_version = version + 1 exactly (not +2, not +0).rcount*rzsess-wb1rZr;rir]rerfrgN) r3r rrkrmrDrErFrGrHrIrJrKrs r0test_wb1_new_version_is_version_plus_one_exactlyrs(" FJ : &C !!*ar!B CF   ""  """" """"""6"""6""" """"""""""rcht}t|}ddddg}t|jdd||j}|j j }d}||u}|stjd |fd ||fd tjvstj|rtj|nd tj|d z}tjd dzd|iz}ttj|dx}}|dd} t!j"| } d} | | v}|stjd|fd| | ftj| dtjvstj| rtj| nddz}tjd| dzd|iz}ttj|dx} }d} | | v}|stjd|fd| | ftj| dtjvstj| rtj| nddz}tjd| dzd|iz}ttj|dx} }| d} d} | | k(}|sltjd|fd| | ftj| tj| dz} dd| iz}ttj|dx} x}} | dd} d} | | k(}|sltjd|fd| | ftj| tj| dz} dd| iz}ttj|dx} x}} y)z9WB2: State stored as JSON with 'version' and 'data' keys.rSz/keyvalrUzsess-wb2rrZN)is not)z%(py0)s is not %(py3)s set_call_argsr>zpipe.set was not called >assert %(py5)srBrhr)in)z%(py1)s in %(py3)sparsedr_r@z('version' key missing from stored JSON: rz%'data' key missing from stored JSON: r;z%(py1)s == %(py4)sr_razassert %(py6)spy6key)r.r rrkr*r" call_argsrDrErFrGrHrI_format_assertmsgrJrKrloads)r+rLror,rrMrNrOrP stored_jsonr @py_assert0rprq @py_format7s r8test_wb2_state_stored_as_json_with_version_and_data_keysrsX"$J : &Cv>?IQi@A   DHH&&M $?= $???=??????=???=??????&???????"1%K ZZ $F S9 SSS9SSS9SSSSSSSSSSSSS"J6( SSSSSSS M6V MMM6VMMM6MMMMMMVMMMVMMMMDVHMMMMMMM ) !!  !!!! !!! !!!!!!!!!! &>% )E) E )))) E))) )))E)))))))rctdddi}t|}t|jddg}|j}d}||u}|st j d|fd ||fd tjvst j|rt j|nd t j|t j|d z}d d |iz}tt j|dx}x}}|j}d}||u}|st j d|fd||fd tjvst j|rt j|nd t j|t j|d z}d d |iz}tt j|dx}x}}|j}d}||k(}|st j d|fd||fd tjvst j|rt j|nd t j|t j|d z}d d |iz}tt j|dx}x}}y)ubWB3: WatchError raised by Redis pipeline.execute() → CommitResult(success=False, conflict=True).kvrzsess-wb3rZFrbrdr]rerfrgNTrjrr;ri)r6r rrkrlrDrErFrGrHrIrJrKrnrmrs r3test_wb3_watch_error_from_pipeline_returns_conflictrs-asCjIJ : &C !!*ar!B CF >>"U">U """">U""""""6"""6""">"""U""""""" ??"d"?d """"?d""""""6"""6"""?"""d"""""""   ""  """" """"""6"""6""" """"""""""rc0dddd}td|}t|}dddd d d d d d ddg}t|jdd||j}|j j dd}tj|}|d}|d}d} || k(} | stjd| fd|| ftj|tj| dz} tjd|dzd| iz} ttj| dx}x} } |d}d } || k(} | stjd| fd|| ftj|tj| dz} tjd|dzd| iz} ttj| dx}x} } d}||v} | stjd| fd ||ftj|dtj vstj"|rtj|ndd!z} tjd"|d#zd$| iz}ttj|dx}} |d%}d} || k(} | stjd| fd|| ftj|tj| dz} tjd&|dzd| iz} ttj| dx}x} } y)'zOWB4: Patch operations (add, replace, remove) correctly transform the data dict.rhrr~)abcrrSz/drrUreplacez/acremovez/c)rVrWzsess-wb4rZrrdr;rrz'add' op failed: z >assert %(py6)srNrz'replace' op failed: r)not in)z%(py1)s not in %(py3)srz'remove' op failed: rrBrzuntouched key 'b' was changed: )r3r rrkr*r"rrrrDrErIrrJrKrFrGrH) initial_datar+rLror,rrrrrprMrqrrOrPs r/test_wb4_patch_operations_correctly_modify_datars-+L' EJ : &C$!4$"5$'I Qi@A   D(($$Q'*K ZZ $F &>D 9669>66696669666666/v66666666 9::9?:::9:::9::::::3D6:::::::: 93d?9993d9993999999d999d999924&9999999 9DD9>DDD9DDD9DDDDDD=dVDDDDDDDDrc~ddddd}tj|j}t}t ||_t |}t|jd\}}d}||k(}|stjd|fd ||fd tjvstj|rtj|nd tj|d z}d d |iz} ttj | dx}}ddd}||k(}|stjd|fd ||fdtjvstj|rtj|ndtj|d z}d d |iz} ttj | dx}}y)zBWB5: get_snapshot correctly deserializes the JSON blob from Redis. betar)alpharrrzsess-wb5r;r=rr>rArBNr)rrrrrr!r rrCrDrErFrGrHrIrJrK) payloadr2r+rLrrrMrNrOrPs r1test_wb5_get_snapshot_deserializes_json_correctlyr2s&DEGjj!((*GJG4JN : &C((45MGT7b=7b77b#b1141 1111411111114111411111111111rcddlm}|tu}|stjd|fd|tfdt j vstj |rtj|nddt j vstj trtjtnddz}dd |iz}ttj|d }y ) z?Package level: RedisMasterState importable from core.coherence.r)r rbz%(py0)s is %(py2)sRMSr rrraN) core.coherencer rDrErFrGrHrIrJrK)rrNrrqs r'test_package_exports_redis_master_staterFsq6 " """"3"""""""3"""3""""""""""""""""""rcddlm}|tu}|stjd|fd|tfdt j vstj |rtj|nddt j vstj trtjtnddz}dd |iz}ttj|d }y ) z;Package level: CommitResult importable from core.coherence.r)r rbrCRr rrraN) rr rDrErFrGrHrIrJrK)rrNrrqs r"test_package_exports_commit_resultrLsm1  222rctddd}|j}d}||u}|stjd|fd||fdt j vstj |rtj|ndtj|tj|dz}d d |iz}ttj|d x}x}}|j}d}||k(}|stjd |fd ||fdt j vstj |rtj|ndtj|tj|dz}d d |iz}ttj|d x}x}}|j}d}||u}|stjd|fd||fdt j vstj |rtj|ndtj|tj|dz}d d |iz}ttj|d x}x}}y )zACommitResult has expected fields: success, new_version, conflict.Tr|F)rlrmrnrbrdrrerfrgNr;rirj) r rlrDrErFrGrHrIrJrKrmrn)rrNrrrprPrss r#test_commit_result_dataclass_fieldsrRsWTq5AA 999 9119 ==A=A =A11=A ::: :11:rctj}d}||k(}|stjd|fd||fdt j vstj trtjtndtj|tj|dz}dd|iz}ttj|dx}x}}y) z-KEY_PREFIX is set to 'genesis:state:master:'.zgenesis:state:master:r;)z2%(py2)s {%(py2)s = %(py0)s.KEY_PREFIX } == %(py5)sr rerfrgN) r KEY_PREFIXrDrErFrGrHrIrJrK)rNrrrprPrss rtest_key_prefix_constantrZsz  & &A*AA &*A AAAA &*AAAAAAA AAA AAA &AAA*AAAAAAAAr__main__u3BB1: Fresh session → get_snapshot returns (0, {})u-BB2: commit_patch correct version → successu+BB3: initialize_state → version=1 successu,BB4: commit_patch wrong version → conflictu3BB5: get_snapshot existing state → correct valuesu-BB6: initialize_state key exists → conflictz&WB1: new_version = version + 1 exactlyz0WB2: State stored as JSON with version+data keysu!WB3: WatchError → conflict=Truez+WB4: Patch operations modify data correctlyz-WB5: get_snapshot deserializes JSON correctlyz4PKG: RedisMasterState importable from core.coherencez0PKG: CommitResult importable from core.coherencezCommitResult dataclass fieldszKEY_PREFIX constant checkz [PASS] rhz [FAIL] z:  /z tests passedz(ALL TESTS PASSED -- Story 6.02 (Track B))rintrdictreturnbytes)rrrr)r8rrr);__doc__ __future__rbuiltinsrF_pytest.assertion.rewrite assertionrewriterDsysrWinsertrpytestr unittest.mockrrrrredis.exceptionsr!core.coherence.redis_master_stater r rrr.r3r6r9rQrtrzrrrrrrrrrrrr__name__ tracebacktestspassedlentotalnamefnprint Exceptionexc print_excexitrrrs # *+ CC'M=Z @68@ $$#! #"#*(#E2 2(#  B z ?@vw 8:op 68bc 79mn >@}~ 8:op 13cd ;=uv ,.ab 68gh 8:kl ?Ahi ;=_` (*MN $&>? E$F JE"b " D IdV$ % aKF " Bvhawm ,-  89 I8 " IdV2cU+ , I   ! ! "s*D44E"9EE"