106 lines
3.7 KiB
Go
106 lines
3.7 KiB
Go
package config
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestLoadFromEnvUsesDefaults(t *testing.T) {
|
|
for _, key := range []string{
|
|
"GO_OMNI_HTTP_ADDR", "GO_OMNI_REQUEST_ID_HEADER", "GO_OMNI_ENABLE_LOGGER",
|
|
"GO_OMNI_TRUSTED_PROXIES",
|
|
"GO_OMNI_LOG_LEVEL", "GO_OMNI_LOG_FORMAT", "GO_OMNI_LOG_OUTPUT",
|
|
"GO_OMNI_LOG_FILE", "GO_OMNI_SERVICE", "GO_OMNI_ENVIRONMENT",
|
|
"GO_OMNI_LOG_ADD_SOURCE", "GO_OMNI_SHUTDOWN_TIMEOUT",
|
|
} {
|
|
t.Setenv(key, "")
|
|
}
|
|
|
|
cfg, err := LoadFromEnv("")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if cfg.Address != ":8080" || cfg.RequestIDHeader != "X-Request-ID" || cfg.ShutdownTimeout != 10*time.Second {
|
|
t.Fatalf("unexpected defaults: %+v", cfg)
|
|
}
|
|
if cfg.EnableLogger || cfg.Log.Level != "info" || cfg.Log.Format != "json" || cfg.Log.Output != "stderr" {
|
|
t.Fatalf("unexpected logger defaults: %+v", cfg.Log)
|
|
}
|
|
}
|
|
|
|
func TestLoadFromEnvOverridesSettings(t *testing.T) {
|
|
t.Setenv("GO_OMNI_HTTP_ADDR", "127.0.0.1:9090")
|
|
t.Setenv("GO_OMNI_REQUEST_ID_HEADER", "X-Correlation-ID")
|
|
t.Setenv("GO_OMNI_TRUSTED_PROXIES", "127.0.0.1, 10.0.0.0/8")
|
|
t.Setenv("GO_OMNI_ENABLE_LOGGER", "true")
|
|
t.Setenv("GO_OMNI_LOG_LEVEL", "debug")
|
|
t.Setenv("GO_OMNI_LOG_FORMAT", "text")
|
|
t.Setenv("GO_OMNI_LOG_OUTPUT", "stdout")
|
|
t.Setenv("GO_OMNI_SERVICE", "billing")
|
|
t.Setenv("GO_OMNI_ENVIRONMENT", "test")
|
|
t.Setenv("GO_OMNI_LOG_ADD_SOURCE", "true")
|
|
t.Setenv("GO_OMNI_SHUTDOWN_TIMEOUT", "3s")
|
|
|
|
cfg, err := LoadFromEnv("")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if cfg.Address != "127.0.0.1:9090" || cfg.RequestIDHeader != "X-Correlation-ID" || !cfg.EnableLogger || len(cfg.TrustedProxies) != 2 || cfg.TrustedProxies[1] != "10.0.0.0/8" {
|
|
t.Fatalf("unexpected application settings: %+v", cfg)
|
|
}
|
|
if cfg.ShutdownTimeout != 3*time.Second || cfg.Log.Level != "debug" || cfg.Log.Format != "text" || cfg.Log.Output != "stdout" || cfg.Log.Service != "billing" || cfg.Log.Environment != "test" || !cfg.Log.AddSource {
|
|
t.Fatalf("unexpected logger settings: %+v", cfg.Log)
|
|
}
|
|
}
|
|
|
|
func TestLoadFromEnvRejectsInvalidValues(t *testing.T) {
|
|
t.Setenv("GO_OMNI_ENABLE_LOGGER", "sometimes")
|
|
if _, err := LoadFromEnv(""); err == nil || !strings.Contains(err.Error(), "ENABLE_LOGGER") {
|
|
t.Fatalf("expected invalid logger flag error, got %v", err)
|
|
}
|
|
|
|
t.Setenv("GO_OMNI_ENABLE_LOGGER", "")
|
|
t.Setenv("GO_OMNI_SHUTDOWN_TIMEOUT", "later")
|
|
if _, err := LoadFromEnv(""); err == nil || !strings.Contains(err.Error(), "SHUTDOWN_TIMEOUT") {
|
|
t.Fatalf("expected invalid timeout error, got %v", err)
|
|
}
|
|
|
|
t.Setenv("GO_OMNI_SHUTDOWN_TIMEOUT", "1s")
|
|
t.Setenv("GO_OMNI_TRUSTED_PROXIES", "proxy.example.com")
|
|
if _, err := LoadFromEnv(""); err == nil || !strings.Contains(err.Error(), "trusted_proxies") {
|
|
t.Fatalf("expected invalid trusted proxy error, got %v", err)
|
|
}
|
|
t.Setenv("GO_OMNI_TRUSTED_PROXIES", "")
|
|
cfg := Default()
|
|
cfg.Address = ""
|
|
if err := cfg.Validate(); err == nil || !strings.Contains(err.Error(), "address") {
|
|
t.Fatalf("expected invalid address error, got %v", err)
|
|
}
|
|
cfg = Default()
|
|
cfg.RequestIDHeader = " "
|
|
if err := cfg.Validate(); err == nil || !strings.Contains(err.Error(), "request_id_header") {
|
|
t.Fatalf("expected invalid request header error, got %v", err)
|
|
}
|
|
}
|
|
|
|
func TestWithConfigAndWithLogConfig(t *testing.T) {
|
|
settings := Default()
|
|
settings.Log.Service = "orders"
|
|
settings.TrustedProxies = []string{"127.0.0.1"}
|
|
cfg := Default()
|
|
WithConfig(settings)(&cfg)
|
|
if cfg.Log.Service != "orders" || len(cfg.TrustedProxies) != 1 {
|
|
t.Fatalf("complete config was not applied: %+v", cfg)
|
|
}
|
|
settings.TrustedProxies[0] = "10.0.0.0/8"
|
|
if cfg.TrustedProxies[0] != "127.0.0.1" {
|
|
t.Fatal("WithConfig should copy trusted proxy settings")
|
|
}
|
|
|
|
WithLogConfig(Default().Log)(&cfg)
|
|
if cfg.Log.Service != "" {
|
|
t.Fatalf("log config was not replaced: %+v", cfg.Log)
|
|
}
|
|
}
|