112 lines
3.6 KiB
Go
112 lines
3.6 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 串应返回错误")
|
||
}
|
||
}
|
||
|
||
// TestTypedWithType 验证新编码（类型进密码、无前缀）：
|
||
// - 往返严格成立 DecodeWithType(EncodeWithType(n, typ), typ) == n；
|
||
// - 同数字不同类别得到完全不相关的串（runway#5 ≠ snap#5），且串本身不含类型字母；
|
||
// - 用错误 typ 解码得到错误主键（≠ n），从而查库自然 404。
|
||
func TestTypedWithType(t *testing.T) {
|
||
Init("test-salt")
|
||
|
||
// 往返
|
||
for _, typ := range []byte{TypeRunway, TypeSnap, TypeRunwayImage, TypeSnapImage} {
|
||
for _, n := range []uint32{0, 1, 5, 999, 1 << 20, (1 << 32) - 1} {
|
||
got, err := DecodeWithType(EncodeWithType(n, typ), typ)
|
||
if err != nil || got != n {
|
||
t.Fatalf("往返不一致: typ=%c n=%d got=%d err=%v", typ, n, got, err)
|
||
}
|
||
}
|
||
}
|
||
|
||
// 同数字不同类别：串不同且不相关，且都不含类型字母
|
||
r := EncodeWithType(5, TypeRunway)
|
||
s := EncodeWithType(5, TypeSnap)
|
||
if r == s {
|
||
t.Fatalf("同数字不同类别应得到不同串: runway=%q snap=%q", r, s)
|
||
}
|
||
if r[0] == TypeRunway || r[0] == TypeSnap || s[0] == TypeRunway || s[0] == TypeSnap {
|
||
t.Fatalf("新编码不应含类型字母前缀: runway=%q snap=%q", r, s)
|
||
}
|
||
// 用错误类别解码得不到原主键
|
||
if wrong, err := DecodeWithType(r, TypeSnap); err != nil || wrong == 5 {
|
||
t.Fatalf("用错误类别解码应得不到原主键: got=%d err=%v", wrong, err)
|
||
}
|
||
// 空串报错
|
||
if _, err := DecodeWithType("", TypeRunway); err == nil {
|
||
t.Fatal("空串应返回错误")
|
||
}
|
||
}
|