fake.go

package service

import (
	"github.com/luizalabs/teresa/pkg/server/app"
	"github.com/luizalabs/teresa/pkg/server/auth"
	"github.com/luizalabs/teresa/pkg/server/database"
	"github.com/luizalabs/teresa/pkg/server/spec"
)

type FakeOperations struct {
	EnableSSLErr             error
	SetStaticIpErr           error
	InfoErr                  error
	InfoValue                *Info
	WhitelistSourceRangesErr error
}

type FakeCloudProviderOperations struct {
	CreateOrUpdateSSLErr      error
	CreateOrUpdateStaticIpErr error
	SSLInfoErr                error
	SSLInfoValue              *SSLInfo
}

type FakeAppOperations struct {
	NegateHasPermission bool
	App                 *app.App
	CheckPermAndGetErr  error
}

type FakeK8sOperations struct {
	UpdateServicePortsErr          error
	IsNotFoundErr                  bool
	ServiceErr                     error
	ServiceValue                   *spec.Service
	SetLoadBalancerSourceRangesErr error
	IsInvalidErr                   bool
	HasIngressValue                bool
	HasIngressErr                  error
}

func (f *FakeOperations) EnableSSL(user *database.User, appName, cert string, only bool) error {
	return f.EnableSSLErr
}

func (f *FakeOperations) SetStaticIp(user *database.User, appName, addressName string) error {
	return f.SetStaticIpErr
}

func (f *FakeOperations) Info(user *database.User, appName string) (*Info, error) {
	return f.InfoValue, f.InfoErr
}

func (f *FakeOperations) WhitelistSourceRanges(user *database.User, appName string, sourceRanges []string) error {
	return f.WhitelistSourceRangesErr
}

func (f *FakeCloudProviderOperations) CreateOrUpdateSSL(appName, cert string, port int) error {
	return f.CreateOrUpdateSSLErr
}

func (f *FakeCloudProviderOperations) CreateOrUpdateStaticIp(appName, addressName string) error {
	return f.CreateOrUpdateSSLErr
}

func (f *FakeCloudProviderOperations) SSLInfo(appName string) (*SSLInfo, error) {
	return f.SSLInfoValue, f.SSLInfoErr
}

func (f *FakeAppOperations) HasPermission(user *database.User, appName string) bool {
	return !f.NegateHasPermission
}

func (f *FakeAppOperations) CheckPermAndGet(user *database.User, appName string) (*app.App, error) {
	err := f.CheckPermAndGetErr
	if f.NegateHasPermission {
		err = auth.ErrPermissionDenied
	}
	return f.App, err
}

func (f *FakeK8sOperations) UpdateServicePorts(namespace, svcName string, ports []spec.ServicePort) error {
	return f.UpdateServicePortsErr
}

func (f *FakeK8sOperations) IsNotFound(err error) bool {
	return f.IsNotFoundErr
}

func (f *FakeK8sOperations) Service(namespace, svcName string) (*spec.Service, error) {
	return f.ServiceValue, f.ServiceErr
}

func (f *FakeK8sOperations) SetLoadBalancerSourceRanges(namespace, svcName string, sourceRanges []string) error {
	return f.SetLoadBalancerSourceRangesErr
}

func (f *FakeK8sOperations) IsInvalid(err error) bool {
	return f.IsInvalidErr
}

func (f *FakeK8sOperations) HasIngress(namespace, name string) (bool, error) {
	return f.HasIngressValue, f.HasIngressErr
}