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.
go-zero/core/conf/properties_test.go

49 lines
999 B
Go

package conf
import (
"os"
"testing"
"github.com/stretchr/testify/assert"
"github.com/tal-tech/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"))
}
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)
}