key_test.go

package nodis

import (
	"os"
	"testing"
	"time"

	"github.com/diiyw/nodis/ds"
	"github.com/diiyw/nodis/fs"
)

func TestKey_Expire(t *testing.T) {
	_ = os.RemoveAll("testdata")
	n := Open(&Options{
		Path:         "testdata",
		TidyDuration: 60 * time.Second,
	})
	n.Set("test", []byte("test1"), false)
	n.Expire("test", 2)
	time.Sleep(time.Second * 2)
	if n.TTL("test") > 0 {
		t.Errorf("Get() = %v, want <= %vs", n.TTL("test"), 0)
	}
}

func TestKey_ExpireXX(t *testing.T) {
	_ = os.RemoveAll("testdata")
	n := Open(&Options{
		Path: "testdata",
	})
	n.Set("test", []byte("test1"), false)
	n.ExpireXX("test", 3)
	if n.TTL("test") != -1 {
		t.Errorf("TTL() = %v, want %v", 0, n.TTL("test"))
	}
	n.SetEX("test2", []byte("test2"), 1)
	n.ExpireXX("test2", 2)
	if int64(n.TTL("test2").Seconds()) < 2 {
		t.Errorf("TTL() = %v, want %vs", n.TTL("test2"), 3)
	}
}

func TestKey_ExpireNX(t *testing.T) {
	_ = os.RemoveAll("testdata")
	n := Open(&Options{
		Path:         "testdata",
		TidyDuration: 60 * time.Second,
	})
	n.Set("test", []byte("test1"), false)
	n.ExpireNX("test", 2)
	if int64(n.TTL("test").Seconds()) == 0 {
		t.Errorf("TTL() = %v, want %vs", n.TTL("test"), 0)
	}
	n.SetEX("test2", []byte("test2"), 2)
	v := n.ExpireNX("test2", 5)
	if v != 0 {
		t.Errorf("ExpireNX() = %v, want %vs", v, 0)
	}
}

func TestKey_ExpireLT(t *testing.T) {
	_ = os.RemoveAll("testdata")
	n := Open(&Options{
		Path:         "testdata",
		TidyDuration: 60 * time.Second,
	})
	n.Set("test", []byte("test1"), false)
	v := n.ExpireLT("test", 2)
	if v != 0 {
		t.Errorf("ExpireLT() = %v, want %v", v, 0)
	}
	n.SetEX("test", []byte("test1"), 5)
	v = n.ExpireLT("test", 1)
	if v == 0 {
		t.Errorf("ExpireLT() = %v, want %v", v, 1)
	}
}

func TestKey_ExpireGT(t *testing.T) {
	_ = os.RemoveAll("testdata")
	n := Open(&Options{
		Path:         "testdata",
		TidyDuration: 60 * time.Second,
	})
	n.Set("test", []byte("test1"), false)
	v := n.ExpireGT("test", 2)
	if v == 0 {
		t.Errorf("ExpireGT() = %v, want not %v", v, 0)
	}
	n.SetEX("test", []byte("test1"), 5)
	v = n.ExpireGT("test", 1)
	if v != 0 {
		t.Errorf("ExpireGT() = %v, want %v", v, 0)
	}
}

func TestKey_ExpireAt(t *testing.T) {
	_ = os.RemoveAll("testdata")
	n := Open(&Options{
		Path:         "testdata",
		TidyDuration: 60 * time.Second,
	})
	n.Set("test", []byte("test1"), false)
	n.ExpireAt("test", time.Now().Add(2*time.Second))
	time.Sleep(2 * time.Second)
	v := n.Get("test")
	if v != nil {
		t.Errorf("Get() = %v, want %v", true, false)
	}
}

func TestKey_TTL(t *testing.T) {
	_ = os.RemoveAll("testdata")
	n := Open(&Options{
		Path:         "testdata",
		TidyDuration: 60 * time.Second,
	})
	n.Set("test", []byte("test1"), false)
	n.Expire("test", 300)
	v := n.TTL("test")
	if v < 299 {
		t.Errorf("TTL() = %v, want > %v", v, 0)
	}
	time.Sleep(2 * time.Second)
	v = n.TTL("test")
	if v < 0 {
		t.Errorf("TTL() = %v, want > %v", v, 0)
	}
}

func TestKey_Rename(t *testing.T) {
	_ = os.RemoveAll("testdata")
	n := Open(&Options{
		Path:         "testdata",
		TidyDuration: 60 * time.Second,
	})
	n.Set("test", []byte("test1"), false)
	n.Rename("test", "test2")
	v := n.Get("test")
	if v != nil {
		t.Errorf("Get() = %v, want %v", true, false)
	}
	v = n.Get("test2")
	if v == nil {
		t.Errorf("Get() = %v, want %v", false, true)
	}
}

func TestKey_RenameNX(t *testing.T) {
	_ = os.RemoveAll("testdata")
	n := Open(&Options{
		Path: "testdata",
	})
	n.Set("test", []byte("test1"), false)
	n.Set("test2", []byte("test2"), false)
	err := n.RenameNX("test", "test2")
	if err == nil {
		t.Errorf("RenameNX() = %v, want %v", err, "key exists")
	}
	err = n.RenameNX("test", "test3")
	if err != nil {
		t.Errorf("RenameNX() = %v, want %v", err, nil)
	}
}

