Michael McGuinness
92334ad1dd
Set Up S3 integration * cfginterfaceandfirsts3funcs * addlocalstack * generallypassesfullsuite * addedmultipleattemptedpings * cleanup * stabiliseplusskip
58 lines
1.6 KiB
Go
58 lines
1.6 KiB
Go
package test
|
|
|
|
import (
|
|
"context"
|
|
"queryorchestration/internal/serviceconfig"
|
|
"queryorchestration/internal/serviceconfig/queue"
|
|
"testing"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/aws"
|
|
"github.com/aws/aws-sdk-go-v2/service/sqs"
|
|
"github.com/aws/aws-sdk-go-v2/service/sqs/types"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func CreateQueue(t *testing.T, ctx context.Context, cfg serviceconfig.ConfigProvider, name string) string {
|
|
queueM, err := cfg.GetQueueClient().CreateQueue(ctx, &sqs.CreateQueueInput{
|
|
QueueName: aws.String(name),
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = cfg.PingQueueByURL(ctx, *queueM.QueueUrl)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
return *queueM.QueueUrl
|
|
}
|
|
|
|
func AssertMessage(t *testing.T, ctx context.Context, cfg serviceconfig.ConfigProvider, params *queue.ReceiveParams) types.Message {
|
|
result, err := cfg.ReceiveFromQueue(ctx, params)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, result.Messages)
|
|
assert.Len(t, result.Messages, 1)
|
|
assert.NotNil(t, result.Messages[0])
|
|
|
|
return result.Messages[0]
|
|
}
|
|
|
|
func AssertMessageBody(t *testing.T, ctx context.Context, cfg serviceconfig.ConfigProvider, url string, body string) {
|
|
message := AssertMessage(t, ctx, cfg, &queue.ReceiveParams{
|
|
QueueURL: url,
|
|
})
|
|
|
|
assert.Equal(t, body, *message.Body)
|
|
}
|
|
|
|
func AssertMessageAttr(t *testing.T, ctx context.Context, cfg serviceconfig.ConfigProvider, url string, name string, value string) {
|
|
message := AssertMessage(t, ctx, cfg, &queue.ReceiveParams{
|
|
QueueURL: url,
|
|
Attributes: []string{name},
|
|
})
|
|
|
|
assert.NotNil(t, message.MessageAttributes[name])
|
|
assert.Equal(t, value, *(message.MessageAttributes[name]).StringValue)
|
|
}
|