larksuite__cli/tests/cli_e2e/drive/drive_copy_dryrun_test.go

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

package drive

import (
	"context"
	"strings"
	"testing"
	"time"

	clie2e "github.com/larksuite/cli/tests/cli_e2e"
	"github.com/stretchr/testify/require"
)

func TestDriveCopyDryRun_DocxURL(t *testing.T) {
	setDriveDryRunConfigEnv(t)

	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
	t.Cleanup(cancel)

	result, err := clie2e.RunCmd(ctx, clie2e.Request{
		Args: []string{
			"drive", "+copy",
			"--url", "https://example.larksuite.com/docx/docxDryRunCopy?from=share",
			"--name", "Copied doc",
			"--folder-token", "https://example.larksuite.com/drive/folder/folderDryRunCopy",
			"--dry-run",
		},
		DefaultAs: "bot",
	})
	require.NoError(t, err)
	result.AssertExitCode(t, 0)

	out := result.Stdout
	if got := clie2e.DryRunGet(out, "api.0.method").String(); got != "POST" {
		t.Fatalf("api.0.method=%q, want POST\nstdout:\n%s", got, out)
	}
	if got := clie2e.DryRunGet(out, "api.0.url").String(); got != "/open-apis/drive/v1/files/docxDryRunCopy/copy" {
		t.Fatalf("api.0.url=%q, want copy endpoint\nstdout:\n%s", got, out)
	}
	if got := clie2e.DryRunGet(out, "api.0.body.name").String(); got != "Copied doc" {
		t.Fatalf("api.0.body.name=%q, want Copied doc\nstdout:\n%s", got, out)
	}
	if got := clie2e.DryRunGet(out, "api.0.body.type").String(); got != "docx" {
		t.Fatalf("api.0.body.type=%q, want docx\nstdout:\n%s", got, out)
	}
	if got := clie2e.DryRunGet(out, "api.0.body.folder_token").String(); got != "folderDryRunCopy" {
		t.Fatalf("api.0.body.folder_token=%q, want token parsed from folder URL\nstdout:\n%s", got, out)
	}
	if got := clie2e.DryRunGet(out, "post_copy_note").String(); !strings.Contains(got, "full_access") {
		t.Fatalf("post_copy_note=%q, want bot auto-grant note\nstdout:\n%s", got, out)
	}
}

func TestDriveCopyDryRun_BareTokenBaseAlias(t *testing.T) {
	setDriveDryRunConfigEnv(t)

	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
	t.Cleanup(cancel)

	result, err := clie2e.RunCmd(ctx, clie2e.Request{
		Args: []string{
			"drive", "+copy",
			"--token", "bitableDryRunCopy",
			"--type", "base",
			"--name", "Copied base",
			"--folder-token", "folderDryRunCopy",
			"--extra", "target_type=bitable",
			"--dry-run",
		},
		DefaultAs: "bot",
	})
	require.NoError(t, err)
	result.AssertExitCode(t, 0)

	out := result.Stdout
	if got := clie2e.DryRunGet(out, "api.0.url").String(); got != "/open-apis/drive/v1/files/bitableDryRunCopy/copy" {
		t.Fatalf("api.0.url=%q, want copy endpoint\nstdout:\n%s", got, out)
	}
	if got := clie2e.DryRunGet(out, "api.0.body.type").String(); got != "bitable" {
		t.Fatalf("api.0.body.type=%q, want bitable (base alias normalized)\nstdout:\n%s", got, out)
	}
	if got := clie2e.DryRunGet(out, "api.0.body.extra.0.key").String(); got != "target_type" {
		t.Fatalf("api.0.body.extra.0.key=%q, want target_type\nstdout:\n%s", got, out)
	}
	if got := clie2e.DryRunGet(out, "api.0.body.extra.0.value").String(); got != "bitable" {
		t.Fatalf("api.0.body.extra.0.value=%q, want bitable\nstdout:\n%s", got, out)
	}
}

func TestDriveCopyDryRun_MySpaceTarget(t *testing.T) {
	setDriveDryRunConfigEnv(t)

	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
	t.Cleanup(cancel)

	result, err := clie2e.RunCmd(ctx, clie2e.Request{
		Args: []string{
			"drive", "+copy",
			"--url", "https://example.larksuite.com/docx/docxDryRunCopy",
			"--name", "Copied doc",
			"--folder-token", "my_space",
			"--dry-run",
		},
		DefaultAs: "bot",
	})
	require.NoError(t, err)
	result.AssertExitCode(t, 0)

	out := result.Stdout
	if got := clie2e.DryRunGet(out, "api.0.method").String(); got != "GET" {
		t.Fatalf("api.0.method=%q, want GET\nstdout:\n%s", got, out)
	}
	if got := clie2e.DryRunGet(out, "api.0.url").String(); got != "/open-apis/drive/explorer/v2/root_folder/meta" {
		t.Fatalf("api.0.url=%q, want root folder meta\nstdout:\n%s", got, out)
	}
	if got := clie2e.DryRunGet(out, "api.1.url").String(); got != "/open-apis/drive/v1/files/docxDryRunCopy/copy" {
		t.Fatalf("api.1.url=%q, want copy endpoint\nstdout:\n%s", got, out)
	}
	if got := clie2e.DryRunGet(out, "api.1.body.folder_token").String(); got != "<root folder token from step 1>" {
		t.Fatalf("api.1.body.folder_token=%q, want placeholder\nstdout:\n%s", got, out)
	}
}

func TestDriveCopyDryRun_WikiURLResolvesResourceThenCopiesToDrive(t *testing.T) {
	setDriveDryRunConfigEnv(t)

	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
	t.Cleanup(cancel)

	result, err := clie2e.RunCmd(ctx, clie2e.Request{
		Args: []string{
			"drive", "+copy",
			"--url", "https://example.larksuite.com/wiki/wikiDryRunCopy",
			"--name", "Copied wiki",
			"--folder-token", "folderDryRunCopy",
			"--dry-run",
		},
		DefaultAs: "bot",
	})
	require.NoError(t, err)
	result.AssertExitCode(t, 0)
	out := result.Stdout
	if got := clie2e.DryRunGet(out, "api.0.url").String(); got != "/open-apis/wiki/v2/spaces/node_by_token" {
		t.Fatalf("api.0.url=%q, want wiki resolver endpoint\nstdout:\n%s", got, out)
	}
	if got := clie2e.DryRunGet(out, "api.0.params.token").String(); got != "wikiDryRunCopy" {
		t.Fatalf("api.0.params.token=%q, want wikiDryRunCopy\nstdout:\n%s", got, out)
	}
	if got := clie2e.DryRunGet(out, "api.1.url").String(); got != "/open-apis/drive/v1/files/<obj_token from step 1>/copy" {
		t.Fatalf("api.1.url=%q, want resolved Drive copy endpoint\nstdout:\n%s", got, out)
	}
	if got := clie2e.DryRunGet(out, "api.1.body.type").String(); got != "<supported obj_type from step 1>" {
		t.Fatalf("api.1.body.type=%q, want resolved-type placeholder\nstdout:\n%s", got, out)
	}
	if got := clie2e.DryRunGet(out, "wiki_source_constraint").String(); got != "obj_type must be supported by Drive copy" {
		t.Fatalf("wiki_source_constraint=%q\nstdout:\n%s", got, out)
	}
}