to_entangle/to_entangle_test.go
2023-06-24 02:39:05 +08:00

62 lines
1.2 KiB
Go

package toentangle
import (
"fmt"
"os"
"testing"
)
func TestToEntangle_Get(t *testing.T) {
// 创建一个 ToEntangle
entangle := New("data")
// 添加一对数据, 使其双向绑定
entangle.Add("a", "b")
entangle.Add("a", "c")
// 获取 a 的全部绑定数据
arr, _ := entangle.Get("a")
fmt.Println(arr)
if len(arr) != 2 {
t.Errorf("Get(\"a\") = %v; want [\"b\", \"c\"]", arr)
}
// 获取 b 的全部绑定数据
arr, _ = entangle.Get("b")
fmt.Println(arr)
if len(arr) != 1 || arr[0] != "a" {
t.Errorf("Get(\"b\") = %v; want [\"a\"]", arr)
}
// 获取 c 的全部绑定数据
arr, _ = entangle.Get("c")
fmt.Println(arr)
if len(arr) != 1 || arr[0] != "a" {
t.Errorf("Get(\"c\") = %v; want [\"a\"]", arr)
}
// 移除所有绑定
entangle.Remove("a", "b")
entangle.Remove("a", "c")
// 获取 a 的全部绑定数据
arr, _ = entangle.Get("a")
fmt.Println(arr)
if len(arr) != 0 {
t.Errorf("Get(\"a\") = %v; want []", arr)
}
// 获取 s 的全部绑定数据
arr, _ = entangle.Get("s")
fmt.Println(arr)
if len(arr) != 0 {
t.Errorf("Get(\"s\") = %v; want []", arr)
}
// 清理 leveldb
entangle.Close()
// 删除 data 文件夹
os.RemoveAll("data")
}