updated ezconf
This commit is contained in:
@@ -3,6 +3,7 @@ package ezconf
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
@@ -240,6 +241,163 @@ func TestGetEnvVars(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestParseEnvVars(t *testing.T) {
|
||||
loader := New()
|
||||
|
||||
// Add a test config function
|
||||
loader.AddConfigFunc("test", func() (interface{}, error) {
|
||||
return "test config", nil
|
||||
})
|
||||
|
||||
// Add current package path
|
||||
loader.AddPackagePath(".")
|
||||
|
||||
// Add an extra env var
|
||||
loader.AddEnvVar(EnvVar{
|
||||
Name: "EXTRA_VAR",
|
||||
Description: "Extra test variable",
|
||||
Default: "extra",
|
||||
})
|
||||
|
||||
err := loader.ParseEnvVars()
|
||||
if err != nil {
|
||||
t.Fatalf("ParseEnvVars failed: %v", err)
|
||||
}
|
||||
|
||||
// Check that env vars were extracted
|
||||
envVars := loader.GetEnvVars()
|
||||
if len(envVars) == 0 {
|
||||
t.Error("expected at least one env var")
|
||||
}
|
||||
|
||||
// Check for extra var
|
||||
foundExtra := false
|
||||
for _, ev := range envVars {
|
||||
if ev.Name == "EXTRA_VAR" {
|
||||
foundExtra = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !foundExtra {
|
||||
t.Error("extra env var not found")
|
||||
}
|
||||
|
||||
// Check that configs are NOT loaded (should be empty)
|
||||
configs := loader.GetAllConfigs()
|
||||
if len(configs) != 0 {
|
||||
t.Errorf("expected no configs loaded after ParseEnvVars, got %d", len(configs))
|
||||
}
|
||||
}
|
||||
|
||||
func TestLoadConfigs(t *testing.T) {
|
||||
loader := New()
|
||||
|
||||
// Add a test config function
|
||||
testCfg := struct {
|
||||
Value string
|
||||
}{Value: "test"}
|
||||
|
||||
loader.AddConfigFunc("test", func() (interface{}, error) {
|
||||
return testCfg, nil
|
||||
})
|
||||
|
||||
// Manually set some env vars (simulating ParseEnvVars already called)
|
||||
loader.envVars = []EnvVar{
|
||||
{Name: "TEST_VAR", Description: "Test variable"},
|
||||
}
|
||||
|
||||
err := loader.LoadConfigs()
|
||||
if err != nil {
|
||||
t.Fatalf("LoadConfigs failed: %v", err)
|
||||
}
|
||||
|
||||
// Check that config was loaded
|
||||
cfg, ok := loader.GetConfig("test")
|
||||
if !ok {
|
||||
t.Error("test config not loaded")
|
||||
}
|
||||
if cfg == nil {
|
||||
t.Error("test config is nil")
|
||||
}
|
||||
_ = cfg // Use the variable to avoid unused variable error
|
||||
|
||||
// Check that env vars are NOT modified (should remain as set)
|
||||
envVars := loader.GetEnvVars()
|
||||
if len(envVars) != 1 {
|
||||
t.Errorf("expected 1 env var, got %d", len(envVars))
|
||||
}
|
||||
}
|
||||
|
||||
func TestLoadConfigs_Error(t *testing.T) {
|
||||
loader := New()
|
||||
|
||||
loader.AddConfigFunc("error", func() (interface{}, error) {
|
||||
return nil, os.ErrNotExist
|
||||
})
|
||||
|
||||
err := loader.LoadConfigs()
|
||||
if err == nil {
|
||||
t.Error("expected error from failing config func")
|
||||
}
|
||||
}
|
||||
|
||||
func TestParseEnvVars_Then_LoadConfigs(t *testing.T) {
|
||||
loader := New()
|
||||
|
||||
// Add a test config function
|
||||
testCfg := struct {
|
||||
Value string
|
||||
}{Value: "test"}
|
||||
|
||||
loader.AddConfigFunc("test", func() (interface{}, error) {
|
||||
return testCfg, nil
|
||||
})
|
||||
|
||||
// Add current package path
|
||||
loader.AddPackagePath(".")
|
||||
|
||||
// Add an extra env var
|
||||
loader.AddEnvVar(EnvVar{
|
||||
Name: "EXTRA_VAR",
|
||||
Description: "Extra test variable",
|
||||
Default: "extra",
|
||||
})
|
||||
|
||||
// First parse env vars
|
||||
err := loader.ParseEnvVars()
|
||||
if err != nil {
|
||||
t.Fatalf("ParseEnvVars failed: %v", err)
|
||||
}
|
||||
|
||||
// Check env vars are extracted but configs are not loaded
|
||||
envVars := loader.GetEnvVars()
|
||||
if len(envVars) == 0 {
|
||||
t.Error("expected env vars to be extracted")
|
||||
}
|
||||
|
||||
configs := loader.GetAllConfigs()
|
||||
if len(configs) != 0 {
|
||||
t.Error("expected no configs loaded yet")
|
||||
}
|
||||
|
||||
// Then load configs
|
||||
err = loader.LoadConfigs()
|
||||
if err != nil {
|
||||
t.Fatalf("LoadConfigs failed: %v", err)
|
||||
}
|
||||
|
||||
// Check both env vars and configs are loaded
|
||||
_, ok := loader.GetConfig("test")
|
||||
if !ok {
|
||||
t.Error("test config not loaded after LoadConfigs")
|
||||
}
|
||||
|
||||
configs = loader.GetAllConfigs()
|
||||
if len(configs) != 1 {
|
||||
t.Errorf("expected 1 config loaded, got %d", len(configs))
|
||||
}
|
||||
}
|
||||
|
||||
func TestLoad_Integration(t *testing.T) {
|
||||
// Integration test with real hlog package
|
||||
hlogPath := filepath.Join("..", "hlog")
|
||||
@@ -269,3 +427,62 @@ func TestLoad_Integration(t *testing.T) {
|
||||
t.Logf(" %s: %s (default: %s, required: %t)", ev.Name, ev.Description, ev.Default, ev.Required)
|
||||
}
|
||||
}
|
||||
|
||||
func TestParseEnvVars_GenerateEnvFile_Integration(t *testing.T) {
|
||||
// Test the new separated ParseEnvVars functionality
|
||||
hlogPath := filepath.Join("..", "hlog")
|
||||
if _, err := os.Stat(hlogPath); os.IsNotExist(err) {
|
||||
t.Skip("hlog package not found, skipping integration test")
|
||||
}
|
||||
|
||||
loader := New()
|
||||
|
||||
// Add hlog package
|
||||
if err := loader.AddPackagePath(hlogPath); err != nil {
|
||||
t.Fatalf("failed to add hlog package: %v", err)
|
||||
}
|
||||
|
||||
// Parse env vars without loading configs (this should work even if required env vars are missing)
|
||||
if err := loader.ParseEnvVars(); err != nil {
|
||||
t.Fatalf("ParseEnvVars failed: %v", err)
|
||||
}
|
||||
|
||||
envVars := loader.GetEnvVars()
|
||||
if len(envVars) == 0 {
|
||||
t.Error("expected env vars from hlog package")
|
||||
}
|
||||
|
||||
// Now test that we can generate an env file without calling Load()
|
||||
tempDir := t.TempDir()
|
||||
envFile := filepath.Join(tempDir, "test-generated.env")
|
||||
|
||||
err := loader.GenerateEnvFile(envFile, false)
|
||||
if err != nil {
|
||||
t.Fatalf("GenerateEnvFile failed: %v", err)
|
||||
}
|
||||
|
||||
// Verify the file was created and contains expected content
|
||||
content, err := os.ReadFile(envFile)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to read generated file: %v", err)
|
||||
}
|
||||
|
||||
output := string(content)
|
||||
if !strings.Contains(output, "# Environment Configuration") {
|
||||
t.Error("expected header in generated file")
|
||||
}
|
||||
|
||||
// Should contain environment variables from hlog
|
||||
foundHlogVar := false
|
||||
for _, ev := range envVars {
|
||||
if strings.Contains(output, ev.Name) {
|
||||
foundHlogVar = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !foundHlogVar {
|
||||
t.Error("expected to find at least one hlog environment variable in generated file")
|
||||
}
|
||||
|
||||
t.Logf("Successfully generated env file with %d variables", len(envVars))
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user