You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
96 lines
2.2 KiB
Go
96 lines
2.2 KiB
Go
package conf
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/zeromicro/go-zero/core/fs"
|
|
)
|
|
|
|
func TestProperties(t *testing.T) {
|
|
text := `app.name = test
|
|
|
|
app.program=app
|
|
|
|
# this is comment
|
|
app.threads = 5`
|
|
tmpfile, err := fs.TempFilenameWithText(text)
|
|
assert.Nil(t, err)
|
|
defer os.Remove(tmpfile)
|
|
|
|
props, err := LoadProperties(tmpfile)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "test", props.GetString("app.name"))
|
|
assert.Equal(t, "app", props.GetString("app.program"))
|
|
assert.Equal(t, 5, props.GetInt("app.threads"))
|
|
|
|
val := props.ToString()
|
|
assert.Contains(t, val, "app.name")
|
|
assert.Contains(t, val, "app.program")
|
|
assert.Contains(t, val, "app.threads")
|
|
}
|
|
|
|
func TestPropertiesEnv(t *testing.T) {
|
|
text := `app.name = test
|
|
|
|
app.program=app
|
|
|
|
app.env1 = ${FOO}
|
|
app.env2 = $none
|
|
|
|
# this is comment
|
|
app.threads = 5`
|
|
tmpfile, err := fs.TempFilenameWithText(text)
|
|
assert.Nil(t, err)
|
|
defer os.Remove(tmpfile)
|
|
|
|
os.Setenv("FOO", "2")
|
|
defer os.Unsetenv("FOO")
|
|
|
|
props, err := LoadProperties(tmpfile, UseEnv())
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "test", props.GetString("app.name"))
|
|
assert.Equal(t, "app", props.GetString("app.program"))
|
|
assert.Equal(t, 5, props.GetInt("app.threads"))
|
|
assert.Equal(t, "2", props.GetString("app.env1"))
|
|
assert.Equal(t, "", props.GetString("app.env2"))
|
|
|
|
val := props.ToString()
|
|
assert.Contains(t, val, "app.name")
|
|
assert.Contains(t, val, "app.program")
|
|
assert.Contains(t, val, "app.threads")
|
|
assert.Contains(t, val, "app.env1")
|
|
assert.Contains(t, val, "app.env2")
|
|
}
|
|
|
|
func TestLoadProperties_badContent(t *testing.T) {
|
|
filename, err := fs.TempFilenameWithText("hello")
|
|
assert.Nil(t, err)
|
|
defer os.Remove(filename)
|
|
_, err = LoadProperties(filename)
|
|
assert.NotNil(t, err)
|
|
assert.True(t, len(err.Error()) > 0)
|
|
}
|
|
|
|
func TestSetString(t *testing.T) {
|
|
key := "a"
|
|
value := "the value of a"
|
|
props := NewProperties()
|
|
props.SetString(key, value)
|
|
assert.Equal(t, value, props.GetString(key))
|
|
}
|
|
|
|
func TestSetInt(t *testing.T) {
|
|
key := "a"
|
|
value := 101
|
|
props := NewProperties()
|
|
props.SetInt(key, value)
|
|
assert.Equal(t, value, props.GetInt(key))
|
|
}
|
|
|
|
func TestLoadBadFile(t *testing.T) {
|
|
_, err := LoadProperties("nosuchfile")
|
|
assert.NotNil(t, err)
|
|
}
|