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  }