6b8c897e37
Town-level services (Mayor, Deacon) now use hq- prefix instead of gt-: - hq-mayor (was gt-mayor) - hq-deacon (was gt-deacon) This distinguishes town-level sessions from rig-level sessions which continue to use gt- prefix (gt-gastown-witness, gt-gastown-crew-max, etc). Changes: - session.MayorSessionName() returns "hq-mayor" - session.DeaconSessionName() returns "hq-deacon" - ParseSessionName() handles both hq- and gt- prefixes - categorizeSession() handles both prefixes - categorizeSessions() accepts both prefixes - Updated all tests and documentation 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
87 lines
2.1 KiB
Go
87 lines
2.1 KiB
Go
package cmd
|
|
|
|
import "testing"
|
|
|
|
func TestCategorizeSessionRig(t *testing.T) {
|
|
tests := []struct {
|
|
session string
|
|
wantRig string
|
|
}{
|
|
// Standard polecat sessions
|
|
{"gt-gastown-slit", "gastown"},
|
|
{"gt-gastown-Toast", "gastown"},
|
|
{"gt-myrig-worker", "myrig"},
|
|
|
|
// Crew sessions
|
|
{"gt-gastown-crew-max", "gastown"},
|
|
{"gt-myrig-crew-user", "myrig"},
|
|
|
|
// Witness sessions (canonical format: gt-<rig>-witness)
|
|
{"gt-gastown-witness", "gastown"},
|
|
{"gt-myrig-witness", "myrig"},
|
|
// Legacy format still works as fallback
|
|
{"gt-witness-gastown", "gastown"},
|
|
{"gt-witness-myrig", "myrig"},
|
|
|
|
// Refinery sessions
|
|
{"gt-gastown-refinery", "gastown"},
|
|
{"gt-myrig-refinery", "myrig"},
|
|
|
|
// Edge cases
|
|
{"gt-a-b", "a"}, // minimum valid
|
|
|
|
// Town-level agents (no rig, use hq- prefix)
|
|
{"hq-mayor", ""},
|
|
{"hq-deacon", ""},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.session, func(t *testing.T) {
|
|
agent := categorizeSession(tt.session)
|
|
gotRig := ""
|
|
if agent != nil {
|
|
gotRig = agent.Rig
|
|
}
|
|
if gotRig != tt.wantRig {
|
|
t.Errorf("categorizeSession(%q).Rig = %q, want %q", tt.session, gotRig, tt.wantRig)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestCategorizeSessionType(t *testing.T) {
|
|
tests := []struct {
|
|
session string
|
|
wantType AgentType
|
|
}{
|
|
// Polecat sessions
|
|
{"gt-gastown-slit", AgentPolecat},
|
|
{"gt-gastown-Toast", AgentPolecat},
|
|
{"gt-myrig-worker", AgentPolecat},
|
|
{"gt-a-b", AgentPolecat},
|
|
|
|
// Non-polecat sessions
|
|
{"gt-gastown-witness", AgentWitness}, // canonical format
|
|
{"gt-witness-gastown", AgentWitness}, // legacy fallback
|
|
{"gt-gastown-refinery", AgentRefinery},
|
|
{"gt-gastown-crew-max", AgentCrew},
|
|
{"gt-myrig-crew-user", AgentCrew},
|
|
|
|
// Town-level agents (hq- prefix)
|
|
{"hq-mayor", AgentMayor},
|
|
{"hq-deacon", AgentDeacon},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.session, func(t *testing.T) {
|
|
agent := categorizeSession(tt.session)
|
|
if agent == nil {
|
|
t.Fatalf("categorizeSession(%q) returned nil", tt.session)
|
|
}
|
|
if agent.Type != tt.wantType {
|
|
t.Errorf("categorizeSession(%q).Type = %v, want %v", tt.session, agent.Type, tt.wantType)
|
|
}
|
|
})
|
|
}
|
|
}
|