74 lines
2.8 KiB
Go
74 lines
2.8 KiB
Go
package multitenant
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
"go.uber.org/zap"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
|
|
fv1 "github.com/fission/fission/pkg/apis/core/v1"
|
|
"github.com/fission/fission/pkg/executor/executortype"
|
|
"github.com/fission/fission/pkg/executor/fscache"
|
|
"github.com/fission/fission/pkg/utils/manager"
|
|
)
|
|
|
|
type fakeExecutorType struct {
|
|
addCalls int
|
|
lastNamespace string
|
|
addErr error
|
|
}
|
|
|
|
func (f *fakeExecutorType) Run(context.Context, manager.Interface) {}
|
|
func (f *fakeExecutorType) GetTypeName(context.Context) fv1.ExecutorType { return fv1.ExecutorTypePoolmgr }
|
|
func (f *fakeExecutorType) GetFuncSvc(context.Context, *fv1.Function) (*fscache.FuncSvc, error) {
|
|
return nil, nil
|
|
}
|
|
func (f *fakeExecutorType) GetFuncSvcFromCache(context.Context, *fv1.Function) (*fscache.FuncSvc, error) {
|
|
return nil, nil
|
|
}
|
|
func (f *fakeExecutorType) DumpDebugInfo(context.Context) error { return nil }
|
|
func (f *fakeExecutorType) DeleteFuncSvcFromCache(context.Context, *fscache.FuncSvc) {}
|
|
func (f *fakeExecutorType) TapService(context.Context, string) error { return nil }
|
|
func (f *fakeExecutorType) UnTapService(context.Context, *metav1.ObjectMeta, string) {}
|
|
func (f *fakeExecutorType) MarkSpecializationFailure(context.Context, *metav1.ObjectMeta) {}
|
|
func (f *fakeExecutorType) IsValid(context.Context, *fscache.FuncSvc) bool { return true }
|
|
func (f *fakeExecutorType) RefreshFuncPods(context.Context, *zap.Logger, fv1.Function) error { return nil }
|
|
func (f *fakeExecutorType) AdoptExistingResources(context.Context) {}
|
|
func (f *fakeExecutorType) CleanupOldExecutorObjects(context.Context) {}
|
|
func (f *fakeExecutorType) AddNamespace(ctx context.Context, ns string, mgr manager.Interface) error {
|
|
f.addCalls++
|
|
f.lastNamespace = ns
|
|
return f.addErr
|
|
}
|
|
|
|
var _ executortype.ExecutorType = (*fakeExecutorType)(nil)
|
|
|
|
func TestRegisterExecutorTypes(t *testing.T) {
|
|
poolmgr := &fakeExecutorType{}
|
|
container := &fakeExecutorType{}
|
|
err := registerExecutorTypes(context.Background(), zap.NewNop(), "tenant-executor-a", map[fv1.ExecutorType]executortype.ExecutorType{
|
|
fv1.ExecutorTypePoolmgr: poolmgr,
|
|
fv1.ExecutorTypeContainer: container,
|
|
}, nil)
|
|
if err != nil {
|
|
t.Fatalf("expected registration success: %v", err)
|
|
}
|
|
if poolmgr.addCalls != 1 || container.addCalls != 1 {
|
|
t.Fatalf("expected all executor types to receive AddNamespace")
|
|
}
|
|
}
|
|
|
|
func TestRegisterExecutorTypesAggregatesErrors(t *testing.T) {
|
|
err := registerExecutorTypes(context.Background(), zap.NewNop(), "tenant-executor-a", map[fv1.ExecutorType]executortype.ExecutorType{
|
|
fv1.ExecutorTypePoolmgr: &fakeExecutorType{addErr: errors.New("poolmgr failed")},
|
|
fv1.ExecutorTypeContainer: &fakeExecutorType{},
|
|
}, nil)
|
|
if err == nil {
|
|
t.Fatalf("expected aggregated error")
|
|
}
|
|
if err.Error() != "poolmgr failed" {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
} |