Files
Yiqun/worker/orca-agent/tests/steps.py
2025-04-24 22:02:03 -03:00

67 lines
1.6 KiB
Python

"""Test step definitions."""
from prometheus_test import TestStep
from functools import partial
from .stages import (
worker_fetch,
worker_task,
worker_pr,
worker_submission,
worker_check,
worker_audit,
update_audit,
)
steps = [
TestStep(
name="worker_fetch",
description="Fetch worker task",
prepare=worker_fetch.prepare,
execute=worker_fetch.execute,
worker="worker1",
),
TestStep(
name="worker_task",
description="Execute worker task",
prepare=worker_task.prepare,
execute=worker_task.execute,
worker="worker1",
),
TestStep(
name="worker_pr",
description="Add PR to worker",
prepare=worker_pr.prepare,
execute=worker_pr.execute,
worker="worker1",
),
TestStep(
name="worker_submission",
description="Submit worker task",
prepare=worker_submission.prepare,
execute=worker_submission.execute,
worker="worker1",
),
TestStep(
name="worker_check",
description="Check worker task",
prepare=worker_check.prepare,
execute=worker_check.execute,
worker="worker1",
),
TestStep(
name="worker_audit",
description="Worker2 audits Worker1",
prepare=partial(worker_audit.prepare, target_name="worker1"),
execute=worker_audit.execute,
worker="worker2",
),
TestStep(
name="update_audit",
description="Update audit results",
prepare=partial(update_audit.prepare, role="worker"),
execute=update_audit.execute,
worker="worker2",
),
]