CLOUDP-322605: Add suport to ISS on atlas deployments pause and start

internal/cli/deployments/options/deployment_opts.go

@@ -45,24 +45,26 @@ import (
 const (
 	spinnerSpeed = 100 * time.Millisecond
 	// based on https://www.mongodb.com/docs/atlas/reference/api-resources-spec/v2/#tag/Clusters/operation/createCluster
-	clusterNamePattern = "^[a-zA-Z0-9][a-zA-Z0-9-]*$"
-	PausedState        = "PAUSED"
-	StoppedState       = "STOPPED"
-	IdleState          = "IDLE"
-	UpdatingState      = "UPDATING"
-	DeletingState      = "DELETING"
-	RestartingState    = "RESTARTING"
-	LocalCluster       = "local"
-	AtlasCluster       = "atlas"
-	CompassConnect     = "compass"
-	MongoshConnect     = "mongosh"
-	VsCodeConnect      = "vscode"
-	PromptTypeMessage  = "What type of deployment would you like to work with?"
-	MaxItemsPerPage    = 500
-	ContainerFilter    = "mongodb-atlas-local=container"
-	bytesInGb          = 1073741824
-	minimumRAM         = 2 * bytesInGb
-	minimumCores       = 2
+	clusterNamePattern      = "^[a-zA-Z0-9][a-zA-Z0-9-]*$"
+	PausedState             = "PAUSED"
+	StoppedState            = "STOPPED"
+	IdleState               = "IDLE"
+	UpdatingState           = "UPDATING"
+	DeletingState           = "DELETING"
+	RestartingState         = "RESTARTING"
+	LocalCluster            = "local"
+	AtlasCluster            = "atlas"
+	CompassConnect          = "compass"
+	MongoshConnect          = "mongosh"
+	VsCodeConnect           = "vscode"
+	PromptTypeMessage       = "What type of deployment would you like to work with?"
+	MaxItemsPerPage         = 500
+	ContainerFilter         = "mongodb-atlas-local=container"
+	ClusterWideScaling      = "CLUSTER_WIDE_SCALING"
+	IndependentShardScaling = "INDEPENDENT_SHARD_SCALING"
+	bytesInGb               = 1073741824
+	minimumRAM              = 2 * bytesInGb
+	minimumCores            = 2
 )
 
 var (

internal/cli/deployments/pause.go

@@ -28,12 +28,15 @@ import (
 	"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/usage"
 	"github.com/spf13/cobra"
 	atlasClustersPinned "go.mongodb.org/atlas-sdk/v20240530005/admin"
+	atlasv2 "go.mongodb.org/atlas-sdk/v20250312003/admin"
 )
 
 //go:generate go tool go.uber.org/mock/mockgen -typed -destination=pause_mock_test.go -package=deployments . ClusterPauser
 
 type ClusterPauser interface {
 	PauseCluster(string, string) (*atlasClustersPinned.AdvancedClusterDescription, error)
+	PauseClusterLatest(string, string) (*atlasv2.ClusterDescription20240805, error)
+	GetClusterAutoScalingConfig(string, string) (*atlasv2.ClusterDescriptionAutoScalingModeConfiguration, error)
 }
 
 type PauseOpts struct {
@@ -104,7 +107,17 @@ func (opts *PauseOpts) RunAtlas() error {
 	opts.StartSpinner()
 	defer opts.StopSpinner()
 
-	r, err := opts.store.PauseCluster(opts.ConfigProjectID(), opts.DeploymentName)
+	clusterAutoScalingConfig, err := opts.store.GetClusterAutoScalingConfig(opts.ConfigProjectID(), opts.DeploymentName)
+	if err != nil || clusterAutoScalingConfig.GetAutoScalingMode() == options.ClusterWideScaling {
+		r, err := opts.store.PauseCluster(opts.ConfigProjectID(), opts.DeploymentName)
+		if err != nil {
+			return err
+		}
+		return opts.Print(r)
+	}
+
+	// If cluster is not cluster wide scaling, we use the latest API version
+	r, err := opts.store.PauseClusterLatest(opts.ConfigProjectID(), opts.DeploymentName)
 	if err != nil {
 		return err
 	}

internal/cli/deployments/pause_test.go

@@ -22,10 +22,12 @@ import (
 	"testing"
 
 	"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/cli"
+	"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/cli/deployments/options"
 	"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/cli/deployments/test/fixture"
 	"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/pointer"
 	"github.com/stretchr/testify/assert"
 	atlasClustersPinned "go.mongodb.org/atlas-sdk/v20240530005/admin"
+	atlasv2 "go.mongodb.org/atlas-sdk/v20250312003/admin"
 	"go.uber.org/mock/gomock"
 )
 
@@ -70,7 +72,7 @@ func TestPause_RunLocal(t *testing.T) {
 	t.Log(buf.String())
 }
 
-func TestPause_RunAtlas(t *testing.T) {
+func TestPause_RunAtlas_clusterWideScaling(t *testing.T) {
 	ctrl := gomock.NewController(t)
 	mockStore := NewMockClusterPauser(ctrl)
 	ctx := t.Context()
@@ -92,6 +94,15 @@ func TestPause_RunAtlas(t *testing.T) {
 
 	deploymentTest.CommonAtlasMocks(projectID)
 
+	mockStore.
+		EXPECT().
+		GetClusterAutoScalingConfig(projectID, deploymentName).
+		Return(
+			&atlasv2.ClusterDescriptionAutoScalingModeConfiguration{
+				AutoScalingMode: pointer.Get(options.ClusterWideScaling),
+			}, nil).
+		Times(1)
+
 	mockStore.
 		EXPECT().
 		PauseCluster(projectID, deploymentName).
@@ -109,6 +120,54 @@ func TestPause_RunAtlas(t *testing.T) {
 	t.Log(buf.String())
 }
 
+func TestPause_RunAtlas_independentShardScaling(t *testing.T) {
+	ctrl := gomock.NewController(t)
+	mockStore := NewMockClusterPauser(ctrl)
+	ctx := t.Context()
+
+	deploymentTest := fixture.NewMockAtlasDeploymentOpts(ctrl, deploymentName)
+
+	buf := new(bytes.Buffer)
+	listOpts := &PauseOpts{
+		store:          mockStore,
+		DeploymentOpts: *deploymentTest.Opts,
+		ProjectOpts: cli.ProjectOpts{
+			ProjectID: projectID,
+		},
+		OutputOpts: cli.OutputOpts{
+			Template:  pauseTemplate,
+			OutWriter: buf,
+		},
+	}
+
+	deploymentTest.CommonAtlasMocks(projectID)
+
+	mockStore.
+		EXPECT().
+		GetClusterAutoScalingConfig(projectID, deploymentName).
+		Return(
+			&atlasv2.ClusterDescriptionAutoScalingModeConfiguration{
+				AutoScalingMode: pointer.Get(string(options.IndependentShardScaling)),
+			}, nil).
+		Times(1)
+
+	mockStore.
+		EXPECT().
+		PauseClusterLatest(projectID, deploymentName).
+		Return(
+			&atlasv2.ClusterDescription20240805{
+				Name: pointer.Get(deploymentName),
+			}, nil).
+		Times(1)
+
+	if err := listOpts.Run(ctx); err != nil {
+		t.Fatalf("Run() unexpected error: %v", err)
+	}
+
+	assert.Equal(t, fmt.Sprintf("Pausing deployment '%s'.\n", deploymentName), buf.String())
+	t.Log(buf.String())
+}
+
 func TestPauseOpts_PostRun(t *testing.T) {
 	ctrl := gomock.NewController(t)
 	deploymentTest := fixture.NewMockLocalDeploymentOpts(ctrl, deploymentName)

internal/cli/deployments/start.go

@@ -26,12 +26,15 @@ import (
 	"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/usage"
 	"github.com/spf13/cobra"
 	atlasClustersPinned "go.mongodb.org/atlas-sdk/v20240530005/admin"
+	atlasv2 "go.mongodb.org/atlas-sdk/v20250312003/admin"
 )
 
 //go:generate go tool go.uber.org/mock/mockgen -typed -destination=start_mock_test.go -package=deployments . ClusterStarter
 
 type ClusterStarter interface {
 	StartCluster(string, string) (*atlasClustersPinned.AdvancedClusterDescription, error)
+	StartClusterLatest(string, string) (*atlasv2.ClusterDescription20240805, error)
+	GetClusterAutoScalingConfig(string, string) (*atlasv2.ClusterDescriptionAutoScalingModeConfiguration, error)
 }
 
 type StartOpts struct {
@@ -83,7 +86,17 @@ func (opts *StartOpts) RunAtlas() error {
 	opts.StartSpinner()
 	defer opts.StopSpinner()
 
-	r, err := opts.store.StartCluster(opts.ConfigProjectID(), opts.DeploymentName)
+	clusterAutoScalingConfig, err := opts.store.GetClusterAutoScalingConfig(opts.ConfigProjectID(), opts.DeploymentName)
+	if err != nil || clusterAutoScalingConfig.GetAutoScalingMode() == options.ClusterWideScaling {
+		r, err := opts.store.StartCluster(opts.ConfigProjectID(), opts.DeploymentName)
+		if err != nil {
+			return err
+		}
+		return opts.Print(r)
+	}
+
+	// If cluster is not cluster wide scaling, we use the latest API version
+	r, err := opts.store.StartClusterLatest(opts.ConfigProjectID(), opts.DeploymentName)
 	if err != nil {
 		return err
 	}