update nodegroup nebius

Author: drewmalin

internal/validation/suite.go

@@ -442,6 +442,33 @@ func RunKubernetesValidation(t *testing.T, config ProviderConfig, opts Kubernete
 		require.NoError(t, err, "ValidateClusterNodeGroups should pass")
 	})
 
+	// Test: Modify Kubernetes Node Group
+	t.Run("ValidateModifyKubernetesNodeGroup", func(t *testing.T) {
+		err := v1.ValidateModifyKubernetesNodeGroup(ctx, client, v1.ModifyNodeGroupArgs{
+			ID:           nodeGroup.ID,
+			MinNodeCount: opts.NodeGroupOpts.MinNodeCount + 1,
+			MaxNodeCount: opts.NodeGroupOpts.MaxNodeCount + 1,
+		})
+		require.NoError(t, err, "ValidateModifyKubernetesNodeGroup should pass")
+	})
+
+	// Test: WaitFor Kubernetes Node Group to Be Available
+	t.Run("WaitForKubernetesNodeGroupToBeAvailable", func(t *testing.T) {
+		err := WaitForResourcePredicate(ctx, WaitForResourcePredicateOpts[*v1.NodeGroup]{
+			GetResource: func() (*v1.NodeGroup, error) {
+				return client.GetNodeGroup(ctx, v1.GetNodeGroupArgs{ID: nodeGroup.ID})
+			},
+			Predicate: func(nodeGroup *v1.NodeGroup) bool {
+				return nodeGroup.Status == v1.NodeGroupStatusAvailable &&
+					nodeGroup.MinNodeCount == opts.NodeGroupOpts.MinNodeCount+1 &&
+					nodeGroup.MaxNodeCount == opts.NodeGroupOpts.MaxNodeCount+1
+			},
+			Timeout:  5 * time.Minute,
+			Interval: 5 * time.Second,
+		})
+		require.NoError(t, err, "WaitForKubernetesNodeGroupToBeAvailable should pass")
+	})
+
 	// Test: Delete Kubernetes Node Group
 	t.Run("ValidateDeleteKubernetesNodeGroup", func(t *testing.T) {
 		err := v1.ValidateDeleteKubernetesNodeGroup(ctx, client, v1.DeleteNodeGroupArgs{

v1/kubernetes.go

@@ -157,13 +157,28 @@ type DeleteClusterArgs struct {
 }
 
 type CloudMaintainKubernetes interface {
+	// Create a new Kubernetes cluster.
 	CreateCluster(ctx context.Context, args CreateClusterArgs) (*Cluster, error)
+
+	// Get a Kubernetes cluster identified by the provided args.
 	GetCluster(ctx context.Context, args GetClusterArgs) (*Cluster, error)
+
+	// Put a user into a Kubernetes cluster.
 	PutUser(ctx context.Context, args PutUserArgs) (*PutUserResponse, error)
+
+	// Create a new Kubernetes node group.
 	CreateNodeGroup(ctx context.Context, args CreateNodeGroupArgs) (*NodeGroup, error)
+
+	// Get a Kubernetes node group identified by the provided args.
 	GetNodeGroup(ctx context.Context, args GetNodeGroupArgs) (*NodeGroup, error)
+
+	// Modify a Kubernetes node group.
 	ModifyNodeGroup(ctx context.Context, args ModifyNodeGroupArgs) error
+
+	// Delete a Kubernetes node group identified by the provided args.
 	DeleteNodeGroup(ctx context.Context, args DeleteNodeGroupArgs) error
+
+	// Delete a Kubernetes cluster identified by the provided args.
 	DeleteCluster(ctx context.Context, args DeleteClusterArgs) error
 }
 

v1/providers/nebius/kubernetes.go

@@ -422,6 +422,74 @@ func parseNebiusNodeGroupStatus(status *nebiusmk8s.NodeGroupStatus) v1.NodeGroup
 	return v1.NodeGroupStatusUnknown
 }
 
+func (c *NebiusClient) ModifyNodeGroup(ctx context.Context, args v1.ModifyNodeGroupArgs) error {
+	nebiusNodeGroupService := c.sdk.Services().MK8S().V1().NodeGroup()
+
+	err := validateModifyNodeGroupArgs(args)
+	if err != nil {
+		return errors.WrapAndTrace(err)
+	}
+
+	nodeGroup, err := nebiusNodeGroupService.Get(ctx, &nebiusmk8s.GetNodeGroupRequest{
+		Id: string(args.ID),
+	})
+	if err != nil {
+		return errors.WrapAndTrace(err)
+	}
+
+	_, err = nebiusNodeGroupService.Update(ctx, &nebiusmk8s.UpdateNodeGroupRequest{
+		Metadata: &nebiuscommon.ResourceMetadata{
+			Id:       nodeGroup.Metadata.Id,
+			Name:     nodeGroup.Metadata.Name,
+			ParentId: nodeGroup.Metadata.ParentId,
+			Labels:   nodeGroup.Metadata.Labels,
+		},
+		Spec: &nebiusmk8s.NodeGroupSpec{
+			Size: &nebiusmk8s.NodeGroupSpec_Autoscaling{
+				Autoscaling: &nebiusmk8s.NodeGroupAutoscalingSpec{
+					MinNodeCount: int64(args.MinNodeCount),
+					MaxNodeCount: int64(args.MaxNodeCount),
+				},
+			},
+			Template: &nebiusmk8s.NodeTemplate{
+				Resources: &nebiusmk8s.ResourcesSpec{
+					Platform: nodeGroup.Spec.GetTemplate().GetResources().GetPlatform(),
+					Size: &nebiusmk8s.ResourcesSpec_Preset{
+						Preset: nodeGroup.Spec.GetTemplate().GetResources().GetPreset(),
+					},
+				},
+				GpuSettings: &nebiusmk8s.GpuSettings{
+					DriversPreset: nodeGroup.Spec.GetTemplate().GetGpuSettings().GetDriversPreset(),
+				},
+				BootDisk: &nebiusmk8s.DiskSpec{
+					Type: nodeGroup.Spec.GetTemplate().GetBootDisk().GetType(),
+					Size: &nebiusmk8s.DiskSpec_SizeGibibytes{
+						SizeGibibytes: nodeGroup.Spec.GetTemplate().GetBootDisk().GetSizeGibibytes(),
+					},
+				},
+			},
+		},
+	})
+	if err != nil {
+		return errors.WrapAndTrace(fmt.Errorf("failed to modify node group: %w", err))
+	}
+
+	return nil
+}
+
+func validateModifyNodeGroupArgs(args v1.ModifyNodeGroupArgs) error {
+	if args.MinNodeCount < 1 {
+		return fmt.Errorf("node group minNodeCount must be greater than 0")
+	}
+	if args.MaxNodeCount < 1 {
+		return fmt.Errorf("node group maxNodeCount must be greater than 0")
+	}
+	if args.MaxNodeCount < args.MinNodeCount {
+		return fmt.Errorf("node group maxNodeCount must be greater than or equal to minNodeCount")
+	}
+	return nil
+}
+
 func (c *NebiusClient) DeleteNodeGroup(ctx context.Context, args v1.DeleteNodeGroupArgs) error {
 	nebiusNodeGroupService := c.sdk.Services().MK8S().V1().NodeGroup()