ed
8f11340b38
Per post_module_taxonomy_de_cruft_20260627 Phase 2 (FR7). Each
'from src.models import X' for a moved class is rewritten to
'from src.<destination> import X':
Ticket, Track, WorkerContext, TrackState, TrackMetadata,
ThinkingSegment, EMPTY_TRACK_STATE -> src.mma
ProjectContext, ProjectMeta, ProjectOutput, ProjectFiles,
ProjectScreenshots, ProjectDiscussion, EMPTY_PROJECT_CONTEXT -> src.project
FileItem, Preset, ContextPreset, ContextFileEntry,
NamedViewPreset -> src.project_files
Tool, ToolPreset -> src.tool_presets
BiasProfile -> src.tool_bias
TextEditorConfig, ExternalEditorConfig,
EMPTY_TEXT_EDITOR_CONFIG -> src.external_editor
Persona -> src.personas
WorkspaceProfile -> src.workspace_manager
MCPServerConfig, MCPConfiguration, VectorStoreConfig,
RAGConfig, load_mcp_config -> src.mcp_client
NOT touched (kept on src.models; Phase 3 or Phase 4 will move them):
GenerateRequest, ConfirmRequest, DEFAULT_TOOL_CATEGORIES, Metadata, PROVIDERS
Migration was performed by the one-time script
scripts/tier2/artifacts/post_module_taxonomy_de_cruft_20260627/migrate_imports.py
which uses a class-to-module map and re.sub() to rewrite each
'from src.models import X' line.
Total: 85 import lines rewritten across 71 files.
Note: this commit depends on the v2 SHIPPED work
(origin/tier2/module_taxonomy_refactor_20260627) being merged into
this branch NEXT. On master (without the v2 SHIPPED commits), the
destination modules do not exist and these imports would fail.
81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
from src.mma import Ticket
|
|
from src.dag_engine import TrackDAG, ExecutionEngine
|
|
|
|
def test_execution_engine_basic_flow():
|
|
t1 = Ticket(id="T1", description="desc", status="todo", assigned_to="worker1")
|
|
t2 = Ticket(id="T2", description="desc", status="todo", assigned_to="worker1", depends_on=["T1"])
|
|
dag = TrackDAG([t1, t2])
|
|
engine = ExecutionEngine(dag)
|
|
|
|
# 1. First tick
|
|
ready = engine.tick()
|
|
assert len(ready) == 1
|
|
assert ready[0].id == "T1"
|
|
assert ready[0].status == "todo" # Not auto-queued yet
|
|
|
|
# 2. Mark T1 in_progress
|
|
# update_task_status updates the underlying Ticket object.
|
|
engine.update_task_status("T1", "in_progress")
|
|
# tick() returns 'todo' tasks that are ready. T1 is in_progress, so it's not 'todo'.
|
|
ready = engine.tick()
|
|
assert len(ready) == 0
|
|
|
|
# 3. Mark T1 complete
|
|
engine.update_task_status("T1", "completed")
|
|
# Now T2 should be ready
|
|
ready = engine.tick()
|
|
assert len(ready) == 1
|
|
assert ready[0].id == "T2"
|
|
|
|
def test_execution_engine_update_nonexistent_task():
|
|
dag = TrackDAG([])
|
|
engine = ExecutionEngine(dag)
|
|
# Should not crash
|
|
engine.update_task_status("NONEXISTENT", "completed")
|
|
|
|
def test_execution_engine_status_persistence():
|
|
t1 = Ticket(id="T1", description="desc", status="todo", assigned_to="worker1")
|
|
dag = TrackDAG([t1])
|
|
engine = ExecutionEngine(dag)
|
|
engine.update_task_status("T1", "in_progress")
|
|
assert t1.status == "in_progress"
|
|
|
|
def test_execution_engine_auto_queue():
|
|
t1 = Ticket(id="T1", description="desc", status="todo", assigned_to="worker1")
|
|
t2 = Ticket(id="T2", description="desc", status="todo", assigned_to="worker1", depends_on=["T1"])
|
|
dag = TrackDAG([t1, t2])
|
|
engine = ExecutionEngine(dag, auto_queue=True)
|
|
|
|
# Tick should return T1
|
|
ready = engine.tick()
|
|
assert len(ready) == 1
|
|
assert ready[0].id == "T1"
|
|
|
|
# Mark T1 complete
|
|
engine.update_task_status("T1", "completed")
|
|
ready = engine.tick()
|
|
assert len(ready) == 1
|
|
assert ready[0].id == "T2"
|
|
|
|
def test_execution_engine_step_mode():
|
|
t1 = Ticket(id="T1", description="desc", status="todo", assigned_to="worker1", step_mode=True)
|
|
dag = TrackDAG([t1])
|
|
engine = ExecutionEngine(dag, auto_queue=True)
|
|
|
|
# Even with auto_queue, step_mode task requires manual approval
|
|
ready = engine.tick()
|
|
assert len(ready) == 1
|
|
assert ready[0].id == "T1"
|
|
assert ready[0].status == "todo"
|
|
|
|
# Manual approval
|
|
engine.approve_task("T1")
|
|
assert t1.status == "in_progress"
|
|
|
|
def test_execution_engine_approve_task():
|
|
t1 = Ticket(id="T1", description="desc", status="todo", assigned_to="worker1")
|
|
dag = TrackDAG([t1])
|
|
engine = ExecutionEngine(dag)
|
|
engine.approve_task("T1")
|
|
assert t1.status == "in_progress"
|