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/clientsync"
	"queryorchestration/internal/test"

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

// AWSConfig provides test configuration for queue operations.
// Note: Query-related configuration has been removed. See remove_query_plan.md.
type AWSConfig struct {
	serviceconfig.BaseConfig
	queue.QueueConfig
	objectstore.ObjectStoreConfig
	clientsync.ClientSyncConfig
}

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

	test.SetQueueClient(t, t.Context(), cfg, acfg.ExternalEndpoint)
	cfg.ClientSyncURL = test.CreateQueue(t, cfg, test.ClientSyncRunnerName)

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

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

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