larksuite--cli/shortcuts/apps/apps_db_env_create_test.go

// Copyright (c) 2026 Lark Technologies Pte. Ltd.
// SPDX-License-Identifier: MIT

package apps

import (
	"encoding/json"
	"strings"
	"testing"

	"github.com/larksuite/cli/internal/httpmock"
)

func TestAppsDBEnvCreate_WithYesPostsSyncData(t *testing.T) {
	factory, stdout, reg := newAppsExecuteFactory(t)
	stub := &httpmock.Stub{
		Method: "POST",
		URL:    "/open-apis/spark/v1/apps/app_x/db_dev_init", // URL 仍走 db_dev_init，CLI 命令名 +db-env-create
		Body: map[string]interface{}{
			"code": 0,
			"data": map[string]interface{}{
				"status":       "initialized",
				"environments": []interface{}{"dev", "online"},
				"data_synced":  true,
			},
		},
	}
	reg.Register(stub)
	if err := runAppsShortcut(t, AppsDBEnvCreate,
		[]string{"+db-env-create", "--app-id", "app_x", "--environment", "dev", "--sync-data", "--yes", "--as", "user"},
		factory, stdout); err != nil {
		t.Fatalf("execute err=%v", err)
	}

	var sent map[string]interface{}
	if err := json.Unmarshal(stub.CapturedBody, &sent); err != nil {
		t.Fatalf("decode body: %v", err)
	}
	if sent["sync_data"] != true {
		t.Fatalf("body.sync_data = %v (want true)", sent["sync_data"])
	}
	if !strings.Contains(stdout.String(), "initialized") {
		t.Fatalf("stdout should include status, got %s", stdout.String())
	}
}

// 不传 --sync-data（默认）→ body.sync_data=false
func TestAppsDBEnvCreate_SyncDataFalseByDefault(t *testing.T) {
	factory, stdout, reg := newAppsExecuteFactory(t)
	stub := &httpmock.Stub{
		Method: "POST",
		URL:    "/open-apis/spark/v1/apps/app_x/db_dev_init",
		Body:   map[string]interface{}{"code": 0, "data": map[string]interface{}{"status": "initialized"}},
	}
	reg.Register(stub)
	if err := runAppsShortcut(t, AppsDBEnvCreate,
		[]string{"+db-env-create", "--app-id", "app_x", "--environment", "dev", "--yes", "--as", "user"},
		factory, stdout); err != nil {
		t.Fatalf("execute err=%v", err)
	}
	var sent map[string]interface{}
	if err := json.Unmarshal(stub.CapturedBody, &sent); err != nil {
		t.Fatalf("decode body: %v", err)
	}
	if sent["sync_data"] != false {
		t.Fatalf("body.sync_data = %v (want false by default)", sent["sync_data"])
	}
}

func TestAppsDBEnvCreate_PrettyEmitsAllFourLines(t *testing.T) {
	factory, stdout, reg := newAppsExecuteFactory(t)
	reg.Register(&httpmock.Stub{
		Method: "POST",
		URL:    "/open-apis/spark/v1/apps/app_x/db_dev_init",
		Body: map[string]interface{}{
			"code": 0,
			"data": map[string]interface{}{
				"status":       "initialized",
				"environments": []interface{}{"dev", "online"},
				"data_synced":  true,
			},
		},
	})
	if err := runAppsShortcut(t, AppsDBEnvCreate,
		[]string{"+db-env-create", "--app-id", "app_x", "--environment", "dev", "--sync-data", "--yes", "--format", "pretty", "--as", "user"},
		factory, stdout); err != nil {
		t.Fatalf("execute err=%v", err)
	}
	got := stdout.String()
	wantLines := []string{
		"✓ Multi-env initialized",
		"Environments: dev, online",
		"Data synced: yes",
		"Note: structure changes in dev now need to be released to online.",
	}
	for _, line := range wantLines {
		if !strings.Contains(got, line) {
			t.Errorf("pretty output missing line %q\ngot:\n%s", line, got)
		}
	}
}

func TestAppsDBEnvCreate_DryRunNoConfirm(t *testing.T) {
	factory, stdout, _ := newAppsExecuteFactory(t)
	if err := runAppsShortcut(t, AppsDBEnvCreate,
		[]string{"+db-env-create", "--app-id", "app_x", "--environment", "dev", "--dry-run", "--as", "user"},
		factory, stdout); err != nil {
		t.Fatalf("dry-run err=%v", err)
	}
	if got := stdout.String(); !strings.Contains(got, "/open-apis/spark/v1/apps/app_x/db_dev_init") {
		t.Fatalf("dry-run missing endpoint: %s", got)
	}
}

// --env 只接受 dev：传 online 应被 enum 校验拒绝。
func TestAppsDBEnvCreate_RejectsNonDevEnv(t *testing.T) {
	factory, stdout, _ := newAppsExecuteFactory(t)
	err := runAppsShortcut(t, AppsDBEnvCreate,
		[]string{"+db-env-create", "--app-id", "app_x", "--environment", "online", "--yes", "--as", "user"},
		factory, stdout)
	if err == nil || !strings.Contains(err.Error(), "env") {
		t.Fatalf("expected env enum rejection, got %v", err)
	}
}