remove bus from login (#44995)
* remove bus from login * remove comments and format correctly * remove the downcast
This commit is contained in:
@@ -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)
|
||||
})
|
||||
}
|
||||
|
||||
@@ -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
|
||||
}
|
||||
Reference in New Issue
Block a user