diff --git a/internal/daemon/common/handler_test.go b/internal/daemon/common/handler_test.go index f2bfede3cc..0c617e0886 100644 --- a/internal/daemon/common/handler_test.go +++ b/internal/daemon/common/handler_test.go @@ -487,10 +487,15 @@ func (b *testMockBroker) RegisterPipeline(def eventlogger.Pipeline, opt ...event return nil } func (b *testMockBroker) StopTimeAt(t time.Time) {} + func (b *testMockBroker) RegisterNode(id eventlogger.NodeID, node eventlogger.Node, opt ...eventlogger.Option) error { return nil } +func (b *testMockBroker) RemoveNode(ctx context.Context, id eventlogger.NodeID) error { + return nil +} + func (b *testMockBroker) RemovePipelineAndNodes(ctx context.Context, t eventlogger.EventType, id eventlogger.PipelineID) (bool, error) { return true, nil } diff --git a/internal/event/eventer.go b/internal/event/eventer.go index 224377b2fe..683816c405 100644 --- a/internal/event/eventer.go +++ b/internal/event/eventer.go @@ -59,6 +59,7 @@ type broker interface { Reopen(ctx context.Context) error StopTimeAt(now time.Time) RegisterNode(id eventlogger.NodeID, node eventlogger.Node, opt ...eventlogger.Option) error + RemoveNode(ctx context.Context, id eventlogger.NodeID) error SetSuccessThreshold(t eventlogger.EventType, successThreshold int) error RegisterPipeline(def eventlogger.Pipeline, opt ...eventlogger.Option) error RemovePipelineAndNodes(ctx context.Context, t eventlogger.EventType, id eventlogger.PipelineID) (bool, error) diff --git a/internal/event/testing.go b/internal/event/testing.go index f1ddabcb40..bf4158a06b 100644 --- a/internal/event/testing.go +++ b/internal/event/testing.go @@ -361,6 +361,10 @@ func (b *testMockBroker) RegisterNode(id eventlogger.NodeID, node eventlogger.No return nil } +func (b *testMockBroker) RemoveNode(ctx context.Context, id eventlogger.NodeID) error { + return nil +} + func (b *testMockBroker) SetSuccessThreshold(t eventlogger.EventType, successThreshold int) error { if b.successThresholds == nil { b.successThresholds = map[eventlogger.EventType]int{}