bf9395e022
CI / license-header (push) Has been skipped
CI / e2e-dry-run (push) Has been skipped
CI / fast-gate (push) Failing after 0s
Test PR Label Logic / test-pr-labels (push) Failing after 1s
Skill Format Check / check-format (push) Failing after 2s
CI / security (push) Failing after 5s
CI / unit-test (push) Has been skipped
CI / lint (push) Has been skipped
CI / script-test (push) Has been skipped
CI / deterministic-gate (push) Has been skipped
CI / coverage (push) Has been skipped
CI / results (push) Has been cancelled
CI / deadcode (push) Has been cancelled
CI / e2e-live (push) Has been cancelled
77 lines
2.5 KiB
Go
77 lines
2.5 KiB
Go
// Copyright (c) 2026 Lark Technologies Pte. Ltd.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package sheets
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
clie2e "github.com/larksuite/cli/tests/cli_e2e"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/tidwall/gjson"
|
|
)
|
|
|
|
// TestSheets_TableGetDefaultDryRun pins the request structure +table-get emits
|
|
// when no --range is given: it must first read get_workbook_structure (to learn
|
|
// each sheet's grid dimensions, which anchor the used-range probe over the full
|
|
// grid) and then read cells via get_cell_ranges. This guards the pro016 / pro025
|
|
// fix — the default read must span internal blank rows/columns, not stop at the
|
|
// A1 current region.
|
|
func TestSheets_TableGetDefaultDryRun(t *testing.T) {
|
|
setSheetsDryRunEnv(t)
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
|
t.Cleanup(cancel)
|
|
|
|
result, err := clie2e.RunCmd(ctx, clie2e.Request{
|
|
Args: []string{
|
|
"sheets", "+table-get",
|
|
"--spreadsheet-token", "shtDryRun",
|
|
"--dry-run",
|
|
},
|
|
DefaultAs: "user",
|
|
})
|
|
require.NoError(t, err)
|
|
result.AssertExitCode(t, 0)
|
|
|
|
out := result.Stdout
|
|
|
|
// api.0 — the structure read that supplies the grid dimensions.
|
|
require.Equal(t, "get_workbook_structure", gjson.Get(out, "api.0.body.tool_name").String(), "stdout:\n%s", out)
|
|
require.Equal(t, "/open-apis/sheet_ai/v2/spreadsheets/shtDryRun/tools/invoke_read",
|
|
gjson.Get(out, "api.0.url").String(), "stdout:\n%s", out)
|
|
|
|
// api.1 — the cells read.
|
|
require.Equal(t, "get_cell_ranges", gjson.Get(out, "api.1.body.tool_name").String(), "stdout:\n%s", out)
|
|
}
|
|
|
|
// TestSheets_TableGetSingleSheetDryRun confirms the single-sheet selector path
|
|
// also reads get_workbook_structure now (previously it did not): the grid
|
|
// dimensions are needed even when only one sheet is read, so the used-range
|
|
// probe can anchor over the full grid.
|
|
func TestSheets_TableGetSingleSheetDryRun(t *testing.T) {
|
|
setSheetsDryRunEnv(t)
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
|
t.Cleanup(cancel)
|
|
|
|
result, err := clie2e.RunCmd(ctx, clie2e.Request{
|
|
Args: []string{
|
|
"sheets", "+table-get",
|
|
"--spreadsheet-token", "shtDryRun",
|
|
"--sheet-name", "Sheet1",
|
|
"--dry-run",
|
|
},
|
|
DefaultAs: "user",
|
|
})
|
|
require.NoError(t, err)
|
|
result.AssertExitCode(t, 0)
|
|
|
|
out := result.Stdout
|
|
require.Equal(t, "get_workbook_structure", gjson.Get(out, "api.0.body.tool_name").String(),
|
|
"single-sheet path must still read the structure for grid dimensions; stdout:\n%s", out)
|
|
require.Equal(t, "get_cell_ranges", gjson.Get(out, "api.1.body.tool_name").String(), "stdout:\n%s", out)
|
|
}
|