update cache and db with mockgen

Author: leo-aa88

pkg/cache/cache.go

@@ -8,7 +8,7 @@ import (
 	"github.com/go-redis/redis/v8"
 )
 
-type CacheInterface interface {
+type Cache interface {
 	Get(ctx context.Context, key string) *redis.StringCmd
 	Set(ctx context.Context, key string, value interface{}, expiration time.Duration) *redis.StatusCmd
 	Keys(context.Context, string) *redis.StringSliceCmd

pkg/cache/cache_mock.go

@@ -3,10 +3,15 @@ package cache
 import (
 	"testing"
 
+	"github.com/golang/mock/gomock"
+
 	"github.com/stretchr/testify/assert"
 )
 
 func TestNewRedisClient(t *testing.T) {
-	redisClient := NewRedisClient()
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+
+	redisClient := NewMockCache(ctrl)
 	assert.NotNil(t, redisClient)
 }

pkg/cache/cache_test.go

@@ -12,14 +12,15 @@ import (
 	"gorm.io/gorm"
 )
 
-type DBInterface interface {
+type Database interface {
 	Offset(offset int) *gorm.DB
 	Limit(limit int) *gorm.DB
 	Find(interface{}, ...interface{}) *gorm.DB
 	Create(value interface{}) *gorm.DB
 	Where(query interface{}, args ...interface{}) *gorm.DB
 	Delete(interface{}, ...interface{}) *gorm.DB
 	Model(model interface{}) *gorm.DB
+	First(dest interface{}, conds ...interface{}) *gorm.DB
 }
 
 func NewDatabase() *gorm.DB {

pkg/database/db.go

@@ -3,10 +3,14 @@ package database
 import (
 	"testing"
 
+	"github.com/golang/mock/gomock"
 	"github.com/stretchr/testify/assert"
 )
 
 func TestNewDatabase(t *testing.T) {
-	db := NewDatabase()
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+
+	db := NewMockDatabase(ctrl)
 	assert.NotNil(t, db)
 }