cedar_test.go

package cedar

import (
	"fmt"
	"os"
	"testing"
)

var (
	cd    *Cedar
	words = []string{
		"a", "aa", "ab", "ac", "abc", "abd",
		"abcd", "abde", "abdf", "abcdef", "abcde",
		"abcdefghijklmn", "bcd", "b", "xyz",
		"中国", "中国北京", "中国上海", "中国广州",
		"中华", "中华文明", "中华民族", "中华人民共和国",
		"this", "this is", "this is a sentence.",
	}
)

func loadTestData() {
	if cd != nil {
		return
	}
	cd = New()
	// cd.Ordered = false

	// add the keys
	for i, word := range words {
		if err := cd.Insert([]byte(word), i); err != nil {
			panic(err)
		}
	}

	for _, word := range words {
		if err := cd.Delete([]byte(word)); err != nil {
			panic(err)
		}
	}

	for i, word := range words {
		if err := cd.Update([]byte(word), i); err != nil {
			panic(err)
		}
	}

	// delete some keys
	for i := 0; i < len(words); i += 4 {
		if err := cd.Delete([]byte(words[i])); err != nil {
			panic(err)
		}
	}
	return
}

func TestBasic(t *testing.T) {
	loadTestData()
	// check the consistency
	checkConsistency(cd)
}

func TestSaveAndLoad(t *testing.T) {
	loadTestData()

	cd.SaveToFile("cedar.gob", "gob")
	defer os.Remove("cedar.gob")
	daGob := New()
	if err := daGob.LoadFromFile("cedar.gob", "gob"); err != nil {
		panic(err)
	}
	checkConsistency(daGob)

	cd.SaveToFile("cedar.json", "json")
	defer os.Remove("cedar.json")
	daJson := New()
	if err := daJson.LoadFromFile("cedar.json", "json"); err != nil {
		panic(err)
	}
	checkConsistency(daJson)
}

func TestPrefixMatch(t *testing.T) {
	var ids []int
	var keys []string
	var values []int

	ids = cd.PrefixMatch([]byte("abcdefg"), 0)
	keys = []string{"ab", "abcd", "abcde", "abcdef"}
	values = []int{2, 6, 10, 9}
	check(cd, ids, keys, values)

	ids = cd.PrefixMatch([]byte("中华人民共和国"), 0)
	keys = []string{"中华", "中华人民共和国"}
	values = []int{19, 22}
	check(cd, ids, keys, values)

	ids = cd.PrefixMatch([]byte("this is a sentence."), 0)
	keys = []string{"this", "this is a sentence."}
	values = []int{23, 25}
	check(cd, ids, keys, values)
}

func check(cd *Cedar, ids []int, keys []string, values []int) {
	if len(ids) != len(keys) {
		panic("wrong prefix match")
	}
	for i, n := range ids {
		key, _ := cd.Key(n)
		val, _ := cd.Value(n)
		if string(key) != keys[i] || val != values[i] {
			panic("wrong prefix match")
		}
	}
}

func TestOrder(t *testing.T) {
	c := New()
	c.Insert([]byte("a"), 1)
	c.Insert([]byte("b"), 3)
	c.Insert([]byte("d"), 6)
	c.Insert([]byte("ab"), 2)
	c.Insert([]byte("c"), 5)
	c.Insert([]byte(""), 0)
	c.Insert([]byte("bb"), 4)
	ids := c.PrefixPredict([]byte(""), 0)
	if len(ids) != 7 {
		panic("wrong order")
	}
	for i, n := range ids {
		val, _ := c.Value(n)
		if i != val {
			panic("wrong order")
		}
	}
}

func TestPrefixPredict(t *testing.T) {
	var ids []int
	var keys []string
	var values []int
	ids = cd.PrefixPredict([]byte("中华"), 0)
	keys = []string{"中华", "中华人民共和国", "中华民族"}
	values = []int{19, 22, 21}
	check(cd, ids, keys, values)

	ids = cd.PrefixPredict([]byte("中国"), 0)
	keys = []string{"中国", "中国上海", "中国广州"}
	values = []int{15, 17, 18}
	check(cd, ids, keys, values)
}

func checkConsistency(cd *Cedar) {
	for i, word := range words {
		id, err := cd.Jump([]byte(word), 0)
		if i%4 == 0 {
			if err == ErrNoPath {
				continue
			}
			_, err := cd.Value(id)
			if err == ErrNoValue {
				continue
			}
			panic("not deleted")
		}
		key, err := cd.Key(id)
		if err != nil {
			panic(err)
		}
		if string(key) != word {
			panic("key error")
		}
		value, err := cd.Value(id)
		if err != nil || value != i {
			fmt.Println(word, i, value, err)
			panic("value error")
		}
	}
}