e0b835156c
Auto-fix via golangci-lint --fix to satisfy gci formatter rules (standard → default → localmodule import grouping). Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
99 lines
2.1 KiB
Go
99 lines
2.1 KiB
Go
package tools
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestSpawnTool_Execute_EmptyTask(t *testing.T) {
|
|
provider := &MockLLMProvider{}
|
|
|
|
manager := NewSubagentManager(provider, "test-model", "/tmp/test", nil, nil, WebSearchToolOptions{})
|
|
|
|
tool := NewSpawnTool(manager)
|
|
|
|
ctx := context.Background()
|
|
|
|
tests := []struct {
|
|
name string
|
|
|
|
args map[string]any
|
|
}{
|
|
{"empty string", map[string]any{"task": ""}},
|
|
|
|
{"whitespace only", map[string]any{"task": " "}},
|
|
|
|
{"tabs and newlines", map[string]any{"task": "\t\n "}},
|
|
|
|
{"missing task key", map[string]any{"label": "test"}},
|
|
|
|
{"wrong type", map[string]any{"task": 123}},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
result := tool.Execute(ctx, tt.args)
|
|
|
|
if result == nil {
|
|
t.Fatal("Result should not be nil")
|
|
}
|
|
|
|
if !result.IsError {
|
|
t.Error("Expected error for invalid task parameter")
|
|
}
|
|
|
|
if !strings.Contains(result.ForLLM, `"task"`) {
|
|
t.Errorf("Error message should mention '\"task\"', got: %s", result.ForLLM)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestSpawnTool_Execute_ValidTask(t *testing.T) {
|
|
provider := &MockLLMProvider{}
|
|
|
|
manager := NewSubagentManager(provider, "test-model", "/tmp/test", nil, nil, WebSearchToolOptions{})
|
|
|
|
tool := NewSpawnTool(manager)
|
|
|
|
ctx := context.Background()
|
|
|
|
args := map[string]any{
|
|
"task": "Write a haiku about coding",
|
|
|
|
"label": "haiku-task",
|
|
}
|
|
|
|
result := tool.Execute(ctx, args)
|
|
|
|
if result == nil {
|
|
t.Fatal("Result should not be nil")
|
|
}
|
|
|
|
if result.IsError {
|
|
t.Errorf("Expected success for valid task, got error: %s", result.ForLLM)
|
|
}
|
|
|
|
if !result.Async {
|
|
t.Error("SpawnTool should return async result")
|
|
}
|
|
}
|
|
|
|
func TestSpawnTool_Execute_NilManager(t *testing.T) {
|
|
tool := NewSpawnTool(nil)
|
|
|
|
ctx := context.Background()
|
|
|
|
args := map[string]any{"task": "test task"}
|
|
|
|
result := tool.Execute(ctx, args)
|
|
|
|
if !result.IsError {
|
|
t.Error("Expected error for nil manager")
|
|
}
|
|
|
|
if !strings.Contains(result.ForLLM, "spawn tool is not available") {
|
|
t.Errorf("Error message should mention spawn tool not available, got: %s", result.ForLLM)
|
|
}
|
|
}
|