authorizerd_mock_test.go

// Copyright 2023 LY Corporation
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package authorizerd

import (
	"context"
	"crypto/x509"
	"time"

	"github.com/AthenZ/athenz-authorizer/v5/access"
	"github.com/AthenZ/athenz-authorizer/v5/jwk"
	"github.com/AthenZ/athenz-authorizer/v5/policy"
	"github.com/AthenZ/athenz-authorizer/v5/pubkey"
	"github.com/AthenZ/athenz-authorizer/v5/role"
	"github.com/pkg/errors"
)

type ConfdMock struct {
	pubkey.Daemon
	confdExp time.Duration
}

func (cm *ConfdMock) Start(ctx context.Context) <-chan error {
	ech := make(chan error, 1)
	go func() {
		time.Sleep(cm.confdExp)
		ech <- errors.New("pubkey error")
	}()
	return ech
}

type PubkeydMock struct {
	StartFunc       func(context.Context) <-chan error
	UpdateFunc      func(context.Context) error
	GetProviderFunc func() pubkey.Provider
}

func (pm *PubkeydMock) Start(ctx context.Context) <-chan error {
	if pm.StartFunc != nil {
		return pm.StartFunc(ctx)
	}
	return nil
}

func (pm *PubkeydMock) Update(ctx context.Context) error {
	if pm.UpdateFunc != nil {
		return pm.UpdateFunc(ctx)
	}
	return nil
}

func (pm *PubkeydMock) GetProvider() pubkey.Provider {
	if pm.GetProviderFunc != nil {
		return pm.GetProviderFunc()
	}
	return nil
}

type PolicydMock struct {
	UpdateFunc                func(context.Context) error
	CheckPolicyRoleFunc       func(ctx context.Context, domain string, roles []string, action, resource string) ([]string, error)
	GetPrincipalCacheLenFunc  func() int
	GetPrincipalCacheSizeFunc func() int64

	policydExp  time.Duration
	policyCache map[string][]*policy.Assertion

	principalCacheLen  int
	principalCacheSize int64
}

func (pdm *PolicydMock) Start(context.Context) <-chan error {
	ech := make(chan error, 1)
	go func() {
		time.Sleep(pdm.policydExp)
		ech <- errors.New("policyd error")
	}()
	return ech
}

func (pdm *PolicydMock) Update(ctx context.Context) error {
	if pdm.UpdateFunc != nil {
		return pdm.UpdateFunc(ctx)
	}
	return nil
}

func (pdm *PolicydMock) CheckPolicy(ctx context.Context, domain string, roles []string, action, resource string) error {
	_, err := pdm.CheckPolicyRoles(ctx, domain, roles, action, resource)
	return err
}

func (pdm *PolicydMock) CheckPolicyRoles(ctx context.Context, domain string, roles []string, action, resource string) ([]string, error) {
	if pdm.CheckPolicyRoleFunc != nil {
		return pdm.CheckPolicyRoleFunc(ctx, domain, roles, action, resource)
	}
	return nil, nil
}

func (pdm *PolicydMock) GetPolicyCache(ctx context.Context) map[string][]*policy.Assertion {
	return pdm.policyCache
}

func (pdm *PolicydMock) GetPrincipalCacheLen() int {
	return pdm.principalCacheLen
}

func (pdm *PolicydMock) GetPrincipalCacheSize() int64 {
	return pdm.principalCacheSize
}

type RoleProcessorMock struct {
	role.Processor
	wantErr error
	rt      *role.Token
}

func (rpm *RoleProcessorMock) ParseAndValidateRoleToken(tok string) (*role.Token, error) {
	return rpm.rt, rpm.wantErr
}

type AccessProcessorMock struct {
	access.Processor
	wantErr error
	atc     *access.OAuth2AccessTokenClaim
}

func (apm *AccessProcessorMock) ParseAndValidateOAuth2AccessToken(cred string, cert *x509.Certificate) (*access.OAuth2AccessTokenClaim, error) {
	return apm.atc, apm.wantErr
}

type JwkdMock struct {
	StartFunc       func(context.Context) <-chan error
	UpdateFunc      func(context.Context) error
	GetProviderFunc func() jwk.Provider
}

func (jm *JwkdMock) Start(ctx context.Context) <-chan error {
	if jm.StartFunc != nil {
		return jm.StartFunc(ctx)
	}
	return nil
}

func (jm *JwkdMock) Update(ctx context.Context) error {
	if jm.UpdateFunc != nil {
		return jm.UpdateFunc(ctx)
	}
	return nil
}

func (jm *JwkdMock) GetProvider() jwk.Provider {
	if jm.GetProviderFunc != nil {
		return jm.GetProviderFunc()
	}
	return nil
}