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

package queue_test

import (
	"testing"

	"queryorchestration/internal/serviceconfig/queue"
	"queryorchestration/internal/test"

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

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

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

	err := cfg.SendToQueue(t.Context(), &queue.SendParams{
		QueueURL: cfg.GetClientSyncURL(),
		Body:     struct{}{},
	})
	require.NoError(t, err)

	msg := test.AssertMessage(t, cfg, &queue.ReceiveParams{
		QueueURL: cfg.GetClientSyncURL(),
	})

	params := &queue.DeleteParams{
		ReceiptHandle: msg.ReceiptHandle,
		QueueURL:      cfg.GetClientSyncURL(),
	}

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

	ares, err := cfg.ReceiveFromQueue(t.Context(), &queue.ReceiveParams{
		QueueURL: cfg.GetClientSyncURL(),
	})
	require.NoError(t, err)
	require.Len(t, ares.Messages, 0)
}