weiguoqiang/cert-manager
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Add RegisterAdditionalSyncFuncs

Browse Source 
Signed-off-by: James Munnelly <james@munnelly.eu>
This commit is contained in:
James Munnelly 2019-08-21 13:42:26 +01:00
parent 1b8a286206
commit ea5c111bca
2 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pkg/controller/test/BUILD.bazel
View File

@ -26,6 +26,7 @@ go_library(
"//vendor/k8s.io/client-go/informers:go_default_library",
"//vendor/k8s.io/client-go/kubernetes/fake:go_default_library",
"//vendor/k8s.io/client-go/testing:go_default_library",
"//vendor/k8s.io/client-go/tools/cache:go_default_library",
"//vendor/k8s.io/utils/clock:go_default_library",
"//vendor/k8s.io/utils/clock/testing:go_default_library",
],

18
pkg/controller/test/context_builder.go
View File

@ -30,6 +30,7 @@ import (
kubeinformers "k8s.io/client-go/informers"
kubefake "k8s.io/client-go/kubernetes/fake"
coretesting "k8s.io/client-go/testing"
"k8s.io/client-go/tools/cache"
"k8s.io/utils/clock"
fakeclock "k8s.io/utils/clock/testing"
@ -72,8 +73,9 @@ type Builder struct {
// test).
CheckFn func(*Builder, ...interface{})
stopCh chan struct{}
requiredReactors map[string]bool
stopCh chan struct{}
requiredReactors map[string]bool
additionalSyncFuncs []cache.InformerSynced
*controller.Context
}
@ -285,6 +287,18 @@ func (b *Builder) Sync() {
if err := mustAllSync(b.SharedInformerFactory.WaitForCacheSync(b.stopCh)); err != nil {
panic("Error waiting for SharedInformerFactory to sync: " + err.Error())
}
if b.additionalSyncFuncs != nil {
cache.WaitForCacheSync(b.stopCh, b.additionalSyncFuncs...)
}
}
// RegisterAdditionalSyncFuncs registers an additional InformerSynced function
// with the builder.
// When the Sync method is called, the builder will also wait for the given
// listers to be synced as well as the listers that were registered with the
// informer factories that the builder provides.
func (b *Builder) RegisterAdditionalSyncFuncs(fns ...cache.InformerSynced) {
b.additionalSyncFuncs = append(b.additionalSyncFuncs, fns...)
}
func (b *Builder) Events() []string {