fix tests
This commit is contained in:
@@ -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()
|
||||||
|
|||||||
Reference in New Issue
Block a user