diff --git a/tests/suite/scale_test.go b/tests/suite/scale_test.go index 6d18986352..80a3a010a7 100644 --- a/tests/suite/scale_test.go +++ b/tests/suite/scale_test.go @@ -50,13 +50,16 @@ var _ = Describe("Scale test", Ordered, Label("nfr", "scale"), func() { ngfPodName string promInstance framework.PrometheusInstance promPortForwardStopCh = make(chan struct{}) + + upstreamServerCount int32 ) const ( - httpListenerCount = 64 - httpsListenerCount = 64 - httpRouteCount = 1000 - upstreamServerCount = 648 + httpListenerCount = 64 + httpsListenerCount = 64 + httpRouteCount = 1000 + ossUpstreamServerCount = 648 + plusUpstreamServerCount = 556 ) BeforeAll(func() { @@ -85,6 +88,12 @@ var _ = Describe("Scale test", Ordered, Label("nfr", "scale"), func() { if !clusterInfo.IsGKE { Expect(promInstance.PortForward(k8sConfig, promPortForwardStopCh)).To(Succeed()) } + + if *plusEnabled { + upstreamServerCount = plusUpstreamServerCount + } else { + upstreamServerCount = ossUpstreamServerCount + } }) BeforeEach(func() { @@ -563,7 +572,10 @@ The logs are attached only if there are errors. ) }) - It(fmt.Sprintf("scales upstream servers to %d", upstreamServerCount), func() { + It(fmt.Sprintf("scales upstream servers to %d for OSS and %d for Plus", + ossUpstreamServerCount, + plusUpstreamServerCount, + ), func() { const testName = "TestScale_UpstreamServers" testResultsDir := filepath.Join(resultsDir, testName)