mirror of
https://github.com/etcd-io/etcd.git
synced 2024-09-27 06:25:44 +00:00
151 lines
3.1 KiB
Go
151 lines
3.1 KiB
Go
package fileSystem
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestSetAndGet(t *testing.T) {
|
|
fs := New()
|
|
setAndGet(fs, "/foobar", t)
|
|
setAndGet(fs, "/foo/bar", t)
|
|
setAndGet(fs, "/foo/foo/bar", t)
|
|
}
|
|
|
|
func TestListDirectory(t *testing.T) {
|
|
fs := New()
|
|
|
|
// create dir /foo
|
|
// set key-value /foo/foo=bar
|
|
fs.Set("/foo/foo", "bar", Permanent, 1, 1)
|
|
|
|
// create dir /foo/fooDir
|
|
// set key-value /foo/fooDir/foo=bar
|
|
fs.Set("/foo/fooDir/foo", "bar", Permanent, 2, 1)
|
|
|
|
e, err := fs.Get("/foo", false, 2, 1)
|
|
|
|
if err != nil {
|
|
t.Fatalf("%v", err)
|
|
}
|
|
|
|
if len(e.KVPairs) != 2 {
|
|
t.Fatalf("wrong number of kv pairs [%d/2]", len(e.KVPairs))
|
|
}
|
|
|
|
if e.KVPairs[0].Key != "/foo/foo" || e.KVPairs[0].Value != "bar" {
|
|
t.Fatalf("wrong kv [/foo/foo/ / %s] -> [bar / %s]", e.KVPairs[0].Key, e.KVPairs[0].Value)
|
|
}
|
|
|
|
if e.KVPairs[1].Key != "/foo/fooDir" || e.KVPairs[1].Dir != true {
|
|
t.Fatalf("wrong kv [/foo/fooDir/ / %s] -> [true / %v]", e.KVPairs[1].Key, e.KVPairs[1].Dir)
|
|
}
|
|
|
|
// test hidden node
|
|
|
|
// create dir /foo/_hidden
|
|
// set key-value /foo/_hidden/foo -> bar
|
|
fs.Set("/foo/_hidden/foo", "bar", Permanent, 3, 1)
|
|
|
|
e, _ = fs.Get("/foo", false, 2, 1)
|
|
|
|
if len(e.KVPairs) != 2 {
|
|
t.Fatalf("hidden node is not hidden! %s", e.KVPairs[2].Key)
|
|
}
|
|
}
|
|
|
|
func TestRemove(t *testing.T) {
|
|
fs := New()
|
|
|
|
fs.Set("/foo", "bar", Permanent, 1, 1)
|
|
_, err := fs.Delete("/foo", false, 1, 1)
|
|
|
|
if err != nil {
|
|
t.Fatalf("cannot delete %s [%s]", "/foo", err.Error())
|
|
}
|
|
|
|
_, err = fs.Get("/foo", false, 1, 1)
|
|
|
|
if err == nil || err.Error() != "Key Not Found" {
|
|
t.Fatalf("can get the node after deletion")
|
|
}
|
|
|
|
fs.Set("/foo/bar", "bar", Permanent, 1, 1)
|
|
fs.Set("/foo/car", "car", Permanent, 1, 1)
|
|
fs.Set("/foo/dar/dar", "dar", Permanent, 1, 1)
|
|
|
|
_, err = fs.Delete("/foo", false, 1, 1)
|
|
|
|
if err == nil {
|
|
t.Fatalf("should not be able to delete a directory without recursive")
|
|
}
|
|
|
|
_, err = fs.Delete("/foo", true, 1, 1)
|
|
|
|
if err != nil {
|
|
t.Fatalf("cannot delete %s [%s]", "/foo", err.Error())
|
|
}
|
|
|
|
_, err = fs.Get("/foo", false, 1, 1)
|
|
|
|
if err == nil || err.Error() != "Key Not Found" {
|
|
t.Fatalf("can get the node after deletion ")
|
|
}
|
|
|
|
}
|
|
|
|
func TestExpire(t *testing.T) {
|
|
fs := New()
|
|
|
|
expire := time.Now().Add(time.Second)
|
|
|
|
fs.Set("/foo", "bar", expire, 1, 1)
|
|
|
|
_, err := fs.InternalGet("/foo", 1, 1)
|
|
|
|
if err != nil {
|
|
t.Fatalf("can not get the node")
|
|
}
|
|
|
|
time.Sleep(time.Second * 2)
|
|
|
|
_, err = fs.InternalGet("/foo", 1, 1)
|
|
|
|
if err == nil {
|
|
t.Fatalf("can get the node after expiration time")
|
|
}
|
|
|
|
fs.Set("/foo", "bar", expire, 1, 1)
|
|
|
|
time.Sleep(time.Millisecond * 50)
|
|
_, err = fs.InternalGet("/foo", 1, 1)
|
|
|
|
if err == nil {
|
|
t.Fatalf("can get the node after expiration time")
|
|
}
|
|
|
|
expire = time.Now().Add(time.Second)
|
|
|
|
fs.Set("/foo", "bar", expire, 1, 1)
|
|
fs.Delete("/foo", false, 1, 1)
|
|
|
|
}
|
|
|
|
func setAndGet(fs *FileSystem, path string, t *testing.T) {
|
|
_, err := fs.Set(path, "bar", Permanent, 1, 1)
|
|
|
|
if err != nil {
|
|
t.Fatalf("cannot set %s=bar [%s]", path, err.Error())
|
|
}
|
|
|
|
e, err := fs.Get(path, false, 1, 1)
|
|
|
|
if err != nil {
|
|
t.Fatalf("cannot get %s [%s]", path, err.Error())
|
|
}
|
|
|
|
if e.Value != "bar" {
|
|
t.Fatalf("expect value of %s is bar [%s]", path, e.Value)
|
|
}
|
|
}
|