server/beego
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix UT

Browse Source
This commit is contained in:
Anker Jam 2021-01-10 14:22:11 +08:00
parent a93e35244a
commit 87ffa0b730
3 changed files with 50 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/web/context/context.go
View File

@ -197,7 +197,7 @@ func (ctx *Context) RenderMethodResult(result interface{}) {
}
// Session return session store of this context of request
func (ctx *Context) Session() (store session.Store,err error){
func (ctx *Context) Session() (store session.Store, err error) {
if ctx.Input != nil {
if ctx.Input.CruSession != nil {
store = ctx.Input.CruSession

62
server/web/context/context_test.go
View File

@ -15,9 +15,7 @@
package context
import (
"github.com/beego/beego/v2/server/web"
"github.com/beego/beego/v2/server/web/filter/session"
webSession "github.com/beego/beego/v2/server/web/session"
"github.com/beego/beego/v2/server/web/session"
"net/http"
"net/http/httptest"
"testing"
@ -49,45 +47,25 @@ func TestXsrfReset_01(t *testing.T) {
}
}
func testRequest(t *testing.T, handler *web.ControllerRegister, path string, method string, code int) {
r, _ := http.NewRequest(method, path, nil)
w := httptest.NewRecorder()
handler.ServeHTTP(w, r)
if w.Code != code {
t.Errorf("%s, %s: %d, supposed to be %d", path, method, w.Code, code)
}
}
func TestContext_Session(t *testing.T) {
handler := web.NewControllerRegister()
handler.InsertFilterChain(
"*",
session.Session(
webSession.ProviderMemory,
webSession.CfgCookieName(`go_session_id`),
webSession.CfgSetCookie(true),
webSession.CfgGcLifeTime(3600),
webSession.CfgMaxLifeTime(3600),
webSession.CfgSecure(false),
webSession.CfgCookieLifeTime(3600),
),
)
handler.InsertFilterChain(
"*",
func(next web.FilterFunc) web.FilterFunc {
return func(ctx *Context) {
if _, err := ctx.Session(); err == nil {
t.Error()
c := NewContext()
if store, err := c.Session(); store != nil || err == nil {
t.FailNow()
}
}
func TestContext_Session1(t *testing.T) {
c := Context{}
if store, err := c.Session(); store != nil || err == nil {
t.FailNow()
}
}
func TestContext_Session2(t *testing.T) {
c := NewContext()
c.Input.CruSession = &session.MemSessionStore{}
if store, err := c.Session(); store == nil || err != nil {
t.FailNow()
}
}
},
)
handler.Any("*", func(ctx *Context) {
ctx.Output.SetStatus(200)
})
testRequest(t, handler, "/dataset1/resource1", "GET", 200)
}

32
server/web/filter/session/filter_test.go
View File

@ -52,3 +52,35 @@ func TestSession(t *testing.T) {
testRequest(t, handler, "/dataset1/resource1", "GET", 200)
}
func TestSession1(t *testing.T) {
handler := web.NewControllerRegister()
handler.InsertFilterChain(
"*",
Session(
session.ProviderMemory,
session.CfgCookieName(`go_session_id`),
session.CfgSetCookie(true),
session.CfgGcLifeTime(3600),
session.CfgMaxLifeTime(3600),
session.CfgSecure(false),
session.CfgCookieLifeTime(3600),
),
)
handler.InsertFilterChain(
"*",
func(next web.FilterFunc) web.FilterFunc {
return func(ctx *webContext.Context) {
if store, err := ctx.Session(); store == nil || err != nil {
t.Error(`store should not be nil`)
}
next(ctx)
}
},
)
handler.Any("*", func(ctx *webContext.Context) {
ctx.Output.SetStatus(200)
})
testRequest(t, handler, "/dataset1/resource1", "GET", 200)
}