70 lines
2.1 KiB
Go
70 lines
2.1 KiB
Go
// Copyright 2026 Alibaba Group
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
|
|
package helpers
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/DingTalk-Real-AI/dingtalk-workspace-cli/pkg/edition"
|
|
)
|
|
|
|
type sheetStyleDryRunCaller struct {
|
|
format string
|
|
calls int
|
|
}
|
|
|
|
func (c *sheetStyleDryRunCaller) CallTool(context.Context, string, string, map[string]any) (*edition.ToolResult, error) {
|
|
c.calls++
|
|
return &edition.ToolResult{}, nil
|
|
}
|
|
|
|
func (c *sheetStyleDryRunCaller) Format() string { return c.format }
|
|
func (*sheetStyleDryRunCaller) DryRun() bool { return true }
|
|
func (*sheetStyleDryRunCaller) Fields() string { return "" }
|
|
func (*sheetStyleDryRunCaller) JQ() string { return "" }
|
|
|
|
func TestRangeBatchSetStyleDryRunNeverCallsRemote(t *testing.T) {
|
|
previousDeps := deps
|
|
t.Cleanup(func() { deps = previousDeps })
|
|
|
|
batchPath := filepath.Join(t.TempDir(), "styles.json")
|
|
if err := os.WriteFile(batchPath, []byte(`[{"sheetId":"Sheet1","range":"A1:B2","fontWeight":"bold"}]`), 0o600); err != nil {
|
|
t.Fatalf("write batch fixture: %v", err)
|
|
}
|
|
|
|
for _, format := range []string{"table", "json"} {
|
|
t.Run(format, func(t *testing.T) {
|
|
caller := &sheetStyleDryRunCaller{format: format}
|
|
InitDeps(caller)
|
|
var output bytes.Buffer
|
|
deps.Out.w = &output
|
|
deps.Out.errW = &output
|
|
|
|
cmd := newRangeBatchSetStyleCmd()
|
|
cmd.SetArgs([]string{"--node", "NODE_ID", "--batch", batchPath})
|
|
if err := cmd.Execute(); err != nil {
|
|
t.Fatalf("batch-set-style dry-run error: %v", err)
|
|
}
|
|
if caller.calls != 0 {
|
|
t.Fatalf("remote CallTool count = %d, want 0", caller.calls)
|
|
}
|
|
preview := output.String()
|
|
for _, want := range []string{"Tool:", "update_range", "Arguments:"} {
|
|
if !strings.Contains(preview, want) {
|
|
t.Fatalf("dry-run preview missing %q:\n%s", want, preview)
|
|
}
|
|
}
|
|
if format == "json" && !strings.Contains(preview, `"dryRun": true`) {
|
|
t.Fatalf("JSON dry-run summary missing typed dryRun evidence:\n%s", preview)
|
|
}
|
|
})
|
|
}
|
|
}
|