sync_trie_test.go

package trie

import (
	"github.com/stretchr/testify/assert"
	"testing"
)

import (
	"fmt"
	"strings"
)

func TestNewSyncTrie(t *testing.T) {

	// 假装是一个Web路由
	trie := NewSync[func() error](func(s string) ([]string, error) {
		slice := make([]string, 0)
		for _, x := range strings.Split(s, "/") {
			if x != "" {
				slice = append(slice, x)
			}
		}
		return slice, nil
	})

	// 增加一个路由
	err := trie.Add("/foo/bar", func() error {
		fmt.Println("路由到了/foo/bar")
		return nil
	})
	assert.Nil(t, err)

	// 尝试寻找不存在的路由
	value, err := trie.Query("/foo")
	assert.NotNil(t, err)
	assert.Nil(t, value)

	// 尝试寻找存在的路由
	handler, err := trie.Query("//foo//bar/")
	assert.Nil(t, err)
	assert.NotNil(t, handler)
	err = handler()
	assert.Nil(t, err)

}

func TestSyncTrieAdd(t *testing.T) {
	trie := NewSync[string]()

	err := trie.Add("test", "测试")
	assert.Nil(t, err)

	slice := trie.ToSlice()
	assert.Equal(t, 1, len(slice))
}

func TestSyncTrieFindTrieNode(t *testing.T) {

}

func TestSyncTrieQuery(t *testing.T) {
	trie := NewSync[string]()

	err := trie.Add("test", "测试")
	assert.Nil(t, err)

	value, err := trie.Query("test")
	assert.Nil(t, err)
	assert.Equal(t, "测试", value)

	value, err = trie.Query("test-001")
	assert.ErrorIs(t, ErrNotFound, err)
}

func TestSyncTrieQueryOrDefault(t *testing.T) {
	trie := NewSync[string]()

	err := trie.Add("test", "测试")
	assert.Nil(t, err)

	value, err := trie.QueryOrDefault("test", "策士")
	assert.Nil(t, err)
	assert.Equal(t, "测试", value)

	value, err = trie.QueryOrDefault("test-001", "策士")
	assert.Nil(t, err)
	assert.Equal(t, "策士", value)

}

func TestSyncTrieRemove(t *testing.T) {
	trie := NewSync[string]()

	err := trie.Upsert("china", "瓷器")
	assert.Nil(t, err)
	err = trie.Upsert("chinese", "中国人")
	assert.Nil(t, err)
	slice := trie.ToSlice()
	assert.Equal(t, 2, len(slice))

	err = trie.Remove("china")
	assert.Nil(t, err)
	slice = trie.ToSlice()
	assert.Equal(t, 1, len(slice))
}

func TestSyncTrieUpsert(t *testing.T) {
	trie := NewSync[string]()

	_ = trie.Upsert("china", "瓷器")
	value, err := trie.Query("china")
	assert.Nil(t, err)
	assert.Equal(t, "瓷器", value)

	_ = trie.Upsert("china", "中国")
	value, err = trie.Query("china")
	assert.Nil(t, err)
	assert.Equal(t, "中国", value)
}

func TestSyncTrieToSlice(t *testing.T) {
	trie := NewSync[string]()
	err := trie.Add("china", "中国")
	assert.Nil(t, err)
	err = trie.Add("chinese", "中国人")
	assert.Nil(t, err)
	slice := trie.ToSlice("")
	assert.Equal(t, 2, len(slice))
}

func TestSyncTrieFindByPrefix(t *testing.T) {
	trie := NewSync[string]()
	_ = trie.Upsert("china", "china")
	_ = trie.Upsert("chinese", "chinese")
	_ = trie.Upsert("channel", "channel")
	_ = trie.Upsert("chan", "chan")
	_ = trie.Upsert("boy", "boy")
	_ = trie.Upsert("CC11001100", "CC11001100")

	slice := trie.QueryByPrefix("chan")
	//t.Log(slice)
	assert.Equal(t, 2, len(slice))

}

func TestSyncTrieContains(t *testing.T) {
	trie := NewSync[string]()
	_ = trie.Upsert("china", "china")
	_ = trie.Upsert("chinese", "chinese")

	exists, err := trie.Contains("china")
	assert.Nil(t, err)
	assert.True(t, exists)

	exists, err = trie.Contains("chi")
	assert.Nil(t, err)
	assert.False(t, exists)
}