test

catalog/consumer_test.go

@@ -1,10 +1,13 @@
 package catalog
 
 import (
+	"encoding/json"
 	"strconv"
 	"testing"
 
+	"git.k6n.net/mats/platform/event"
 	"git.k6n.net/mats/platform/money"
+	amqp "github.com/rabbitmq/amqp091-go"
 )
 
 func snapMeta(phase string, epoch int64, seq, count int) map[string]string {
@@ -223,3 +226,53 @@ func TestProjectionStore_BadPayloadErrors(t *testing.T) {
 		t.Fatal("delta with bad payload should error")
 	}
 }
+
+func TestMakeAmqpHandler(t *testing.T) {
+	s := newStore(t, nil)
+	handler := MakeAmqpHandler(s)
+
+	// Build a mock AMQP delivery containing an event envelope.
+	payload, err := EncodeUpdates([]ProjectionUpdate{{Projection: proj("A", 123_45)}})
+	if err != nil {
+		t.Fatal(err)
+	}
+	ev := event.Event{
+		Type:    event.CatalogProjectionPublished,
+		Payload: payload,
+		Meta:    snapMeta(SnapshotBegin, 1, 0, 1),
+	}
+	envelope, err := json.Marshal(ev)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	d := amqp.Delivery{
+		Body: envelope,
+	}
+
+	// 1. Process Begin frame
+	if err := handler(d); err != nil {
+		t.Fatalf("handler failed on Begin: %v", err)
+	}
+
+	// 2. Process Chunk frame
+	ev.Meta = snapMeta(SnapshotChunk, 1, 0, 0)
+	envelope, _ = json.Marshal(ev)
+	if err := handler(amqp.Delivery{Body: envelope}); err != nil {
+		t.Fatalf("handler failed on Chunk: %v", err)
+	}
+
+	// 3. Process End frame
+	ev.Meta = snapMeta(SnapshotEnd, 1, 1, 1)
+	ev.Payload = nil
+	envelope, _ = json.Marshal(ev)
+	if err := handler(amqp.Delivery{Body: envelope}); err != nil {
+		t.Fatalf("handler failed on End: %v", err)
+	}
+
+	// Validate results in store
+	p, ok := s.Get("A")
+	if !ok || p.PriceIncVat != 123_45 {
+		t.Fatalf("A = %+v ok=%v, want price 12345", p, ok)
+	}
+}