ed/manual_slop
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

WIP: PAIN

Browse Source
This commit is contained in:
Edward R. Gonzalez
2026-03-05 14:24:03 -05:00
parent e81843b11b
commit 0e3b479bd6
27 changed files with 684 additions and 772 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
tests/test_execution_engine.py
View File

@@ -3,8 +3,8 @@ from src.models import Ticket
from src.dag_engine import TrackDAG, ExecutionEngine
def test_execution_engine_basic_flow():
t1 = Ticket(id="T1", description="desc", status="todo")
t2 = Ticket(id="T2", description="desc", status="todo", depends_on=["T1"])
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)
@@ -15,13 +15,15 @@ def test_execution_engine_basic_flow():
assert ready[0].status == "todo" # Not auto-queued yet
# 2. Mark T1 in_progress
ready[0].status = "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) == 1
assert ready[0].id == "T1"
assert len(ready) == 0
# 3. Mark T1 complete
ready[0].status = "completed"
engine.update_task_status("T1", "completed")
# Now T2 should be ready
ready = engine.tick()
assert len(ready) == 1
assert ready[0].id == "T2"
@@ -33,15 +35,15 @@ def test_execution_engine_update_nonexistent_task():
engine.update_task_status("NONEXISTENT", "completed")
def test_execution_engine_status_persistence():
t1 = Ticket(id="T1", description="desc", status="todo")
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")
t2 = Ticket(id="T2", description="desc", status="todo", depends_on=["T1"])
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)
@@ -51,13 +53,13 @@ def test_execution_engine_auto_queue():
assert ready[0].id == "T1"
# Mark T1 complete
t1.status = "completed"
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", step_mode=True)
t1 = Ticket(id="T1", description="desc", status="todo", assigned_to="worker1", step_mode=True)
dag = TrackDAG([t1])
engine = ExecutionEngine(dag, auto_queue=True)
@@ -72,7 +74,7 @@ def test_execution_engine_step_mode():
assert t1.status == "in_progress"
def test_execution_engine_approve_task():
t1 = Ticket(id="T1", description="desc", status="todo")
t1 = Ticket(id="T1", description="desc", status="todo", assigned_to="worker1")
dag = TrackDAG([t1])
engine = ExecutionEngine(dag)
engine.approve_task("T1")