grafana/pkg/services/multildap/multidap_mock.go

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
}