69 lines
1.2 KiB
Go
69 lines
1.2 KiB
Go
|
package config
|
||
|
|
||
|
import (
|
||
|
"github.com/spf13/pflag"
|
||
|
"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,
|
||
|
},
|
||
|
}
|
||
|
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")
|
||
|
}
|
||
|
})
|
||
|
}
|
||
|
}
|