Files
Yiqun/worker/orca-agent/tests/steps.py
2025-04-24 11:52:26 -03:00

51 lines
1.2 KiB
Python

"""Test step definitions."""
from prometheus_test import TestStep
from functools import partial
from .stages import (
worker_fetch,
worker_task,
worker_submission,
worker_audit,
audit_results,
)
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_submission",
description="Submit worker task",
prepare=worker_submission.prepare,
execute=worker_submission.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="audit_results",
description="Update audit results",
prepare=partial(audit_results.prepare, role="worker"),
execute=audit_results.execute,
worker="worker1",
),
]