36 lines
723 B
Go
36 lines
723 B
Go
package middleware
|
|
|
|
import (
|
|
"github.com/HFO4/cloudreve/pkg/util"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/assert"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
)
|
|
|
|
func TestMockHelper(t *testing.T) {
|
|
asserts := assert.New(t)
|
|
MockHelperFunc := MockHelper()
|
|
rec := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(rec)
|
|
c.Request, _ = http.NewRequest("GET", "/test", nil)
|
|
|
|
// 写入session
|
|
{
|
|
SessionMock["test"] = "pass"
|
|
Session("test")(c)
|
|
MockHelperFunc(c)
|
|
asserts.Equal("pass", util.GetSession(c, "test").(string))
|
|
}
|
|
|
|
// 写入context
|
|
{
|
|
ContextMock["test"] = "pass"
|
|
MockHelperFunc(c)
|
|
test, exist := c.Get("test")
|
|
asserts.True(exist)
|
|
asserts.Equal("pass", test.(string))
|
|
|
|
}
|
|
}
|