Files
gastown/internal/cmd/statusline_test.go
joe 4bcf50bf1c Revert to simple gt-mayor/gt-deacon session names
Reverts the session naming changes from PR #70. Multi-town support
on a single machine is not a real use case - rigs provide project
isolation, and true isolation should use containers/VMs.

Changes:
- MayorSessionName() and DeaconSessionName() no longer take townName parameter
- ParseSessionName() handles simple gt-mayor and gt-deacon formats
- Removed Town field from AgentIdentity and AgentSession structs
- Updated all callers and tests

Generated with Claude Code

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
2026-01-03 14:33:24 -08:00

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)
{"gt-mayor", ""},
{"gt-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
{"gt-mayor", AgentMayor},
{"gt-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)
}
})
}
}