From 0e001ab1a5f1522c2d73ffcab35bf851405379fe Mon Sep 17 00:00:00 2001
From: Hemant Kumar <hekumar@redhat.com>
Date: Wed, 11 Mar 2026 08:34:29 -0400
Subject: [PATCH] Fix e2e tests to handle explicit driver opt-in for scheduling

---
 .../csimock/csi_volume_limit_scaling.go       | 25 +++++++++++++------
 1 file changed, 17 insertions(+), 8 deletions(-)

diff --git a/test/e2e/storage/csimock/csi_volume_limit_scaling.go b/test/e2e/storage/csimock/csi_volume_limit_scaling.go
index 0ce6a0d42bb..8e55fa0cada 100644
--- a/test/e2e/storage/csimock/csi_volume_limit_scaling.go
+++ b/test/e2e/storage/csimock/csi_volume_limit_scaling.go
@@ -44,20 +44,27 @@ var _ = utils.SIGDescribe("CSI Mock VolumeLimitScaling scheduling", framework.Wi
 
 	f.Context("VolumeLimitScaling scheduling", func() {
 		tests := []struct {
-			name              string
-			csiDriverPresent  bool
-			expectSchedulable bool
+			name                          string
+			csiDriverPresent              bool
+			preventPodSchedulingIfMissing *bool
+			expectSchedulable             bool
 		}{
 			{
-				name:              "blocks scheduling when driver not installed and CSIDriver is present",
-				csiDriverPresent:  true,
-				expectSchedulable: false,
+				name:                          "blocks scheduling when driver not installed and CSIDriver opts in",
+				csiDriverPresent:              true,
+				preventPodSchedulingIfMissing: new(true),
+				expectSchedulable:             false,
 			},
 			{
 				name:              "allows scheduling when driver not installed and CSIDriver object is not present",
 				csiDriverPresent:  false,
 				expectSchedulable: true,
 			},
+			{
+				name:              "allows scheduling when driver not installed and CSIDriver does not opt in",
+				csiDriverPresent:  true,
+				expectSchedulable: true,
+			},
 		}
 
 		for _, t := range tests {
@@ -68,7 +75,9 @@ var _ = utils.SIGDescribe("CSI Mock VolumeLimitScaling scheduling", framework.Wi
 				fakeDriver := fmt.Sprintf("csi-mock-uninstalled-%s", f.Namespace.Name)
 				csiDriver := &storagev1.CSIDriver{
 					ObjectMeta: metav1.ObjectMeta{Name: fakeDriver},
-					Spec:       storagev1.CSIDriverSpec{},
+					Spec: storagev1.CSIDriverSpec{
+						PreventPodSchedulingIfMissing: tc.preventPodSchedulingIfMissing,
+					},
 				}
 
 				if tc.csiDriverPresent {
@@ -138,7 +147,7 @@ var _ = utils.SIGDescribe("CSI Mock VolumeLimitScaling scheduling", framework.Wi
 			}
 
 			// Compose It with feature tags
-			args := []interface{}{tc.name, testFunc}
+			args := []any{tc.name, testFunc}
 			framework.It(args...)
 		}
 	})