From 8dfeed77cd088d1996170ff16852f5ed1714b4ca Mon Sep 17 00:00:00 2001
From: Wainer dos Santos Moschetta <wainersm@redhat.com>
Date: Wed, 2 Jul 2025 17:22:13 -0300
Subject: [PATCH] tests/k8s: add handler for Job in set_node()

Set the node in the spec template of a Job manifest, allowing to use
set_node() on tests like k8s-parallel.bats

Signed-off-by: Wainer dos Santos Moschetta <wainersm@redhat.com>
---
 tests/integration/kubernetes/lib.sh | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/tests/integration/kubernetes/lib.sh b/tests/integration/kubernetes/lib.sh
index bfc4c3c3d9..c726a0c14c 100644
--- a/tests/integration/kubernetes/lib.sh
+++ b/tests/integration/kubernetes/lib.sh
@@ -340,10 +340,19 @@ set_container_command() {
 set_node() {
 	local yaml="$1"
 	local node="$2"
+	local kind
+	local spec
 	[ -n "$node" ] || return 1
 
+	kind="$(yq -r '.kind' "${yaml}")"
+	if [[ "${kind}" = "Job" ]]; then
+		spec=".spec.template.spec.nodeName"
+	else
+		spec=".spec.nodeName"
+	fi
+
   yq -i \
-    ".spec.nodeName = \"$node\"" \
+    "${spec} = \"$node\"" \
     "${yaml}"
 }