From a5d495eb6ddd4b93925c5dbac080047b259c40d8 Mon Sep 17 00:00:00 2001
From: Jigisha Patil <89548848+jigisha620@users.noreply.github.com>
Date: Fri, 12 Jul 2024 07:37:51 -0700
Subject: [PATCH] test: Fix time out in scale tests (#6451)

---
 test/suites/scale/deprovisioning_test.go | 10 ++++++++++
 test/suites/scale/provisioning_test.go   |  5 +++++
 2 files changed, 15 insertions(+)

diff --git a/test/suites/scale/deprovisioning_test.go b/test/suites/scale/deprovisioning_test.go
index c86f08d1c85f..438c0568ed9c 100644
--- a/test/suites/scale/deprovisioning_test.go
+++ b/test/suites/scale/deprovisioning_test.go
@@ -101,6 +101,11 @@ var _ = Describe("Deprovisioning", Label(debug.NoWatch), Label(debug.NoEvents),
 				},
 			},
 		}...)
+		nodePool.Spec.Disruption.Budgets = []corev1beta1.Budget{
+			{
+				Nodes: "70%",
+			},
+		}
 		deploymentOptions = test.DeploymentOptions{
 			PodOptions: test.PodOptions{
 				ResourceRequirements: v1.ResourceRequirements{
@@ -297,6 +302,11 @@ var _ = Describe("Deprovisioning", Label(debug.NoWatch), Label(debug.NoEvents),
 			env.MeasureDeprovisioningDurationFor(func() {
 				By("enabling deprovisioning across nodePools")
 				for _, p := range nodePoolMap {
+					p.Spec.Disruption.Budgets = []corev1beta1.Budget{
+						{
+							Nodes: "70%",
+						},
+					}
 					env.ExpectCreatedOrUpdated(p)
 				}
 				env.ExpectUpdated(driftNodeClass)
diff --git a/test/suites/scale/provisioning_test.go b/test/suites/scale/provisioning_test.go
index 44fea2a8d659..cffd85133962 100644
--- a/test/suites/scale/provisioning_test.go
+++ b/test/suites/scale/provisioning_test.go
@@ -48,6 +48,11 @@ var _ = Describe("Provisioning", Label(debug.NoWatch), Label(debug.NoEvents), fu
 		nodeClass = env.DefaultEC2NodeClass()
 		nodePool = env.DefaultNodePool(nodeClass)
 		nodePool.Spec.Limits = nil
+		nodePool.Spec.Disruption.Budgets = []corev1beta1.Budget{
+			{
+				Nodes: "70%",
+			},
+		}
 		test.ReplaceRequirements(nodePool, corev1beta1.NodeSelectorRequirementWithMinValues{
 			NodeSelectorRequirement: v1.NodeSelectorRequirement{
 				Key:      v1beta1.LabelInstanceHypervisor,