1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
package config
import (
"os"
"testing"
)
func TestNewConfigFromEnv_AllSet(t *testing.T) {
t.Setenv("ALBATROSS_DB_HOST", "localhost")
t.Setenv("ALBATROSS_DB_PORT", "5432")
t.Setenv("ALBATROSS_DB_USER", "user")
t.Setenv("ALBATROSS_DB_PASSWORD", "pass")
t.Setenv("ALBATROSS_DB_NAME", "testdb")
t.Setenv("ALBATROSS_BASE_PATH", "/app")
t.Setenv("ALBATROSS_IS_LOCAL", "1")
conf, err := NewConfigFromEnv()
if err != nil {
t.Fatalf("unexpected error: %v", err)
}
if conf.DBHost != "localhost" {
t.Errorf("expected DBHost 'localhost', got %q", conf.DBHost)
}
if conf.DBPort != "5432" {
t.Errorf("expected DBPort '5432', got %q", conf.DBPort)
}
if conf.DBUser != "user" {
t.Errorf("expected DBUser 'user', got %q", conf.DBUser)
}
if conf.DBPassword != "pass" {
t.Errorf("expected DBPassword 'pass', got %q", conf.DBPassword)
}
if conf.DBName != "testdb" {
t.Errorf("expected DBName 'testdb', got %q", conf.DBName)
}
if conf.BasePath != "/app" {
t.Errorf("expected BasePath '/app', got %q", conf.BasePath)
}
if !conf.IsLocal {
t.Error("expected IsLocal true")
}
}
func TestNewConfigFromEnv_IsLocalFalse(t *testing.T) {
t.Setenv("ALBATROSS_DB_HOST", "localhost")
t.Setenv("ALBATROSS_DB_PORT", "5432")
t.Setenv("ALBATROSS_DB_USER", "user")
t.Setenv("ALBATROSS_DB_PASSWORD", "pass")
t.Setenv("ALBATROSS_DB_NAME", "testdb")
t.Setenv("ALBATROSS_BASE_PATH", "/app")
// ALBATROSS_IS_LOCAL not set
conf, err := NewConfigFromEnv()
if err != nil {
t.Fatalf("unexpected error: %v", err)
}
if conf.IsLocal {
t.Error("expected IsLocal false when env not set")
}
}
func TestNewConfigFromEnv_MissingRequired(t *testing.T) {
tests := []struct {
name string
envVars map[string]string
}{
{
name: "missing DB_HOST",
envVars: map[string]string{
"ALBATROSS_DB_PORT": "5432",
"ALBATROSS_DB_USER": "user",
"ALBATROSS_DB_PASSWORD": "pass",
"ALBATROSS_DB_NAME": "testdb",
"ALBATROSS_BASE_PATH": "/app",
},
},
{
name: "missing DB_PORT",
envVars: map[string]string{
"ALBATROSS_DB_HOST": "localhost",
"ALBATROSS_DB_USER": "user",
"ALBATROSS_DB_PASSWORD": "pass",
"ALBATROSS_DB_NAME": "testdb",
"ALBATROSS_BASE_PATH": "/app",
},
},
{
name: "missing DB_USER",
envVars: map[string]string{
"ALBATROSS_DB_HOST": "localhost",
"ALBATROSS_DB_PORT": "5432",
"ALBATROSS_DB_PASSWORD": "pass",
"ALBATROSS_DB_NAME": "testdb",
"ALBATROSS_BASE_PATH": "/app",
},
},
{
name: "missing DB_PASSWORD",
envVars: map[string]string{
"ALBATROSS_DB_HOST": "localhost",
"ALBATROSS_DB_PORT": "5432",
"ALBATROSS_DB_USER": "user",
"ALBATROSS_DB_NAME": "testdb",
"ALBATROSS_BASE_PATH": "/app",
},
},
{
name: "missing DB_NAME",
envVars: map[string]string{
"ALBATROSS_DB_HOST": "localhost",
"ALBATROSS_DB_PORT": "5432",
"ALBATROSS_DB_USER": "user",
"ALBATROSS_DB_PASSWORD": "pass",
"ALBATROSS_BASE_PATH": "/app",
},
},
{
name: "missing BASE_PATH",
envVars: map[string]string{
"ALBATROSS_DB_HOST": "localhost",
"ALBATROSS_DB_PORT": "5432",
"ALBATROSS_DB_USER": "user",
"ALBATROSS_DB_PASSWORD": "pass",
"ALBATROSS_DB_NAME": "testdb",
},
},
}
allKeys := []string{
"ALBATROSS_DB_HOST",
"ALBATROSS_DB_PORT",
"ALBATROSS_DB_USER",
"ALBATROSS_DB_PASSWORD",
"ALBATROSS_DB_NAME",
"ALBATROSS_BASE_PATH",
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
for _, k := range allKeys {
if v, ok := tt.envVars[k]; ok {
t.Setenv(k, v)
} else {
t.Setenv(k, "")
os.Unsetenv(k)
}
}
_, err := NewConfigFromEnv()
if err == nil {
t.Error("expected error for missing env var, got nil")
}
})
}
}
|
