forked from p93542168/wheat-cache
25 lines
458 B
Go
25 lines
458 B
Go
package conf
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/spf13/viper"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestConf(t *testing.T) {
|
|
// 外部导入 conf.yaml 需要导入 conf 包
|
|
|
|
// get 使用, 读取 public_conf 配置文件
|
|
h := viper.Get("storage.host")
|
|
require.Equal(t, h, "127.0.0.1")
|
|
|
|
h = viper.Get("env")
|
|
require.Equal(t, h, "dev")
|
|
|
|
// set 使用
|
|
viper.Set("host", "1222")
|
|
host := viper.GetString("host")
|
|
require.Equal(t, host, "1222")
|
|
}
|