pomerium/internal/grpc/authorize/client/mock.go

package client

import (
	"context"

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

var _ Authorizer = &MockAuthorize{}

// 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.State) (bool, error) {
	return a.AuthorizeResponse, a.AuthorizeError
}

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