fix tests

This commit is contained in:
LeonardoTrapani
2025-09-07 19:09:52 +02:00
parent 03e5e6e66b
commit 00dcfa3b9b
+14
View File
@@ -262,13 +262,20 @@ func TestConfig_Load(t *testing.T) {
t.Run("creates default config when none exists", func(t *testing.T) { t.Run("creates default config when none exists", func(t *testing.T) {
tempDir := t.TempDir() tempDir := t.TempDir()
originalConfigDir := os.Getenv("XDG_CONFIG_HOME") originalConfigDir := os.Getenv("XDG_CONFIG_HOME")
originalAPIKey := os.Getenv("OPENAI_API_KEY")
os.Setenv("XDG_CONFIG_HOME", tempDir) os.Setenv("XDG_CONFIG_HOME", tempDir)
os.Setenv("OPENAI_API_KEY", "test-api-key") // Set test API key for validation
defer func() { defer func() {
if originalConfigDir == "" { if originalConfigDir == "" {
os.Unsetenv("XDG_CONFIG_HOME") os.Unsetenv("XDG_CONFIG_HOME")
} else { } else {
os.Setenv("XDG_CONFIG_HOME", originalConfigDir) os.Setenv("XDG_CONFIG_HOME", originalConfigDir)
} }
if originalAPIKey == "" {
os.Unsetenv("OPENAI_API_KEY")
} else {
os.Setenv("OPENAI_API_KEY", originalAPIKey)
}
}() }()
config, err := Load() config, err := Load()
@@ -362,13 +369,20 @@ func TestConfig_SaveDefaultConfig(t *testing.T) {
// Override the config path by setting environment variable // Override the config path by setting environment variable
tempDir := t.TempDir() tempDir := t.TempDir()
originalConfigDir := os.Getenv("XDG_CONFIG_HOME") originalConfigDir := os.Getenv("XDG_CONFIG_HOME")
originalAPIKey := os.Getenv("OPENAI_API_KEY")
os.Setenv("XDG_CONFIG_HOME", tempDir) os.Setenv("XDG_CONFIG_HOME", tempDir)
os.Setenv("OPENAI_API_KEY", "test-api-key") // Set test API key for validation
defer func() { defer func() {
if originalConfigDir == "" { if originalConfigDir == "" {
os.Unsetenv("XDG_CONFIG_HOME") os.Unsetenv("XDG_CONFIG_HOME")
} else { } else {
os.Setenv("XDG_CONFIG_HOME", originalConfigDir) os.Setenv("XDG_CONFIG_HOME", originalConfigDir)
} }
if originalAPIKey == "" {
os.Unsetenv("OPENAI_API_KEY")
} else {
os.Setenv("OPENAI_API_KEY", originalAPIKey)
}
}() }()
err := SaveDefaultConfig() err := SaveDefaultConfig()