pipe_test.go

package ziggurat

import (
	"context"
	"testing"
	"time"
)

func TestPipeHandlers(t *testing.T) {
	mw1 := func(next Handler) Handler {
		return HandlerFunc(func(ctx context.Context, messageEvent *Event) {
			me := Event{
				Value:             []byte("foo"),
				Key:               nil,
				ProducerTimestamp: time.Time{},
				ReceivedTimestamp: time.Time{},
				EventType:         "",
			}

			next.Handle(ctx, &me)
		})
	}
	mw2 := func(next Handler) Handler {
		return HandlerFunc(func(ctx context.Context, messageEvent *Event) {
			byteValue := append(messageEvent.Value, []byte("-bar")...)
			me := Event{
				RoutingPath:       "",
				Value:             byteValue,
				Key:               nil,
				ProducerTimestamp: time.Time{},
				ReceivedTimestamp: time.Time{},
				EventType:         "",
			}

			next.Handle(ctx, &me)
		})
	}
	actualHandler := HandlerFunc(func(ctx context.Context, event *Event) {
		if string(event.Value) != "foo-bar" {
			t.Errorf("expected message to be %s,but got %s", "foo-bar", string(event.Value))
		}
	})
	finalHandler := pipe(actualHandler, mw1, mw2)
	finalHandler.Handle(context.Background(), &Event{})
}