From 61bbff9c249dc3f239650decf4079b6783ad4222 Mon Sep 17 00:00:00 2001
From: Tim Gross <tgross@hashicorp.com>
Date: Tue, 11 Mar 2025 08:48:40 -0400
Subject: [PATCH] upgrade testing: Variables, Workload Identity, and Task API
 (#25229)

Add an upgrade test workload for that continuously writes to a Nomad
Variable. In order to run this workload, we'll need to deploy a
Workload-Associated ACL policy. So this extends the `run_workloads` module to
allow for a "pre script" to be run before a given job is deployed. We can use
that as a model for other test workloads.

Ref: https://hashicorp.atlassian.net/browse/NET-12217
---
 enos/enos-scenario-upgrade.hcl                |   7 ++
 .../run_workloads/jobs/writes-vars.nomad.hcl  | 100 ++++++++++++++++++
 .../scripts/configure-variables-acls.sh       |  17 +++
 3 files changed, 124 insertions(+)
 create mode 100644 enos/modules/run_workloads/jobs/writes-vars.nomad.hcl
 create mode 100755 enos/modules/run_workloads/scripts/configure-variables-acls.sh

diff --git a/enos/enos-scenario-upgrade.hcl b/enos/enos-scenario-upgrade.hcl
index bf74c0345..580406600 100644
--- a/enos/enos-scenario-upgrade.hcl
+++ b/enos/enos-scenario-upgrade.hcl
@@ -176,6 +176,13 @@ scenario "upgrade" {
           pre_script  = "scripts/create-consul-intention.sh"
         }
 
+        writes_variable = {
+          job_spec    = "jobs/writes-vars.nomad.hcl"
+          alloc_count = 1
+          type        = "service"
+          pre_script  = "scripts/configure-variables-acls.sh"
+        }
+
       }
     }
 
diff --git a/enos/modules/run_workloads/jobs/writes-vars.nomad.hcl b/enos/modules/run_workloads/jobs/writes-vars.nomad.hcl
new file mode 100644
index 000000000..4a0026af3
--- /dev/null
+++ b/enos/modules/run_workloads/jobs/writes-vars.nomad.hcl
@@ -0,0 +1,100 @@
+# Copyright (c) HashiCorp, Inc.
+# SPDX-License-Identifier: BUSL-1.1
+
+variable "alloc_count" {
+  type    = number
+  default = 1
+}
+
+# a job that continuously writes a counter value to a Nomad Variable and reads
+# it back out. This exercises Workload Identity and the Task API.
+job "writes-vars" {
+
+  group "group" {
+
+    count = var.alloc_count
+
+    # need a service port so we can have a health check, but it's not used here
+    network {
+      port "web" {
+        to = 8001
+      }
+    }
+
+    service {
+      provider = "consul"
+      name     = "writes-vars-checker"
+      port     = "web"
+      task     = "task"
+
+      check {
+        type     = "script"
+        interval = "10s"
+        timeout  = "1s"
+        command  = "/bin/sh"
+        args     = ["/local/read-script.sh"]
+
+        # this check will read from the Task API, so we need to ensure that we
+        # can tolerate the listener going away during client upgrades
+        check_restart {
+          limit = 10
+        }
+      }
+    }
+
+
+    task "task" {
+      driver = "docker"
+
+      config {
+        image   = "curlimages/curl:latest"
+        command = "/bin/sh"
+        args    = ["/local/write-script.sh"]
+      }
+
+      template {
+        destination = "local/write-script.sh"
+        data        = <<EOT
+count=0
+while :
+do
+  count=$(( count + 1 ))
+  body=$(printf '{"Items": {"count": "%d" }}' $count)
+  echo "sending ==> $body"
+  curl --unix-socket "${NOMAD_SECRETS_DIR}/api.sock" \
+       -H "Authorization: Bearer ${NOMAD_TOKEN}" \
+       -verbose \
+       --fail-with-body \
+       -d "$body" \
+       http://localhost/v1/var/nomad/jobs/writes-vars
+  sleep 1
+done
+
+EOT
+
+      }
+
+      template {
+        destination = "local/read-script.sh"
+        data        = <<EOT
+curl --unix-socket "${NOMAD_SECRETS_DIR}/api.sock" \
+     -H "Authorization: Bearer ${NOMAD_TOKEN}" \
+     -verbose \
+     http://localhost/v1/var/nomad/jobs/writes-vars
+
+EOT
+
+      }
+
+      identity {
+        env = true
+      }
+
+      resources {
+        cpu    = 100
+        memory = 64
+      }
+    }
+
+  }
+}
diff --git a/enos/modules/run_workloads/scripts/configure-variables-acls.sh b/enos/modules/run_workloads/scripts/configure-variables-acls.sh
new file mode 100755
index 000000000..74fdb88f6
--- /dev/null
+++ b/enos/modules/run_workloads/scripts/configure-variables-acls.sh
@@ -0,0 +1,17 @@
+#!/usr/bin/env bash
+# Copyright (c) HashiCorp, Inc.
+# SPDX-License-Identifier: BUSL-1.1
+
+set -euo pipefail
+
+nomad acl policy apply \
+   -namespace default -job writes-vars \
+   writes-vars-policy - <<EOF
+namespace "default" {
+  variables {
+    path "nomad/jobs/writes-vars" {
+      capabilities = ["write", "read"]
+    }
+  }
+}
+EOF