func TestKey_Keys(t *testing.T) {
	_ = os.RemoveAll("testdata")
	n := Open(&Options{
		Path:         "testdata",
		TidyDuration: 60 * time.Second,
	})
	n.Set("test1", []byte("test1"), false)
	n.Set("test2", []byte("test2"), false)
	n.Set("test3", []byte("test3"), false)
	keys := n.Keys("test*")
	if len(keys) != 3 {
		t.Errorf("Keys() = %v, want %v", len(keys), 3)
	}
}

func TestKey_Type(t *testing.T) {
	_ = os.RemoveAll("testdata")
	n := Open(&Options{
		Path:         "testdata",
		TidyDuration: 60 * time.Second,
		Filesystem:   &fs.Disk{},
	})
	n.Set("test1", []byte("test1"), false)
	n.LPush("test2", []byte("test2"))
	n.ZAdd("test3", "test3", 10)
	n.SAdd("test4", "test4")
	n.HSet("test5", "test5", []byte("test5"))
	if n.Type("test1") != "string" {
		t.Errorf("Type() = %v, want %v", n.Type("test1"), "string")
	}
	if n.Type("test2") != "list" {
		t.Errorf("Type() = %v, want %v", n.Type("test2"), "int")
	}
	if n.Type("test3") != "zset" {
		t.Errorf("Type() = %v, want %v", n.Type("test3"), "float64")
	}
	if n.Type("test4") != "set" {
		t.Errorf("Type() = %v, want %v", n.Type("test4"), "string")
	}
	if n.Type("test5") != "hash" {
		t.Errorf("Type() = %v, want %v", n.Type("test5"), "bool")
	}
	err := n.Close()
	if err != nil {
		t.Fatalf("Close() = %v, want %v", err, nil)
	}
	n = Open(&Options{
		Path:         "testdata",
		TidyDuration: 60 * time.Second,
		Filesystem:   &fs.Disk{},
	})
	if n.Type("test1") != "string" {
		t.Errorf("Type() = %v, want %v", n.Type("test1"), "string")
	}
	if n.Type("test2") != "list" {
		t.Errorf("Type() = %v, want %v", n.Type("test2"), "int")
	}
	if n.Type("test3") != "zset" {
		t.Errorf("Type() = %v, want %v", n.Type("test3"), "float64")
	}
	if n.Type("test4") != "set" {
		t.Errorf("Type() = %v, want %v", n.Type("test4"), "string")
	}
	if n.Type("test5") != "hash" {
		t.Errorf("Type() = %v, want %v", n.Type("test5"), "hash")
	}
}

func TestKey_Scan(t *testing.T) {
	_ = os.RemoveAll("testdata")
	n := Open(&Options{
		Path:         "testdata",
		TidyDuration: 60 * time.Second,
	})
	n.Set("test1", []byte("test1"), false)
	n.Set("test2", []byte("test2"), false)
	n.Set("test3", []byte("test3"), false)
	n.Set("test4", []byte("test4"), false)
	n.Set("test5", []byte("test5"), false)
	n.Set("test6", []byte("test6"), false)
	n.Set("test7", []byte("test7"), false)
	n.Set("test8", []byte("test8"), false)
	n.Set("test9", []byte("test9"), false)
	n.Set("test10", []byte("test10"), false)
	n.Set("test11", []byte("test11"), false)
	n.Set("test12", []byte("test12"), false)
	n.Set("test13", []byte("test13"), false)
	n.Set("test14", []byte("test14"), false)
	n.Set("test15", []byte("test15"), false)
	n.Set("test16", []byte("test16"), false)
	n.Set("test17", []byte("test17"), false)
	n.Set("test18", []byte("test18"), false)
	n.Set("test19", []byte("test19"), false)
	n.Set("test20", []byte("test20"), false)
	n.Set("test21", []byte("test21"), false)
	n.Set("test22", []byte("test22"), false)
	n.Set("test23", []byte("test23"), false)
	n.Set("test24", []byte("test24"), false)
	n.Set("test25", []byte("test25"), false)
	n.Set("test26", []byte("test26"), false)
	n.Set("test27", []byte("test27"), false)
	n.Set("test28", []byte("test28"), false)
	n.Set("test29", []byte("test29"), false)
	n.Set("test30", []byte("test30"), false)
	n.Set("test31", []byte("test31"), false)
	_, result := n.Scan(0, "test*", 10, ds.String)
	if len(result) != 10 {
		t.Errorf("Scan() = %v, want %v", len(result), 10)
	}
	_, result = n.Scan(32, "test*", 10, ds.String)
	if len(result) != 0 {
		t.Errorf("Scan() = %v, want %v", len(result), 0)
	}
	_, result = n.Scan(0, "test*", 32, ds.String)
	if len(result) != 31 {
		t.Errorf("Scan() = %v, want %v", len(result), 31)
	}
	_, result = n.Scan(0, "test*", 31, ds.String)
	if len(result) != 31 {
		t.Errorf("Scan() = %v, want %v", len(result), 31)
	}
	_, result = n.Scan(23, "test*", 10, ds.String)
	if len(result) != 9 {
		t.Errorf("Scan() = %v, want %v", len(result), 9)
	}
}

func TestKey_Exists(t *testing.T) {
	_ = os.RemoveAll("testdata")
	n := Open(&Options{
		Path:         "testdata",
		TidyDuration: 60 * time.Second,
	})
	n.Set("test", []byte("test1"), false)
	if n.Exists("test") != 1 {
		t.Errorf("Exists() = %v, want %v", n.Exists("test"), true)
	}
	n.Del("test")
	if n.Exists("test") == 1 {
		t.Errorf("Exists() = %v, want %v", n.Exists("test"), false)
	}
}