5853f80c48
Some checks failed
CI Docker Mining Proof / Linux agent hashrate proof (push) Has been cancelled
Prosecutor and Public Defender use real fleet telemetry only; Judge dispatches L4 commands and emits full transcripts via seer_events and emberwake_court_debate when ai_control_enabled.
76 lines
2.4 KiB
Go
76 lines
2.4 KiB
Go
package ai
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestShouldUseSurgicalReplayPartialFailureWithTrace(t *testing.T) {
|
|
trace := SurgicalTraceContext{SessionID: "sess-abc", HopIndex: 1, HopCount: 3}
|
|
snap := AgentSnapshot{
|
|
LOTLAttempts: []TierAttempt{
|
|
{Tier: "vuln_recon", OK: true},
|
|
{Tier: "docker", OK: false, Error: "daemon missing"},
|
|
},
|
|
}
|
|
if !ShouldUseSurgicalReplay(trace, snap) {
|
|
t.Fatal("expected surgical replay for partial spread failure with trace")
|
|
}
|
|
}
|
|
|
|
func TestShouldUseSurgicalReplaySkipsFullCourtStuck(t *testing.T) {
|
|
trace := SurgicalTraceContext{SessionID: "sess-abc"}
|
|
snap := AgentSnapshot{Stuck: true, LOTLAttempts: stuckSpreadAttempts()}
|
|
if ShouldUseSurgicalReplay(trace, snap) {
|
|
t.Fatal("expected court path, not surgical replay, when fully stuck")
|
|
}
|
|
}
|
|
|
|
func TestShouldUseSurgicalReplayRequiresTrace(t *testing.T) {
|
|
snap := AgentSnapshot{
|
|
LOTLAttempts: []TierAttempt{{Tier: "docker", OK: false}},
|
|
}
|
|
if ShouldUseSurgicalReplay(SurgicalTraceContext{}, snap) {
|
|
t.Fatal("expected false without pathtrace session")
|
|
}
|
|
}
|
|
|
|
func TestBuildSurgicalReplayPromptMentionsFailedTier(t *testing.T) {
|
|
bundle := BuildSurgicalDiagnosticBundle(
|
|
AgentSnapshot{AgentID: "a1", Name: "host", JoinLane: "do_peer"},
|
|
SurgicalTraceContext{SessionID: "trace-123", HopIndex: 0, HopCount: 2, DiscoverError: "no smb"},
|
|
"wsl 3/3 failed", PersonaBalanced, false,
|
|
)
|
|
bundle.FailedTier = "docker"
|
|
bundle.FailedError = "not installed"
|
|
bundle.FailedAttempts = []TierAttempt{{Tier: "docker", OK: false, Error: "not installed"}}
|
|
_, user := BuildSurgicalReplayPrompt(bundle)
|
|
if !strings.Contains(user, "docker") {
|
|
t.Fatalf("missing failed tier in prompt: %s", user)
|
|
}
|
|
if !strings.Contains(user, "trace-123") {
|
|
t.Fatalf("missing session id in prompt: %s", user)
|
|
}
|
|
if !strings.Contains(user, "no smb") {
|
|
t.Fatalf("missing discover error in prompt: %s", user)
|
|
}
|
|
}
|
|
|
|
func TestSelectSurgicalCommandPicksFirstActionable(t *testing.T) {
|
|
cmd, ok := SelectSurgicalCommand([]Command{
|
|
{Type: CmdNoop},
|
|
{Type: CmdSkipTier, Args: map[string]interface{}{"tier": "docker"}},
|
|
{Type: CmdRestartMining},
|
|
})
|
|
if !ok || cmd.Type != CmdSkipTier {
|
|
t.Fatalf("got %+v ok=%v", cmd, ok)
|
|
}
|
|
}
|
|
|
|
func TestExpandSurgicalCommandSkipTier(t *testing.T) {
|
|
out := ExpandSurgicalCommand(Command{Type: CmdSkipTier, Args: map[string]interface{}{"tier": "wsl"}})
|
|
if len(out) != 1 || out[0].Type != CmdReorderTiers {
|
|
t.Fatalf("got %+v", out)
|
|
}
|
|
}
|