You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
109 lines
2.6 KiB
109 lines
2.6 KiB
6 years ago
|
package paladin
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"io/ioutil"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestNewFile(t *testing.T) {
|
||
|
|
// test data
|
||
|
|
path := "/tmp/test_conf/"
|
||
|
|
assert.Nil(t, os.MkdirAll(path, 0700))
|
||
|
|
assert.Nil(t, ioutil.WriteFile(path+"test.toml", []byte(`
|
||
|
|
text = "hello"
|
||
|
|
number = 100
|
||
|
|
slice = [1, 2, 3]
|
||
|
|
sliceStr = ["1", "2", "3"]
|
||
|
|
`), 0644))
|
||
|
|
// test client
|
||
|
|
cli, err := NewFile(path + "test.toml")
|
||
|
|
assert.Nil(t, err)
|
||
|
|
assert.NotNil(t, cli)
|
||
|
|
// test map
|
||
|
|
m := Map{}
|
||
|
|
text, err := cli.Get("test.toml").String()
|
||
|
|
assert.Nil(t, err)
|
||
|
|
assert.Nil(t, m.Set(text), "text")
|
||
|
|
s, err := m.Get("text").String()
|
||
|
|
assert.Nil(t, err)
|
||
|
|
assert.Equal(t, s, "hello", "text")
|
||
|
|
n, err := m.Get("number").Int64()
|
||
|
|
assert.Nil(t, err)
|
||
|
|
assert.Equal(t, n, int64(100), "number")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNewFilePath(t *testing.T) {
|
||
|
|
// test data
|
||
|
|
path := "/tmp/test_conf/"
|
||
|
|
assert.Nil(t, os.MkdirAll(path, 0700))
|
||
|
|
assert.Nil(t, ioutil.WriteFile(path+"test.toml", []byte(`
|
||
|
|
text = "hello"
|
||
|
|
number = 100
|
||
|
|
`), 0644))
|
||
|
|
assert.Nil(t, ioutil.WriteFile(path+"abc.toml", []byte(`
|
||
|
|
text = "hello"
|
||
|
|
number = 100
|
||
|
|
`), 0644))
|
||
|
|
// test client
|
||
|
|
cli, err := NewFile(path)
|
||
|
|
assert.Nil(t, err)
|
||
|
|
assert.NotNil(t, cli)
|
||
|
|
// test map
|
||
|
|
m := Map{}
|
||
|
|
text, err := cli.Get("test.toml").String()
|
||
|
|
assert.Nil(t, err)
|
||
|
|
assert.Nil(t, m.Set(text), "text")
|
||
|
|
s, err := m.Get("text").String()
|
||
|
|
assert.Nil(t, err, s)
|
||
|
|
assert.Equal(t, s, "hello", "text")
|
||
|
|
n, err := m.Get("number").Int64()
|
||
|
|
assert.Nil(t, err, s)
|
||
|
|
assert.Equal(t, n, int64(100), "number")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestFileEvent(t *testing.T) {
|
||
|
|
// test data
|
||
|
|
path := "/tmp/test_conf_event/"
|
||
|
|
assert.Nil(t, os.MkdirAll(path, 0700))
|
||
|
|
assert.Nil(t, ioutil.WriteFile(path+"test.toml", []byte(`
|
||
|
|
text = "hello"
|
||
|
|
number = 100
|
||
|
|
`), 0644))
|
||
|
|
assert.Nil(t, ioutil.WriteFile(path+"abc.toml", []byte(`
|
||
|
|
text = "hello"
|
||
|
|
number = 100
|
||
|
|
`), 0644))
|
||
|
|
// test client
|
||
|
|
cli, err := NewFile(path)
|
||
|
|
assert.Nil(t, err)
|
||
|
|
assert.NotNil(t, cli)
|
||
|
|
ch := cli.WatchEvent(context.Background(), "test.toml", "abc.toml")
|
||
|
|
time.Sleep(time.Millisecond)
|
||
|
|
ioutil.WriteFile(path+"test.toml", []byte(`hello`), 0644)
|
||
|
|
timeout := time.NewTimer(time.Second)
|
||
|
|
select {
|
||
|
|
case <-timeout.C:
|
||
|
|
t.Fatalf("run test timeout")
|
||
|
|
case ev := <-ch:
|
||
|
|
assert.Equal(t, EventUpdate, ev.Event)
|
||
|
|
assert.Equal(t, "hello", ev.Value)
|
||
|
|
}
|
||
|
|
ioutil.WriteFile(path+"abc.toml", []byte(`test`), 0644)
|
||
|
|
select {
|
||
|
|
case <-timeout.C:
|
||
|
|
t.Fatalf("run test timeout")
|
||
|
|
case ev := <-ch:
|
||
|
|
assert.Equal(t, EventUpdate, ev.Event)
|
||
|
|
assert.Equal(t, "test", ev.Value)
|
||
|
|
}
|
||
|
|
content1, _ := cli.Get("test.toml").String()
|
||
|
|
assert.Equal(t, "hello", content1)
|
||
|
|
content2, _ := cli.Get("abc.toml").String()
|
||
|
|
assert.Equal(t, "test", content2)
|
||
|
|
}
|