mirror of
https://github.com/Ed1s0nZ/CyberStrikeAI.git
synced 2026-08-01 16:38:48 +02:00
42 lines
1.2 KiB
Go
42 lines
1.2 KiB
Go
package mcp
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
)
|
|
|
|
func TestExecutionServiceBackgroundWaitResultCompletesWaitTool(t *testing.T) {
|
|
service := NewExecutionService(nil, nil)
|
|
handle, err := service.Submit(context.Background(), ExecutionRequest{
|
|
ToolName: "wait_tool_execution",
|
|
Run: func(context.Context) (*ToolResult, error) {
|
|
return &ToolResult{
|
|
Content: []Content{{Type: "text", Text: `{
|
|
"execution_id": "3eaaa391-050b-4be1-a870-48a855923cb7",
|
|
"tool": "exec",
|
|
"status": "running"
|
|
}
|
|
|
|
本次等待已到达 timeout_seconds，上述 execution 仍未完成。可继续等待、取消，或采用其他步骤。`}},
|
|
IsError: true,
|
|
}, nil
|
|
},
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("Submit: %v", err)
|
|
}
|
|
snap, err := service.Wait(context.Background(), handle.ID, 0)
|
|
if err != nil {
|
|
t.Fatalf("Wait: %v", err)
|
|
}
|
|
if snap == nil || snap.Execution == nil {
|
|
t.Fatal("missing execution snapshot")
|
|
}
|
|
if snap.Execution.Status != ToolExecutionStatusCompleted {
|
|
t.Fatalf("status = %q, want %q", snap.Execution.Status, ToolExecutionStatusCompleted)
|
|
}
|
|
if snap.Execution.Result == nil || !snap.Execution.Result.IsError {
|
|
t.Fatal("model-facing result should remain IsError")
|
|
}
|
|
}
|