/
handler.go
38 lines (29 loc) · 938 Bytes
/
handler.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
package eventtest
import (
"context"
"github.com/ONSdigital/dp-observation-importer/event"
"github.com/ONSdigital/log.go/v2/log"
)
var _ event.Handler = (*EventHandler)(nil)
// NewEventHandler returns a new mock event handler to capture event
func NewEventHandler() *EventHandler {
events := make([]*event.ObservationExtracted, 0)
eventUpdated := make(chan bool)
return &EventHandler{
Events: events,
EventUpdated: eventUpdated,
}
}
// EventHandler provides a mock implementation that captures events to check.
type EventHandler struct {
Events []*event.ObservationExtracted
Error error
EventUpdated chan bool
}
// Handle captures the given event and stores it for later assertions
func (handler *EventHandler) Handle(ctx context.Context, events []*event.ObservationExtracted) error {
log.Info(ctx, "handle called")
handler.Events = events
handler.EventUpdated <- true
return handler.Error
}