redis-shake工具
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 

90 lines
2.5 KiB

package commands
import (
"testing"
)
func testEq(a, b []string) bool {
if len(a) != len(b) {
return false
}
for i := range a {
if a[i] != b[i] {
return false
}
}
return true
}
func TestCalcKeys(t *testing.T) {
// SET
cmd, group, keys := CalcKeys([]string{"SET", "key", "value"})
if cmd != "SET" || group != "STRING" || !testEq(keys, []string{"key"}) {
t.Errorf("CalcKeys(SET key value) failed. cmd=%s, group=%s, keys=%v", cmd, group, keys)
}
// MSET
cmd, group, keys = CalcKeys([]string{"MSET", "key1", "value1", "key2", "value2"})
if cmd != "MSET" || group != "STRING" || !testEq(keys, []string{"key1", "key2"}) {
t.Errorf("CalcKeys(MSET key1 value1 key2 value2) failed. cmd=%s, group=%s, keys=%v", cmd, group, keys)
}
// XADD
cmd, group, keys = CalcKeys([]string{"XADD", "key", "*", "field1", "value1", "field2", "value2"})
if cmd != "XADD" || group != "STREAM" || !testEq(keys, []string{"key"}) {
t.Errorf("CalcKeys(XADD key * field1 value1 field2 value2) failed. cmd=%s, group=%s, keys=%v", cmd, group, keys)
}
// ZUNIONSTORE
cmd, group, keys = CalcKeys([]string{"ZUNIONSTORE", "key", "2", "key1", "key2"})
if cmd != "ZUNIONSTORE" || group != "SORTED_SET" || !testEq(keys, []string{"key", "key1", "key2"}) {
t.Errorf("CalcKeys(ZUNIONSTORE key 2 key1 key2) failed. cmd=%s, group=%s, keys=%v", cmd, group, keys)
}
}
func TestKeyHash(t *testing.T) {
ret := keyHash("abcde")
if ret != 16097 {
t.Errorf("keyHash(abcde) = %x", ret)
}
ret = keyHash("abcde{")
if ret != 14689 {
t.Errorf("keyHash(abcde{) = %x", ret)
}
ret = keyHash("abcde}")
if ret != 6567 {
t.Errorf("keyHash(abcde}) = %x", ret)
}
ret = keyHash("{abcde}")
if ret != 16097 {
t.Errorf("keyHash({abcde}) = %x", ret)
}
ret = keyHash("same")
if ret != 13447 {
t.Errorf("keyHash(same) = %x", ret)
}
ret = keyHash("abcdefghi{same}abcdefghi")
if ret != 13447 {
t.Errorf("keyHash(abcdefghi{same}abcdefghi) = %x", ret)
}
ret = keyHash("123456789{same}123456789")
if ret != 13447 {
t.Errorf("keyHash(123456789{same}123456789) = %x", ret)
}
ret = keyHash("1234我是你89{same}12我就456789")
if ret != 13447 {
t.Errorf("keyHash(1234我是你89{same}12我就456789) = %x", ret)
}
ret = keyHash("你你你{你你你}你你你")
if ret != 15023 {
t.Errorf("keyHash(1234我是你89{same}12我就456789) = %x", ret)
}
b := make([]byte, 0)
for i := 0; i < 256; i++ {
b = append(b, byte(i))
}
ret = keyHash(string(b))
if ret != 16155 {
t.Errorf("keyHash(%s) = %x", string(b), ret)
}
}