CLOUDP-240596: fix search index create watcher for Atlas deployments

internal/cli/deployments/search/indexes/create.go

@@ -160,13 +160,25 @@ func (opts *CreateOpts) status(ctx context.Context) (string, error) {
 	return status, nil
 }
 
-func (opts *CreateOpts) watch(ctx context.Context) (any, bool, error) {
+func (opts *CreateOpts) watchLocal(ctx context.Context) (any, bool, error) {
 	state, err := opts.status(ctx)
 	if err != nil {
 		return nil, false, err
 	}
 	if state == "READY" {
-		return nil, true, nil
+		opts.index.Status = &state
+		return opts.index, true, nil
+	}
+	return nil, false, nil
+}
+
+func (opts *CreateOpts) watchAtlas(_ context.Context) (any, bool, error) {
+	index, err := opts.store.SearchIndex(opts.ConfigProjectID(), opts.DeploymentName, *opts.index.IndexID)
+	if err != nil {
+		return nil, false, err
+	}
+	if index.GetStatus() == "STEADY" {
+		return index, true, nil
 	}
 	return nil, false, nil
 }
@@ -177,11 +189,19 @@ func (opts *CreateOpts) PostRun(ctx context.Context) error {
 		return opts.Print(opts.index)
 	}
 
-	if _, err := opts.Watch(func() (any, bool, error) {
-		return opts.watch(ctx)
-	}); err != nil {
+	watch := opts.watchLocal
+	if opts.IsAtlasDeploymentType() {
+		watch = opts.watchAtlas
+	}
+
+	watchResult, err := opts.Watch(func() (any, bool, error) {
+		return watch(ctx)
+	})
+
+	if err != nil {
 		return err
 	}
+	opts.index = watchResult.(*admin.ClusterSearchIndex)
 
 	if err := opts.Print(opts.index); err != nil {
 		return err

internal/store/search.go

@@ -26,6 +26,7 @@ type SearchIndexLister interface {
 
 type SearchIndexCreator interface {
 	CreateSearchIndexes(string, string, *atlasv2.ClusterSearchIndex) (*atlasv2.ClusterSearchIndex, error)
+	SearchIndex(string, string, string) (*atlasv2.ClusterSearchIndex, error)
 }
 
 type SearchIndexDescriber interface {

test/e2e/atlas/deployments_atlas_test.go

@@ -144,7 +144,7 @@ func TestDeploymentsAtlas(t *testing.T) {
 	})
 	require.NoError(t, watchCluster(g.projectID, clusterName))
 
-	t.Run("Create Index", func(t *testing.T) {
+	t.Run("Create Search Index", func(t *testing.T) {
 		cmd := exec.Command(cliPath,
 			deploymentEntity,
 			searchEntity,
@@ -159,6 +159,7 @@ func TestDeploymentsAtlas(t *testing.T) {
 			databaseNameAtlas,
 			"--collection",
 			collectionNameAtlas,
+			"--watch",
 		)
 		cmd.Env = os.Environ()