2021-09-05 17:08:13 +08:00
|
|
|
package conf
|
|
|
|
|
|
|
|
import (
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/spf13/viper"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
)
|
|
|
|
|
2021-09-27 11:29:47 +08:00
|
|
|
func TestConf(t *testing.T) {
|
|
|
|
// 外部导入 conf.yaml 需要导入 conf 包
|
2021-10-06 18:55:23 +08:00
|
|
|
// 每次迁移文件时, 使用 sudo make init-conf来将yam文件迁移到指定的文件夹下
|
2021-09-27 11:29:47 +08:00
|
|
|
// get 使用, 读取 public_conf 配置文件
|
|
|
|
h := viper.Get("storage.host")
|
|
|
|
require.Equal(t, h, "127.0.0.1")
|
2021-09-05 17:08:13 +08:00
|
|
|
|
2021-09-27 11:29:47 +08:00
|
|
|
h = viper.Get("env")
|
|
|
|
require.Equal(t, h, "dev")
|
|
|
|
|
|
|
|
// set 使用
|
2021-09-05 17:08:13 +08:00
|
|
|
viper.Set("host", "1222")
|
|
|
|
host := viper.GetString("host")
|
|
|
|
require.Equal(t, host, "1222")
|
|
|
|
}
|