45 lines
1.1 KiB
Go
45 lines
1.1 KiB
Go
package store
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestUser_EncodeID(t *testing.T) {
|
|
tbl := []struct {
|
|
id string
|
|
hash string
|
|
}{
|
|
{"myid", "6e34471f84557e1713012d64a7477c71bfdac631"},
|
|
{"", "da39a3ee5e6b4b0d3255bfef95601890afd80709"},
|
|
{"blah blah", "135a1e01bae742c4a576b20fd41a683f6483ca43"},
|
|
}
|
|
|
|
for i, tt := range tbl {
|
|
assert.Equal(t, tt.hash, EncodeID(tt.id), "case #%d", i)
|
|
}
|
|
}
|
|
|
|
func TestUser_HashIP(t *testing.T) {
|
|
tbl := []struct {
|
|
ip string
|
|
hash1, hash2 string
|
|
}{
|
|
{"127.0.0.1", "ae12fe3b5f129b5cc4cdd2b136b7b7947c4d2741", "dbc7c999343f003f189f70aaf52cc04443f90790"},
|
|
{"8.8.8.8", "8cee77c27e32a2b5aec95c29888ac9946618d9a2", "70a46afce9633f010b06e129b8ad08243a1c4da9"},
|
|
{"8cee77c27e32a2b5aec95c29888ac9946618d9a2", "8cee77c27e32a2b5aec95c29888ac9946618d9a2", "8cee77c27e32a2b5aec95c29888ac9946618d9a2"},
|
|
{"", "", ""},
|
|
}
|
|
|
|
for i, tt := range tbl {
|
|
u := User{IP: tt.ip}
|
|
u.HashIP("")
|
|
assert.Equal(t, tt.hash1, u.IP, "case #%d", i)
|
|
|
|
u = User{IP: tt.ip}
|
|
u.HashIP("123456")
|
|
assert.Equal(t, tt.hash2, u.IP, "case #%d", i)
|
|
}
|
|
}
|