remove bus from login (#44995)

* remove bus from login

* remove comments and format correctly

* remove the downcast
This commit is contained in:
ying-jeanne
2022-02-09 17:17:09 +01:00
committed by GitHub
parent 071ff0b399
commit 78eb06ec02
6 changed files with 80 additions and 62 deletions
@@ -19,35 +19,6 @@ import (
"github.com/stretchr/testify/require"
)
type fakeMultiLDAP struct {
multildap.MultiLDAP
ID int64
userCalled bool
loginCalled bool
}
func (m *fakeMultiLDAP) Login(query *models.LoginUserQuery) (
*models.ExternalUserInfo, error,
) {
m.loginCalled = true
result := &models.ExternalUserInfo{
UserId: m.ID,
}
return result, nil
}
func (m *fakeMultiLDAP) User(login string) (
*models.ExternalUserInfo,
ldap.ServerConfig,
error,
) {
m.userCalled = true
result := &models.ExternalUserInfo{
UserId: m.ID,
}
return result, ldap.ServerConfig{}, nil
}
const hdrName = "markelog"
func prepareMiddleware(t *testing.T, remoteCache *remotecache.RemoteCache, cb func(*http.Request, *setting.Cfg)) *AuthProxy {
@@ -152,7 +123,7 @@ func TestMiddlewareContext_ldap(t *testing.T) {
return true
}
stub := &fakeMultiLDAP{
stub := &multildap.MultiLDAPmock{
ID: id,
}
@@ -179,7 +150,7 @@ func TestMiddlewareContext_ldap(t *testing.T) {
require.NoError(t, err)
assert.Equal(t, id, gotID)
assert.True(t, stub.userCalled)
assert.True(t, stub.UserCalled)
})
t.Run("Gets nice error if LDAP is enabled, but not configured", func(t *testing.T) {
@@ -205,7 +176,7 @@ func TestMiddlewareContext_ldap(t *testing.T) {
auth := prepareMiddleware(t, cache, nil)
stub := &fakeMultiLDAP{
stub := &multildap.MultiLDAPmock{
ID: id,
}
@@ -217,6 +188,6 @@ func TestMiddlewareContext_ldap(t *testing.T) {
require.EqualError(t, err, "failed to get the user")
assert.NotEqual(t, id, gotID)
assert.False(t, stub.loginCalled)
assert.False(t, stub.LoginCalled)
})
}
+40
View File
@@ -0,0 +1,40 @@
package multildap
import (
"github.com/grafana/grafana/pkg/models"
"github.com/grafana/grafana/pkg/services/ldap"
)
type MultiLDAPmock struct {
MultiLDAP
ID int64
UserCalled bool
LoginCalled bool
UserInfo *models.User
AuthModule string
ExpectedErr error
}
func (m *MultiLDAPmock) Login(query *models.LoginUserQuery) (
*models.ExternalUserInfo, error,
) {
m.LoginCalled = true
query.User = m.UserInfo
query.AuthModule = m.AuthModule
result := &models.ExternalUserInfo{
UserId: m.ID,
}
return result, m.ExpectedErr
}
func (m *MultiLDAPmock) User(login string) (
*models.ExternalUserInfo,
ldap.ServerConfig,
error,
) {
m.UserCalled = true
result := &models.ExternalUserInfo{
UserId: m.ID,
}
return result, ldap.ServerConfig{}, nil
}