github.com/cloudreve/Cloudreve/v3@v3.0.0-20240224133659-3edb00a6484c/middleware/mock_test.go (about) 1 package middleware 2 3 import ( 4 "net/http" 5 "net/http/httptest" 6 "testing" 7 8 "github.com/cloudreve/Cloudreve/v3/pkg/util" 9 "github.com/gin-gonic/gin" 10 "github.com/stretchr/testify/assert" 11 ) 12 13 func TestMockHelper(t *testing.T) { 14 asserts := assert.New(t) 15 MockHelperFunc := MockHelper() 16 rec := httptest.NewRecorder() 17 c, _ := gin.CreateTestContext(rec) 18 c.Request, _ = http.NewRequest("GET", "/test", nil) 19 20 // 写入session 21 { 22 SessionMock["test"] = "pass" 23 Session("test")(c) 24 MockHelperFunc(c) 25 asserts.Equal("pass", util.GetSession(c, "test").(string)) 26 } 27 28 // 写入context 29 { 30 ContextMock["test"] = "pass" 31 MockHelperFunc(c) 32 test, exist := c.Get("test") 33 asserts.True(exist) 34 asserts.Equal("pass", test.(string)) 35 36 } 37 }