From bae0eb91d448996a0f67f2c75ebf0c3f2891b2ae Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Maciej=20Skocze=C5=84?= <mskoczen@google.com>
Date: Tue, 1 Oct 2024 13:53:04 +0000
Subject: [PATCH] Use barrier instead of sleep when possible in scheduler_perf
 test cases

---
 .../scheduler_perf/config/performance-config.yaml   | 13 ++++---------
 1 file changed, 4 insertions(+), 9 deletions(-)

diff --git a/test/integration/scheduler_perf/config/performance-config.yaml b/test/integration/scheduler_perf/config/performance-config.yaml
index 892c2437ad4..dd67ca05251 100644
--- a/test/integration/scheduler_perf/config/performance-config.yaml
+++ b/test/integration/scheduler_perf/config/performance-config.yaml
@@ -1449,8 +1449,8 @@
     podTemplatePath: config/templates/gated-pod.yaml
     skipWaitToCompletion: true
     # Wait to make sure gated pods are enqueued in scheduler.
-  - opcode: sleep
-    durationParam: $sleepDuration
+  - opcode: barrier
+    stageRequirement: Attempted
     # Create pods that will be gradually deleted after being scheduled.
   - opcode: createPods
     countParam: $deletingPods
@@ -1471,7 +1471,6 @@
       gatedPods: 10
       deletingPods: 10
       measurePods: 10
-      sleepDuration: 1s
   - name: 1Node_10000GatedPods
     labels: [performance, short]
     threshold: 130
@@ -1479,7 +1478,6 @@
       gatedPods: 10000
       deletingPods: 20000
       measurePods: 20000
-      sleepDuration: 5s
 
 - name: SchedulingGatedPodsWithPodAffinityImpactForThroughput
   defaultPodTemplatePath: config/templates/pod-with-label.yaml
@@ -1491,9 +1489,8 @@
     countParam: $gatedPods
     podTemplatePath: config/templates/gated-pod-with-pod-affinity.yaml
     skipWaitToCompletion: true
-  - opcode: sleep
-    # To produce a stable scheduler_perf result, here we make sure all gated Pods are stored in the scheduling queue.
-    durationParam: $sleepDuration
+  - opcode: barrier
+    stageRequirement: Attempted
   - opcode: createPods
     # The scheduling of those Pods will result in many cluster events (AssignedPodAdded)
     # and each of them will be processed by the scheduling queue.
@@ -1506,14 +1503,12 @@
     params:
       gatedPods: 10
       measurePods: 10
-      sleepDuration: 1s
   - name: 1Node_10000GatedPods
     labels: [performance, short]
     threshold: 110
     params:
       gatedPods: 10000
       measurePods: 20000
-      sleepDuration: 5s
 
 # This test case simulates the scheduling when pods selected to schedule have deletionTimestamp set.
 # There was a memory leak related to this path of code fixed in: