25 lines
510 B
Go
25 lines
510 B
Go
|
package middleware
|
||
|
|
|
||
|
|
import (
|
||
|
|
"cloudreve/pkg/util"
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
)
|
||
|
|
|
||
|
|
// SessionMock 测试时模拟Session
|
||
|
|
var SessionMock = make(map[string]interface{})
|
||
|
|
|
||
|
|
// ContextMock 测试时模拟Context
|
||
|
|
var ContextMock = make(map[string]interface{})
|
||
|
|
|
||
|
|
// MockHelper 单元测试助手中间件
|
||
|
|
func MockHelper() gin.HandlerFunc {
|
||
|
|
return func(c *gin.Context) {
|
||
|
|
// 将SessionMock写入会话
|
||
|
|
util.SetSession(c, SessionMock)
|
||
|
|
for key, value := range ContextMock {
|
||
|
|
c.Set(key, value)
|
||
|
|
}
|
||
|
|
c.Next()
|
||
|
|
}
|
||
|
|
}
|