6209a49d54
Error: Ran 1 stop hook
⎿ Stop hook error: Failed with non-blocking status code: Error: --session flag required (or set GT_SESSION env var, or GT_RIG/GT_ROLE)
Usage:
gt costs record [flags]
deriveSessionName() now falls back to gt-{role} when GT_ROLE is mayor
or deacon but GT_TOWN is not set. Previously this case returned empty
string, causing the Stop hook to fail.
115 lines
2.2 KiB
Go
115 lines
2.2 KiB
Go
package cmd
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestDeriveSessionName(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
envVars map[string]string
|
|
expected string
|
|
}{
|
|
{
|
|
name: "polecat session",
|
|
envVars: map[string]string{
|
|
"GT_ROLE": "polecat",
|
|
"GT_RIG": "gastown",
|
|
"GT_POLECAT": "toast",
|
|
},
|
|
expected: "gt-gastown-toast",
|
|
},
|
|
{
|
|
name: "crew session",
|
|
envVars: map[string]string{
|
|
"GT_ROLE": "crew",
|
|
"GT_RIG": "gastown",
|
|
"GT_CREW": "max",
|
|
},
|
|
expected: "gt-gastown-crew-max",
|
|
},
|
|
{
|
|
name: "witness session",
|
|
envVars: map[string]string{
|
|
"GT_ROLE": "witness",
|
|
"GT_RIG": "gastown",
|
|
},
|
|
expected: "gt-gastown-witness",
|
|
},
|
|
{
|
|
name: "refinery session",
|
|
envVars: map[string]string{
|
|
"GT_ROLE": "refinery",
|
|
"GT_RIG": "gastown",
|
|
},
|
|
expected: "gt-gastown-refinery",
|
|
},
|
|
{
|
|
name: "mayor session",
|
|
envVars: map[string]string{
|
|
"GT_ROLE": "mayor",
|
|
"GT_TOWN": "ai",
|
|
},
|
|
expected: "gt-ai-mayor",
|
|
},
|
|
{
|
|
name: "deacon session",
|
|
envVars: map[string]string{
|
|
"GT_ROLE": "deacon",
|
|
"GT_TOWN": "ai",
|
|
},
|
|
expected: "gt-ai-deacon",
|
|
},
|
|
{
|
|
name: "mayor session without GT_TOWN",
|
|
envVars: map[string]string{
|
|
"GT_ROLE": "mayor",
|
|
},
|
|
expected: "gt-mayor",
|
|
},
|
|
{
|
|
name: "deacon session without GT_TOWN",
|
|
envVars: map[string]string{
|
|
"GT_ROLE": "deacon",
|
|
},
|
|
expected: "gt-deacon",
|
|
},
|
|
{
|
|
name: "no env vars",
|
|
envVars: map[string]string{},
|
|
expected: "",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
// Save and clear relevant env vars
|
|
saved := make(map[string]string)
|
|
envKeys := []string{"GT_ROLE", "GT_RIG", "GT_POLECAT", "GT_CREW", "GT_TOWN"}
|
|
for _, key := range envKeys {
|
|
saved[key] = os.Getenv(key)
|
|
os.Unsetenv(key)
|
|
}
|
|
defer func() {
|
|
// Restore env vars
|
|
for key, val := range saved {
|
|
if val != "" {
|
|
os.Setenv(key, val)
|
|
}
|
|
}
|
|
}()
|
|
|
|
// Set test env vars
|
|
for key, val := range tt.envVars {
|
|
os.Setenv(key, val)
|
|
}
|
|
|
|
result := deriveSessionName()
|
|
if result != tt.expected {
|
|
t.Errorf("deriveSessionName() = %q, want %q", result, tt.expected)
|
|
}
|
|
})
|
|
}
|
|
}
|