forked from dunglas/mercure
-
Notifications
You must be signed in to change notification settings - Fork 3
/
config_test.go
59 lines (45 loc) · 1.8 KB
/
config_test.go
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
package mercure
import (
"os"
"testing"
"github.com/spf13/pflag"
"github.com/spf13/viper"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestMissingConfig(t *testing.T) {
err := ValidateConfig(viper.New())
require.EqualError(t, err, `invalid config: one of "jwt_key" or "publisher_jwt_key" configuration parameter must be defined`)
}
func TestMissingKeyFile(t *testing.T) {
v := viper.New()
v.Set("jwt_key", "abc")
v.Set("cert_file", "foo")
err := ValidateConfig(v)
require.EqualError(t, err, `invalid config: if the "cert_file" configuration parameter is defined, "key_file" must be defined too`)
}
func TestMissingCertFile(t *testing.T) {
v := viper.New()
v.Set("jwt_key", "abc")
v.Set("key_file", "foo")
err := ValidateConfig(v)
require.EqualError(t, err, `invalid config: if the "key_file" configuration parameter is defined, "cert_file" must be defined too`)
}
func TestSetFlags(t *testing.T) {
v := viper.New()
fs := pflag.NewFlagSet("test", pflag.PanicOnError)
SetFlags(fs, v)
assert.Subset(t, v.AllKeys(), []string{"cert_file", "compress", "demo", "jwt_algorithm", "transport_url", "acme_hosts", "acme_cert_dir", "subscriber_jwt_key", "jwt_key", "allow_anonymous", "debug", "read_timeout", "publisher_jwt_algorithm", "write_timeout", "key_file", "use_forwarded_headers", "subscriber_jwt_algorithm", "addr", "publisher_jwt_key", "heartbeat_interval", "cors_allowed_origins", "publish_allowed_origins", "subscriptions", "dispatch_timeout"})
}
func TestInitConfig(t *testing.T) {
t.Setenv("JWT_KEY", "foo")
defer os.Unsetenv("JWT_KEY")
v := viper.New()
InitConfig(v)
assert.Equal(t, "foo", v.GetString("jwt_key"))
}
func TestMetricsAreDisabledByDefault(t *testing.T) {
v := viper.New()
SetConfigDefaults(v)
assert.False(t, v.GetBool("metrics_enabled"))
}