From 5b012028443b407f7fdcd6cd1a98618989bae092 Mon Sep 17 00:00:00 2001
From: HFO4 <912394456@qq.com>
Date: Mon, 9 Dec 2019 19:40:24 +0800
Subject: [PATCH] Test: redis pool connection

---
 pkg/cache/redis_test.go | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/pkg/cache/redis_test.go b/pkg/cache/redis_test.go
index 41c9f14..ebdd69b 100644
--- a/pkg/cache/redis_test.go
+++ b/pkg/cache/redis_test.go
@@ -7,13 +7,27 @@ import (
 	"github.com/rafaeljusto/redigomock"
 	"github.com/stretchr/testify/assert"
 	"testing"
+	"time"
 )
 
 func TestNewRedisStore(t *testing.T) {
 	asserts := assert.New(t)
 
-	store := NewRedisStore(10, "tcp", ":2333", "", "0")
+	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) {