jay brown
127 lines
2.7 KiB
Go
127 lines
2.7 KiB
Go
package test
|
|
|
|
import (
|
|
"regexp"
|
|
"testing"
|
|
|
|
"queryorchestration/internal/serviceconfig"
|
|
"queryorchestration/internal/serviceconfig/objectstore"
|
|
"queryorchestration/internal/serviceconfig/queue"
|
|
|
|
awsc "queryorchestration/internal/serviceconfig/aws"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/aws"
|
|
"github.com/aws/aws-sdk-go-v2/service/sqs/types"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type TestConfig struct {
|
|
serviceconfig.BaseConfig
|
|
objectstore.ObjectStoreConfig
|
|
}
|
|
|
|
func TestCreateQueue(t *testing.T) {
|
|
if testing.Short() {
|
|
t.SkipNow()
|
|
}
|
|
ctx := t.Context()
|
|
|
|
cfg := &TestConfig{}
|
|
|
|
a := CreateAWSContainer(t, cfg)
|
|
SetQueueClient(t, ctx, cfg, a.ExternalEndpoint)
|
|
|
|
url := CreateQueue(t, cfg, "myname")
|
|
assert.Equal(t, "http://localstack:4566/queue/us-east-1/000000000000/mynametestcreatequeue", url)
|
|
}
|
|
|
|
func TestAssertMessageWait(t *testing.T) {
|
|
if testing.Short() {
|
|
t.SkipNow()
|
|
}
|
|
ctx := t.Context()
|
|
|
|
cfg := &TestConfig{}
|
|
|
|
a := CreateAWSContainer(t, cfg)
|
|
SetQueueClient(t, ctx, cfg, a.ExternalEndpoint)
|
|
url := CreateQueue(t, cfg, "myname")
|
|
|
|
err := cfg.SendToQueue(ctx, &queue.SendParams{
|
|
QueueURL: url,
|
|
Body: "body",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
msg := AssertMessage(t, cfg, &queue.ReceiveParams{
|
|
QueueURL: url,
|
|
})
|
|
assert.NotNil(t, msg)
|
|
}
|
|
|
|
func TestAssertMessageBodyWait(t *testing.T) {
|
|
if testing.Short() {
|
|
t.SkipNow()
|
|
}
|
|
ctx := t.Context()
|
|
|
|
cfg := &TestConfig{}
|
|
|
|
a := CreateAWSContainer(t, cfg)
|
|
SetQueueClient(t, ctx, cfg, a.ExternalEndpoint)
|
|
url := CreateQueue(t, cfg, "myname")
|
|
|
|
err := cfg.SendToQueue(ctx, &queue.SendParams{
|
|
QueueURL: url,
|
|
Body: "body",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
AssertMessageBody(t, cfg, url, regexp.MustCompile("\"body\""))
|
|
}
|
|
|
|
func TestAssertMessageAttrWait(t *testing.T) {
|
|
if testing.Short() {
|
|
t.SkipNow()
|
|
}
|
|
ctx := t.Context()
|
|
|
|
cfg := &TestConfig{}
|
|
|
|
a := CreateAWSContainer(t, cfg)
|
|
SetQueueClient(t, ctx, cfg, a.ExternalEndpoint)
|
|
url := CreateQueue(t, cfg, "myname")
|
|
|
|
name := "name"
|
|
value := "value"
|
|
|
|
err := cfg.SendToQueue(ctx, &queue.SendParams{
|
|
QueueURL: url,
|
|
Body: "body",
|
|
Attributes: map[string]types.MessageAttributeValue{
|
|
name: {
|
|
DataType: aws.String("String"),
|
|
StringValue: aws.String(value),
|
|
},
|
|
},
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
AssertMessageAttr(t, cfg, url, name, regexp.MustCompile(value))
|
|
}
|
|
|
|
func TestGetQueueURL(t *testing.T) {
|
|
cfg := &awsc.AWSConfig{
|
|
AWSRegion: "us-east-1",
|
|
}
|
|
assert.Equal(t, "http://localstack:4566/queue/us-east-1/000000000000/docinitrunnertestgetqueueurl", GetQueueURL(t, cfg, DocInitRunnerName))
|
|
}
|
|
|
|
func TestGetQueueARN(t *testing.T) {
|
|
cfg := &awsc.AWSConfig{
|
|
AWSRegion: "us-east-1",
|
|
}
|
|
assert.Equal(t, "arn:aws:sqs:us-east-1:000000000000:docinitrunnertestgetqueuearn", GetQueueArn(t, cfg, DocInitRunnerName))
|
|
}
|