mirror of
https://github.com/grafana/grafana.git
synced 2025-07-31 05:02:35 +08:00
update auth proxy
This commit is contained in:
@ -9,6 +9,7 @@ import (
|
||||
|
||||
ms "github.com/go-macaron/session"
|
||||
"github.com/grafana/grafana/pkg/bus"
|
||||
"github.com/grafana/grafana/pkg/login"
|
||||
m "github.com/grafana/grafana/pkg/models"
|
||||
"github.com/grafana/grafana/pkg/services/session"
|
||||
"github.com/grafana/grafana/pkg/setting"
|
||||
@ -182,6 +183,11 @@ func TestMiddlewareContext(t *testing.T) {
|
||||
return nil
|
||||
})
|
||||
|
||||
login.UpsertUser = func(ctx *m.ReqContext, cmd *m.UpsertUserCommand) error {
|
||||
cmd.User = &m.User{Id: 12}
|
||||
return nil
|
||||
}
|
||||
|
||||
sc.fakeReq("GET", "/")
|
||||
sc.req.Header.Add("X-WEBAUTH-USER", "torkelo")
|
||||
sc.exec()
|
||||
@ -208,10 +214,10 @@ func TestMiddlewareContext(t *testing.T) {
|
||||
}
|
||||
})
|
||||
|
||||
bus.AddHandler("test", func(cmd *m.CreateUserCommand) error {
|
||||
cmd.Result = m.User{Id: 33}
|
||||
login.UpsertUser = func(ctx *m.ReqContext, cmd *m.UpsertUserCommand) error {
|
||||
cmd.User = &m.User{Id: 33}
|
||||
return nil
|
||||
})
|
||||
}
|
||||
|
||||
sc.fakeReq("GET", "/")
|
||||
sc.req.Header.Add("X-WEBAUTH-USER", "torkelo")
|
||||
@ -270,6 +276,11 @@ func TestMiddlewareContext(t *testing.T) {
|
||||
return nil
|
||||
})
|
||||
|
||||
login.UpsertUser = func(ctx *m.ReqContext, cmd *m.UpsertUserCommand) error {
|
||||
cmd.User = &m.User{Id: 33}
|
||||
return nil
|
||||
}
|
||||
|
||||
sc.fakeReq("GET", "/")
|
||||
sc.req.Header.Add("X-WEBAUTH-USER", "torkelo")
|
||||
sc.req.RemoteAddr = "[2001::23]:12345"
|
||||
|
Reference in New Issue
Block a user