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  }