51 lines
4.1 KiB
Python
51 lines
4.1 KiB
Python
#!/usr/bin/env python3
|
|
import importlib.util,json,tempfile,unittest
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
S=Path(__file__).with_name('command_router.py'); spec=importlib.util.spec_from_file_location('router',S); M=importlib.util.module_from_spec(spec); spec.loader.exec_module(M)
|
|
|
|
def event(persona='ICE-P-ZY001', session='s1'):
|
|
return {"parent_persona_id":persona,"session_id":session,"channel_id":"ICE-CH-HB001","current_intent_sha256":"a"*64,"model_runtime":"codex"}
|
|
|
|
class Tests(unittest.TestCase):
|
|
def test_every_slot_is_model_backed_and_owned_by_parent(self):
|
|
with tempfile.TemporaryDirectory() as t, patch.object(M,'runtime_available',return_value=True):
|
|
v=M.activate(event(),Path(t)); self.assertEqual(len(v['team']),8)
|
|
self.assertTrue(all(x['model_cognition_required'] and x['controller']=='ICE-P-ZY001' for x in v['team']))
|
|
self.assertTrue(all(x['runtime_state']=='DORMANT_READY' and not x['model_context_loaded'] for x in v['team']))
|
|
self.assertEqual(v['active_agent_count'],0)
|
|
def test_persona_switch_rebinds_all_slots_and_invalidates_generation(self):
|
|
personas=iter(sorted(M.registered_personas()))
|
|
first=next(personas); second=next(x for x in personas if x!=first)
|
|
with tempfile.TemporaryDirectory() as t, patch.object(M,'runtime_available',return_value=True):
|
|
state=Path(t); a=M.activate(event(first,'s1'),state); b=M.activate(event(second,'s2'),state)
|
|
self.assertEqual(b['generation'],a['generation']+1); self.assertTrue(b['prior_generation_invalidated'])
|
|
self.assertTrue(all(x['parent_persona_id']==second and second in x['child_agent_id'] for x in b['team']))
|
|
with self.assertRaisesRegex(M.CommandError,'STALE_TEAM_GENERATION'): M.dispatch({"generation":a['generation'],"child_agent_id":a['team'][0]['child_agent_id'],"task":"x"},state)
|
|
def test_unknown_persona_and_missing_provider_fail_closed(self):
|
|
with tempfile.TemporaryDirectory() as t:
|
|
with self.assertRaisesRegex(M.CommandError,'NOT_REGISTERED'): M.activate(event('PER-NOT-REAL'),Path(t))
|
|
with patch.object(M,'runtime_available',return_value=False), self.assertRaisesRegex(M.CommandError,'PROVIDER_UNAVAILABLE'): M.activate(event(),Path(t))
|
|
def test_dispatch_activates_only_selected_slot_then_releases_it(self):
|
|
with tempfile.TemporaryDirectory() as t, patch.object(M,'runtime_available',return_value=True):
|
|
state=Path(t); current=M.activate(event(),state); selected=current['team'][0]
|
|
def fake(*args,**kwargs):
|
|
during=M.load(state/'CURRENT.json')
|
|
self.assertEqual(during['active_agent_count'],1)
|
|
self.assertEqual([x['slot_id'] for x in during['team'] if x['runtime_state']=='ACTIVE'],[selected['slot_id']])
|
|
return __import__('subprocess').CompletedProcess(args[0],0,'{}','')
|
|
with patch.object(M.subprocess,'run',side_effect=fake):
|
|
receipt=M.dispatch({'generation':current['generation'],'child_agent_id':selected['child_agent_id'],'task':'read only'},state)
|
|
after=M.load(state/'CURRENT.json')
|
|
self.assertEqual(receipt['final_slot_state'],'DORMANT_READY'); self.assertEqual(after['active_agent_count'],0)
|
|
def test_failed_limb_emits_symptom_for_treatment_team(self):
|
|
with tempfile.TemporaryDirectory() as t, patch.object(M,'runtime_available',return_value=True):
|
|
state=Path(t); current=M.activate(event(),state); selected=current['team'][0]
|
|
failed=__import__('subprocess').CompletedProcess([],1,'','broken')
|
|
with patch.object(M.subprocess,'run',return_value=failed):
|
|
receipt=M.dispatch({'generation':current['generation'],'child_agent_id':selected['child_agent_id'],'task':'read only'},state)
|
|
after=M.load(state/'CURRENT.json')
|
|
self.assertEqual(receipt['final_slot_state'],'SYMPTOM_REPORTED')
|
|
self.assertEqual(after['treatment_signal']['treatment_router'],'TCS-AGENT-TREATMENT-ROUTER-001')
|
|
|
|
if __name__=='__main__': unittest.main()
|