github.com/cloudreve/Cloudreve/v3@v3.0.0-20240224133659-3edb00a6484c/middleware/mock.go (about) 1 package middleware 2 3 import ( 4 "github.com/cloudreve/Cloudreve/v3/pkg/util" 5 "github.com/gin-gonic/gin" 6 ) 7 8 // SessionMock 测试时模拟Session 9 var SessionMock = make(map[string]interface{}) 10 11 // ContextMock 测试时模拟Context 12 var ContextMock = make(map[string]interface{}) 13 14 // MockHelper 单元测试助手中间件 15 func MockHelper() gin.HandlerFunc { 16 return func(c *gin.Context) { 17 // 将SessionMock写入会话 18 util.SetSession(c, SessionMock) 19 for key, value := range ContextMock { 20 c.Set(key, value) 21 } 22 c.Next() 23 } 24 }