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) {
tempDir := t.TempDir()
originalConfigDir := os.Getenv("XDG_CONFIG_HOME")
originalAPIKey := os.Getenv("OPENAI_API_KEY")
os.Setenv("XDG_CONFIG_HOME", tempDir)
os.Setenv("OPENAI_API_KEY", "test-api-key") // Set test API key for validation
defer func() {
if originalConfigDir == "" {
os.Unsetenv("XDG_CONFIG_HOME")
} else {
os.Setenv("XDG_CONFIG_HOME", originalConfigDir)
}
if originalAPIKey == "" {
os.Unsetenv("OPENAI_API_KEY")
} else {
os.Setenv("OPENAI_API_KEY", originalAPIKey)
}
}()
config, err := Load()
@@ -362,13 +369,20 @@ func TestConfig_SaveDefaultConfig(t *testing.T) {
// Override the config path by setting environment variable
tempDir := t.TempDir()
originalConfigDir := os.Getenv("XDG_CONFIG_HOME")
originalAPIKey := os.Getenv("OPENAI_API_KEY")
os.Setenv("XDG_CONFIG_HOME", tempDir)
os.Setenv("OPENAI_API_KEY", "test-api-key") // Set test API key for validation
defer func() {
if originalConfigDir == "" {
os.Unsetenv("XDG_CONFIG_HOME")
} else {
os.Setenv("XDG_CONFIG_HOME", originalConfigDir)
}
if originalAPIKey == "" {
os.Unsetenv("OPENAI_API_KEY")
} else {
os.Setenv("OPENAI_API_KEY", originalAPIKey)
}
}()
err := SaveDefaultConfig()