Herman/Yiqun
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

use new version of test framework with updated worker management

Browse Source
This commit is contained in:
Laura Abro
2025-04-30 10:04:49 -03:00
parent b1c53693a5
commit aa936faf92
6 changed files with 32 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
worker/orca-agent/tests/stages/worker_audit.py
View File

@ -23,8 +23,8 @@ def prepare(runner, worker, target_name):
# "taskId": runner.get("task_id"),
# "roundNumber": runner.get("current_round"),
# "prUrl": pr_url,
# "stakingKey": worker.staking_public_key,
# "pubKey": worker.public_key,
# "stakingKey": worker.get_key("staking_public"),
# "pubKey": worker.get_key("main_public"),
# }
# Structure the payload according to what the server expects
@ -47,13 +47,15 @@ def prepare(runner, worker, target_name):
# "prUrl": pr_url,
# "repoOwner": submission_data.get("repoOwner"),
# "repoName": submission_data.get("repoName"),
# "githubUsername": worker.env.get("GITHUB_USERNAME"),
# "stakingKey": worker.staking_public_key,
# "pubKey": worker.public_key,
# "githubUsername": worker.get_env("GITHUB_USERNAME"),
# "stakingKey": worker.get_key("staking_public"),
# "pubKey": worker.get_key("main_public"),
# "stakingSignature": create_signature(
# worker.staking_signing_key, auditor_payload
# worker.get_key("staking_signing"), auditor_payload
# ),
# "publicSignature": create_signature(
# worker.get_key("main_signing"), auditor_payload
# ),
# "publicSignature": create_signature(worker.public_signing_key, auditor_payload),
# }
return {"submission": submission_data}

4
worker/orca-agent/tests/stages/worker_check.py
View File

@ -11,9 +11,9 @@ def prepare(runner, worker):
return None
return {
"stakingKey": worker.staking_public_key,
"stakingKey": worker.get_key("staking_public"),
"roundNumber": runner.get("current_round"),
"githubUsername": worker.env.get("GITHUB_USERNAME"),
"githubUsername": worker.get_env("GITHUB_USERNAME"),
"prUrl": pr_url,
}

16
worker/orca-agent/tests/stages/worker_fetch.py
View File

@ -11,18 +11,20 @@ def prepare(runner, worker):
"taskId": runner.get("task_id"),
"roundNumber": runner.get("current_round"),
"action": "fetch-todo",
"githubUsername": worker.env.get("GITHUB_USERNAME"),
"stakingKey": worker.staking_public_key,
"pubKey": worker.public_key,
"githubUsername": worker.get_env("GITHUB_USERNAME"),
"stakingKey": worker.get_key("staking_public"),
"pubKey": worker.get_key("main_public"),
}
return {
"taskId": runner.get("task_id"),
"roundNumber": runner.get("current_round"),
"stakingKey": worker.staking_public_key,
"pubKey": worker.public_key,
"stakingSignature": create_signature(worker.staking_signing_key, payload),
"publicSignature": create_signature(worker.public_signing_key, payload),
"stakingKey": worker.get_key("staking_public"),
"pubKey": worker.get_key("main_public"),
"stakingSignature": create_signature(
worker.get_key("staking_signing"), payload
),
"publicSignature": create_signature(worker.get_key("main_signing"), payload),
}

8
worker/orca-agent/tests/stages/worker_pr.py
View File

@ -13,12 +13,12 @@ def prepare(runner, worker):
"action": "add-todo-pr",
"roundNumber": runner.get("current_round"),
"prUrl": pr_url,
"stakingKey": worker.staking_public_key,
"pubKey": worker.public_key,
"stakingKey": worker.get_key("staking_public"),
"pubKey": worker.get_key("main_public"),
}
return {
"signature": create_signature(worker.staking_signing_key, payload),
"stakingKey": worker.staking_public_key,
"signature": create_signature(worker.get_key("staking_signing"), payload),
"stakingKey": worker.get_key("staking_public"),
}

12
worker/orca-agent/tests/stages/worker_submission.py
View File

@ -21,17 +21,19 @@ def prepare(runner, worker):
submitter_payload = {
"taskId": runner.get("task_id"),
"roundNumber": runner.get("current_round"),
"stakingKey": worker.staking_public_key,
"pubKey": worker.public_key,
"stakingKey": worker.get_key("staking_public"),
"pubKey": worker.get_key("main_public"),
"action": "audit",
**submission_data,
}
return {
**submission_data,
"signature": create_signature(worker.staking_signing_key, submitter_payload),
"stakingKey": worker.staking_public_key,
"pubKey": worker.public_key,
"signature": create_signature(
worker.get_key("staking_signing"), submitter_payload
),
"stakingKey": worker.get_key("staking_public"),
"pubKey": worker.get_key("main_public"),
}