requeue_message_test.go

package requeue_test

import (
	"testing"

	"github.com/nickpoorman/nats-requeue/flatbuf"
	"github.com/nickpoorman/nats-requeue/protocol"
	"github.com/stretchr/testify/assert"
)

func TestRequeueMessage_FlatbufMutate(t *testing.T) {
	msg := protocol.DefaultRequeueMessage()
	msg.Retries = 5
	msg.TTL = 10000
	msg.Delay = 20000
	msg.BackoffStrategy = protocol.BackoffStrategy_Exponential
	msg.OriginalSubject = "foo.bar"
	msg.OriginalPayload = []byte("my awesome message")

	msgBytes := msg.Bytes()

	fb := flatbuf.GetRootAsRequeueMessage(msgBytes, 0)

	assert.Equal(t, fb.Retries(), uint64(5))
	assert.True(t, fb.MutateRetries(4))
	assert.Equal(t, fb.Retries(), uint64(4))

	// Assert that fb was modifying the underlying bytes.
	fb2 := flatbuf.GetRootAsRequeueMessage(msgBytes, 0)
	assert.Equal(t, fb2.Retries(), uint64(4))
}