From 0cd06ccaf128df7341eb3cd71e554aa975f8da3e Mon Sep 17 00:00:00 2001
From: Nishchay <awesomenix@users.noreply.github.com>
Date: Tue, 6 Jan 2026 10:30:22 -0800
Subject: [PATCH] fix: oops! accidently disabled running test pod

---
 e2e/README.md     | 19 +++++++++----------
 e2e/validation.go |  4 +++-
 2 files changed, 12 insertions(+), 11 deletions(-)

diff --git a/e2e/README.md b/e2e/README.md
index 816bb5cd064..925078ff172 100644
--- a/e2e/README.md
+++ b/e2e/README.md
@@ -43,16 +43,15 @@ sequenceDiagram
     AgentBakerCode-->>-E2E: VM Configuration
     E2E->>+ARM: Create VM using fetched VM Config in cluster network
     ARM-->>-E2E: VM instance
-    E2E->>+KubeAPI: Create test Pod
-    KubeAPI->>+TestPod: Perform healthcheck
-    TestPod-->>-KubeAPI: Healthcheck OK
-    KubeAPI-->>-E2E: Test Pod ready
-    E2E->>+KubeAPI: Execute test validators
-    KubeAPI->>+DebugPod: Execute test validator
-    DebugPod->>+VM: Execute test validator
-    VM-->>-DebugPod: Test results
-    DebugPod-->>-KubeAPI: Test results
-    KubeAPI-->>-E2E: Final results
+    E2E->>+Bastion: Create SSH Tunnel
+    Bastion->>+VM: Forward SSH Connection
+    E2E->>VM: Healthcheck via SSH Tunnel
+    VM-->>E2E: Healthcheck OK
+    E2E->>+KubeAPI: Verify Node Ready
+    KubeAPI-->>-E2E: Node Ready
+    E2E->>VM: Execute test validators via SSH Tunnel
+    VM-->>-E2E: Test results
+    Bastion-->>-E2E: Close SSH Tunnel
 ```
 
 ## Running Locally
diff --git a/e2e/validation.go b/e2e/validation.go
index aa98bf4e40d..523b8993066 100644
--- a/e2e/validation.go
+++ b/e2e/validation.go
@@ -51,12 +51,14 @@ func validatePodRunning(ctx context.Context, s *Scenario, pod *corev1.Pod) error
 
 func ValidatePodRunningWithRetry(ctx context.Context, s *Scenario, pod *corev1.Pod, maxRetries int) {
 	var err error
-	for i := 0; i < maxRetries && err != nil; i++ {
+	for i := range maxRetries {
 		err = validatePodRunning(ctx, s, pod)
 		if err != nil {
 			time.Sleep(1 * time.Second)
 			s.T.Logf("retrying pod %q validation (%d/%d)", pod.Name, i+1, maxRetries)
+			continue
 		}
+		break
 	}
 	require.NoErrorf(s.T, err, "failed to validate pod running %q", pod.Name)
 }