i<dZddlZddlmcmZddlZddlZddl Z ddl Z ddl m cm ZddlmZmZmZmZddlZe j*j-ddddlmZddlmZmZddlmZddlmZm Z dd l!m"Z"Gd d Z#Gd d Z$GddZ%GddZ&GddZ'GddZ(GddZ)GddZ*GddZ+GddZ,y)u VERIFICATION_STAMP Story: 2.08 (Track B) Verified By: parallel-builder Verified At: 2026-02-25T00:00:00Z Tests: 12/12 Coverage: 100% Module 2 White-Box Test Suite — JIT Hydration Pipeline Story 2.08 — Track B Tests internal implementation paths: - fast_extract: timing contract (<5ms), no I/O, no HTTP - scatter_gather: asyncio.gather concurrency, individual per-layer timeout - JITHydrationInterceptor: priority, metadata.name, interceptor chain contract - build_envelope: latency_ms formatting, mandatory 4 XML child tags - RedisL1Client.bump_version: atomic INCR pattern (not read-modify-write) - RedisL1State.version: defaults to 0 ALL external I/O is mocked. No live Redis, Qdrant, or KG required. N) AsyncMock MagicMockcallpatchz/mnt/e/genesis-system) fast_extract) RedisL1Client RedisL1State)scatter_gather_memory) MemoryContextbuild_envelope)JITHydrationInterceptorceZdZdZdZy)TestFastExtractTimingcdddd}tj}t|tj|z dz}d}||k}|stjd|fd||fd t j vstj|rtj|nd tj|d z}tjd |d d dzd|iz}ttj|dx}}y)u;fast_extract runs in <5ms — it is pure CPU with zero I/O.z8Build the redis schema at core/memory/redis_l1_schema.pyz/Use async def get_state and class RedisL1Clientz'Wire redis and qdrant services together)prompt descriptiontask@@@<z%(py0)s < %(py3)s elapsed_mspy0py3zfast_extract took z.2fu-ms — must stay under 5ms (pure CPU, no I/O) >assert %(py5)spy5N) time monotonicr @pytest_ar_call_reprcompare @py_builtinslocals_should_repr_global_name _saferepr_format_assertmsgAssertionError_format_explanation)selfpayloadstartr @py_assert2 @py_assert1 @py_format4 @py_format6s 8/mnt/e/genesis-system/tests/memory/test_jit_hydration.py%test_fast_extract_completes_under_5msz;TestFastExtractTiming.test_fast_extract_completes_under_5ms.sQL=   Wnn&.&8  zC  zC   6         !C 00] ^     c`tj}ti\}}tj|z dz}d}||k}|stjd|fd||fdt j vstj|rtj|ndtj|dz}dd|iz}ttj|d x}}g}||k(}|stjd |fd ||fd t j vstj|rtj|nd tj|dz}dd|iz}ttj|d x}}d }||k(}|stjd |fd ||fdt j vstj|rtj|ndtj|dz}dd|iz}ttj|d x}}y )zassert not %(py5)s {%(py5)s = %(py2)s {%(py2)s = %(py0)s.iscoroutine }(%(py3)s) }asyncioresultrpy2rrN) rrD iscoroutiner!r'r#r$r%r&r(r))r*rEr. @py_assert4 @py_assert6 @py_format7s r1(test_fast_extract_makes_no_network_callszassert not %(py5)s {%(py5)s = %(py2)s {%(py2)s = %(py0)s.iscoroutinefunction }(%(py3)s) }rDrrFN) rDiscoroutinefunctionrr!r'r#r$r%r&r(r))r*r.rIrJrKs r1-test_fast_extract_is_not_a_coroutine_functionzATestFastExtractNoIO.test_fast_extract_is_not_a_coroutine_functionSs.. .|< << <  C   6      /   6  0<  0<  =      r3cFtddi\}}|Dcgc]}|j}}d}||v}|stjd|fd||ftj|dt j vstj|rtj|nddz}dd |iz}ttj|d x}}d }||v}|stjd|fd||ftj|dt j vstj|rtj|nddz}dd |iz}ttj|d x}}d }||v}|stjd|fd||ftj|dt j vstj|rtj|nddz}dd |iz}ttj|d x}}y cc}w) zKfast_extract correctly identifies known Genesis services from payload text.rz@Connect to redis and qdrant for memory storage with telnyx voiceredisinz%(py1)s in %(py3)s entity_lowerpy1rr5rNqdranttelnyx) rlowerr!r"r&r#r$r%r(r)) r*r9r;erU @py_assert0r-r/r0s r1)test_fast_extract_extracts_known_servicesz=TestFastExtractNoIO.test_fast_extract_extracts_known_servicesYsG' X)  &,44a 4 4&w,&&&&w,&&&w&&&&&&,&&&,&&&&&&&'x<''''x<'''x''''''<'''<''''''''x<''''x<'''x''''''<'''<'''''''5sHctddi\}}d|D}t|}|sddtjvst j trt j tndt j |t j |dz}tt j|dx}}d|D}t|}|sddtjvst j trt j tndt j |t j |dz}tt j|dx}}y) z8fast_extract identifies .py file paths from the payload.rzCEdit core/memory/redis_l1_schema.py and core/memory/fast_extract.pyc3$K|]}d|v yw)zredis_l1_schema.pyNr@.0r[s r1 zLTestFastExtractNoIO.test_fast_extract_extracts_file_paths..hs?'1,?z,assert %(py4)s {%(py4)s = %(py0)s(%(py2)s) }any)rrGpy4Nc3$K|]}d|v yw)zfast_extract.pyNr@r`s r1rbzLTestFastExtractNoIO.test_fast_extract_extracts_file_paths..isr?rLrOr]rirvr@r3r1rBrBJsW (="r3rBcDeZdZejj dZy)TestScatterGatherConcurrencyc  Kg tj fd}tdtd5tdtd5tdtd5td|5t d d gd d d{ddddddddddddt }d}||k\}|s t jd|fd||fdtjvst jt rt jt nddtjvst j rt j ndt j|t j|dz}t jddzd|iz}tt j|dx}x}} d}d}||k(}|slt jd|fd||ft j|t j|dz} dd| iz}tt j|dx}x}}y7#1swYxYw#1swYxYw#1swYxYw#1swYxYww)zLscatter_gather internally calls asyncio.gather to fire all 3 layers at once.cdKjt||i|d{S7w)N)appendrn)coroskwargs gather_callsoriginal_gathers r1 spy_gatherzdTestScatterGatherConcurrency.test_asyncio_gather_is_called_during_scatter_gather..spy_gather~s/   E +(%:6:: ::s &0.0*core.memory.scatter_gather._fetch_redis_l1N) new_callable return_value'core.memory.scatter_gather._fetch_kg_l2+core.memory.scatter_gather._fetch_qdrant_l3z)core.memory.scatter_gather.asyncio.gather side_effectz task-concxr;rl timeout_ms)>=)z0%(py3)s {%(py3)s = %(py0)s(%(py1)s) } >= %(py6)srnr~roz5asyncio.gather must be called for concurrent fetching >assert %(py8)srqrr6)z%(py1)s == %(py4)s)rWrezassert %(py6)srp)rDgatherrrr rnr!r"r#r$r%r&r'r(r)) r*rr-rtrIrKrur\rgrhr~rs @@r13test_asyncio_gather_is_called_during_scatter_gatherzPTestScatterGatherConcurrency.test_asyncio_gather_is_called_during_scatter_gatherxs !.. ; >Yei j V ;Yei j V ?ifj k V =: V  V ( cUHQTU U U  V V V V< ^A^ A%^^^ A^^^^^^s^^^s^^^^^^<^^^<^^^ ^^^A^^^'^^^^^^^^A#!#!####!######!####### V  V V V V V V V Vsw+I;I.I!I#I 7I 8I <II! I.F0I;I I II I!!I+ &I..I83I;N)r=r>r?pytestmarkrDrr@r3r1rxrxvs [[$$r3rxcDeZdZejj dZy)"TestScatterGatherIndividualTimeoutc  Kg g d} fd} fd}td|5td|5td|5tdd gd d d {}d d d d d d d d d  s{tjddzddt j vstj rtj ndiz}ttj| s{tjddzddt j vstj rtj ndiz}ttj|j}d }||u}|stjd|fd||fdt j vstj |rtj|ndtj|tj|dz} dd| iz} ttj| d x}x}}|j}d }||u}|stjd|fd||fdt j vstj |rtj|ndtj|tj|dz} dd| iz} ttj| d x}x}}|j}d }||u}|stjd|fd||fdt j vstj |rtj|ndtj|tj|dz} dd| iz} ttj| d x}x}}y 7m#1swYmxYw#1swYrxYw#1swYwxYww)zCA slow L1 does not delay L2/L3 beyond the per-layer timeout budget.cJKtjdd{y7w)N?never)rDsleep)_task_ids r1slow_l1zaTestScatterGatherIndividualTimeout.test_per_layer_timeout_enforced_independently..slow_l1s --$ $ $ %s #!#c0Kjdyw)NTzKG Topology: fast entityr{) _entities l2_calleds r1fast_l2zaTestScatterGatherIndividualTimeout.test_per_layer_timeout_enforced_independently..fast_l2s   T ".c0Kjdyw)NTzLearned Constraints: fast scarr)_intent l3_calleds r1fast_l3zaTestScatterGatherIndividualTimeout.test_per_layer_timeout_enforced_independently..fast_l3s   T "4rrrrrz task-timeoutyr;2rNz+L2 must execute independently of L1 timeoutz >assert %(py0)srrz+L3 must execute independently of L1 timeoutris)z5%(py2)s {%(py2)s = %(py0)s.working_state } is %(py5)sctxrrGrassert %(py7)spy7is not)z7%(py2)s {%(py2)s = %(py0)s.kg_topology } is not %(py5)s)z?%(py2)s {%(py2)s = %(py0)s.learned_constraints } is not %(py5)s)rr r!r'r#r$r%r&r(r) working_stater" kg_topologylearned_constraints) r*rrrr @py_format1r.rIrgr0 @py_format8rrs @@r1-test_per_layer_timeout_enforced_independentlyzPTestScatterGatherIndividualTimeout.test_per_layer_timeout_enforced_independentlysw     / 5 >G T ^ ;G T ^ ?W U ^ .nseXZ\]]C  ^ ^ ^GGGGGGGGGyGGGyGGGGGGGGGGGGGGyGGGyGGGGG  (D( D(((( D((((((s(((s((( (((D(((((((*d*d****d******s***s******d*******&&2d2&d2222&d222222s222s222&222d2222222^  ^ ^ ^ ^ ^ ^s]O*OOOO OO O(MO*OO OO OO'"O*N)r=r>r?rrrDrr@r3r1rrs [[!3!3r3rceZdZdZy)TestInterceptorPrioritycFt}|j}|j}d}||k(}|stjd|fd||fdt j vstj|rtj|ndtj|tj|tj|dz}tjd|jjdzd|iz}ttj|d x}x}x}}y ) z=JITHydrationInterceptor.metadata.priority must be exactly 10. r6)zN%(py4)s {%(py4)s = %(py2)s {%(py2)s = %(py0)s.metadata }.priority } == %(py7)s interceptorrrGrerz#Expected priority=10, got priority= >assert %(py9)spy9N) r metadatapriorityr!r"r#r$r%r&r'r(r)r*rr.rgrJrtr @py_format10s r1test_interceptor_priority_is_10z7TestInterceptorPriority.test_interceptor_priority_is_10s-/ ## #,,  ,2  ,   6      $  -  13  2+2F2F2O2O1P Q      r3N)r=r>r?rr@r3r1rrs r3rceZdZdZdZy)TestInterceptorMetadataNamecHt}|j}|j}d}||k(}|stjd|fd||fdt j vstj|rtj|ndtj|tj|tj|dz}tjd|jjddzd |iz}ttj|d x}x}x}}y ) z>JITHydrationInterceptor.metadata.name must be 'jit_hydration'. jit_hydrationr6)zJ%(py4)s {%(py4)s = %(py2)s {%(py2)s = %(py0)s.metadata }.name } == %(py7)srrz)Expected name='jit_hydration', got name=''rrN) r rnamer!r"r#r$r%r&r'r(r)rs r1/test_interceptor_metadata_name_is_jit_hydrationzKTestInterceptorMetadataName.test_interceptor_metadata_name_is_jit_hydrations-/ ## #(( O (O;  (O   6      $  )  -<  8 8L8L8Q8Q7RRS T      r3ct}|j}|j}d}||u}|stjd|fd||fdt j vstj|rtj|ndtj|tj|tj|dz}dd|iz}ttj|dx}x}x}}y) zr?rrr@r3r1rrs  4r3rceZdZdZdZdZy)TestEnvelopeLatencyFormattingcftdddd}t|}tj|}|j d}d}||u}|st j d|fd||fdtjvst j|rt j|ndt j|d z}d d |iz}tt j|dx}}|j} d }| |k(}|st j d |fd| |fdtjvst j| rt j| ndt j|d z}t jd| ddzd |iz}tt j|dx}}y)z=build_envelope formats latency_ms to exactly 1 decimal place.okg$(@rrr latency_msHYDRATION_LATENCY_MSNr)z%(py0)s is not %(py3)s latency_elrr5rz12.3r6r8 latency_textzExpected latency '12.3', got 'rr)r r ET fromstringfindr!r"r#r$r%r&r(r)textr') r*rxml_strrootrr-r.r/r0rs r1%test_latency_formatted_to_one_decimalzCTestEnvelopeLatencyFormatting.test_latency_formatted_to_one_decimalsG $!   !%}}W%YY56 !%%z%%%%z%%%%%%z%%%z%%%%%%%%%%! % |v%  |v   6       &  -\N! <     r3ctdddd}t|}tj|}|j d}|j }d}||k(}|st jd|fd||fdtjvst j|rt j|ndt j|t j|d z}d d |iz} tt j| dx}x}}y) z6latency_ms=0.0 renders as '0.0' (not '0', not '0.00').Nrrz0.0r6z,%(py2)s {%(py2)s = %(py0)s.text } == %(py5)srrrrr r rrrrr!r"r#r$r%r&r(r) r*rrrrr.rIrgr0rs r1.test_zero_latency_formatted_as_zero_point_zerozLTestEnvelopeLatencyFormatting.test_zero_latency_formatted_as_zero_point_zeros $   !%}}W%YY56 '%'%''''%''''''z'''z''''''%'''''''r3ctdddd}t|}tj|}|j d}|j }d}||k(}|st jd|fd||fdtjvst j|rt j|ndt j|t j|d z}d d |iz} tt j| d x}x}}y ) zr?rrrr@r3r1rrs & ( +r3rceZdZdZdZy)TestEnvelopeFourTagsc tdddd}t|}tj|}t |}t |}d}||k(}|s3t jd|fd||fd tjvst jt rt jt nd d tjvst j|rt j|nd t j|t j|d z}t jd t |d |D cgc]} | jc} dzd|iz} tt j| dx}x}}ycc} w)zJbuild_envelope produces exactly 4 child elements under ZERO_AMNESIA_STATE.wskglcrrr6rmrnchildrenrozExpected 4 child elements, got z: rrqN)r r rrlistrnr!r"r#r$r%r&r'tagr(r)) r*rrrrr-rtrIrKcrus r1,test_envelope_always_has_four_child_elementszATestEnvelopeFourTags.test_envelope_always_has_four_child_elementssY $   !%}}W%:8}  }!   }   6       6        !"  .c(m_Bx?X!?X>Y Z      ?XsF ctdddd}t|}tj|}|Dcgc]}|j}}d}||v}|st j d|fd||ft j|dtjvst j|rt j|nddz}d d |iz} tt j| dx}}d }||v}|st j d|fd||ft j|dtjvst j|rt j|nddz}d d |iz} tt j| dx}}d }||v}|st j d|fd||ft j|dtjvst j|rt j|nddz}d d |iz} tt j| dx}}d }||v}|st j d|fd||ft j|dtjvst j|rt j|nddz}d d |iz} tt j| dx}}ycc}w)z7The 4 child elements have the exact expected tag names.NrrWORKING_CONTEXTrRrTtagsrVr5rTOPOLOGICAL_BLAST_RADIUSLEARNED_CONSTRAINTSr) r r rrrr!r"r&r#r$r%r(r)) r*rrrchildrr\r-r/r0s r1)test_envelope_child_tag_names_are_correctz>TestEnvelopeFourTags.test_envelope_child_tag_names_are_correct$s $   !%}}W%'+,e ,, ( D(((( D((( ((((((D(((D((((((()1)T1111)T111)111111T111T1111111$,$,,,,$,,,$,,,,,,,,,,,,,,,,%-%----%---%---------------- -sKN)r=r>r?rrr@r3r1rrs   .r3rcDeZdZejj dZy)TestRedisL1BumpVersionAtomiccTKtd}t}td|_ttd|_||_|j dd{}d}||k(}|stjd |fd ||fd tjvstj|rtj|nd tj|d z}d d|iz}ttj|dx}}|jj|jjdd}d} | |v}|stjd|fd| |ftj| dtjvstj|rtj|nddz}d d|iz}ttj|dx} }|j }d} || } | sddtjvstj|rtj|ndtj|tj| tj| dz} ttj| dx}x} } y7Fw)uKbump_version must use INCR for atomicity — NOT a read-then-write pattern.zredis://fake:6379) redis_url)rz&get must NOT be called in bump_versionrz task-atomicNr6r8 new_versionrr5rrrRrTincr_keyrVz:versionzJassert %(py6)s {%(py6)s = %(py2)s {%(py2)s = %(py0)s.endswith }(%(py4)s) })rrGrerp)rrincrr(get_redis bump_versionr!r"r#r$r%r&r)assert_awaited_once call_argsendswith) r*clientmock_rrr-r.r/r0rr\rgrtrKs r1.test_bump_version_calls_redis_incr_not_get_setzKTestRedisL1BumpVersionAtomic.test_bump_version_calls_redis_incr_not_get_set;s)<=Q/ >:b+cd  "// >> {a{a{{a '');;((+A.(}((((}(((}((((((((((((((((  ,, ,,,,,,,,x,,,x,,, ,,,,,,,,,,,,,?sAJ(J% IJ(N)r=r>r?rrrDrr@r3r1rr9s [[--r3rc$eZdZdZdZdZdZy)TestRedisL1StateVersionDefaultctddd}|j}d}||k(}|stjd|fd||fdt j vstj |rtj|ndtj|tj|dz}tjd |jd zd |iz}ttj|d x}x}}y ) z;RedisL1State.version field defaults to 0 when not provided.z t-defaults-1task_id session_idactive_task_idrr6z/%(py2)s {%(py2)s = %(py0)s.version } == %(py5)sstaterz*Expected version=0 (default), got version=z >assert %(py7)srN) r versionr!r"r#r$r%r&r'r(r)r*rr.rIrgr0rs r1test_version_default_is_zeroz;TestRedisL1StateVersionDefault.test_version_default_is_zeroWs&  }}  }!  }   6        !"  9 H      r3ctdddd}|j}d}||k(}|stjd|fd||fdt j vstj |rtj|ndtj|tj|dz}d d |iz}ttj|d x}x}}y ) zCRedisL1State.version can be explicitly set to any non-negative int.z t-versionedr *)rrrrr6rrrrrN) r rr!r"r#r$r%r&r(r)rs r1"test_version_can_be_set_explicitlyzATestRedisL1StateVersionDefault.test_version_can_be_set_explicitlybs!(   }}""}""""}""""""u"""u"""}""""""""""r3c4tddd}|j}g}||k(}|stjd|fd||fdt j vstj |rtj|ndtj|tj|dz}dd |iz}ttj|d x}x}}|j}t|t}|sd d t j vstj trtjtnd dt j vstj |rtj|ndtj|d t j vstj trtjtnd tj|dz} ttj| d x}}y )zARedisL1State.focus_entities defaults to an empty list (not None).zt-fezs-2r r6)z6%(py2)s {%(py2)s = %(py0)s.focus_entities } == %(py5)srrrrNzYassert %(py6)s {%(py6)s = %(py0)s(%(py3)s {%(py3)s = %(py1)s.focus_entities }, %(py4)s) } isinstancer)rrWrrerp) r focus_entitiesr!r"r#r$r%r&r(r)rr) r*rr.rIrgr0rr-rtrKs r1*test_focus_entities_defaults_to_empty_listzITestRedisL1StateVersionDefault.test_focus_entities_defaults_to_empty_listls)!  ##)r)#r))))#r))))))u)))u)))#)))r)))))))..5z.55555555z555z555555%555%555.5555555555555555555r3ctddd}|j}d}||u}|stjd|fd||fdt j vstj |rtj|ndtj|tj|dz}d d |iz}ttj|dx}x}}y) z-RedisL1State.active_blocker defaults to None.zt-abzs-3r Nr)z6%(py2)s {%(py2)s = %(py0)s.active_blocker } is %(py5)srrrr) r active_blockerr!r"r#r$r%r&r(r)rs r1$test_active_blocker_defaults_to_nonezCTestRedisL1StateVersionDefault.test_active_blocker_defaults_to_nonevs!  ##+t+#t++++#t++++++u+++u+++#+++t+++++++r3N)r=r>r?rrrrr@r3r1r r Us  #6,r3r )-__doc__builtinsr#_pytest.assertion.rewrite assertionrewriter!rDinspectsysrxml.etree.ElementTreeetree ElementTreer unittest.mockrrrrrpathinsertcore.memory.fast_extractrcore.memory.redis_l1_schemarr core.memory.scatter_gatherr !core.memory.zero_amnesia_enveloper r %core.memory.jit_hydration_interceptorr rrBrxrrrrrrr r@r3r1r2s* "";; *+1C<KI<%"%"X$$:$3$3V   4 4&-+-+h..L--8(,(,r3