fix sonar Part3

client/cache/cache_test.go

@@ -21,13 +21,13 @@ import (
 	"sync"
 	"testing"
 	"time"
+
+	"github.com/stretchr/testify/assert"
 )
 
 func TestCacheIncr(t *testing.T) {
 	bm, err := NewCache("memory", `{"interval":20}`)
-	if err != nil {
-		t.Error("init err")
-	}
+	assert.Nil(t, err)
 	// timeoutDuration := 10 * time.Second
 
 	bm.Put(context.Background(), "edwardhey", 0, time.Second*20)
@@ -48,9 +48,7 @@ func TestCacheIncr(t *testing.T) {
 
 func TestCache(t *testing.T) {
 	bm, err := NewCache("memory", `{"interval":1}`)
-	if err != nil {
-		t.Error("init err")
-	}
+	assert.Nil(t, err)
 	timeoutDuration := 5 * time.Second
 	if err = bm.Put(context.Background(), "astaxie", 1, timeoutDuration); err != nil {
 		t.Error("set Error", err)
@@ -81,70 +79,48 @@ func TestCache(t *testing.T) {
 	testDecrOverFlow(t, bm, timeoutDuration)
 
 	bm.Delete(context.Background(), "astaxie")
-	if res, _ := bm.IsExist(context.Background(), "astaxie"); res {
-		t.Error("delete err")
-	}
+	res, _ := bm.IsExist(context.Background(), "astaxie")
+	assert.False(t, res)
 
-	// test GetMulti
-	if err = bm.Put(context.Background(), "astaxie", "author", timeoutDuration); err != nil {
-		t.Error("set Error", err)
-	}
-	if res, _ := bm.IsExist(context.Background(), "astaxie"); !res {
-		t.Error("check err")
-	}
-	if v, _ := bm.Get(context.Background(), "astaxie"); v.(string) != "author" {
-		t.Error("get err")
-	}
+	assert.Nil(t, bm.Put(context.Background(), "astaxie", "author", timeoutDuration))
 
-	if err = bm.Put(context.Background(), "astaxie1", "author1", timeoutDuration); err != nil {
-		t.Error("set Error", err)
-	}
-	if res, _ := bm.IsExist(context.Background(), "astaxie1"); !res {
-		t.Error("check err")
-	}
+	res, _ = bm.IsExist(context.Background(), "astaxie")
+	assert.True(t, res)
+
+	v, _ := bm.Get(context.Background(), "astaxie")
+	assert.Equal(t, "author", v)
+
+	assert.Nil(t, bm.Put(context.Background(), "astaxie1", "author1", timeoutDuration))
+
+	res, _ = bm.IsExist(context.Background(), "astaxie1")
+	assert.True(t, res)
 
 	vv, _ := bm.GetMulti(context.Background(), []string{"astaxie", "astaxie1"})
-	if len(vv) != 2 {
-		t.Error("GetMulti ERROR")
-	}
-	if vv[0].(string) != "author" {
-		t.Error("GetMulti ERROR")
-	}
-	if vv[1].(string) != "author1" {
-		t.Error("GetMulti ERROR")
-	}
+	assert.Equal(t, 2, len(vv))
+	assert.Equal(t, "author", vv[0])
+	assert.Equal(t,"author1", vv[1])
+
+
 
 	vv, err = bm.GetMulti(context.Background(), []string{"astaxie0", "astaxie1"})
-	if len(vv) != 2 {
-		t.Error("GetMulti ERROR")
-	}
-	if vv[0] != nil {
-		t.Error("GetMulti ERROR")
-	}
-	if vv[1].(string) != "author1" {
-		t.Error("GetMulti ERROR")
-	}
-	if err != nil && err.Error() != "key [astaxie0] error: the key isn't exist" {
-		t.Error("GetMulti ERROR")
-	}
+	assert.Equal(t, 2, len(vv))
+	assert.Nil(t, vv[0])
+	assert.Equal(t, "author1", vv[1])
+
+	assert.NotNil(t, err)
+	assert.Equal(t, "key [astaxie0] error: key not exist", err.Error())
 }
 
 func TestFileCache(t *testing.T) {
 	bm, err := NewCache("file", `{"CachePath":"cache","FileSuffix":".bin","DirectoryLevel":"2","EmbedExpiry":"0"}`)
-	if err != nil {
-		t.Error("init err")
-	}
+	assert.Nil(t, err)
 	timeoutDuration := 10 * time.Second
-	if err = bm.Put(context.Background(), "astaxie", 1, timeoutDuration); err != nil {
-		t.Error("set Error", err)
-	}
-	if res, _ := bm.IsExist(context.Background(), "astaxie"); !res {
-		t.Error("check err")
-	}
+	assert.Nil(t,  bm.Put(context.Background(), "astaxie", 1, timeoutDuration))
 
-	if v, _ := bm.Get(context.Background(), "astaxie"); v.(int) != 1 {
-		t.Error("get err")
-	}
+	res, _ := bm.IsExist(context.Background(), "astaxie")
+	assert.True(t, res)
+	v, _ := bm.Get(context.Background(), "astaxie")
+	assert.Equal(t, 1, v)
 
 	// test different integer type for incr & decr
 	testMultiTypeIncrDecr(t, bm, timeoutDuration)
@@ -154,54 +130,35 @@ func TestFileCache(t *testing.T) {
 	testDecrOverFlow(t, bm, timeoutDuration)
 
 	bm.Delete(context.Background(), "astaxie")
-	if res, _ := bm.IsExist(context.Background(), "astaxie"); res {
-		t.Error("delete err")
-	}
+	res, _ = bm.IsExist(context.Background(), "astaxie")
+	assert.False(t, res)
 
 	// test string
-	if err = bm.Put(context.Background(), "astaxie", "author", timeoutDuration); err != nil {
-		t.Error("set Error", err)
-	}
-	if res, _ := bm.IsExist(context.Background(), "astaxie"); !res {
-		t.Error("check err")
-	}
-	if v, _ := bm.Get(context.Background(), "astaxie"); v.(string) != "author" {
-		t.Error("get err")
-	}
+	assert.Nil(t, bm.Put(context.Background(), "astaxie", "author", timeoutDuration))
+	res, _ = bm.IsExist(context.Background(), "astaxie")
+	assert.True(t, res)
+
+	v, _ = bm.Get(context.Background(), "astaxie")
+	assert.Equal(t, "author", v)
 
 	// test GetMulti
-	if err = bm.Put(context.Background(), "astaxie1", "author1", timeoutDuration); err != nil {
-		t.Error("set Error", err)
-	}
-	if res, _ := bm.IsExist(context.Background(), "astaxie1"); !res {
-		t.Error("check err")
-	}
+	assert.Nil(t, bm.Put(context.Background(), "astaxie1", "author1", timeoutDuration))
+
+	res, _ = bm.IsExist(context.Background(), "astaxie1")
+	assert.True(t, res)
 
 	vv, _ := bm.GetMulti(context.Background(), []string{"astaxie", "astaxie1"})
-	if len(vv) != 2 {
-		t.Error("GetMulti ERROR")
-	}
-	if vv[0].(string) != "author" {
-		t.Error("GetMulti ERROR")
-	}
-	if vv[1].(string) != "author1" {
-		t.Error("GetMulti ERROR")
-	}
+	assert.Equal(t, 2, len(vv))
+	assert.Equal(t, "author", vv[0])
+	assert.Equal(t, "author1", vv[1])
 
 	vv, err = bm.GetMulti(context.Background(), []string{"astaxie0", "astaxie1"})
-	if len(vv) != 2 {
-		t.Error("GetMulti ERROR")
-	}
-	if vv[0] != nil {
-		t.Error("GetMulti ERROR")
-	}
-	if vv[1].(string) != "author1" {
-		t.Error("GetMulti ERROR")
-	}
-	if err == nil {
-		t.Error("GetMulti ERROR")
-	}
+	assert.Equal(t, 2, len(vv))
 
+	assert.Nil(t, vv[0])
+
+	assert.Equal(t, "author1", vv[1])
+	assert.NotNil(t, err)
 	os.RemoveAll("cache")
 }
 
@@ -215,53 +172,33 @@ func testMultiTypeIncrDecr(t *testing.T, c Cache, timeout time.Duration) {
 }
 
 func testIncrDecr(t *testing.T, c Cache, beforeIncr interface{}, afterIncr interface{}, timeout time.Duration) {
-	var err error
 	ctx := context.Background()
 	key := "incDecKey"
-	if err = c.Put(ctx, key, beforeIncr, timeout); err != nil {
-		t.Error("Get Error", err)
-	}
 
-	if err = c.Incr(ctx, key); err != nil {
-		t.Error("Incr Error", err)
-	}
+	assert.Nil(t, c.Put(ctx, key, beforeIncr, timeout))
+	assert.Nil(t, c.Incr(ctx, key))
 
-	if v, _ := c.Get(ctx, key); v != afterIncr {
-		t.Error("Get Error")
-	}
 
-	if err = c.Decr(ctx, key); err != nil {
-		t.Error("Decr Error", err)
-	}
+	v, _ := c.Get(ctx, key)
+	assert.Equal(t, afterIncr, v)
 
-	if v, _ := c.Get(ctx, key); v != beforeIncr {
-		t.Error("Get Error")
-	}
+	assert.Nil(t, c.Decr(ctx, key))
 
-	if err := c.Delete(ctx, key); err != nil {
-		t.Error("Delete Error")
-	}
+	v, _ = c.Get(ctx, key)
+	assert.Equal(t, v, beforeIncr)
+	assert.Nil(t, c.Delete(ctx, key))
 }
 
 func testIncrOverFlow(t *testing.T, c Cache, timeout time.Duration) {
-	var err error
 	ctx := context.Background()
 	key := "incKey"
 
+	assert.Nil(t, c.Put(ctx, key, int64(math.MaxInt64), timeout))
 	// int64
-	if err = c.Put(ctx, key, int64(math.MaxInt64), timeout); err != nil {
-		t.Error("Put Error: ", err.Error())
-		return
-	}
 	defer func() {
-		if err = c.Delete(ctx, key); err != nil {
-			t.Errorf("Delete error: %s", err.Error())
-		}
+		assert.Nil(t, c.Delete(ctx, key))
 	}()
-	if err = c.Incr(ctx, key); err == nil {
-		t.Error("Incr error")
-		return
-	}
+	assert.NotNil(t, c.Incr(ctx, key))
 }
 
 func testDecrOverFlow(t *testing.T, c Cache, timeout time.Duration) {

client/cache/calc_utils_test.go

@@ -4,6 +4,8 @@ import (
 	"math"
 	"strconv"
 	"testing"
+
+	"github.com/stretchr/testify/assert"
 )
 
 func TestIncr(t *testing.T) {
@@ -11,116 +13,62 @@ func TestIncr(t *testing.T) {
 	var originVal interface{} = int(1)
 	var updateVal interface{} = int(2)
 	val, err := incr(originVal)
-	if err != nil {
-		t.Errorf("incr failed, err: %s", err.Error())
-		return
-	}
-	if val != updateVal {
-		t.Errorf("incr failed, expect %v, but %v actually", updateVal, val)
-		return
-	}
+	assert.Nil(t, err)
+	assert.Equal(t, updateVal, val)
 	_, err = incr(int(1<<(strconv.IntSize-1) - 1))
-	if err == nil {
-		t.Error("incr failed")
-		return
-	}
+	assert.NotNil(t, err)
 
 	// int32
 	originVal = int32(1)
 	updateVal = int32(2)
 	val, err = incr(originVal)
-	if err != nil {
-		t.Errorf("incr failed, err: %s", err.Error())
-		return
-	}
-	if val != updateVal {
-		t.Errorf("incr failed, expect %v, but %v actually", updateVal, val)
-		return
-	}
+	assert.Nil(t, err)
+	assert.Equal(t, updateVal, val)
 	_, err = incr(int32(math.MaxInt32))
-	if err == nil {
-		t.Error("incr failed")
-		return
-	}
-
+	assert.NotNil(t, err)
 	// int64
 	originVal = int64(1)
 	updateVal = int64(2)
 	val, err = incr(originVal)
-	if err != nil {
-		t.Errorf("incr failed, err: %s", err.Error())
-		return
-	}
-	if val != updateVal {
-		t.Errorf("incr failed, expect %v, but %v actually", updateVal, val)
-		return
-	}
+	assert.Nil(t, err)
+
+	assert.Equal(t, updateVal, val)
+
 	_, err = incr(int64(math.MaxInt64))
-	if err == nil {
-		t.Error("incr failed")
-		return
-	}
+	assert.NotNil(t, err)
 
 	// uint
 	originVal = uint(1)
 	updateVal = uint(2)
 	val, err = incr(originVal)
-	if err != nil {
-		t.Errorf("incr failed, err: %s", err.Error())
-		return
-	}
-	if val != updateVal {
-		t.Errorf("incr failed, expect %v, but %v actually", updateVal, val)
-		return
-	}
-	_, err = incr(uint(1<<(strconv.IntSize) - 1))
-	if err == nil {
-		t.Error("incr failed")
-		return
-	}
+	assert.Nil(t, err)
 
+	assert.Equal(t, updateVal, val)
+
+	_, err = incr(uint(1<<(strconv.IntSize) - 1))
+	assert.NotNil(t, err)
 	// uint32
 	originVal = uint32(1)
 	updateVal = uint32(2)
 	val, err = incr(originVal)
-	if err != nil {
-		t.Errorf("incr failed, err: %s", err.Error())
-		return
-	}
-	if val != updateVal {
-		t.Errorf("incr failed, expect %v, but %v actually", updateVal, val)
-		return
-	}
+	assert.Nil(t, err)
+	assert.Equal(t, updateVal, val)
+
 	_, err = incr(uint32(math.MaxUint32))
-	if err == nil {
-		t.Error("incr failed")
-		return
-	}
+	assert.NotNil(t, err)
 
 	// uint64
 	originVal = uint64(1)
 	updateVal = uint64(2)
 	val, err = incr(originVal)
-	if err != nil {
-		t.Errorf("incr failed, err: %s", err.Error())
-		return
-	}
-	if val != updateVal {
-		t.Errorf("incr failed, expect %v, but %v actually", updateVal, val)
-		return
-	}
+	assert.Nil(t, err)
+	assert.Equal(t, updateVal, val)
 	_, err = incr(uint64(math.MaxUint64))
-	if err == nil {
-		t.Error("incr failed")
-		return
-	}
+	assert.NotNil(t, err)
 
 	// other type
 	_, err = incr("string")
-	if err == nil {
-		t.Error("incr failed")
-		return
-	}
+	assert.NotNil(t, err)
 }
 
 func TestDecr(t *testing.T) {
@@ -128,114 +76,58 @@ func TestDecr(t *testing.T) {
 	var originVal interface{} = int(2)
 	var updateVal interface{} = int(1)
 	val, err := decr(originVal)
-	if err != nil {
-		t.Errorf("decr failed, err: %s", err.Error())
-		return
-	}
-	if val != updateVal {
-		t.Errorf("decr failed, expect %v, but %v actually", updateVal, val)
-		return
-	}
+	assert.Nil(t, err)
+	assert.Equal(t, updateVal, val)
 	_, err = decr(int(-1 << (strconv.IntSize - 1)))
-	if err == nil {
-		t.Error("decr failed")
-		return
-	}
+
+	assert.NotNil(t, err)
 
 	// int32
 	originVal = int32(2)
 	updateVal = int32(1)
 	val, err = decr(originVal)
-	if err != nil {
-		t.Errorf("decr failed, err: %s", err.Error())
-		return
-	}
-	if val != updateVal {
-		t.Errorf("decr failed, expect %v, but %v actually", updateVal, val)
-		return
-	}
+
+	assert.Nil(t, err)
+	assert.Equal(t, updateVal, val)
 	_, err = decr(int32(math.MinInt32))
-	if err == nil {
-		t.Error("decr failed")
-		return
-	}
+	assert.NotNil(t, err)
 
 	// int64
 	originVal = int64(2)
 	updateVal = int64(1)
 	val, err = decr(originVal)
-	if err != nil {
-		t.Errorf("decr failed, err: %s", err.Error())
-		return
-	}
-	if val != updateVal {
-		t.Errorf("decr failed, expect %v, but %v actually", updateVal, val)
-		return
-	}
+	assert.Nil(t, err)
+	assert.Equal(t, updateVal, val)
 	_, err = decr(int64(math.MinInt64))
-	if err == nil {
-		t.Error("decr failed")
-		return
-	}
-
+	assert.NotNil(t, err)
 	// uint
 	originVal = uint(2)
 	updateVal = uint(1)
 	val, err = decr(originVal)
-	if err != nil {
-		t.Errorf("decr failed, err: %s", err.Error())
-		return
-	}
-	if val != updateVal {
-		t.Errorf("decr failed, expect %v, but %v actually", updateVal, val)
-		return
-	}
+	assert.Nil(t, err)
+	assert.Equal(t, updateVal, val)
 	_, err = decr(uint(0))
-	if err == nil {
-		t.Error("decr failed")
-		return
-	}
+	assert.NotNil(t, err)
 
 	// uint32
 	originVal = uint32(2)
 	updateVal = uint32(1)
 	val, err = decr(originVal)
-	if err != nil {
-		t.Errorf("decr failed, err: %s", err.Error())
-		return
-	}
-	if val != updateVal {
-		t.Errorf("decr failed, expect %v, but %v actually", updateVal, val)
-		return
-	}
+	assert.Nil(t, err)
+	assert.Equal(t, updateVal, val)
 	_, err = decr(uint32(0))
-	if err == nil {
-		t.Error("decr failed")
-		return
-	}
+	assert.NotNil(t, err)
 
 	// uint64
 	originVal = uint64(2)
 	updateVal = uint64(1)
 	val, err = decr(originVal)
-	if err != nil {
-		t.Errorf("decr failed, err: %s", err.Error())
-		return
-	}
-	if val != updateVal {
-		t.Errorf("decr failed, expect %v, but %v actually", updateVal, val)
-		return
-	}
+	assert.Nil(t, err)
+	assert.Equal(t, updateVal, val)
 	_, err = decr(uint64(0))
-	if err == nil {
-		t.Error("decr failed")
-		return
-	}
+	assert.NotNil(t, err)
 
 	// other type
 	_, err = decr("string")
-	if err == nil {
-		t.Error("decr failed")
-		return
-	}
+	assert.NotNil(t, err)
 }

client/cache/conv_test.go

@@ -16,128 +16,74 @@ package cache
 
 import (
 	"testing"
+
+	"github.com/stretchr/testify/assert"
 )
 
 func TestGetString(t *testing.T) {
 	var t1 = "test1"
-	if "test1" != GetString(t1) {
-		t.Error("get string from string error")
-	}
-	var t2 = []byte("test2")
-	if "test2" != GetString(t2) {
-		t.Error("get string from byte array error")
-	}
-	var t3 = 1
-	if "1" != GetString(t3) {
-		t.Error("get string from int error")
-	}
-	var t4 int64 = 1
-	if "1" != GetString(t4) {
-		t.Error("get string from int64 error")
-	}
-	var t5 = 1.1
-	if "1.1" != GetString(t5) {
-		t.Error("get string from float64 error")
-	}
 
-	if "" != GetString(nil) {
-		t.Error("get string from nil error")
-	}
+	assert.Equal(t, "test1", GetString(t1))
+	var t2 = []byte("test2")
+	assert.Equal(t, "test2", GetString(t2))
+	var t3 = 1
+	assert.Equal(t, "1", GetString(t3))
+	var t4 int64 = 1
+	assert.Equal(t, "1", GetString(t4))
+	var t5 = 1.1
+	assert.Equal(t, "1.1", GetString(t5))
+	assert.Equal(t, "", GetString(nil))
 }
 
 func TestGetInt(t *testing.T) {
 	var t1 = 1
-	if 1 != GetInt(t1) {
-		t.Error("get int from int error")
-	}
+	assert.Equal(t, 1, GetInt(t1))
 	var t2 int32 = 32
-	if 32 != GetInt(t2) {
-		t.Error("get int from int32 error")
-	}
+	assert.Equal(t, 32, GetInt(t2))
+
 	var t3 int64 = 64
-	if 64 != GetInt(t3) {
-		t.Error("get int from int64 error")
-	}
+	assert.Equal(t, 64, GetInt(t3))
 	var t4 = "128"
-	if 128 != GetInt(t4) {
-		t.Error("get int from num string error")
-	}
-	if 0 != GetInt(nil) {
-		t.Error("get int from nil error")
-	}
+
+	assert.Equal(t, 128, GetInt(t4))
+	assert.Equal(t, 0, GetInt(nil))
 }
 
 func TestGetInt64(t *testing.T) {
 	var i int64 = 1
 	var t1 = 1
-	if i != GetInt64(t1) {
-		t.Error("get int64 from int error")
-	}
+	assert.Equal(t, i, GetInt64(t1))
 	var t2 int32 = 1
-	if i != GetInt64(t2) {
-		t.Error("get int64 from int32 error")
-	}
+
+	assert.Equal(t, i, GetInt64(t2))
 	var t3 int64 = 1
-	if i != GetInt64(t3) {
-		t.Error("get int64 from int64 error")
-	}
+	assert.Equal(t, i, GetInt64(t3))
 	var t4 = "1"
-	if i != GetInt64(t4) {
-		t.Error("get int64 from num string error")
-	}
-	if 0 != GetInt64(nil) {
-		t.Error("get int64 from nil")
-	}
+	assert.Equal(t, i, GetInt64(t4))
+	assert.Equal(t, int64(0), GetInt64(nil))
 }
 
 func TestGetFloat64(t *testing.T) {
 	var f = 1.11
 	var t1 float32 = 1.11
-	if f != GetFloat64(t1) {
-		t.Error("get float64 from float32 error")
-	}
+	assert.Equal(t, f, GetFloat64(t1))
 	var t2 = 1.11
-	if f != GetFloat64(t2) {
-		t.Error("get float64 from float64 error")
-	}
+	assert.Equal(t, f, GetFloat64(t2))
 	var t3 = "1.11"
-	if f != GetFloat64(t3) {
-		t.Error("get float64 from string error")
-	}
+	assert.Equal(t, f, GetFloat64(t3))
 
 	var f2 float64 = 1
 	var t4 = 1
-	if f2 != GetFloat64(t4) {
-		t.Error("get float64 from int error")
-	}
+	assert.Equal(t, f2, GetFloat64(t4))
 
-	if 0 != GetFloat64(nil) {
-		t.Error("get float64 from nil error")
-	}
+	assert.Equal(t, float64(0), GetFloat64(nil))
 }
 
 func TestGetBool(t *testing.T) {
 	var t1 = true
-	if !GetBool(t1) {
-		t.Error("get bool from bool error")
-	}
+	assert.True(t, GetBool(t1))
 	var t2 = "true"
-	if !GetBool(t2) {
-		t.Error("get bool from string error")
-	}
-	if GetBool(nil) {
-		t.Error("get bool from nil error")
-	}
-}
+	assert.True(t, GetBool(t2))
 
-func byteArrayEquals(a []byte, b []byte) bool {
-	if len(a) != len(b) {
-		return false
-	}
-	for i, v := range a {
-		if v != b[i] {
-			return false
-		}
-	}
-	return true
+	assert.False(t, GetBool(nil))
 }

client/cache/memcache/memcache.go

@@ -84,7 +84,7 @@ func (rc *Cache) GetMulti(ctx context.Context, keys []string) ([]interface{}, er
 	keysErr := make([]string, 0)
 	for i, ki := range keys {
 		if _, ok := mv[ki]; !ok {
-			keysErr = append(keysErr, fmt.Sprintf("key [%s] error: %s", ki, "the key isn't exist"))
+			keysErr = append(keysErr, fmt.Sprintf("key [%s] error: %s", ki, "key not exist"))
 			continue
 		}
 		rv[i] = mv[ki].Value

client/cache/memcache/memcache_test.go

@@ -23,6 +23,7 @@ import (
 	"time"
 
 	_ "github.com/bradfitz/gomemcache/memcache"
+	"github.com/stretchr/testify/assert"
 
 	"github.com/beego/beego/v2/client/cache"
 )
@@ -34,78 +35,63 @@ func TestMemcacheCache(t *testing.T) {
 	}
 
 	bm, err := cache.NewCache("memcache", fmt.Sprintf(`{"conn": "%s"}`, addr))
-	if err != nil {
-		t.Error("init err")
-	}
+	assert.Nil(t, err)
+
 	timeoutDuration := 10 * time.Second
-	if err = bm.Put(context.Background(), "astaxie", "1", timeoutDuration); err != nil {
-		t.Error("set Error", err)
-	}
-	if res, _ := bm.IsExist(context.Background(), "astaxie"); !res {
-		t.Error("check err")
-	}
+
+	assert.Nil(t, bm.Put(context.Background(), "astaxie", "1", timeoutDuration))
+	res, _ := bm.IsExist(context.Background(), "astaxie")
+	assert.True(t, res)
 
 	time.Sleep(11 * time.Second)
 
-	if res, _ := bm.IsExist(context.Background(), "astaxie"); res {
-		t.Error("check err")
-	}
-	if err = bm.Put(context.Background(), "astaxie", "1", timeoutDuration); err != nil {
-		t.Error("set Error", err)
-	}
+	res, _ = bm.IsExist(context.Background(), "astaxie")
+	assert.False(t, res)
+
+	assert.Nil(t, bm.Put(context.Background(), "astaxie", "1", timeoutDuration))
 
 	val, _ := bm.Get(context.Background(), "astaxie")
-	if v, err := strconv.Atoi(string(val.([]byte))); err != nil || v != 1 {
-		t.Error("get err")
-	}
+	v, err := strconv.Atoi(string(val.([]byte)))
+	assert.Nil(t, err)
+	assert.Equal(t, 1, v)
 
-	if err = bm.Incr(context.Background(), "astaxie"); err != nil {
-		t.Error("Incr Error", err)
-	}
+	assert.Nil(t, bm.Incr(context.Background(), "astaxie"))
 
 	val, _ = bm.Get(context.Background(), "astaxie")
-	if v, err := strconv.Atoi(string(val.([]byte))); err != nil || v != 2 {
-		t.Error("get err")
-	}
+	v, err = strconv.Atoi(string(val.([]byte)))
+	assert.Nil(t, err)
+	assert.Equal(t, 2, v)
 
-	if err = bm.Decr(context.Background(), "astaxie"); err != nil {
-		t.Error("Decr Error", err)
-	}
+	assert.Nil(t, bm.Decr(context.Background(), "astaxie"))
 
 	val, _ = bm.Get(context.Background(), "astaxie")
-	if v, err := strconv.Atoi(string(val.([]byte))); err != nil || v != 1 {
-		t.Error("get err")
-	}
+	v, err = strconv.Atoi(string(val.([]byte)))
+	assert.Nil(t, err)
+	assert.Equal(t, 1, v)
 	bm.Delete(context.Background(), "astaxie")
-	if res, _ := bm.IsExist(context.Background(), "astaxie"); res {
-		t.Error("delete err")
-	}
 
+	res, _ = bm.IsExist(context.Background(), "astaxie")
+	assert.False(t, res)
+
+	assert.Nil(t,bm.Put(context.Background(), "astaxie", "author", timeoutDuration) )
 	// test string
-	if err = bm.Put(context.Background(), "astaxie", "author", timeoutDuration); err != nil {
-		t.Error("set Error", err)
-	}
-	if res, _ := bm.IsExist(context.Background(), "astaxie"); !res {
-		t.Error("check err")
-	}
+	res, _ = bm.IsExist(context.Background(), "astaxie")
+	assert.True(t, res)
 
 	val, _ = bm.Get(context.Background(), "astaxie")
-	if v := val.([]byte); string(v) != "author" {
-		t.Error("get err")
-	}
+	vs := val.([]byte)
+	assert.Equal(t, "author", string(vs))
 
 	// test GetMulti
-	if err = bm.Put(context.Background(), "astaxie1", "author1", timeoutDuration); err != nil {
-		t.Error("set Error", err)
-	}
-	if res, _ := bm.IsExist(context.Background(), "astaxie1"); !res {
-		t.Error("check err")
-	}
+	assert.Nil(t, bm.Put(context.Background(), "astaxie1", "author1", timeoutDuration))
+
+
+	res, _ = bm.IsExist(context.Background(), "astaxie1")
+	assert.True(t, res)
 
 	vv, _ := bm.GetMulti(context.Background(), []string{"astaxie", "astaxie1"})
-	if len(vv) != 2 {
-		t.Error("GetMulti ERROR")
-	}
+	assert.Equal(t, 2, len(vv))
+
 	if string(vv[0].([]byte)) != "author" && string(vv[0].([]byte)) != "author1" {
 		t.Error("GetMulti ERROR")
 	}
@@ -114,21 +100,14 @@ func TestMemcacheCache(t *testing.T) {
 	}
 
 	vv, err = bm.GetMulti(context.Background(), []string{"astaxie0", "astaxie1"})
-	if len(vv) != 2 {
-		t.Error("GetMulti ERROR")
-	}
-	if vv[0] != nil {
-		t.Error("GetMulti ERROR")
-	}
-	if string(vv[1].([]byte)) != "author1" {
-		t.Error("GetMulti ERROR")
-	}
-	if err != nil && err.Error() == "key [astaxie0] error: key isn't exist" {
-		t.Error("GetMulti ERROR")
-	}
+	assert.Equal(t, 2, len(vv))
+	assert.Nil(t, vv[0])
 
+	assert.Equal(t, "author1", string(vv[1].([]byte)))
+
+	assert.NotNil(t, err)
+	assert.Equal(t, "key [astaxie0] error: key not exist", err.Error())
+
+	assert.Nil(t, bm.ClearAll(context.Background()))
 	// test clear all
-	if err = bm.ClearAll(context.Background()); err != nil {
-		t.Error("clear all err")
-	}
 }

client/cache/memory.go

@@ -29,6 +29,8 @@ var (
 	DefaultEvery = 60 // 1 minute
 )
 
+const keyNotExistMsg = "key not exist"
+
 // MemoryItem stores memory cache item.
 type MemoryItem struct {
 	val         interface{}
@@ -70,7 +72,7 @@ func (bc *MemoryCache) Get(ctx context.Context, key string) (interface{}, error)
 		}
 		return itm.val, nil
 	}
-	return nil, errors.New("the key isn't exist")
+	return nil, errors.New(keyNotExistMsg)
 }
 
 // GetMulti gets caches from memory.
@@ -112,7 +114,8 @@ func (bc *MemoryCache) Delete(ctx context.Context, key string) error {
 	bc.Lock()
 	defer bc.Unlock()
 	if _, ok := bc.items[key]; !ok {
-		return errors.New("key not exist")
+		
+		return errors.New(keyNotExistMsg)
 	}
 	delete(bc.items, key)
 	if _, ok := bc.items[key]; ok {
@@ -128,7 +131,7 @@ func (bc *MemoryCache) Incr(ctx context.Context, key string) error {
 	defer bc.Unlock()
 	itm, ok := bc.items[key]
 	if !ok {
-		return errors.New("key not exist")
+		return errors.New(keyNotExistMsg)
 	}
 
 	val, err := incr(itm.val)
@@ -145,7 +148,7 @@ func (bc *MemoryCache) Decr(ctx context.Context, key string) error {
 	defer bc.Unlock()
 	itm, ok := bc.items[key]
 	if !ok {
-		return errors.New("key not exist")
+		return errors.New(keyNotExistMsg)
 	}
 
 	val, err := decr(itm.val)

client/cache/redis/redis_test.go

@@ -35,96 +35,74 @@ func TestRedisCache(t *testing.T) {
 	}
 
 	bm, err := cache.NewCache("redis", fmt.Sprintf(`{"conn": "%s"}`, redisAddr))
-	if err != nil {
-		t.Error("init err")
-	}
-	timeoutDuration := 10 * time.Second
-	if err = bm.Put(context.Background(), "astaxie", 1, timeoutDuration); err != nil {
-		t.Error("set Error", err)
-	}
-	if res, _ := bm.IsExist(context.Background(), "astaxie"); !res {
-		t.Error("check err")
-	}
+	assert.Nil(t, err)
+	timeoutDuration := 3 * time.Second
 
-	time.Sleep(11 * time.Second)
+	assert.Nil(t, bm.Put(context.Background(), "astaxie", 1, timeoutDuration))
+
+
+	res, _ := bm.IsExist(context.Background(), "astaxie")
+	assert.True(t, res)
+
+	time.Sleep(5 * time.Second)
+
+	res, _ = bm.IsExist(context.Background(), "astaxie")
+	assert.False(t, res)
+
+	assert.Nil(t, bm.Put(context.Background(), "astaxie", 1, timeoutDuration))
 
-	if res, _ := bm.IsExist(context.Background(), "astaxie"); res {
-		t.Error("check err")
-	}
-	if err = bm.Put(context.Background(), "astaxie", 1, timeoutDuration); err != nil {
-		t.Error("set Error", err)
-	}
 
 	val, _ := bm.Get(context.Background(), "astaxie")
-	if v, _ := redis.Int(val, err); v != 1 {
-		t.Error("get err")
-	}
+	v, _ := redis.Int(val, err)
+	assert.Equal(t, 1, v)
 
-	if err = bm.Incr(context.Background(), "astaxie"); err != nil {
-		t.Error("Incr Error", err)
-	}
+	assert.Nil(t, bm.Incr(context.Background(), "astaxie"))
 	val, _ = bm.Get(context.Background(), "astaxie")
-	if v, _ := redis.Int(val, err); v != 2 {
-		t.Error("get err")
-	}
+	v, _ = redis.Int(val, err)
+	assert.Equal(t, 2, v)
 
-	if err = bm.Decr(context.Background(), "astaxie"); err != nil {
-		t.Error("Decr Error", err)
-	}
+	assert.Nil(t, bm.Decr(context.Background(), "astaxie"))
 
 	val, _ = bm.Get(context.Background(), "astaxie")
-	if v, _ := redis.Int(val, err); v != 1 {
-		t.Error("get err")
-	}
+	v, _ = redis.Int(val, err)
+	assert.Equal(t, 1, v)
 	bm.Delete(context.Background(), "astaxie")
-	if res, _ := bm.IsExist(context.Background(), "astaxie"); res {
-		t.Error("delete err")
-	}
 
+	res, _ = bm.IsExist(context.Background(), "astaxie")
+	assert.False(t, res)
+
+	assert.Nil(t, bm.Put(context.Background(), "astaxie", "author", timeoutDuration))
 	// test string
-	if err = bm.Put(context.Background(), "astaxie", "author", timeoutDuration); err != nil {
-		t.Error("set Error", err)
-	}
-	if res, _ := bm.IsExist(context.Background(), "astaxie"); !res {
-		t.Error("check err")
-	}
+
+	res, _ = bm.IsExist(context.Background(), "astaxie")
+	assert.True(t, res)
 
 	val, _ = bm.Get(context.Background(), "astaxie")
-	if v, _ := redis.String(val, err); v != "author" {
-		t.Error("get err")
-	}
+	vs, _ := redis.String(val, err)
+	assert.Equal(t, "author", vs)
 
 	// test GetMulti
-	if err = bm.Put(context.Background(), "astaxie1", "author1", timeoutDuration); err != nil {
-		t.Error("set Error", err)
-	}
-	if res, _ := bm.IsExist(context.Background(), "astaxie1"); !res {
-		t.Error("check err")
-	}
+	assert.Nil(t, bm.Put(context.Background(), "astaxie1", "author1", timeoutDuration))
+
+	res, _ = bm.IsExist(context.Background(), "astaxie1")
+	assert.True(t, res)
 
 	vv, _ := bm.GetMulti(context.Background(), []string{"astaxie", "astaxie1"})
-	if len(vv) != 2 {
-		t.Error("GetMulti ERROR")
-	}
-	if v, _ := redis.String(vv[0], nil); v != "author" {
-		t.Error("GetMulti ERROR")
-	}
-	if v, _ := redis.String(vv[1], nil); v != "author1" {
-		t.Error("GetMulti ERROR")
-	}
+	assert.Equal(t, 2, len(vv))
+	vs, _ = redis.String(vv[0], nil)
+	assert.Equal(t, "author", vs)
+
+	vs, _ = redis.String(vv[1], nil)
+	assert.Equal(t, "author1", vs)
 
 	vv, _ = bm.GetMulti(context.Background(), []string{"astaxie0", "astaxie1"})
-	if vv[0] != nil {
-		t.Error("GetMulti ERROR")
-	}
-	if v, _ := redis.String(vv[1], nil); v != "author1" {
-		t.Error("GetMulti ERROR")
-	}
+	assert.Nil(t, vv[0])
+
+	vs, _ = redis.String(vv[1], nil)
+	assert.Equal(t, "author1", vs)
 
 	// test clear all
-	if err = bm.ClearAll(context.Background()); err != nil {
-		t.Error("clear all err")
-	}
+	assert.Nil(t, bm.ClearAll(context.Background()))
 }
 
 func TestCache_Scan(t *testing.T) {
@@ -137,35 +115,24 @@ func TestCache_Scan(t *testing.T) {
 
 	// init
 	bm, err := cache.NewCache("redis", fmt.Sprintf(`{"conn": "%s"}`, addr))
-	if err != nil {
-		t.Error("init err")
-	}
+
+	assert.Nil(t, err)
 	// insert all
 	for i := 0; i < 100; i++ {
-		if err = bm.Put(context.Background(), fmt.Sprintf("astaxie%d", i), fmt.Sprintf("author%d", i), timeoutDuration); err != nil {
-			t.Error("set Error", err)
-		}
+		assert.Nil(t, bm.Put(context.Background(), fmt.Sprintf("astaxie%d", i), fmt.Sprintf("author%d", i), timeoutDuration))
 	}
 	time.Sleep(time.Second)
 	// scan all for the first time
 	keys, err := bm.(*Cache).Scan(DefaultKey + ":*")
-	if err != nil {
-		t.Error("scan Error", err)
-	}
+	assert.Nil(t, err)
 
 	assert.Equal(t, 100, len(keys), "scan all error")
 
 	// clear all
-	if err = bm.ClearAll(context.Background()); err != nil {
-		t.Error("clear all err")
-	}
+	assert.Nil(t, bm.ClearAll(context.Background()))
 
 	// scan all for the second time
 	keys, err = bm.(*Cache).Scan(DefaultKey + ":*")
-	if err != nil {
-		t.Error("scan Error", err)
-	}
-	if len(keys) != 0 {
-		t.Error("scan all err")
-	}
+	assert.Nil(t, err)
+	assert.Equal(t, 0, len(keys))
 }

client/cache/ssdb/ssdb.go

@@ -63,7 +63,7 @@ func (rc *Cache) GetMulti(ctx context.Context, keys []string) ([]interface{}, er
 	keysErr := make([]string, 0)
 	for i, ki := range keys {
 		if _, ok := keyIdx[ki]; !ok {
-			keysErr = append(keysErr, fmt.Sprintf("key [%s] error: %s", ki, "the key isn't exist"))
+			keysErr = append(keysErr, fmt.Sprintf("key [%s] error: %s", ki, "key not exist"))
 			continue
 		}
 		values[i] = res[keyIdx[ki]+1]

client/cache/ssdb/ssdb_test.go

@@ -8,6 +8,8 @@ import (
 	"testing"
 	"time"
 
+	"github.com/stretchr/testify/assert"
+
 	"github.com/beego/beego/v2/client/cache"
 )
 
@@ -19,114 +21,80 @@ func TestSsdbcacheCache(t *testing.T) {
 	}
 
 	ssdb, err := cache.NewCache("ssdb", fmt.Sprintf(`{"conn": "%s"}`, ssdbAddr))
-	if err != nil {
-		t.Error("init err")
-	}
+	assert.Nil(t, err)
 
 	// test put and exist
-	if res, _ := ssdb.IsExist(context.Background(), "ssdb"); res {
-		t.Error("check err")
-	}
-	timeoutDuration := 10 * time.Second
+	res, _ := ssdb.IsExist(context.Background(), "ssdb")
+	assert.False(t, res)
+	timeoutDuration := 3 * time.Second
 	// timeoutDuration := -10*time.Second   if timeoutDuration is negtive,it means permanent
-	if err = ssdb.Put(context.Background(), "ssdb", "ssdb", timeoutDuration); err != nil {
-		t.Error("set Error", err)
-	}
-	if res, _ := ssdb.IsExist(context.Background(), "ssdb"); !res {
-		t.Error("check err")
-	}
+
+	assert.Nil(t,  ssdb.Put(context.Background(), "ssdb", "ssdb", timeoutDuration))
+
+	res, _ = ssdb.IsExist(context.Background(), "ssdb")
+	assert.True(t, res)
 
 	// Get test done
-	if err = ssdb.Put(context.Background(), "ssdb", "ssdb", timeoutDuration); err != nil {
-		t.Error("set Error", err)
-	}
+	assert.Nil(t, ssdb.Put(context.Background(), "ssdb", "ssdb", timeoutDuration))
 
-	if v, _ := ssdb.Get(context.Background(), "ssdb"); v != "ssdb" {
-		t.Error("get Error")
-	}
+	v, _ := ssdb.Get(context.Background(), "ssdb")
+	assert.Equal(t, "ssdb", v)
 
 	// inc/dec test done
-	if err = ssdb.Put(context.Background(), "ssdb", "2", timeoutDuration); err != nil {
-		t.Error("set Error", err)
-	}
-	if err = ssdb.Incr(context.Background(), "ssdb"); err != nil {
-		t.Error("incr Error", err)
-	}
+	assert.Nil(t, ssdb.Put(context.Background(), "ssdb", "2", timeoutDuration))
+
+	assert.Nil(t, ssdb.Incr(context.Background(), "ssdb"))
 
 	val, _ := ssdb.Get(context.Background(), "ssdb")
-	if v, err := strconv.Atoi(val.(string)); err != nil || v != 3 {
-		t.Error("get err")
-	}
+	v, err = strconv.Atoi(val.(string))
+	assert.Nil(t, err)
+	assert.Equal(t, 3, v)
 
-	if err = ssdb.Decr(context.Background(), "ssdb"); err != nil {
-		t.Error("decr error")
-	}
+	assert.Nil(t, ssdb.Decr(context.Background(), "ssdb"))
 
 	// test del
-	if err = ssdb.Put(context.Background(), "ssdb", "3", timeoutDuration); err != nil {
-		t.Error("set Error", err)
-	}
+	assert.Nil(t, ssdb.Put(context.Background(), "ssdb", "3", timeoutDuration))
 
 	val, _ = ssdb.Get(context.Background(), "ssdb")
-	if v, err := strconv.Atoi(val.(string)); err != nil || v != 3 {
-		t.Error("get err")
-	}
-	if err := ssdb.Delete(context.Background(), "ssdb"); err == nil {
-		if e, _ := ssdb.IsExist(context.Background(), "ssdb"); e {
-			t.Error("delete err")
-		}
-	}
+	v, err = strconv.Atoi(val.(string))
+	assert.Equal(t, 3, v)
+	assert.Nil(t, err)
 
+	assert.NotNil(t, ssdb.Delete(context.Background(), "ssdb"))
+	assert.Nil(t, ssdb.Put(context.Background(), "ssdb", "ssdb", -10*time.Second))
 	// test string
-	if err = ssdb.Put(context.Background(), "ssdb", "ssdb", -10*time.Second); err != nil {
-		t.Error("set Error", err)
-	}
-	if res, _ := ssdb.IsExist(context.Background(), "ssdb"); !res {
-		t.Error("check err")
-	}
-	if v, _ := ssdb.Get(context.Background(), "ssdb"); v.(string) != "ssdb" {
-		t.Error("get err")
-	}
+
+	res, _ = ssdb.IsExist(context.Background(), "ssdb")
+	assert.True(t, res)
+
+	v, _ = ssdb.Get(context.Background(), "ssdb")
+	assert.Equal(t, "ssdb", v.(string))
 
 	// test GetMulti done
-	if err = ssdb.Put(context.Background(), "ssdb1", "ssdb1", -10*time.Second); err != nil {
-		t.Error("set Error", err)
-	}
-	if res, _ := ssdb.IsExist(context.Background(), "ssdb1"); !res {
-		t.Error("check err")
-	}
+	assert.Nil(t, ssdb.Put(context.Background(), "ssdb1", "ssdb1", -10*time.Second))
+
+	res, _ = ssdb.IsExist(context.Background(), "ssdb1")
+	assert.True(t, res)
 	vv, _ := ssdb.GetMulti(context.Background(), []string{"ssdb", "ssdb1"})
-	if len(vv) != 2 {
-		t.Error("getmulti error")
-	}
-	if vv[0].(string) != "ssdb" {
-		t.Error("getmulti error")
-	}
-	if vv[1].(string) != "ssdb1" {
-		t.Error("getmulti error")
-	}
+	assert.Equal(t, 2, len(vv))
+
+	assert.Equal(t, "ssdb", vv[0])
+	assert.Equal(t, "ssdb1", vv[1])
 
 	vv, err = ssdb.GetMulti(context.Background(), []string{"ssdb", "ssdb11"})
-	if len(vv) != 2 {
-		t.Error("getmulti error")
-	}
-	if vv[0].(string) != "ssdb" {
-		t.Error("getmulti error")
-	}
-	if vv[1] != nil {
-		t.Error("getmulti error")
-	}
-	if err != nil && err.Error() != "key [ssdb11] error: the key isn't exist" {
-		t.Error("getmulti error")
-	}
+
+	assert.Equal(t, 2, len(vv))
+
+	assert.Equal(t, "ssdb", vv[0])
+	assert.Nil(t,  vv[1])
+
+	assert.NotNil(t, err)
+	assert.Equal(t, "key [ssdb11] error: key not exist", err.Error())
 
 	// test clear all done
-	if err = ssdb.ClearAll(context.Background()); err != nil {
-		t.Error("clear all err")
-	}
+	assert.Nil(t, ssdb.ClearAll(context.Background()))
 	e1, _ := ssdb.IsExist(context.Background(), "ssdb")
 	e2, _ := ssdb.IsExist(context.Background(), "ssdb1")
-	if e1 || e2 {
-		t.Error("check err")
-	}
+	assert.False(t, e1)
+	assert.False(t, e2)
 }