ledisdb/ledis/scan_test.go

package ledis

import (
	"fmt"
	"github.com/siddontang/ledisdb/store"
	"testing"
)

func TestDBScan(t *testing.T) {
	db := getTestDB()

	db.FlushAll()

	if v, err := db.Scan(nil, 10, true); err != nil {
		t.Fatal(err)
	} else if len(v) != 0 {
		t.Fatal(len(v))
	}

	db.Set([]byte("a"), []byte{})
	db.Set([]byte("b"), []byte{})
	db.Set([]byte("c"), []byte{})

	if v, err := db.Scan(nil, 1, true); err != nil {
		t.Fatal(err)
	} else if len(v) != 1 {
		t.Fatal(len(v))
	}

	if v, err := db.Scan([]byte("a"), 2, false); err != nil {
		t.Fatal(err)
	} else if len(v) != 2 {
		t.Fatal(len(v))
	}

	if v, err := db.Scan(nil, 3, true); err != nil {
		t.Fatal(err)
	} else if len(v) != 3 {
		t.Fatal(len(v))
	}
}

func TestDBHScan(t *testing.T) {
	db := getTestDB()

	db.hFlush()

	k1 := []byte("k1")
	db.HSet(k1, []byte("1"), []byte{})

	k2 := []byte("k2")
	db.HSet(k2, []byte("2"), []byte{})

	k3 := []byte("k3")
	db.HSet(k3, []byte("3"), []byte{})

	if v, err := db.HScan(nil, 1, true); err != nil {
		t.Fatal(err)
	} else if len(v) != 1 {
		t.Fatal("invalid length ", len(v))
	} else if string(v[0]) != "k1" {
		t.Fatal("invalid value ", string(v[0]))
	}

	if v, err := db.HScan(k1, 2, true); err != nil {
		t.Fatal(err)
	} else if len(v) != 2 {
		t.Fatal("invalid length ", len(v))
	} else if string(v[0]) != "k1" {
		t.Fatal("invalid value ", string(v[0]))
	} else if string(v[1]) != "k2" {
		t.Fatal("invalid value ", string(v[1]))
	}

	if v, err := db.HScan(k1, 2, false); err != nil {
		t.Fatal(err)
	} else if len(v) != 2 {
		t.Fatal("invalid length ", len(v))
	} else if string(v[0]) != "k2" {
		t.Fatal("invalid value ", string(v[0]))
	} else if string(v[1]) != "k3" {
		t.Fatal("invalid value ", string(v[1]))
	}

}

func TestDBZScan(t *testing.T) {
	db := getTestDB()

	db.zFlush()

	k1 := []byte("k1")
	db.ZAdd(k1, ScorePair{1, []byte("m")})

	k2 := []byte("k2")
	db.ZAdd(k2, ScorePair{2, []byte("m")})

	k3 := []byte("k3")
	db.ZAdd(k3, ScorePair{3, []byte("m")})

	if v, err := db.ZScan(nil, 1, true); err != nil {
		t.Fatal(err)
	} else if len(v) != 1 {
		t.Fatal("invalid length ", len(v))
	} else if string(v[0]) != "k1" {
		t.Fatal("invalid value ", string(v[0]))
	}

	if v, err := db.ZScan(k1, 2, true); err != nil {
		t.Fatal(err)
	} else if len(v) != 2 {
		t.Fatal("invalid length ", len(v))
	} else if string(v[0]) != "k1" {
		t.Fatal("invalid value ", string(v[0]))
	} else if string(v[1]) != "k2" {
		t.Fatal("invalid value ", string(v[1]))
	}

	if v, err := db.ZScan(k1, 2, false); err != nil {
		t.Fatal(err)
	} else if len(v) != 2 {
		t.Fatal("invalid length ", len(v))
	} else if string(v[0]) != "k2" {
		t.Fatal("invalid value ", string(v[0]))
	} else if string(v[1]) != "k3" {
		t.Fatal("invalid value ", string(v[1]))
	}

}

