pomerium/proxy/clients/mock_clients.go

package clients // import "github.com/pomerium/pomerium/proxy/clients"

import (
	"context"

	"github.com/pomerium/pomerium/internal/sessions"
)

// MockAuthenticate provides a mocked implementation of the authenticator interface.
type MockAuthenticate struct {
	RedeemError      error
	RedeemResponse   *sessions.SessionState
	RefreshResponse  *sessions.SessionState
	RefreshError     error
	ValidateResponse bool
	ValidateError    error
	CloseError       error
}

// Redeem is a mocked authenticator client function.
func (a MockAuthenticate) Redeem(ctx context.Context, code string) (*sessions.SessionState, error) {
	return a.RedeemResponse, a.RedeemError
}

// Refresh is a mocked authenticator client function.
func (a MockAuthenticate) Refresh(ctx context.Context, s *sessions.SessionState) (*sessions.SessionState, error) {
	return a.RefreshResponse, a.RefreshError
}

// Validate is a mocked authenticator client function.
func (a MockAuthenticate) Validate(ctx context.Context, idToken string) (bool, error) {
	return a.ValidateResponse, a.ValidateError
}

// Close is a mocked authenticator client function.
func (a MockAuthenticate) Close() error { return a.CloseError }

// MockAuthorize provides a mocked implementation of the authorizer interface.
type MockAuthorize struct {
	AuthorizeResponse bool
	AuthorizeError    error
	IsAdminResponse   bool
	IsAdminError      error
	CloseError        error
}

// Close is a mocked authorizer client function.
func (a MockAuthorize) Close() error { return a.CloseError }

// Authorize is a mocked authorizer client function.
func (a MockAuthorize) Authorize(ctx context.Context, route string, s *sessions.SessionState) (bool, error) {
	return a.AuthorizeResponse, a.AuthorizeError
}

// IsAdmin is a mocked IsAdmin function.
func (a MockAuthorize) IsAdmin(ctx context.Context, s *sessions.SessionState) (bool, error) {
	return a.IsAdminResponse, a.IsAdminError
}