From cfd4ee1756f1ccf7e8b1805d3790a5cb94e1d9fe Mon Sep 17 00:00:00 2001
From: Juanadelacuesta <8647634+Juanadelacuesta@users.noreply.github.com>
Date: Fri, 14 Mar 2025 17:57:26 +0100
Subject: [PATCH] fix: add missing variables for drain module

---
 enos/enos-scenario-upgrade.hcl | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/enos/enos-scenario-upgrade.hcl b/enos/enos-scenario-upgrade.hcl
index 10de93b14..f3d776164 100644
--- a/enos/enos-scenario-upgrade.hcl
+++ b/enos/enos-scenario-upgrade.hcl
@@ -417,12 +417,21 @@ scenario "upgrade" {
   step "drain_client" {
     depends_on = [step.upgrade_second_client]
 
-    description = <<-EOF 
+    description = <<-EOF
     Selects one client to drain, waits for all allocs to be rescheduled and
     brings back the node eligibility
     EOF
 
     module = module.drain_client
+    variables {
+      # connecting to the Nomad API
+      nomad_addr     = step.provision_cluster.nomad_addr
+      ca_file        = step.provision_cluster.ca_file
+      cert_file      = step.provision_cluster.cert_file
+      key_file       = step.provision_cluster.key_file
+      nomad_token    = step.provision_cluster.nomad_token
+      nodes_to_drain = 1
+    }
   }
 
   step "upgrade_third_client" {