31 lines
862 B
Go
31 lines
862 B
Go
|
/*
|
||
|
|
Copyright 2020 VMware, Inc.
|
||
|
|
SPDX-License-Identifier: Apache-2.0
|
||
|
|
*/
|
||
|
|
|
||
|
|
package testutil
|
||
|
|
|
||
|
|
import "github.com/suzerain-io/controller-go"
|
||
|
|
|
||
|
|
type ObservableWithInformerOption struct {
|
||
|
|
InformerToFilterMap map[controller.InformerGetter]controller.Filter
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewObservableWithInformerOption() *ObservableWithInformerOption {
|
||
|
|
return &ObservableWithInformerOption{
|
||
|
|
InformerToFilterMap: make(map[controller.InformerGetter]controller.Filter),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (i *ObservableWithInformerOption) WithInformer(
|
||
|
|
getter controller.InformerGetter,
|
||
|
|
filter controller.Filter,
|
||
|
|
opt controller.InformerOption) controller.Option {
|
||
|
|
i.InformerToFilterMap[getter] = filter
|
||
|
|
return controller.WithInformer(getter, filter, opt)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (i *ObservableWithInformerOption) GetFilterForInformer(getter controller.InformerGetter) controller.Filter {
|
||
|
|
return i.InformerToFilterMap[getter]
|
||
|
|
}
|