From 5e6911d4f8be12cf3b34c7b7a8dc16960ff5c189 Mon Sep 17 00:00:00 2001 From: Laura Abro Date: Thu, 24 Apr 2025 12:28:37 -0300 Subject: [PATCH] add missing steps --- worker/orca-agent/tests/steps.py | 24 ++++++++++++++++++++---- 1 file changed, 20 insertions(+), 4 deletions(-) diff --git a/worker/orca-agent/tests/steps.py b/worker/orca-agent/tests/steps.py index e669361..1cb9120 100644 --- a/worker/orca-agent/tests/steps.py +++ b/worker/orca-agent/tests/steps.py @@ -5,9 +5,11 @@ from functools import partial from .stages import ( worker_fetch, worker_task, + worker_pr, worker_submission, + worker_check, worker_audit, - audit_results, + update_audit, ) @@ -26,6 +28,13 @@ steps = [ 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", @@ -33,6 +42,13 @@ steps = [ 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", @@ -41,10 +57,10 @@ steps = [ worker="worker2", ), TestStep( - name="audit_results", + name="update_audit", description="Update audit results", - prepare=partial(audit_results.prepare, role="worker"), - execute=audit_results.execute, + prepare=partial(update_audit.prepare, role="worker"), + execute=update_audit.execute, worker="worker1", ), ]