/* Copyright 2018 The Knative Authors 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 duck import ( "context" "errors" "testing" "time" "k8s.io/apimachinery/pkg/runtime/schema" "k8s.io/client-go/tools/cache" ) func TestEnqueueInformerFactory(t *testing.T) { called := false want := cache.ResourceEventHandlerFuncs{ AddFunc: func(obj interface{}) { called = true }, } fsii := &fakeSharedIndexInformer{t: t} fif := &FixedInformerFactory{inf: fsii} eif := &EnqueueInformerFactory{ Delegate: fif, EventHandler: want, } gvr := schema.GroupVersionResource{ Group: "testing.knative.dev", Version: "v3", Resource: "caches", } inf, _, err := eif.Get(context.Background(), gvr) if err != nil { t.Fatal("Get() =", err) } if inf != fsii { t.Fatalf("Get() = %v, wanted %v", inf, fsii) } got, ok := fsii.eventHandler.(cache.ResourceEventHandlerFuncs) if !ok { t.Errorf("eventHandler = %T, wanted %T", fsii.eventHandler, want) } if called { t.Error("Want not called, got called") } got.AddFunc(nil) if !called { t.Error("Want called, got not called") } if got.UpdateFunc != nil { t.Error("UpdateFunc = non-nil, wanted nil") } if got.DeleteFunc != nil { t.Error("DeleteFunc = non-nil, wanted nil") } } func TestEnqueueInformerFactoryWithFailure(t *testing.T) { want := errors.New("expected error") fif := &FixedInformerFactory{err: want} eif := &EnqueueInformerFactory{ Delegate: fif, EventHandler: cache.ResourceEventHandlerFuncs{ AddFunc: func(obj interface{}) { t.Error("Unexpected call to AddFunc.") }, UpdateFunc: func(old, new interface{}) { t.Error("Unexpected call to UpdateFunc.") }, }, } gvr := schema.GroupVersionResource{ Group: "testing.knative.dev", Version: "v3", Resource: "caches", } inf, _, got := eif.Get(context.Background(), gvr) if !errors.Is(got, want) { t.Fatalf("Get() = %v, wanted %v", got, want) } if inf != nil { t.Fatal("Get() = non nil, wanted nil") } } type FixedInformerFactory struct { inf cache.SharedIndexInformer lister cache.GenericLister err error } var _ InformerFactory = (*FixedInformerFactory)(nil) func (fif *FixedInformerFactory) Get(ctx context.Context, gvr schema.GroupVersionResource) (cache.SharedIndexInformer, cache.GenericLister, error) { return fif.inf, fif.lister, fif.err } type fakeSharedIndexInformer struct { t *testing.T eventHandler cache.ResourceEventHandler } var _ cache.SharedIndexInformer = (*fakeSharedIndexInformer)(nil) func (fsii *fakeSharedIndexInformer) AddEventHandler(handler cache.ResourceEventHandler) { fsii.eventHandler = handler } func (fsii *fakeSharedIndexInformer) AddEventHandlerWithResyncPeriod(handler cache.ResourceEventHandler, resyncPeriod time.Duration) { fsii.t.Fatalf("NYI: AddEventHandlerWithResyncPeriod") } func (fsii *fakeSharedIndexInformer) GetStore() cache.Store { fsii.t.Fatalf("NYI: GetStore") return nil } func (fsii *fakeSharedIndexInformer) GetController() cache.Controller { fsii.t.Fatalf("NYI: GetController") return nil } func (fsii *fakeSharedIndexInformer) Run(stopCh <-chan struct{}) { fsii.t.Fatalf("NYI: Run") } func (fsii *fakeSharedIndexInformer) HasSynced() bool { fsii.t.Fatalf("NYI: HadSynced") return false } func (fsii *fakeSharedIndexInformer) LastSyncResourceVersion() string { fsii.t.Fatalf("NYI: LastSyncResourceVersion") return "" } func (fsii *fakeSharedIndexInformer) AddIndexers(indexers cache.Indexers) error { fsii.t.Fatalf("NYI: AddIndexers") return nil } func (fsii *fakeSharedIndexInformer) GetIndexer() cache.Indexer { fsii.t.Fatalf("NYI: GetIndexer") return nil }