mirror of https://github.com/zhufuyi/sponge
109 lines
2.0 KiB
Go
109 lines
2.0 KiB
Go
package cache
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/zhufuyi/sponge/pkg/encoding"
|
|
|
|
"github.com/alicebob/miniredis/v2"
|
|
"github.com/go-redis/redis/v8"
|
|
)
|
|
|
|
// InitTestRedis 实例化一个可以用于单元测试的redis
|
|
func InitTestRedis() *redis.Client {
|
|
var mr, err = miniredis.Run()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
// 打开下面命令可以测试链接关闭的情况
|
|
// defer mr.Close()
|
|
|
|
fmt.Println("mini redis addr:", mr.Addr())
|
|
return redis.NewClient(&redis.Options{
|
|
Addr: mr.Addr(),
|
|
})
|
|
}
|
|
|
|
func Test_redisCache_SetGet(t *testing.T) {
|
|
// 实例化redis客户端
|
|
redisClient := InitTestRedis()
|
|
|
|
// 实例化redis cache
|
|
cache := NewRedisCache(redisClient, "unit-test", encoding.JSONEncoding{}, func() interface{} {
|
|
return new(int64)
|
|
})
|
|
|
|
ctx := context.Background()
|
|
|
|
// test set
|
|
type setArgs struct {
|
|
key string
|
|
value interface{}
|
|
expiration time.Duration
|
|
}
|
|
|
|
value := "val-001"
|
|
setTests := []struct {
|
|
name string
|
|
cache Cache
|
|
args setArgs
|
|
wantErr bool
|
|
}{
|
|
{
|
|
"test redis set",
|
|
cache,
|
|
setArgs{"key-001", &value, 60 * time.Second},
|
|
false,
|
|
},
|
|
}
|
|
|
|
for _, tt := range setTests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
c := tt.cache
|
|
if err := c.Set(ctx, tt.args.key, tt.args.value, tt.args.expiration); (err != nil) != tt.wantErr {
|
|
t.Errorf("Set() error = %v, wantErr %v", err, tt.wantErr)
|
|
}
|
|
})
|
|
}
|
|
|
|
// test get
|
|
type args struct {
|
|
key string
|
|
}
|
|
|
|
tests := []struct {
|
|
name string
|
|
cache Cache
|
|
args args
|
|
wantVal interface{}
|
|
wantErr bool
|
|
}{
|
|
{
|
|
"test redis get",
|
|
cache,
|
|
args{"key-001"},
|
|
"val-001",
|
|
false,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
c := tt.cache
|
|
var gotVal interface{}
|
|
err := c.Get(ctx, tt.args.key, &gotVal)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("Get() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
t.Log("gotval", gotVal)
|
|
if !reflect.DeepEqual(gotVal, tt.wantVal) {
|
|
t.Errorf("Get() gotVal = %v, want %v", gotVal, tt.wantVal)
|
|
}
|
|
})
|
|
}
|
|
}
|