snapshot of https://github.com/ThreeDotsLabs/watermill.git (master, 2026-09-15; blobs > 131072B omitted)

message/router/middleware/instant_ack_test.go

@@ -0,0 +1,33 @@
+package middleware
+
+import (
+	"testing"
+
+	"github.com/ThreeDotsLabs/watermill/message"
+	"github.com/pkg/errors"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestInstantAck(t *testing.T) {
+	producedMessages := message.Messages{message.NewMessage("2", nil)}
+	producedErr := errors.New("foo")
+
+	h := InstantAck(func(msg *message.Message) (messages []*message.Message, e error) {
+		return producedMessages, producedErr
+	})
+
+	msg := message.NewMessage("1", nil)
+
+	handlerMessages, handlerErr := h(msg)
+	assert.EqualValues(t, producedMessages, handlerMessages)
+	assert.Equal(t, producedErr, handlerErr)
+
+	select {
+	case <-msg.Acked():
+	// ok
+	case <-msg.Nacked():
+		t.Fatal("expected ack, not nack")
+	default:
+		t.Fatal("no ack received")
+	}
+}