From 5ebc9033162f021b57bd0038efa57f6c1d9108d6 Mon Sep 17 00:00:00 2001
From: machichima <nary12321@gmail.com>
Date: Fri, 6 Mar 2026 13:21:50 +0800
Subject: [PATCH 1/2] feat: draft test scripts

Signed-off-by: machichima <nary12321@gmail.com>
---
 runs/test/scripts/get_action_details.sh   | 22 +++++++++++++
 runs/test/scripts/get_run_details.sh      | 18 ++++++++++
 runs/test/scripts/list_actions.sh         | 24 ++++++++++++++
 runs/test/scripts/list_runs.sh            | 23 +++++++++++++
 runs/test/scripts/watch_actions.sh        | 18 ++++++++++
 runs/test/scripts/watch_cluster_events.sh | 24 ++++++++++++++
 runs/test/scripts/watch_groups.sh         | 40 +++++++++++++++++++++++
 runs/test/scripts/watch_run_details.sh    | 18 ++++++++++
 runs/test/scripts/watch_runs.sh           | 25 ++++++++++++++
 9 files changed, 212 insertions(+)
 create mode 100755 runs/test/scripts/get_action_details.sh
 create mode 100755 runs/test/scripts/get_run_details.sh
 create mode 100755 runs/test/scripts/list_actions.sh
 create mode 100755 runs/test/scripts/list_runs.sh
 create mode 100755 runs/test/scripts/watch_actions.sh
 create mode 100755 runs/test/scripts/watch_cluster_events.sh
 create mode 100755 runs/test/scripts/watch_groups.sh
 create mode 100755 runs/test/scripts/watch_run_details.sh
 create mode 100755 runs/test/scripts/watch_runs.sh

