From 65cf0f57aa3d3896ec96a2b68e63da3f368136cd Mon Sep 17 00:00:00 2001
From: HFO4 <912394456@qq.com>
Date: Wed, 18 Mar 2020 09:47:06 +0800
Subject: [PATCH] Fix: failed test due to policy cache

---
 middleware/auth_test.go    | 4 ++--
 middleware/option_test.go  | 2 --
 middleware/session_test.go | 1 +
 3 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/middleware/auth_test.go b/middleware/auth_test.go
index 32fdd7f..2f60f6c 100644
--- a/middleware/auth_test.go
+++ b/middleware/auth_test.go
@@ -654,7 +654,7 @@ func TestOneDriveCallbackAuth(t *testing.T) {
 		mock.ExpectQuery("SELECT(.+)users(.+)").
 			WillReturnRows(sqlmock.NewRows([]string{"id", "group_id"}).AddRow(1, 1))
 		mock.ExpectQuery("SELECT(.+)groups(.+)").
-			WillReturnRows(sqlmock.NewRows([]string{"id", "policies"}).AddRow(1, "[522]"))
+			WillReturnRows(sqlmock.NewRows([]string{"id", "policies"}).AddRow(1, "[657]"))
 		mock.ExpectQuery("SELECT(.+)policies(.+)").
 			WillReturnRows(sqlmock.NewRows([]string{"id", "access_key", "secret_key"}).AddRow(2, "123", "123"))
 		c, _ := gin.CreateTestContext(rec)
@@ -699,7 +699,7 @@ func TestCOSCallbackAuth(t *testing.T) {
 		mock.ExpectQuery("SELECT(.+)users(.+)").
 			WillReturnRows(sqlmock.NewRows([]string{"id", "group_id"}).AddRow(1, 1))
 		mock.ExpectQuery("SELECT(.+)groups(.+)").
-			WillReturnRows(sqlmock.NewRows([]string{"id", "policies"}).AddRow(1, "[522]"))
+			WillReturnRows(sqlmock.NewRows([]string{"id", "policies"}).AddRow(1, "[702]"))
 		mock.ExpectQuery("SELECT(.+)policies(.+)").
 			WillReturnRows(sqlmock.NewRows([]string{"id", "access_key", "secret_key"}).AddRow(2, "123", "123"))
 		c, _ := gin.CreateTestContext(rec)
diff --git a/middleware/option_test.go b/middleware/option_test.go
index a0af92d..97d8d5e 100644
--- a/middleware/option_test.go
+++ b/middleware/option_test.go
@@ -62,7 +62,6 @@ func TestIsFunctionEnabled(t *testing.T) {
 		c.Params = []gin.Param{}
 		c.Request, _ = http.NewRequest("POST", "/api/v3/file/dellete/1", nil)
 		TestFunc(c)
-		asserts.NoError(mock.ExpectationsWereMet())
 		asserts.True(c.IsAborted())
 	}
 	// 开启
@@ -72,7 +71,6 @@ func TestIsFunctionEnabled(t *testing.T) {
 		c.Params = []gin.Param{}
 		c.Request, _ = http.NewRequest("POST", "/api/v3/file/dellete/1", nil)
 		TestFunc(c)
-		asserts.NoError(mock.ExpectationsWereMet())
 		asserts.False(c.IsAborted())
 	}
 
diff --git a/middleware/session_test.go b/middleware/session_test.go
index cc4c23f..674f4bd 100644
--- a/middleware/session_test.go
+++ b/middleware/session_test.go
@@ -24,6 +24,7 @@ func TestSession(t *testing.T) {
 		asserts.Panics(func() {
 			Session("2333")
 		})
+		conf.RedisConfig.Server = ""
 	}
 
 }