fix tests

beam-cloud · Oct 30, 2024 · 8258e20 · 8258e20
1 parent f2a991d
commit 8258e20
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/pkg/abstractions/endpoint/autoscaler_test.go b/pkg/abstractions/endpoint/autoscaler_test.go
@@ -17,6 +17,12 @@ func TestDeploymentScaleFuncWithDefaults(t *testing.T) {
 				ExternalId: "test",
 			},
 		},
+		AppConfig: types.AppConfig{},
+	}
+	autoscaledInstance.AppConfig.GatewayService = types.GatewayServiceConfig{
+		StubLimits: types.StubLimits{
+			MaxReplicas: 10,
+		},
 	}
 	autoscaledInstance.StubConfig = &types.StubConfigV1{}
 	autoscaledInstance.StubConfig.Autoscaler = &types.Autoscaler{
@@ -64,6 +70,13 @@ func TestDeploymentScaleFuncWithMaxTasksPerContainer(t *testing.T) {
 				ExternalId: "test",
 			},
 		},
+		AppConfig: types.AppConfig{},
+	}
+
+	autoscaledInstance.AppConfig.GatewayService = types.GatewayServiceConfig{
+		StubLimits: types.StubLimits{
+			MaxReplicas: 10,
+		},
 	}
 	autoscaledInstance.StubConfig = &types.StubConfigV1{}
 	autoscaledInstance.StubConfig.Autoscaler = &types.Autoscaler{

diff --git a/pkg/abstractions/taskqueue/autoscaler_test.go b/pkg/abstractions/taskqueue/autoscaler_test.go
@@ -16,6 +16,12 @@ func TestDeploymentScaleFuncWithDefaults(t *testing.T) {
 		MaxContainers:     1,
 		TasksPerContainer: 1,
 	}
+	autoscaledInstance.AppConfig = types.AppConfig{}
+	autoscaledInstance.AppConfig.GatewayService = types.GatewayServiceConfig{
+		StubLimits: types.StubLimits{
+			MaxReplicas: 10,
+		},
+	}
 
 	instance := &taskQueueInstance{}
 	instance.AutoscaledInstance = autoscaledInstance
@@ -56,6 +62,12 @@ func TestDeploymentScaleFuncWithMaxTasksPerContainer(t *testing.T) {
 		MaxContainers:     3,
 		TasksPerContainer: 1,
 	}
+	autoscaledInstance.AppConfig = types.AppConfig{}
+	autoscaledInstance.AppConfig.GatewayService = types.GatewayServiceConfig{
+		StubLimits: types.StubLimits{
+			MaxReplicas: 10,
+		},
+	}
 
 	// Make sure we scale up to max containers
 	instance := &taskQueueInstance{}