forked from Zhuguoping/jackal
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig_test.go
91 lines (76 loc) · 1.81 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
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
/*
* Copyright (c) 2018 Miguel Ángel Ortuño.
* See the LICENSE file for more information.
*/
package storage
import (
"testing"
"github.com/ortuman/jackal/storage/badgerdb"
"github.com/ortuman/jackal/storage/mysql"
"github.com/stretchr/testify/require"
yaml "gopkg.in/yaml.v2"
)
func TestStorageConfig(t *testing.T) {
cfg := Config{}
memCfg := `
type: memory
`
err := yaml.Unmarshal([]byte(memCfg), &cfg)
require.Nil(t, err)
require.Equal(t, Memory, cfg.Type)
mySQLCfg := `
type: mysql
mysql:
host: 127.0.0.1
user: jackal
password: password
database: jackaldb
pool_size: 16
`
err = yaml.Unmarshal([]byte(mySQLCfg), &cfg)
require.Nil(t, err)
require.Equal(t, MySQL, cfg.Type)
require.Equal(t, "jackal", cfg.MySQL.User)
require.Equal(t, "password", cfg.MySQL.Password)
require.Equal(t, "jackaldb", cfg.MySQL.Database)
require.Equal(t, 16, cfg.MySQL.PoolSize)
mySQLCfg2 := `
type: mysql
mysql:
host: 127.0.0.1
user: jackal
password: password
database: jackaldb
`
err = yaml.Unmarshal([]byte(mySQLCfg2), &cfg)
require.Nil(t, err)
require.Equal(t, MySQL, cfg.Type)
require.Equal(t, mysql.DefaultPoolSize, cfg.MySQL.PoolSize)
invalidMySQLCfg := `
type: mysql
`
err = yaml.Unmarshal([]byte(invalidMySQLCfg), &cfg)
require.NotNil(t, err)
invalidCfg := `
type: invalid
`
err = yaml.Unmarshal([]byte(invalidCfg), &cfg)
require.NotNil(t, err)
// Test if BadgerDB config unmarshaller sets defaults
badgerCfg := `
type: badgerdb
badgerdb: {}
`
err = yaml.Unmarshal([]byte(badgerCfg), &cfg)
require.Nil(t, err)
require.NotNil(t, cfg.BadgerDB)
require.Equal(t, cfg.BadgerDB.DataDir, badgerdb.DefaultDataDir)
}
func TestStorageBadConfig(t *testing.T) {
cfg := Config{}
memCfg := `
type
`
err := yaml.Unmarshal([]byte(memCfg), &cfg)
require.NotNil(t, err)
}