dynamomq_test.go

package dynamomq_test

import (
	"time"

	"github.com/vvatanabe/dynamomq"
	"github.com/vvatanabe/dynamomq/internal/clock"
	"github.com/vvatanabe/dynamomq/internal/constant"
	"github.com/vvatanabe/dynamomq/internal/test"
)

func MarkAsProcessing[T any](m *dynamomq.Message[T], now time.Time) {
	ts := clock.FormatRFC3339Nano(now)
	m.UpdatedAt = ts
	m.ReceivedAt = ts
	m.InvisibleUntilAt = clock.FormatRFC3339Nano(now.Add(constant.DefaultVisibilityTimeout))
}

func MarkAsMovedToDLQ[T any](m *dynamomq.Message[T], now time.Time) {
	ts := clock.FormatRFC3339Nano(now)
	m.QueueType = dynamomq.QueueTypeDLQ
	m.ReceiveCount = 0
	m.UpdatedAt = ts
	m.SentAt = ts
	m.ReceivedAt = ""
	m.InvisibleUntilAt = ""
}

func MarkAsRestoredFromDLQ[T any](m *dynamomq.Message[T], now time.Time) {
	ts := clock.FormatRFC3339Nano(now)
	m.QueueType = dynamomq.QueueTypeStandard
	m.ReceiveCount = 0
	m.UpdatedAt = ts
	m.SentAt = ts
	m.ReceivedAt = ""
	m.InvisibleUntilAt = ""
}

func NewTestMessageItemAsReady(id string, now time.Time) *dynamomq.Message[test.MessageData] {
	return dynamomq.NewMessage[test.MessageData](id, test.NewMessageData(id), now)
}

func NewTestMessageItemAsProcessing(id string, now time.Time) *dynamomq.Message[test.MessageData] {
	m := dynamomq.NewMessage[test.MessageData](id, test.NewMessageData(id), now)
	MarkAsProcessing(m, now)
	return m
}

func NewTestMessageItemAsDLQ(id string, now time.Time) *dynamomq.Message[test.MessageData] {
	m := dynamomq.NewMessage[test.MessageData](id, test.NewMessageData(id), now)
	MarkAsMovedToDLQ(m, now)
	return m
}

func NewMessageFromReadyToProcessing(id string,
	readyTime time.Time, processingTime time.Time) *dynamomq.ReceiveMessageOutput[test.MessageData] {
	m := NewTestMessageItemAsReady(id, readyTime)
	MarkAsProcessing(m, processingTime)
	m.Version = 2
	m.ReceiveCount = 1
	m.InvisibleUntilAt = clock.FormatRFC3339Nano(processingTime.Add(constant.DefaultVisibilityTimeout))
	r := &dynamomq.ReceiveMessageOutput[test.MessageData]{
		ReceivedMessage: m,
	}
	return r
}