From d446f8d9ab7f10a708173affb783fe2110b88115 Mon Sep 17 00:00:00 2001
From: John Gardiner Myers <jgmyers@proofpoint.com>
Date: Sat, 17 Jul 2021 14:54:13 -0700
Subject: [PATCH] Accommodate older destination kops versions in upgrade-ab
 scenario

---
 tests/e2e/scenarios/upgrade-ab/run-test.sh | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/tests/e2e/scenarios/upgrade-ab/run-test.sh b/tests/e2e/scenarios/upgrade-ab/run-test.sh
index 5409d6b5ee..fd09acca96 100755
--- a/tests/e2e/scenarios/upgrade-ab/run-test.sh
+++ b/tests/e2e/scenarios/upgrade-ab/run-test.sh
@@ -52,7 +52,11 @@ KOPS_B=$(kops-download-from-base)
 
 KOPS="${KOPS_B}"
 
-"${KOPS_B}" edit cluster "${CLUSTER_NAME}" "--set=cluster.spec.kubernetesVersion=${K8S_VERSION_B}"
+if [[ "${KOPS_B}" =~ v1.2[01] ]]; then
+  "${KOPS_B}" set cluster "${CLUSTER_NAME}" "cluster.spec.kubernetesVersion=${K8S_VERSION_B}"
+else
+  "${KOPS_B}" edit cluster "${CLUSTER_NAME}" "--set=cluster.spec.kubernetesVersion=${K8S_VERSION_B}"
+fi
 
 "${KOPS_B}" update cluster
 "${KOPS_B}" update cluster --admin --yes