Support toml config
This commit is contained in:
Ming Deng
380
core/config/toml/toml_test.go
Normal file
380
core/config/toml/toml_test.go
Normal file
@@ -0,0 +1,380 @@
|
||||
// Copyright 2020
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package toml
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
|
||||
"github.com/astaxie/beego/core/config"
|
||||
)
|
||||
|
||||
func TestConfig_Parse(t *testing.T) {
|
||||
// file not found
|
||||
cfg := &Config{}
|
||||
_, err := cfg.Parse("invalid_file_name.txt")
|
||||
assert.NotNil(t, err)
|
||||
}
|
||||
|
||||
func TestConfig_ParseData(t *testing.T) {
|
||||
data := `
|
||||
name="Tom"
|
||||
`
|
||||
cfg := &Config{}
|
||||
c, err := cfg.ParseData([]byte(data))
|
||||
assert.Nil(t, err)
|
||||
assert.NotNil(t, c)
|
||||
}
|
||||
|
||||
func TestConfigContainer_Bool(t *testing.T) {
|
||||
data := `
|
||||
Man=true
|
||||
Woman="true"
|
||||
`
|
||||
cfg := &Config{}
|
||||
c, err := cfg.ParseData([]byte(data))
|
||||
assert.Nil(t, err)
|
||||
assert.NotNil(t, c)
|
||||
|
||||
val, err := c.Bool(context.Background(), "Man")
|
||||
assert.Nil(t, err)
|
||||
assert.True(t, val)
|
||||
|
||||
_, err = c.Bool(context.Background(), "Woman")
|
||||
assert.NotNil(t, err)
|
||||
assert.Equal(t, config.InvalidValueTypeError, err)
|
||||
}
|
||||
|
||||
func TestConfigContainer_DefaultBool(t *testing.T) {
|
||||
data := `
|
||||
Man=true
|
||||
Woman="false"
|
||||
`
|
||||
cfg := &Config{}
|
||||
c, err := cfg.ParseData([]byte(data))
|
||||
assert.Nil(t, err)
|
||||
assert.NotNil(t, c)
|
||||
|
||||
val := c.DefaultBool(context.Background(), "Man11", true)
|
||||
assert.True(t, val)
|
||||
|
||||
val = c.DefaultBool(context.Background(), "Man", false)
|
||||
assert.True(t, val)
|
||||
|
||||
val = c.DefaultBool(context.Background(), "Woman", true)
|
||||
assert.True(t, val)
|
||||
}
|
||||
|
||||
func TestConfigContainer_DefaultFloat(t *testing.T) {
|
||||
data := `
|
||||
Price=12.3
|
||||
PriceInvalid="12.3"
|
||||
`
|
||||
cfg := &Config{}
|
||||
c, err := cfg.ParseData([]byte(data))
|
||||
assert.Nil(t, err)
|
||||
assert.NotNil(t, c)
|
||||
|
||||
val := c.DefaultFloat(context.Background(), "Price", 11.2)
|
||||
assert.Equal(t, 12.3, val)
|
||||
|
||||
val = c.DefaultFloat(context.Background(), "Price11", 11.2)
|
||||
assert.Equal(t, 11.2, val)
|
||||
|
||||
val = c.DefaultFloat(context.Background(), "PriceInvalid", 11.2)
|
||||
assert.Equal(t, 11.2, val)
|
||||
}
|
||||
|
||||
func TestConfigContainer_DefaultInt(t *testing.T) {
|
||||
data := `
|
||||
Age=12
|
||||
AgeInvalid="13"
|
||||
`
|
||||
cfg := &Config{}
|
||||
c, err := cfg.ParseData([]byte(data))
|
||||
assert.Nil(t, err)
|
||||
assert.NotNil(t, c)
|
||||
|
||||
val := c.DefaultInt(context.Background(), "Age", 11)
|
||||
assert.Equal(t, 12, val)
|
||||
|
||||
val = c.DefaultInt(context.Background(), "Price11", 11)
|
||||
assert.Equal(t, 11, val)
|
||||
|
||||
val = c.DefaultInt(context.Background(), "PriceInvalid", 11)
|
||||
assert.Equal(t, 11, val)
|
||||
}
|
||||
|
||||
func TestConfigContainer_DefaultString(t *testing.T) {
|
||||
data := `
|
||||
Name="Tom"
|
||||
NameInvalid=13
|
||||
`
|
||||
cfg := &Config{}
|
||||
c, err := cfg.ParseData([]byte(data))
|
||||
assert.Nil(t, err)
|
||||
assert.NotNil(t, c)
|
||||
|
||||
val := c.DefaultString(context.Background(), "Name", "Jerry")
|
||||
assert.Equal(t, "Tom", val)
|
||||
|
||||
val = c.DefaultString(context.Background(), "Name11", "Jerry")
|
||||
assert.Equal(t, "Jerry", val)
|
||||
|
||||
val = c.DefaultString(context.Background(), "NameInvalid", "Jerry")
|
||||
assert.Equal(t, "Jerry", val)
|
||||
}
|
||||
|
||||
func TestConfigContainer_DefaultStrings(t *testing.T) {
|
||||
data := `
|
||||
Name=["Tom", "Jerry"]
|
||||
NameInvalid="Tom"
|
||||
`
|
||||
cfg := &Config{}
|
||||
c, err := cfg.ParseData([]byte(data))
|
||||
assert.Nil(t, err)
|
||||
assert.NotNil(t, c)
|
||||
|
||||
val := c.DefaultStrings(context.Background(), "Name", []string{"Jerry"})
|
||||
assert.Equal(t, []string{"Tom", "Jerry"}, val)
|
||||
|
||||
val = c.DefaultStrings(context.Background(), "Name11", []string{"Jerry"})
|
||||
assert.Equal(t, []string{"Jerry"}, val)
|
||||
|
||||
val = c.DefaultStrings(context.Background(), "NameInvalid", []string{"Jerry"})
|
||||
assert.Equal(t, []string{"Jerry"}, val)
|
||||
}
|
||||
|
||||
func TestConfigContainer_DIY(t *testing.T) {
|
||||
data := `
|
||||
Name=["Tom", "Jerry"]
|
||||
`
|
||||
cfg := &Config{}
|
||||
c, err := cfg.ParseData([]byte(data))
|
||||
assert.Nil(t, err)
|
||||
assert.NotNil(t, c)
|
||||
|
||||
_, err = c.DIY(context.Background(), "Name")
|
||||
assert.Nil(t, err)
|
||||
}
|
||||
|
||||
func TestConfigContainer_Float(t *testing.T) {
|
||||
data := `
|
||||
Price=12.3
|
||||
PriceInvalid="12.3"
|
||||
`
|
||||
cfg := &Config{}
|
||||
c, err := cfg.ParseData([]byte(data))
|
||||
assert.Nil(t, err)
|
||||
assert.NotNil(t, c)
|
||||
|
||||
val, err := c.Float(context.Background(), "Price")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, 12.3, val)
|
||||
|
||||
_, err = c.Float(context.Background(), "Price11")
|
||||
assert.Equal(t, config.KeyNotFoundError, err)
|
||||
|
||||
_, err = c.Float(context.Background(), "PriceInvalid")
|
||||
assert.Equal(t, config.InvalidValueTypeError, err)
|
||||
}
|
||||
|
||||
func TestConfigContainer_Int(t *testing.T) {
|
||||
data := `
|
||||
Age=12
|
||||
AgeInvalid="13"
|
||||
`
|
||||
cfg := &Config{}
|
||||
c, err := cfg.ParseData([]byte(data))
|
||||
assert.Nil(t, err)
|
||||
assert.NotNil(t, c)
|
||||
|
||||
val, err := c.Int(context.Background(), "Age")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, 12, val)
|
||||
|
||||
_, err = c.Int(context.Background(), "Age11")
|
||||
assert.Equal(t, config.KeyNotFoundError, err)
|
||||
|
||||
_, err = c.Int(context.Background(), "AgeInvalid")
|
||||
assert.Equal(t, config.InvalidValueTypeError, err)
|
||||
}
|
||||
|
||||
func TestConfigContainer_GetSection(t *testing.T) {
|
||||
data := `
|
||||
[servers]
|
||||
|
||||
# You can indent as you please. Tabs or spaces. TOML don't care.
|
||||
[servers.alpha]
|
||||
ip = "10.0.0.1"
|
||||
dc = "eqdc10"
|
||||
|
||||
[servers.beta]
|
||||
ip = "10.0.0.2"
|
||||
dc = "eqdc10"
|
||||
`
|
||||
cfg := &Config{}
|
||||
c, err := cfg.ParseData([]byte(data))
|
||||
assert.Nil(t, err)
|
||||
assert.NotNil(t, c)
|
||||
|
||||
m, err := c.GetSection(context.Background(), "servers")
|
||||
assert.Nil(t, err)
|
||||
assert.NotNil(t, m)
|
||||
assert.Equal(t, 2, len(m))
|
||||
}
|
||||
|
||||
func TestConfigContainer_String(t *testing.T) {
|
||||
data := `
|
||||
Name="Tom"
|
||||
NameInvalid=13
|
||||
[Person]
|
||||
Name="Jerry"
|
||||
`
|
||||
cfg := &Config{}
|
||||
c, err := cfg.ParseData([]byte(data))
|
||||
assert.Nil(t, err)
|
||||
assert.NotNil(t, c)
|
||||
|
||||
val, err := c.String(context.Background(), "Name")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "Tom", val)
|
||||
|
||||
_, err = c.String(context.Background(), "Name11")
|
||||
assert.Equal(t, config.KeyNotFoundError, err)
|
||||
|
||||
_, err = c.String(context.Background(), "NameInvalid")
|
||||
assert.Equal(t, config.InvalidValueTypeError, err)
|
||||
|
||||
val, err = c.String(context.Background(), "Person.Name")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "Jerry", val)
|
||||
}
|
||||
|
||||
func TestConfigContainer_Strings(t *testing.T) {
|
||||
data := `
|
||||
Name=["Tom", "Jerry"]
|
||||
NameInvalid="Tom"
|
||||
`
|
||||
cfg := &Config{}
|
||||
c, err := cfg.ParseData([]byte(data))
|
||||
assert.Nil(t, err)
|
||||
assert.NotNil(t, c)
|
||||
|
||||
val, err := c.Strings(context.Background(), "Name")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, []string{"Tom", "Jerry"}, val)
|
||||
|
||||
_, err = c.Strings(context.Background(), "Name11")
|
||||
assert.Equal(t, config.KeyNotFoundError, err)
|
||||
|
||||
_, err = c.Strings(context.Background(), "NameInvalid")
|
||||
assert.Equal(t, config.InvalidValueTypeError, err)
|
||||
}
|
||||
|
||||
func TestConfigContainer_Set(t *testing.T) {
|
||||
data := `
|
||||
Name=["Tom", "Jerry"]
|
||||
NameInvalid="Tom"
|
||||
`
|
||||
cfg := &Config{}
|
||||
c, err := cfg.ParseData([]byte(data))
|
||||
assert.Nil(t, err)
|
||||
assert.NotNil(t, c)
|
||||
|
||||
err = c.Set(context.Background(), "Age", "11")
|
||||
assert.Nil(t, err)
|
||||
age, err := c.String(context.Background(), "Age")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "11", age)
|
||||
}
|
||||
|
||||
func TestConfigContainer_SubAndMushall(t *testing.T) {
|
||||
data := `
|
||||
[servers]
|
||||
|
||||
# You can indent as you please. Tabs or spaces. TOML don't care.
|
||||
[servers.alpha]
|
||||
ip = "10.0.0.1"
|
||||
dc = "eqdc10"
|
||||
|
||||
[servers.beta]
|
||||
ip = "10.0.0.2"
|
||||
dc = "eqdc10"
|
||||
`
|
||||
cfg := &Config{}
|
||||
c, err := cfg.ParseData([]byte(data))
|
||||
assert.Nil(t, err)
|
||||
assert.NotNil(t, c)
|
||||
|
||||
sub, err := c.Sub(context.Background(), "servers")
|
||||
assert.Nil(t, err)
|
||||
assert.NotNil(t, sub)
|
||||
|
||||
sub, err = sub.Sub(context.Background(), "alpha")
|
||||
assert.Nil(t, err)
|
||||
assert.NotNil(t, sub)
|
||||
ip, err := sub.String(context.Background(), "ip")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "10.0.0.1", ip)
|
||||
|
||||
svr := &Server{}
|
||||
err = sub.Unmarshaler(context.Background(), "", svr)
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "10.0.0.1", svr.Ip)
|
||||
|
||||
svr = &Server{}
|
||||
err = c.Unmarshaler(context.Background(), "servers.alpha", svr)
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "10.0.0.1", svr.Ip)
|
||||
}
|
||||
|
||||
func TestConfigContainer_SaveConfigFile(t *testing.T) {
|
||||
filename := "test_config.toml"
|
||||
path := os.TempDir() + string(os.PathSeparator) + filename
|
||||
data := `
|
||||
[servers]
|
||||
|
||||
# You can indent as you please. Tabs or spaces. TOML don't care.
|
||||
[servers.alpha]
|
||||
ip = "10.0.0.1"
|
||||
dc = "eqdc10"
|
||||
|
||||
[servers.beta]
|
||||
ip = "10.0.0.2"
|
||||
dc = "eqdc10"
|
||||
`
|
||||
cfg := &Config{}
|
||||
c, err := cfg.ParseData([]byte(data))
|
||||
|
||||
fmt.Println(path)
|
||||
|
||||
assert.Nil(t, err)
|
||||
assert.NotNil(t, c)
|
||||
|
||||
sub, err := c.Sub(context.Background(), "servers")
|
||||
assert.Nil(t, err)
|
||||
|
||||
err = sub.SaveConfigFile(context.Background(), path)
|
||||
assert.Nil(t, err)
|
||||
}
|
||||
|
||||
type Server struct {
|
||||
Ip string `toml:"ip"`
|
||||
}
|
||||
Reference in New Issue
Block a user