81bfe48ed3
Adds shell integration for automatic Gas Town context detection. Features: - `gt enable` / `gt disable` - Global on/off switch - `gt shell install|remove|status` - Shell integration management - `gt rig quick-add [path]` - One-command project setup - `gt uninstall` - Clean removal with options - Shell hook auto-sets GT_TOWN_ROOT/GT_RIG on cd Implementation: - XDG-compliant state storage (~/.local/state/gastown/) - Safe RC file manipulation with block markers - Environment overrides (GASTOWN_DISABLED/ENABLED) - Doctor check for global state validation Co-authored-by: Sohail Mohammad <sohailm25@gmail.com>
132 lines
3.3 KiB
Go
132 lines
3.3 KiB
Go
// ABOUTME: Tests for global state management.
|
|
// ABOUTME: Verifies enable/disable toggle and XDG path resolution.
|
|
|
|
package state
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func TestStateDir(t *testing.T) {
|
|
home, _ := os.UserHomeDir()
|
|
expected := filepath.Join(home, ".local", "state", "gastown")
|
|
|
|
os.Unsetenv("XDG_STATE_HOME")
|
|
if got := StateDir(); got != expected {
|
|
t.Errorf("StateDir() = %q, want %q", got, expected)
|
|
}
|
|
|
|
os.Setenv("XDG_STATE_HOME", "/custom/state")
|
|
defer os.Unsetenv("XDG_STATE_HOME")
|
|
if got := StateDir(); got != "/custom/state/gastown" {
|
|
t.Errorf("StateDir() with XDG = %q, want /custom/state/gastown", got)
|
|
}
|
|
}
|
|
|
|
func TestConfigDir(t *testing.T) {
|
|
home, _ := os.UserHomeDir()
|
|
expected := filepath.Join(home, ".config", "gastown")
|
|
|
|
os.Unsetenv("XDG_CONFIG_HOME")
|
|
if got := ConfigDir(); got != expected {
|
|
t.Errorf("ConfigDir() = %q, want %q", got, expected)
|
|
}
|
|
|
|
os.Setenv("XDG_CONFIG_HOME", "/custom/config")
|
|
defer os.Unsetenv("XDG_CONFIG_HOME")
|
|
if got := ConfigDir(); got != "/custom/config/gastown" {
|
|
t.Errorf("ConfigDir() with XDG = %q, want /custom/config/gastown", got)
|
|
}
|
|
}
|
|
|
|
func TestCacheDir(t *testing.T) {
|
|
home, _ := os.UserHomeDir()
|
|
expected := filepath.Join(home, ".cache", "gastown")
|
|
|
|
os.Unsetenv("XDG_CACHE_HOME")
|
|
if got := CacheDir(); got != expected {
|
|
t.Errorf("CacheDir() = %q, want %q", got, expected)
|
|
}
|
|
|
|
os.Setenv("XDG_CACHE_HOME", "/custom/cache")
|
|
defer os.Unsetenv("XDG_CACHE_HOME")
|
|
if got := CacheDir(); got != "/custom/cache/gastown" {
|
|
t.Errorf("CacheDir() with XDG = %q, want /custom/cache/gastown", got)
|
|
}
|
|
}
|
|
|
|
func TestIsEnabled_EnvOverride(t *testing.T) {
|
|
os.Setenv("GASTOWN_DISABLED", "1")
|
|
defer os.Unsetenv("GASTOWN_DISABLED")
|
|
if IsEnabled() {
|
|
t.Error("IsEnabled() should return false when GASTOWN_DISABLED=1")
|
|
}
|
|
|
|
os.Unsetenv("GASTOWN_DISABLED")
|
|
os.Setenv("GASTOWN_ENABLED", "1")
|
|
defer os.Unsetenv("GASTOWN_ENABLED")
|
|
if !IsEnabled() {
|
|
t.Error("IsEnabled() should return true when GASTOWN_ENABLED=1")
|
|
}
|
|
}
|
|
|
|
func TestIsEnabled_DisabledOverridesEnabled(t *testing.T) {
|
|
os.Setenv("GASTOWN_DISABLED", "1")
|
|
os.Setenv("GASTOWN_ENABLED", "1")
|
|
defer os.Unsetenv("GASTOWN_DISABLED")
|
|
defer os.Unsetenv("GASTOWN_ENABLED")
|
|
|
|
if IsEnabled() {
|
|
t.Error("GASTOWN_DISABLED should take precedence over GASTOWN_ENABLED")
|
|
}
|
|
}
|
|
|
|
func TestEnableDisable(t *testing.T) {
|
|
tmpDir := t.TempDir()
|
|
os.Setenv("XDG_STATE_HOME", tmpDir)
|
|
defer os.Unsetenv("XDG_STATE_HOME")
|
|
os.Unsetenv("GASTOWN_DISABLED")
|
|
os.Unsetenv("GASTOWN_ENABLED")
|
|
|
|
if err := Enable("1.0.0"); err != nil {
|
|
t.Fatalf("Enable() failed: %v", err)
|
|
}
|
|
|
|
if !IsEnabled() {
|
|
t.Error("IsEnabled() should return true after Enable()")
|
|
}
|
|
|
|
s, err := Load()
|
|
if err != nil {
|
|
t.Fatalf("Load() failed: %v", err)
|
|
}
|
|
if s.Version != "1.0.0" {
|
|
t.Errorf("State.Version = %q, want %q", s.Version, "1.0.0")
|
|
}
|
|
if s.MachineID == "" {
|
|
t.Error("State.MachineID should not be empty")
|
|
}
|
|
|
|
if err := Disable(); err != nil {
|
|
t.Fatalf("Disable() failed: %v", err)
|
|
}
|
|
|
|
if IsEnabled() {
|
|
t.Error("IsEnabled() should return false after Disable()")
|
|
}
|
|
}
|
|
|
|
func TestGenerateMachineID(t *testing.T) {
|
|
id1 := generateMachineID()
|
|
id2 := generateMachineID()
|
|
|
|
if len(id1) != 8 {
|
|
t.Errorf("generateMachineID() length = %d, want 8", len(id1))
|
|
}
|
|
if id1 == id2 {
|
|
t.Error("generateMachineID() should generate unique IDs")
|
|
}
|
|
}
|