From 2964332f971aff2ee64175049fe6d0b225f98047 Mon Sep 17 00:00:00 2001
From: Ian Booth <ian.booth@canonical.com>
Date: Fri, 22 Mar 2024 13:16:47 +1000
Subject: [PATCH] Only run k8s smoke tests on 4.0 or later

---
 jobs/ci-run/integration/gen/test-smoke_k8s.yml | 16 +++++++++++-----
 tools/gen-wire-tests/juju.config               |  2 ++
 tools/gen-wire-tests/main.go                   | 11 +++++++++--
 3 files changed, 22 insertions(+), 7 deletions(-)

diff --git a/jobs/ci-run/integration/gen/test-smoke_k8s.yml b/jobs/ci-run/integration/gen/test-smoke_k8s.yml
index abf398d..814c913 100644
--- a/jobs/ci-run/integration/gen/test-smoke_k8s.yml
+++ b/jobs/ci-run/integration/gen/test-smoke_k8s.yml
@@ -101,10 +101,16 @@
             OPERATOR_IMAGE_ACCOUNT=${PARAM_OPERATOR_IMAGE_ACCOUNT}
       - wait-for-cloud-init
       - prepare-integration-test
-      - run-integration-test-microk8s:
-            test_name: 'smoke_k8s'
-            setup_steps: ''
-            task_name: ''
-            skip_tasks: ''
+      - conditional-step:
+          condition-kind: regex-match
+          regex: "^[5-9].*|^4\\.([0-9]|\\d{2,})(\\.|-).*"
+          label: "${JUJU_VERSION}"
+          on-evaluation-failure: "dont-run"
+          steps:
+            - run-integration-test-microk8s:
+                  test_name: 'smoke_k8s'
+                  setup_steps: ''
+                  task_name: ''
+                  skip_tasks: ''
     publishers:
       - integration-artifacts
diff --git a/tools/gen-wire-tests/juju.config b/tools/gen-wire-tests/juju.config
index b9f8c93..068d2ff 100644
--- a/tools/gen-wire-tests/juju.config
+++ b/tools/gen-wire-tests/juju.config
@@ -28,6 +28,8 @@ folders:
       3.4
     test_actions_params:
       4.0
+    smoke_k8s-test_deploy:
+      4.0
   timeout:
     secrets_iaas:
       test_secrets_vault: 60
diff --git a/tools/gen-wire-tests/main.go b/tools/gen-wire-tests/main.go
index cffc490..fb2c688 100644
--- a/tools/gen-wire-tests/main.go
+++ b/tools/gen-wire-tests/main.go
@@ -278,6 +278,9 @@ func writeJobDefinitions(
 		if introduced, ok := config.Folders.Introduced[task]; ok {
 			minVersions[task] = minVersionRegex[introduced]
 		}
+		if introduced, ok := config.Folders.Introduced[suiteName+"-"+task]; ok {
+			minVersions[suiteName+"-"+task] = minVersionRegex[introduced]
+		}
 	}
 
 	if err := t.Execute(f, struct {
@@ -553,10 +556,14 @@ const Template = `
             OPERATOR_IMAGE_ACCOUNT=${PARAM_OPERATOR_IMAGE_ACCOUNT}
       - wait-for-cloud-init
       - prepare-integration-test
-{{- if index $.MinVersions $task_name }}
+{{- if or (index $.MinVersions $task_name) (index $.MinVersions (printf "%s-%s" $.SuiteName $task_name)) }}
+      {{- $cond := index $.MinVersions $task_name -}}
+      {{- if eq $cond "" }}
+        {{- $cond = index $.MinVersions (printf "%s-%s" $.SuiteName $task_name) -}}
+      {{- end }}
       - conditional-step:
           condition-kind: regex-match
-          regex: "{{ index $.MinVersions $task_name }}"
+          regex: "{{ $cond }}"
           label: "${JUJU_VERSION}"
           on-evaluation-failure: "dont-run"
           steps: