linkerd2/controller/destination/test_helper.go

33 lines
851 B
Go

package destination
import (
"context"
common "github.com/runconduit/conduit/controller/gen/common"
)
// implements the updateListener interface
type collectUpdateListener struct {
added []common.TcpAddress
removed []common.TcpAddress
context context.Context
}
func (c *collectUpdateListener) Update(add []common.TcpAddress, remove []common.TcpAddress) {
c.added = append(c.added, add...)
c.removed = append(c.removed, remove...)
}
func (c *collectUpdateListener) Done() <-chan struct{} {
return c.context.Done()
}
func (c *collectUpdateListener) NoEndpoints(exists bool) {}
func (c *collectUpdateListener) SetServiceId(id *serviceId) {}
func newCollectUpdateListener() (*collectUpdateListener, context.CancelFunc) {
ctx, cancelFn := context.WithCancel(context.Background())
return &collectUpdateListener{context: ctx}, cancelFn
}