fix(cloudprovider/externalgrpc): GetOptions returns `MaxNodeProvisi…

…onTime` as 0 seconds Signed-off-by: vadasambar <[email protected]> chore(cloudprovider/externalgrpc): fix CI failing because of wrong header - CI complains if I remove or change the copyright header Signed-off-by: vadasambar <[email protected]> chore(cloudprovider/externalgrpc): fix spacing for license headers Signed-off-by: vadasambar <[email protected]> chore(cloudprovider/externalgrpc): remove extra space in copyright header Signed-off-by: vadasambar <[email protected]> test(cloudprovider/externalgrpc): add assert for `MaxNodeProvisionTime` - add extra assert for err test case Signed-off-by: vadasambar <[email protected]>
kubernetes · Jul 7, 2023 · 76092a2 · 76092a2
1 parent b569db4
commit 76092a2
Show file tree

Hide file tree

Showing 6 changed files with 244 additions and 212 deletions.
diff --git a/cluster-autoscaler/cloudprovider/externalgrpc/README.md b/cluster-autoscaler/cloudprovider/externalgrpc/README.md
@@ -56,8 +56,8 @@ To regenerate the gRPC code:
 1. install `protoc` and `protoc-gen-go-grpc`:
 
 ```bash
-go install google.golang.org/protobuf/cmd/protoc-gen-go@v1.26
-go install google.golang.org/grpc/cmd/protoc-gen-go-grpc@v1.1
+go install google.golang.org/protobuf/cmd/protoc-gen-go@v1.28
+go install google.golang.org/grpc/cmd/protoc-gen-go-grpc@v1.2
 ```
 
 2. generate gRPC client and server code:

diff --git a/cluster-autoscaler/cloudprovider/externalgrpc/externalgrpc_node_group.go b/cluster-autoscaler/cloudprovider/externalgrpc/externalgrpc_node_group.go
@@ -263,6 +263,9 @@ func (n *NodeGroup) GetOptions(defaults config.NodeGroupAutoscalingOptions) (*co
 			ScaleDownUnreadyTime: &metav1.Duration{
 				Duration: defaults.ScaleDownUnreadyTime,
 			},
+			MaxNodeProvisionTime: &metav1.Duration{
+				Duration: defaults.MaxNodeProvisionTime,
+			},
 		},
 	})
 	if err != nil {
@@ -278,6 +281,7 @@ func (n *NodeGroup) GetOptions(defaults config.NodeGroupAutoscalingOptions) (*co
 		ScaleDownGpuUtilizationThreshold: pbOpts.GetScaleDownGpuUtilizationThreshold(),
 		ScaleDownUnneededTime:            pbOpts.GetScaleDownUnneededTime().Duration,
 		ScaleDownUnreadyTime:             pbOpts.GetScaleDownUnreadyTime().Duration,
+		MaxNodeProvisionTime:             pbOpts.GetMaxNodeProvisionTime().Duration,
 	}
 	return opts, nil
 }
diff --git a/cluster-autoscaler/cloudprovider/externalgrpc/externalgrpc_node_group_test.go b/cluster-autoscaler/cloudprovider/externalgrpc/externalgrpc_node_group_test.go
@@ -226,6 +226,7 @@ func TestCloudProvider_GetOptions(t *testing.T) {
 				ScaleDownGpuUtilizationThreshold: 0.7,
 				ScaleDownUnneededTime:            &v1.Duration{Duration: time.Minute},
 				ScaleDownUnreadyTime:             &v1.Duration{Duration: time.Hour},
+				MaxNodeProvisionTime:             &v1.Duration{Duration: time.Minute},
 			},
 		},
 		nil,
@@ -240,6 +241,7 @@ func TestCloudProvider_GetOptions(t *testing.T) {
 		ScaleDownGpuUtilizationThreshold: 0.7,
 		ScaleDownUnneededTime:            time.Minute,
 		ScaleDownUnreadyTime:             time.Hour,
+		MaxNodeProvisionTime:             time.Minute,
 	}
 
 	opts, err := ng1.GetOptions(defaultsOpts)
@@ -248,6 +250,8 @@ func TestCloudProvider_GetOptions(t *testing.T) {
 	assert.Equal(t, 0.7, opts.ScaleDownGpuUtilizationThreshold)
 	assert.Equal(t, time.Minute, opts.ScaleDownUnneededTime)
 	assert.Equal(t, time.Hour, opts.ScaleDownUnreadyTime)
+	assert.Equal(t, time.Hour, opts.ScaleDownUnreadyTime)
+	assert.Equal(t, time.Minute, opts.MaxNodeProvisionTime)
 
 	// test grpc error
 	m.On(
@@ -264,8 +268,9 @@ func TestCloudProvider_GetOptions(t *testing.T) {
 		client: client,
 	}
 
-	_, err = ng2.GetOptions(defaultsOpts)
+	opts, err = ng2.GetOptions(defaultsOpts)
 	assert.Error(t, err)
+	assert.Nil(t, opts)
 
 	// test no opts
 	m.On(