query-orchestration/internal/serviceconfig/queue/send_test.go

package queue_test

import (
	"regexp"
	"testing"

	"queryorchestration/internal/serviceconfig"
	"queryorchestration/internal/serviceconfig/objectstore"
	"queryorchestration/internal/serviceconfig/queue"
	"queryorchestration/internal/serviceconfig/queue/query"
	"queryorchestration/internal/test"

	"github.com/stretchr/testify/require"
)

type AWSConfig struct {
	serviceconfig.BaseConfig
	queue.QueueConfig
	objectstore.ObjectStoreConfig
	query.QueryConfig
}

func TestSend(t *testing.T) {
	cfg := &AWSConfig{}
	acfg := test.CreateAWSContainer(t, cfg)

	test.SetQueueClient(t, t.Context(), cfg, acfg.ExternalEndpoint)
	cfg.QueryURL = test.CreateQueue(t, cfg, test.QueryRunnerName)

	params := &queue.SendParams{
		QueueURL: cfg.GetQueryURL(),
		Body:     struct{}{},
	}

	err := cfg.SendToQueue(t.Context(), params)
	require.NoError(t, err)

	test.AssertMessageBody(t, cfg, cfg.GetQueryURL(), regexp.MustCompile("{}"))
}