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.
cloudreve/pkg/cache/redis_test.go

325 lines
7.3 KiB

package cache
import (
"errors"
"fmt"
"github.com/gomodule/redigo/redis"
"github.com/rafaeljusto/redigomock"
"github.com/stretchr/testify/assert"
"testing"
"time"
)
func TestNewRedisStore(t *testing.T) {
asserts := assert.New(t)
store := NewRedisStore(10, "tcp", "", "", "0")
asserts.NotNil(store)
conn, err := store.pool.Dial()
asserts.Nil(conn)
asserts.Error(err)
testConn := redigomock.NewConn()
cmd := testConn.Command("PING").Expect("PONG")
err = store.pool.TestOnBorrow(testConn, time.Now())
if testConn.Stats(cmd) != 1 {
fmt.Println("Command was not used")
return
}
asserts.NoError(err)
}
func TestRedisStore_Set(t *testing.T) {
asserts := assert.New(t)
conn := redigomock.NewConn()
pool := &redis.Pool{
Dial: func() (redis.Conn, error) { return conn, nil },
MaxIdle: 10,
}
store := &RedisStore{pool: pool}
// 正常情况
{
cmd := conn.Command("SET", "test", redigomock.NewAnyData()).ExpectStringSlice("OK")
err := store.Set("test", "test val", -1)
asserts.NoError(err)
if conn.Stats(cmd) != 1 {
fmt.Println("Command was not used")
return
}
}
// 带有TTL
// 正常情况
{
cmd := conn.Command("SETEX", "test", 10, redigomock.NewAnyData()).ExpectStringSlice("OK")
err := store.Set("test", "test val", 10)
asserts.NoError(err)
if conn.Stats(cmd) != 1 {
fmt.Println("Command was not used")
return
}
}
// 序列化出错
{
value := struct {
Key string
}{
Key: "123",
}
err := store.Set("test", value, -1)
asserts.Error(err)
}
// 命令执行失败
{
conn.Clear()
cmd := conn.Command("SET", "test", redigomock.NewAnyData()).ExpectError(errors.New("error"))
err := store.Set("test", "test val", -1)
asserts.Error(err)
if conn.Stats(cmd) != 1 {
fmt.Println("Command was not used")
return
}
}
// 获取连接失败
{
store.pool = &redis.Pool{
Dial: func() (redis.Conn, error) { return nil, errors.New("error") },
MaxIdle: 10,
}
err := store.Set("test", "123", -1)
asserts.Error(err)
}
}
func TestRedisStore_Get(t *testing.T) {
asserts := assert.New(t)
conn := redigomock.NewConn()
pool := &redis.Pool{
Dial: func() (redis.Conn, error) { return conn, nil },
MaxIdle: 10,
}
store := &RedisStore{pool: pool}
// 正常情况
{
expectVal, _ := serializer("test val")
cmd := conn.Command("GET", "test").Expect(expectVal)
val, ok := store.Get("test")
if conn.Stats(cmd) != 1 {
fmt.Println("Command was not used")
return
}
asserts.True(ok)
asserts.Equal("test val", val.(string))
}
// Key不存在
{
conn.Clear()
cmd := conn.Command("GET", "test").Expect(nil)
val, ok := store.Get("test")
if conn.Stats(cmd) != 1 {
fmt.Println("Command was not used")
return
}
asserts.False(ok)
asserts.Nil(val)
}
// 解码错误
{
conn.Clear()
cmd := conn.Command("GET", "test").Expect([]byte{0x20})
val, ok := store.Get("test")
if conn.Stats(cmd) != 1 {
fmt.Println("Command was not used")
return
}
asserts.False(ok)
asserts.Nil(val)
}
// 获取连接失败
{
store.pool = &redis.Pool{
Dial: func() (redis.Conn, error) { return nil, errors.New("error") },
MaxIdle: 10,
}
val, ok := store.Get("test")
asserts.False(ok)
asserts.Nil(val)
}
}
func TestRedisStore_Gets(t *testing.T) {
asserts := assert.New(t)
conn := redigomock.NewConn()
pool := &redis.Pool{
Dial: func() (redis.Conn, error) { return conn, nil },
MaxIdle: 10,
}
store := &RedisStore{pool: pool}
// 全部命中
{
conn.Clear()
value1, _ := serializer("1")
value2, _ := serializer("2")
cmd := conn.Command("MGET", "test_1", "test_2").ExpectSlice(
value1, value2)
res, missed := store.Gets([]string{"1", "2"}, "test_")
if conn.Stats(cmd) != 1 {
fmt.Println("Command was not used")
return
}
asserts.Len(missed, 0)
asserts.Len(res, 2)
asserts.Equal("1", res["1"].(string))
asserts.Equal("2", res["2"].(string))
}
// 命中一个
{
conn.Clear()
value2, _ := serializer("2")
cmd := conn.Command("MGET", "test_1", "test_2").ExpectSlice(
nil, value2)
res, missed := store.Gets([]string{"1", "2"}, "test_")
if conn.Stats(cmd) != 1 {
fmt.Println("Command was not used")
return
}
asserts.Len(missed, 1)
asserts.Len(res, 1)
asserts.Equal("1", missed[0])
asserts.Equal("2", res["2"].(string))
}
// 命令出错
{
conn.Clear()
cmd := conn.Command("MGET", "test_1", "test_2").ExpectError(errors.New("error"))
res, missed := store.Gets([]string{"1", "2"}, "test_")
if conn.Stats(cmd) != 1 {
fmt.Println("Command was not used")
return
}
asserts.Len(missed, 2)
asserts.Len(res, 0)
}
// 连接出错
{
conn.Clear()
store.pool = &redis.Pool{
Dial: func() (redis.Conn, error) { return nil, errors.New("error") },
MaxIdle: 10,
}
res, missed := store.Gets([]string{"1", "2"}, "test_")
asserts.Len(missed, 2)
asserts.Len(res, 0)
}
}
func TestRedisStore_Sets(t *testing.T) {
asserts := assert.New(t)
conn := redigomock.NewConn()
pool := &redis.Pool{
Dial: func() (redis.Conn, error) { return conn, nil },
MaxIdle: 10,
}
store := &RedisStore{pool: pool}
// 正常
{
cmd := conn.Command("MSET", redigomock.NewAnyData(), redigomock.NewAnyData(), redigomock.NewAnyData(), redigomock.NewAnyData()).ExpectSlice("OK")
err := store.Sets(map[string]interface{}{"1": "1", "2": "2"}, "test_")
asserts.NoError(err)
if conn.Stats(cmd) != 1 {
fmt.Println("Command was not used")
return
}
}
// 序列化失败
{
conn.Clear()
value := struct {
Key string
}{
Key: "123",
}
err := store.Sets(map[string]interface{}{"1": value, "2": "2"}, "test_")
asserts.Error(err)
}
// 执行失败
{
cmd := conn.Command("MSET", redigomock.NewAnyData(), redigomock.NewAnyData(), redigomock.NewAnyData(), redigomock.NewAnyData()).ExpectError(errors.New("error"))
err := store.Sets(map[string]interface{}{"1": "1", "2": "2"}, "test_")
asserts.Error(err)
if conn.Stats(cmd) != 1 {
fmt.Println("Command was not used")
return
}
}
// 连接失败
{
conn.Clear()
store.pool = &redis.Pool{
Dial: func() (redis.Conn, error) { return nil, errors.New("error") },
MaxIdle: 10,
}
err := store.Sets(map[string]interface{}{"1": "1", "2": "2"}, "test_")
asserts.Error(err)
}
}
func TestRedisStore_Delete(t *testing.T) {
asserts := assert.New(t)
conn := redigomock.NewConn()
pool := &redis.Pool{
Dial: func() (redis.Conn, error) { return conn, nil },
MaxIdle: 10,
}
store := &RedisStore{pool: pool}
// 正常
{
cmd := conn.Command("DEL", redigomock.NewAnyData(), redigomock.NewAnyData(), redigomock.NewAnyData(), redigomock.NewAnyData()).ExpectSlice("OK")
err := store.Delete([]string{"1", "2", "3", "4"}, "test_")
asserts.NoError(err)
if conn.Stats(cmd) != 1 {
fmt.Println("Command was not used")
return
}
}
// 命令执行失败
{
conn.Clear()
cmd := conn.Command("DEL", redigomock.NewAnyData(), redigomock.NewAnyData(), redigomock.NewAnyData(), redigomock.NewAnyData()).ExpectError(errors.New("error"))
err := store.Delete([]string{"1", "2", "3", "4"}, "test_")
asserts.Error(err)
if conn.Stats(cmd) != 1 {
fmt.Println("Command was not used")
return
}
}
// 连接失败
{
conn.Clear()
store.pool = &redis.Pool{
Dial: func() (redis.Conn, error) { return nil, errors.New("error") },
MaxIdle: 10,
}
err := store.Delete([]string{"1", "2", "3", "4"}, "test_")
asserts.Error(err)
}
}