func TestDBLScan(t *testing.T) {
	db := getTestDB()

	db.lFlush()

	k1 := []byte("k1")
	if _, err := db.LPush(k1, []byte("elem")); err != nil {
		t.Fatal(err.Error())
	}

	k2 := []byte("k2")
	if _, err := db.LPush(k2, []byte("elem")); err != nil {
		t.Fatal(err.Error())
	}

	k3 := []byte("k3")
	if _, err := db.LPush(k3, []byte("elem")); err != nil {
		t.Fatal(err.Error())
	}

	if v, err := db.LScan(nil, 1, true); err != nil {
		t.Fatal(err)
	} else if len(v) != 1 {
		t.Fatal("invalid length ", len(v))
	} else if string(v[0]) != "k1" {
		t.Fatal("invalid value ", string(v[0]))
	}

	if v, err := db.LScan(k1, 2, true); err != nil {
		t.Fatal(err)
	} else if len(v) != 2 {
		t.Fatal("invalid length ", len(v))
	} else if string(v[0]) != "k1" {
		t.Fatal("invalid value ", string(v[0]))
	} else if string(v[1]) != "k2" {
		t.Fatal("invalid value ", string(v[1]))
	}

	if v, err := db.LScan(k1, 2, false); err != nil {
		t.Fatal(err)
	} else if len(v) != 2 {
		t.Fatal("invalid length ", len(v))
	} else if string(v[0]) != "k2" {
		t.Fatal("invalid value ", string(v[0]))
	} else if string(v[1]) != "k3" {
		t.Fatal("invalid value ", string(v[1]))
	}

}

func TestDBBScan(t *testing.T) {
	db := getTestDB()

	db.bFlush()

	k1 := []byte("k1")
	if _, err := db.BSetBit(k1, 1, 1); err != nil {
		t.Fatal(err.Error())
	}

	k2 := []byte("k2")
	if _, err := db.BSetBit(k2, 1, 1); err != nil {
		t.Fatal(err.Error())
	}
	k3 := []byte("k3")

	if _, err := db.BSetBit(k3, 1, 0); err != nil {
		t.Fatal(err.Error())
	}

	ek1 := db.bEncodeMetaKey(k1)
	fmt.Printf("%x\n", ek1)
	ek2 := db.bEncodeMetaKey(k2)
	fmt.Printf("%x\n", ek2)
	ek3 := db.bEncodeMetaKey(k3)
	fmt.Printf("%x\n", ek3)

	start := db.bEncodeMetaKey(nil)
	fmt.Printf("start: %x\n", start)
	end := db.bEncodeMetaKey(nil)
	end[len(end)-1] = BitMetaType + 1
	fmt.Printf("end: %x\n", end)

	it := db.db.RangeLimitIterator(start, end, store.RangeClose, 0, 4)

	for ; it.Valid(); it.Next() {
		fmt.Printf("%x\n", it.RawKey())
	}
	it.Close()

	if v, err := db.BScan(nil, 1, true); err != nil {
		t.Fatal(err)
	} else if len(v) != 1 {
		t.Fatal("invalid length ", len(v))
	} else if string(v[0]) != "k1" {
		t.Fatal("invalid value ", string(v[0]))
	}

	if v, err := db.BScan(k1, 2, true); err != nil {
		t.Fatal(err)
	} else if len(v) != 2 {
		t.Fatal("invalid length ", len(v))
	} else if string(v[0]) != "k1" {
		t.Fatal("invalid value ", string(v[0]))
	} else if string(v[1]) != "k2" {
		t.Fatal("invalid value ", string(v[1]))
	}

	if v, err := db.BScan(k1, 2, false); err != nil {
		t.Fatal(err)
	} else if len(v) != 2 {
		t.Fatal("invalid length ", len(v))
	} else if string(v[0]) != "k2" {
		t.Fatal("invalid value ", string(v[0]))
	} else if string(v[1]) != "k3" {
		t.Fatal("invalid value ", string(v[1]))
	}

}