mirror of
https://github.com/grafana/grafana.git
synced 2025-02-20 11:48:34 -06:00
7c6dd1868b
* LDAP: Add API endpoint to query the LDAP server(s) status| This endpoint returns the current status(es) of the configured LDAP server(s). The status of each server is verified by dialling and if no error is returned we assume the server is operational. This is the last piece I'll produce as an API before moving into #18759 and see the view come to life.
120 lines
2.5 KiB
Go
120 lines
2.5 KiB
Go
package multildap
|
|
|
|
import (
|
|
"github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/services/ldap"
|
|
)
|
|
|
|
// MockLDAP represents testing struct for ldap testing
|
|
type MockLDAP struct {
|
|
dialCalledTimes int
|
|
loginCalledTimes int
|
|
closeCalledTimes int
|
|
usersCalledTimes int
|
|
bindCalledTimes int
|
|
|
|
dialErrReturn error
|
|
|
|
loginErrReturn error
|
|
loginReturn *models.ExternalUserInfo
|
|
|
|
bindErrReturn error
|
|
|
|
usersErrReturn error
|
|
usersFirstReturn []*models.ExternalUserInfo
|
|
usersRestReturn []*models.ExternalUserInfo
|
|
}
|
|
|
|
// Login test fn
|
|
func (mock *MockLDAP) Login(*models.LoginUserQuery) (*models.ExternalUserInfo, error) {
|
|
|
|
mock.loginCalledTimes = mock.loginCalledTimes + 1
|
|
return mock.loginReturn, mock.loginErrReturn
|
|
}
|
|
|
|
// Users test fn
|
|
func (mock *MockLDAP) Users([]string) ([]*models.ExternalUserInfo, error) {
|
|
mock.usersCalledTimes = mock.usersCalledTimes + 1
|
|
|
|
if mock.usersCalledTimes == 1 {
|
|
return mock.usersFirstReturn, mock.usersErrReturn
|
|
}
|
|
|
|
return mock.usersRestReturn, mock.usersErrReturn
|
|
}
|
|
|
|
// UserBind test fn
|
|
func (mock *MockLDAP) UserBind(string, string) error {
|
|
return nil
|
|
}
|
|
|
|
// Dial test fn
|
|
func (mock *MockLDAP) Dial() error {
|
|
mock.dialCalledTimes = mock.dialCalledTimes + 1
|
|
return mock.dialErrReturn
|
|
}
|
|
|
|
// Close test fn
|
|
func (mock *MockLDAP) Close() {
|
|
mock.closeCalledTimes = mock.closeCalledTimes + 1
|
|
}
|
|
|
|
func (mock *MockLDAP) Bind() error {
|
|
mock.bindCalledTimes++
|
|
return mock.bindErrReturn
|
|
}
|
|
|
|
// MockMultiLDAP represents testing struct for multildap testing
|
|
type MockMultiLDAP struct {
|
|
LoginCalledTimes int
|
|
UsersCalledTimes int
|
|
UserCalledTimes int
|
|
PingCalledTimes int
|
|
|
|
UsersResult []*models.ExternalUserInfo
|
|
}
|
|
|
|
func (mock *MockMultiLDAP) Ping() ([]*ServerStatus, error) {
|
|
mock.PingCalledTimes = mock.PingCalledTimes + 1
|
|
|
|
return nil, nil
|
|
}
|
|
|
|
// Login test fn
|
|
func (mock *MockMultiLDAP) Login(query *models.LoginUserQuery) (
|
|
*models.ExternalUserInfo, error,
|
|
) {
|
|
mock.LoginCalledTimes = mock.LoginCalledTimes + 1
|
|
return nil, nil
|
|
}
|
|
|
|
// Users test fn
|
|
func (mock *MockMultiLDAP) Users(logins []string) (
|
|
[]*models.ExternalUserInfo, error,
|
|
) {
|
|
mock.UsersCalledTimes = mock.UsersCalledTimes + 1
|
|
return mock.UsersResult, nil
|
|
}
|
|
|
|
// User test fn
|
|
func (mock *MockMultiLDAP) User(login string) (
|
|
*models.ExternalUserInfo, ldap.ServerConfig, error,
|
|
) {
|
|
mock.UserCalledTimes = mock.UserCalledTimes + 1
|
|
return nil, ldap.ServerConfig{}, nil
|
|
}
|
|
|
|
func setup() *MockLDAP {
|
|
mock := &MockLDAP{}
|
|
|
|
newLDAP = func(config *ldap.ServerConfig) ldap.IServer {
|
|
return mock
|
|
}
|
|
|
|
return mock
|
|
}
|
|
|
|
func teardown() {
|
|
newLDAP = ldap.New
|
|
}
|