dex/connector/ldapcluster/ldapcluster_test.go

package ldapcluster

import (
	"context"
	"errors"
	"testing"

	log "github.com/sirupsen/logrus"
	"github.com/stretchr/testify/assert"

	"github.com/dexidp/dex/connector"
)

type MockConn struct {
	status bool
}

func (c MockConn) Login(ctx context.Context, s connector.Scopes, username, password string) (ident connector.Identity, validPass bool, err error) {
	if c.status {
		return connector.Identity{}, false, nil
	}
	return connector.Identity{}, false, errors.New("failed")
}

func (c MockConn) Refresh(ctx context.Context, s connector.Scopes, ident connector.Identity) (connector.Identity, error) {
	if c.status {
		return connector.Identity{}, nil
	}
	return connector.Identity{}, errors.New("failed")
}

func (c MockConn) Prompt() string {
	return "name:"
}

func TestLoginSingle(t *testing.T) {
	var ctx context.Context
	var s connector.Scopes

	var c1 MockConn
	c1.status = true

	var lcc ldapClusterConnector
	lcc.MemberConnectors = append(lcc.MemberConnectors, c1)
	lcc.activeMemberIdx = 0

	var logger *log.Logger = log.New()
	lcc.logger = logger
	_, _, e := lcc.Login(ctx, s, "testuser", "password")
	assert.Equal(t, e, nil)
}

func TestLoginMultiple(t *testing.T) {
	var ctx context.Context
	var s connector.Scopes

	var c1 MockConn
	c1.status = false

	var c2 MockConn
	c2.status = true

	var lcc ldapClusterConnector
	lcc.MemberConnectors = append(lcc.MemberConnectors, c1)
	lcc.activeMemberIdx = 0
	lcc.MemberConnectors = append(lcc.MemberConnectors, c2)

	var logger *log.Logger = log.New()
	lcc.logger = logger
	_, _, e := lcc.Login(ctx, s, "testuser", "password")
	assert.Equal(t, e, nil)
	assert.Equal(t, lcc.activeMemberIdx, 1)
}

func TestLoginMultiple2(t *testing.T) {
	var ctx context.Context
	var s connector.Scopes

	var c1 MockConn
	c1.status = false

	var c2 MockConn
	c2.status = false

	var c3 MockConn
	c3.status = true

	var lcc ldapClusterConnector
	lcc.MemberConnectors = append(lcc.MemberConnectors, c1)
	lcc.activeMemberIdx = 0
	lcc.MemberConnectors = append(lcc.MemberConnectors, c2)
	lcc.MemberConnectors = append(lcc.MemberConnectors, c3)

	var logger *log.Logger = log.New()
	lcc.logger = logger
	_, _, e := lcc.Login(ctx, s, "testuser", "password")
	assert.Equal(t, e, nil)
	assert.Equal(t, lcc.activeMemberIdx, 2)
}

func TestRefreshMultiple(t *testing.T) {
	var ctx context.Context
	var s connector.Scopes

	var c1 MockConn
	c1.status = true

	var c2 MockConn
	c2.status = false

	var c3 MockConn
	c3.status = false

	var lcc ldapClusterConnector
	lcc.MemberConnectors = append(lcc.MemberConnectors, c1)
	lcc.activeMemberIdx = 1
	lcc.MemberConnectors = append(lcc.MemberConnectors, c2)
	lcc.MemberConnectors = append(lcc.MemberConnectors, c3)

	var logger *log.Logger = log.New()
	lcc.logger = logger
	_, e := lcc.Refresh(ctx, s, connector.Identity{})
	assert.Equal(t, e, nil)
	assert.Equal(t, lcc.activeMemberIdx, 0)
}