From 4928053cbc77b73a6e237fa1f51b4e73741f905b Mon Sep 17 00:00:00 2001
From: Anton Troshin <anton@diagrid.io>
Date: Wed, 19 Feb 2025 19:51:48 -0600
Subject: [PATCH] Fix scheduler pod count for 1.15 version when testing master
 and latest

Signed-off-by: Anton Troshin <anton@diagrid.io>
---
 tests/e2e/common/common.go | 31 +++++++++++++++++++------------
 1 file changed, 19 insertions(+), 12 deletions(-)

diff --git a/tests/e2e/common/common.go b/tests/e2e/common/common.go
index 9f755768..e609b0a6 100644
--- a/tests/e2e/common/common.go
+++ b/tests/e2e/common/common.go
@@ -47,17 +47,21 @@ const (
 	ClusterRoles
 	ClusterRoleBindings
 
-	numHAPodsWithScheduler    = 16
-	numHAPodsOld              = 13
-	numNonHAPodsWithScheduler = 6
-	numNonHAPodsOld           = 5
+	numHAPodsWithScheduler      = 16
+	numHAPodsOld                = 13
+	numNonHAPodsWithHAScheduler = 8
+	numNonHAPodsWithScheduler   = 6
+	numNonHAPodsOld             = 5
 
 	thirdPartyDevNamespace = "default"
 	devRedisReleaseName    = "dapr-dev-redis"
 	devZipkinReleaseName   = "dapr-dev-zipkin"
 )
 
-var VersionWithScheduler = semver.MustParse("1.14.0-rc.1")
+var (
+	VersionWithScheduler   = semver.MustParse("1.14.0-rc.1")
+	VersionWithHAScheduler = semver.MustParse("1.15.0-rc.1")
+)
 
 type VersionDetails struct {
 	RuntimeVersion       string
@@ -1232,23 +1236,26 @@ func getVersionedNumberOfPods(isHAEnabled bool, details VersionDetails) (int, er
 			return 0, err
 		}
 
-		if rv.LessThan(VersionWithScheduler) {
-			return numHAPodsOld, nil
+		if rv.GreaterThanEqual(VersionWithScheduler) {
+			return numHAPodsWithScheduler, nil
 		}
-		return numHAPodsWithScheduler, nil
+		return numHAPodsOld, nil
 	} else {
 		if details.UseDaprLatestVersion {
-			return numNonHAPodsWithScheduler, nil
+			return numNonHAPodsWithHAScheduler, nil
 		}
 		rv, err := semver.NewVersion(details.RuntimeVersion)
 		if err != nil {
 			return 0, err
 		}
 
-		if rv.LessThan(VersionWithScheduler) {
-			return numNonHAPodsOld, nil
+		if rv.GreaterThanEqual(VersionWithHAScheduler) {
+			return numNonHAPodsWithHAScheduler, nil
 		}
-		return numNonHAPodsWithScheduler, nil
+		if rv.GreaterThanEqual(VersionWithScheduler) {
+			return numNonHAPodsWithScheduler, nil
+		}
+		return numNonHAPodsOld, nil
 	}
 }