Peter Kurfer
91f0cf6963
- use `Unmarshal` method of viper - move config loading, defaulting and stuff to config package - move serve to an extra command and move keep only global flags in root command - add startup logic to onInit method in root command - update mocks - clean config structs - adopt changes in plugins - update default config
88 lines
1.7 KiB
Go
88 lines
1.7 KiB
Go
package config
|
|
|
|
import (
|
|
"github.com/spf13/pflag"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func Test_config_ReadConfig(t *testing.T) {
|
|
type args struct {
|
|
flags *pflag.FlagSet
|
|
config string
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
matcher func(Config) bool
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "Test endpoints config",
|
|
args: args{
|
|
flags: pflag.NewFlagSet("", pflag.ContinueOnError),
|
|
config: `
|
|
endpoints:
|
|
plainHttp:
|
|
handler: http_mock
|
|
listenAddress: 0.0.0.0
|
|
ports:
|
|
- 80
|
|
- 8080
|
|
options: {}
|
|
proxy:
|
|
handler: http_proxy
|
|
listenAddress: 0.0.0.0
|
|
ports:
|
|
- 3128
|
|
options:
|
|
target:
|
|
ipAddress: 127.0.0.1
|
|
port: 80
|
|
`,
|
|
},
|
|
matcher: func(c Config) bool {
|
|
if len(c.EndpointConfigs()) < 1 {
|
|
t.Error("Expected EndpointConfigs to be set but is empty")
|
|
return false
|
|
}
|
|
|
|
return true
|
|
},
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "Test loading of flags by adding",
|
|
args: args{
|
|
flags: func() *pflag.FlagSet {
|
|
flags := pflag.NewFlagSet("", pflag.ContinueOnError)
|
|
flags.String("plugins-directory", os.TempDir(), "")
|
|
|
|
return flags
|
|
}(),
|
|
config: ``,
|
|
},
|
|
matcher: func(c Config) bool {
|
|
if c.PluginsDir() != os.TempDir() {
|
|
t.Errorf("PluginsDir() Expected = %s, Got = %s", os.TempDir(), c.PluginsDir())
|
|
}
|
|
return true
|
|
},
|
|
wantErr: false,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
CreateConfig(tt.args.flags)
|
|
cfg := Instance()
|
|
if err := cfg.ReadConfigString(tt.args.config, "yaml"); (err != nil) != tt.wantErr {
|
|
t.Errorf("ReadConfig() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
|
|
if !tt.matcher(cfg) {
|
|
t.Error("matcher error")
|
|
}
|
|
})
|
|
}
|
|
}
|