diff --git a/runs/test/scripts/get_action_details.sh b/runs/test/scripts/get_action_details.sh
new file mode 100755
index 00000000000..8e531339ff6
--- /dev/null
+++ b/runs/test/scripts/get_action_details.sh
@@ -0,0 +1,22 @@
+#!/bin/bash
+
+ENDPOINT="${ENDPOINT:-http://localhost:8090}"
+ORG="${ORG:-testorg}"
+PROJECT="${PROJECT:-testproject}"
+DOMAIN="${DOMAIN:-development}"
+RUN_NAME="${RUN_NAME:?Error: RUN_NAME environment variable is required}"
+ACTION="${ACTION:-$RUN_NAME}"
+
+buf curl --schema . "$ENDPOINT/flyteidl2.workflow.RunService/GetActionDetails" --data @- <<EOF
+{
+    "action_id": {
+        "run": {
+            "org": "$ORG",
+            "project": "$PROJECT",
+            "domain": "$DOMAIN",
+            "name": "$RUN_NAME"
+        },
+        "name": "$ACTION"
+    }
+}
+EOF
diff --git a/runs/test/scripts/get_run_details.sh b/runs/test/scripts/get_run_details.sh
new file mode 100755
index 00000000000..f3f29368b41
--- /dev/null
+++ b/runs/test/scripts/get_run_details.sh
@@ -0,0 +1,18 @@
+#!/bin/bash
+
+ENDPOINT="${ENDPOINT:-http://localhost:8090}"
+ORG="${ORG:-testorg}"
+PROJECT="${PROJECT:-testproject}"
+DOMAIN="${DOMAIN:-development}"
+RUN_NAME="${RUN_NAME:?Error: RUN_NAME environment variable is required}"
+
+buf curl --schema . "$ENDPOINT/flyteidl2.workflow.RunService/GetRunDetails" --data @- <<EOF
+{
+    "run_id": {
+        "org": "$ORG",
+        "project": "$PROJECT",
+        "domain": "$DOMAIN",
+        "name": "$RUN_NAME"
+    }
+}
+EOF
diff --git a/runs/test/scripts/list_actions.sh b/runs/test/scripts/list_actions.sh
new file mode 100755
index 00000000000..2ce6b25b911
--- /dev/null
+++ b/runs/test/scripts/list_actions.sh
@@ -0,0 +1,24 @@
+#!/bin/bash
+
+ENDPOINT="${ENDPOINT:-http://localhost:8090}"
+ORG="${ORG:-testorg}"
+PROJECT="${PROJECT:-testproject}"
+DOMAIN="${DOMAIN:-development}"
+RUN_NAME="${RUN_NAME:?Error: RUN_NAME environment variable is required}"
+LIMIT="${LIMIT:-10}"
+TOKEN="${TOKEN:-}"
+
+buf curl --schema . "$ENDPOINT/flyteidl2.workflow.RunService/ListActions" --data @- <<EOF
+{
+    "request": {
+        "limit": $LIMIT,
+        "token": "$TOKEN"
+    },
+    "run_id": {
+        "org": "$ORG",
+        "project": "$PROJECT",
+        "domain": "$DOMAIN",
+        "name": "$RUN_NAME"
+    }
+}
+EOF
diff --git a/runs/test/scripts/list_runs.sh b/runs/test/scripts/list_runs.sh
new file mode 100755
index 00000000000..876e6eb49a8
--- /dev/null
+++ b/runs/test/scripts/list_runs.sh
@@ -0,0 +1,23 @@
+#!/bin/bash
+
+ENDPOINT="${ENDPOINT:-http://localhost:8090}"
+ORG="${ORG:-testorg}"
+PROJECT="${PROJECT:-testproject}"
+DOMAIN="${DOMAIN:-development}"
+LIMIT="${LIMIT:-10}"
+TOKEN="${TOKEN:-}"
+
+# List runs by project with basic pagination support
+buf curl --schema . "$ENDPOINT/flyteidl2.workflow.RunService/ListRuns" --data @- <<EOF
+{
+    "request": {
+        "limit": $LIMIT,
+        "token": "$TOKEN"
+    },
+    "project_id": {
+        "organization": "$ORG",
+        "name": "$PROJECT",
+        "domain": "$DOMAIN"
+    }
+}
+EOF
diff --git a/runs/test/scripts/watch_actions.sh b/runs/test/scripts/watch_actions.sh
new file mode 100755
index 00000000000..06f39ff7e20
--- /dev/null
+++ b/runs/test/scripts/watch_actions.sh
@@ -0,0 +1,18 @@
+#!/bin/bash
+
+ENDPOINT="${ENDPOINT:-http://localhost:8090}"
+ORG="${ORG:-testorg}"
+PROJECT="${PROJECT:-testproject}"
+DOMAIN="${DOMAIN:-development}"
+RUN_NAME="${RUN_NAME:?Error: RUN_NAME environment variable is required}"
+
+buf curl --schema . "$ENDPOINT/flyteidl2.workflow.RunService/WatchActions" --data @- <<EOF
+{
+    "run_id": {
+        "org": "$ORG",
+        "project": "$PROJECT",
+        "domain": "$DOMAIN",
+        "name": "$RUN_NAME"
+    }
+}
+EOF
diff --git a/runs/test/scripts/watch_cluster_events.sh b/runs/test/scripts/watch_cluster_events.sh
new file mode 100755
index 00000000000..961770013ce
--- /dev/null
+++ b/runs/test/scripts/watch_cluster_events.sh
@@ -0,0 +1,24 @@
+#!/bin/bash
+
+ENDPOINT="${ENDPOINT:-http://localhost:8090}"
+ORG="${ORG:-testorg}"
+PROJECT="${PROJECT:-testproject}"
+DOMAIN="${DOMAIN:-development}"
+RUN_NAME="${RUN_NAME:?Error: RUN_NAME environment variable is required}"
+ACTION="${ACTION:-$RUN_NAME}"
+ATTEMPT="${ATTEMPT:-1}"
+
+buf curl --schema . "$ENDPOINT/flyteidl2.workflow.RunService/WatchClusterEvents" --data @- <<EOF
+{
+    "id": {
+        "run": {
+            "org": "$ORG",
+            "project": "$PROJECT",
+            "domain": "$DOMAIN",
+            "name": "$RUN_NAME"
+        },
+        "name": "$ACTION"
+    },
+    "attempt": $ATTEMPT
+}
+EOF
diff --git a/runs/test/scripts/watch_groups.sh b/runs/test/scripts/watch_groups.sh
new file mode 100755
index 00000000000..e992a1f8588
--- /dev/null
+++ b/runs/test/scripts/watch_groups.sh
@@ -0,0 +1,40 @@
+#!/bin/bash
+
+ENDPOINT="${ENDPOINT:-http://localhost:8090}"
+ORG="${ORG:-testorg}"
+PROJECT="${PROJECT:-testproject}"
+DOMAIN="${DOMAIN:-development}"
+LIMIT="${LIMIT:-20}"
+START_DATE="${START_DATE:-2026-01-01T00:00:00Z}"
+END_DATE="${END_DATE:-}"
+
+if [ -n "$END_DATE" ]; then
+buf curl --schema . "$ENDPOINT/flyteidl2.workflow.RunService/WatchGroups" --data @- <<EOF
+{
+    "project_id": {
+        "organization": "$ORG",
+        "name": "$PROJECT",
+        "domain": "$DOMAIN"
+    },
+    "start_date": "$START_DATE",
+    "end_date": "$END_DATE",
+    "request": {
+        "limit": $LIMIT
+    }
+}
+EOF
+else
+buf curl --schema . "$ENDPOINT/flyteidl2.workflow.RunService/WatchGroups" --data @- <<EOF
+{
+    "project_id": {
+        "organization": "$ORG",
+        "name": "$PROJECT",
+        "domain": "$DOMAIN"
+    },
+    "start_date": "$START_DATE",
+    "request": {
+        "limit": $LIMIT
+    }
+}
+EOF
+fi
diff --git a/runs/test/scripts/watch_run_details.sh b/runs/test/scripts/watch_run_details.sh
new file mode 100755
index 00000000000..5adfabcc999
--- /dev/null
+++ b/runs/test/scripts/watch_run_details.sh
@@ -0,0 +1,18 @@
+#!/bin/bash
+
+ENDPOINT="${ENDPOINT:-http://localhost:8090}"
+ORG="${ORG:-testorg}"
+PROJECT="${PROJECT:-testproject}"
+DOMAIN="${DOMAIN:-development}"
+RUN_NAME="${RUN_NAME:?Error: RUN_NAME environment variable is required}"
+
+buf curl --schema . "$ENDPOINT/flyteidl2.workflow.RunService/WatchRunDetails" --data @- <<EOF
+{
+    "run_id": {
+        "org": "$ORG",
+        "project": "$PROJECT",
+        "domain": "$DOMAIN",
+        "name": "$RUN_NAME"
+    }
+}
+EOF
diff --git a/runs/test/scripts/watch_runs.sh b/runs/test/scripts/watch_runs.sh
new file mode 100755
index 00000000000..36597c1a584
--- /dev/null
+++ b/runs/test/scripts/watch_runs.sh
@@ -0,0 +1,25 @@
+#!/bin/bash
+
+ENDPOINT="${ENDPOINT:-http://localhost:8090}"
+ORG="${ORG:-testorg}"
+PROJECT="${PROJECT:-testproject}"
+DOMAIN="${DOMAIN:-development}"
+SCOPE="${SCOPE:-project}" # project|org
+
+if [ "$SCOPE" = "org" ]; then
+buf curl --schema . "$ENDPOINT/flyteidl2.workflow.RunService/WatchRuns" --data @- <<EOF
+{
+    "org": "$ORG"
+}
+EOF
+else
+buf curl --schema . "$ENDPOINT/flyteidl2.workflow.RunService/WatchRuns" --data @- <<EOF
+{
+    "project_id": {
+        "organization": "$ORG",
+        "name": "$PROJECT",
+        "domain": "$DOMAIN"
+    }
+}
+EOF
+fi

