diff --git a/models/group_test.go b/models/group_test.go index 1adb9e6..525a801 100644 --- a/models/group_test.go +++ b/models/group_test.go @@ -46,12 +46,12 @@ func TestGroup_AfterFind(t *testing.T) { } err := testCase.AfterFind() asserts.NoError(err) - asserts.Equal(testCase.PolicyList, []int{1}) + asserts.Equal(testCase.PolicyList, []uint{1}) testCase.Policies = "[1,2,3,4,5]" err = testCase.AfterFind() asserts.NoError(err) - asserts.Equal(testCase.PolicyList, []int{1, 2, 3, 4, 5}) + asserts.Equal(testCase.PolicyList, []uint{1, 2, 3, 4, 5}) testCase.Policies = "[1,2,3,4,5" err = testCase.AfterFind() @@ -60,5 +60,5 @@ func TestGroup_AfterFind(t *testing.T) { testCase.Policies = "[]" err = testCase.AfterFind() asserts.NoError(err) - asserts.Equal(testCase.PolicyList, []int{}) + asserts.Equal(testCase.PolicyList, []uint{}) } diff --git a/routers/router.go b/routers/router.go index d16dfb6..13c6259 100644 --- a/routers/router.go +++ b/routers/router.go @@ -7,18 +7,9 @@ import ( "github.com/gin-gonic/gin" ) -var r *gin.Engine - -// SetupTestEngine 设置测试用gin.Engine -func SetupTestEngine(engine *gin.Engine) { - r = engine -} - // InitRouter 初始化路由 func InitRouter() *gin.Engine { - if r == nil { - r = gin.Default() - } + r := gin.Default() /* 中间件