test_helper.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
}