65 lines
2.4 KiB
Go
65 lines
2.4 KiB
Go
package helpers
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/DingTalk-Real-AI/dingtalk-workspace-cli/internal/executor"
|
|
)
|
|
|
|
// docExportStubRunner 驱动一次「提交→查询命中 SUCCESS（无 downloadUrl）」的导出流程，
|
|
// 走到 writeCommandPayload 但不触发实际下载（无网络）。
|
|
type docExportStubRunner struct{}
|
|
|
|
func (docExportStubRunner) Run(_ context.Context, inv executor.Invocation) (executor.Result, error) {
|
|
switch inv.Tool {
|
|
case "submit_export_job":
|
|
return executor.Result{Response: map[string]any{"jobId": "job-123"}}, nil
|
|
case "query_export_job":
|
|
// SUCCESS 但不带 downloadUrl → 跳过 asynctask.Download，仍输出结构化 payload
|
|
return executor.Result{Response: map[string]any{"status": "SUCCESS"}}, nil
|
|
default:
|
|
return executor.Result{}, nil
|
|
}
|
|
}
|
|
|
|
// TestDocExportProgressGoesToStdout 守护 doc export 的评测契约：导出进度文案需要
|
|
// 出现在 stdout，便于 agent 在执行过程中看到 submit → poll → download 的状态。
|
|
func TestDocExportProgressGoesToStdout(t *testing.T) {
|
|
cmd := docHandler{}.Command(docExportStubRunner{})
|
|
var stdout, stderr bytes.Buffer
|
|
cmd.SetOut(&stdout)
|
|
cmd.SetErr(&stderr)
|
|
cmd.SetArgs([]string{"export", "--node", "nodeABC123", "--output", "/tmp/dws-export-progress-test.docx"})
|
|
|
|
if err := cmd.Execute(); err != nil {
|
|
t.Fatalf("Execute() error = %v\nstderr:\n%s", err, stderr.String())
|
|
}
|
|
|
|
for _, marker := range []string{"[1/3]", "提交导出任务", "jobId: job-123", "[2/3]"} {
|
|
if !strings.Contains(stdout.String(), marker) {
|
|
t.Fatalf("期望进度标记 %q 出现在 stdout，实际 stdout:\n%s", marker, stdout.String())
|
|
}
|
|
}
|
|
if strings.Contains(stderr.String(), "[1/3]") {
|
|
t.Fatalf("进度不应出现在 stderr，实际 stderr:\n%s", stderr.String())
|
|
}
|
|
|
|
// stdout 同时包含进度与最终 payload；解析末尾 JSON，确保结构化结果仍输出。
|
|
jsonStart := strings.LastIndex(stdout.String(), "{")
|
|
if jsonStart < 0 {
|
|
t.Fatalf("stdout 未包含最终 JSON payload:\n%s", stdout.String())
|
|
}
|
|
out := strings.TrimSpace(stdout.String()[jsonStart:])
|
|
var payload map[string]any
|
|
if err := json.Unmarshal([]byte(out), &payload); err != nil {
|
|
t.Fatalf("stdout 末尾不是可解析 JSON: err=%v\nstdout:\n%s", err, stdout.String())
|
|
}
|
|
if payload["jobId"] != "job-123" {
|
|
t.Fatalf("payload.jobId = %#v, want job-123; stdout:\n%s", payload["jobId"], stdout.String())
|
|
}
|
|
}
|