From 665badeddfc8881a9d41f4888f9803b2cdff9781 Mon Sep 17 00:00:00 2001
From: machichima <nary12321@gmail.com>
Date: Fri, 6 Mar 2026 13:51:44 +0800
Subject: [PATCH 2/2] feat: add abort script

Signed-off-by: machichima <nary12321@gmail.com>
---
 runs/test/scripts/abort_action.sh | 24 ++++++++++++++++++++++++
 runs/test/scripts/abort_run.sh    | 20 ++++++++++++++++++++
 2 files changed, 44 insertions(+)
 create mode 100755 runs/test/scripts/abort_action.sh
 create mode 100755 runs/test/scripts/abort_run.sh

diff --git a/runs/test/scripts/abort_action.sh b/runs/test/scripts/abort_action.sh
new file mode 100755
index 00000000000..a4ae7c85249
--- /dev/null
+++ b/runs/test/scripts/abort_action.sh
@@ -0,0 +1,24 @@
+#!/bin/bash
+
+ENDPOINT="${ENDPOINT:-http://localhost:8090}"
+ORG="${ORG:-testorg}"
+PROJECT="${PROJECT:-testproject}"
+DOMAIN="${DOMAIN:-development}"
+RUN_NAME="${RUN_NAME:?Error: RUN_NAME environment variable is required}"
+ACTION="${ACTION:-$RUN_NAME}"
+REASON="${REASON:-User requested abort}"
+
+buf curl --schema . "$ENDPOINT/flyteidl2.workflow.RunService/AbortAction" --data @- <<EOF
+{
+    "action_id": {
+        "run": {
+            "org": "$ORG",
+            "project": "$PROJECT",
+            "domain": "$DOMAIN",
+            "name": "$RUN_NAME"
+        },
+        "name": "$ACTION"
+    },
+    "reason": "$REASON"
+}
+EOF
diff --git a/runs/test/scripts/abort_run.sh b/runs/test/scripts/abort_run.sh
new file mode 100755
index 00000000000..1715a402c86
--- /dev/null
+++ b/runs/test/scripts/abort_run.sh
@@ -0,0 +1,20 @@
+#!/bin/bash
+
+ENDPOINT="${ENDPOINT:-http://localhost:8090}"
+ORG="${ORG:-testorg}"
+PROJECT="${PROJECT:-testproject}"
+DOMAIN="${DOMAIN:-development}"
+RUN_NAME="${RUN_NAME:?Error: RUN_NAME environment variable is required}"
+REASON="${REASON:-User requested abort}"
+
+buf curl --schema . "$ENDPOINT/flyteidl2.workflow.RunService/AbortRun" --data @- <<EOF
+{
+    "run_id": {
+        "org": "$ORG",
+        "project": "$PROJECT",
+        "domain": "$DOMAIN",
+        "name": "$RUN_NAME"
+    },
+    "reason": "$REASON"
+}
+EOF