Files
CyberStrikeAI/internal/workflow/tool_output_budget_test.go
T
2026-07-10 16:42:35 +08:00

24 lines
685 B
Go

package workflow
import (
"strings"
"testing"
)
func TestTruncateWorkflowToolOutputBoundsBytesAndKeepsExecutionReference(t *testing.T) {
out := truncateWorkflowToolOutput(strings.Repeat("响应正文", 1000), 256, "exec-123")
if len(out) > 256 {
t.Fatalf("workflow output bytes=%d, want <=256", len(out))
}
if !strings.Contains(out, "exec-123") || !strings.Contains(out, "truncated") {
t.Fatalf("missing truncation reference: %q", out)
}
}
func TestTruncateWorkflowToolOutputLeavesBoundedContentUntouched(t *testing.T) {
const want = "small-result"
if got := truncateWorkflowToolOutput(want, 256, "exec-123"); got != want {
t.Fatalf("got %q want %q", got, want)
}
}