76 lines
2.2 KiB
Go
76 lines
2.2 KiB
Go
package hashid
|
|
|
|
import "testing"
|
|
|
|
func TestRoundtrip(t *testing.T) {
|
|
// 覆盖边界与大量随机值，验证 Decode(Encode(n)) == n 严格成立
|
|
cases := []uint32{0, 1, 2, 3, 7, 15, 16, 255, 256, 4095, 4096, 65535, 65536, 1 << 31, (1 << 32) - 1}
|
|
for _, n := range cases {
|
|
got, err := Decode(Encode(n))
|
|
if err != nil {
|
|
t.Fatalf("Decode(Encode(%d)) 返回错误: %v", n, err)
|
|
}
|
|
if got != n {
|
|
t.Fatalf("往返不一致: 输入 %d, 得到 %d", n, got)
|
|
}
|
|
}
|
|
|
|
// 随机大批量
|
|
for n := uint32(1); n < 5000; n++ {
|
|
got, err := Decode(Encode(n))
|
|
if err != nil || got != n {
|
|
t.Fatalf("往返不一致: 输入 %d, 得到 %d (err=%v)", n, got, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestNonSequential(t *testing.T) {
|
|
Init("test-salt")
|
|
a, b, c := Encode(100), Encode(101), Encode(102)
|
|
if a == b || b == c || a == c {
|
|
t.Fatalf("相邻 id 编码结果出现了相等: %s %s %s", a, b, c)
|
|
}
|
|
}
|
|
|
|
func TestInvalid(t *testing.T) {
|
|
if _, err := Decode(""); err == nil {
|
|
t.Fatal("空串应返回错误")
|
|
}
|
|
if _, err := Decode("!!!"); err == nil {
|
|
t.Fatal("含非法字符应返回错误")
|
|
}
|
|
}
|
|
|
|
// TestTyped 验证带类型前缀的编解码：同数字不同前缀得到不同串，且能正确还原类型与数字主键。
|
|
func TestTyped(t *testing.T) {
|
|
Init("test-salt")
|
|
|
|
r := EncodeTyped(TypeRunway, 5)
|
|
s := EncodeTyped(TypeSnap, 5)
|
|
if r == s {
|
|
t.Fatalf("同数字不同前缀应得到不同串: runway=%q snap=%q", r, s)
|
|
}
|
|
if r[0] != TypeRunway || s[0] != TypeSnap {
|
|
t.Fatalf("前缀字符不正确: runway=%q snap=%q", r, s)
|
|
}
|
|
|
|
tr, nr, err := DecodeTyped(r)
|
|
if err != nil || tr != TypeRunway || nr != 5 {
|
|
t.Fatalf("DecodeTyped(runway) 失败: type=%c num=%d err=%v", tr, nr, err)
|
|
}
|
|
ts, ns, err := DecodeTyped(s)
|
|
if err != nil || ts != TypeSnap || ns != 5 {
|
|
t.Fatalf("DecodeTyped(snap) 失败: type=%c num=%d err=%v", ts, ns, err)
|
|
}
|
|
|
|
// 无前缀旧串仍可用 Decode 还原（兼容历史链接）
|
|
old := Encode(5)
|
|
if no, err := Decode(old); err != nil || no != 5 {
|
|
t.Fatalf("旧无前缀串解码失败: num=%d err=%v", no, err)
|
|
}
|
|
// 长度不足 2 视为非法
|
|
if _, _, err := DecodeTyped("r"); err == nil {
|
|
t.Fatal("长度不足 2 的 typed 串应返回错误")
|
|
}
|
|
}
|