Michael McGuinness
7001ca854c
Queuing Changes and Cfg Testing * staarting * staarting * startedpush * note * save * mocking * removederrs * fixtests * cleanuperrs * newenvsetup * preppingtests * queue * mmovetocfgpassunittests * sortoutconfig * passinginteg * deps * fixtests
145 lines
2.9 KiB
Go
145 lines
2.9 KiB
Go
package test
|
|
|
|
import (
|
|
"context"
|
|
"queryorchestration/internal/server/queue"
|
|
"testing"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/aws"
|
|
"github.com/aws/aws-sdk-go-v2/service/sqs/types"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestCreateQueueContainer(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("Skipping long test in short mode")
|
|
}
|
|
ctx := context.Background()
|
|
cfg := CreateBaseConfig()
|
|
|
|
qcfg, cleanup := createQueueContainer(t, ctx, &CreateQueueConfig{
|
|
Cfg: cfg,
|
|
})
|
|
assert.NotNil(t, qcfg)
|
|
assert.NotNil(t, cleanup)
|
|
|
|
cleanup()
|
|
}
|
|
|
|
func TestCreateQueueClient(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("Skipping long test in short mode")
|
|
}
|
|
ctx := context.Background()
|
|
cfg := CreateBaseConfig()
|
|
|
|
cleanup := CreateQueueClient(t, ctx, &CreateQueueConfig{
|
|
Cfg: cfg,
|
|
})
|
|
assert.NotNil(t, cfg.QueueClient)
|
|
assert.NotNil(t, cleanup)
|
|
|
|
cleanup()
|
|
}
|
|
|
|
func TestCreateQueue(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("Skipping long test in short mode")
|
|
}
|
|
ctx := context.Background()
|
|
|
|
cfg := CreateBaseConfig()
|
|
|
|
cleanup := CreateQueueClient(t, ctx, &CreateQueueConfig{
|
|
Cfg: cfg,
|
|
})
|
|
defer cleanup()
|
|
|
|
url := CreateQueue(t, ctx, cfg, "myname")
|
|
assert.Equal(t, "http://localstack:4566/queue/us-east-1/000000000000/myname", url)
|
|
}
|
|
|
|
func TestAssertMessageWait(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("Skipping long test in short mode")
|
|
}
|
|
ctx := context.Background()
|
|
|
|
cfg := CreateBaseConfig()
|
|
|
|
cleanup := CreateQueueClient(t, ctx, &CreateQueueConfig{
|
|
Cfg: cfg,
|
|
})
|
|
defer cleanup()
|
|
url := CreateQueue(t, ctx, cfg, "myname")
|
|
|
|
qucfg := &queue.Config{
|
|
URL: url,
|
|
Client: cfg.QueueClient,
|
|
}
|
|
|
|
err := queue.Send(ctx, qucfg, "body", map[string]types.MessageAttributeValue{})
|
|
assert.NoError(t, err)
|
|
|
|
msg := AssertMessageWait(t, ctx, qucfg, []string{})
|
|
assert.NotNil(t, msg)
|
|
}
|
|
|
|
func TestAssertMessageBodyWait(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("Skipping long test in short mode")
|
|
}
|
|
ctx := context.Background()
|
|
|
|
cfg := CreateBaseConfig()
|
|
|
|
cleanup := CreateQueueClient(t, ctx, &CreateQueueConfig{
|
|
Cfg: cfg,
|
|
})
|
|
defer cleanup()
|
|
url := CreateQueue(t, ctx, cfg, "myname")
|
|
|
|
qucfg := &queue.Config{
|
|
URL: url,
|
|
Client: cfg.QueueClient,
|
|
}
|
|
|
|
err := queue.Send(ctx, qucfg, "body", map[string]types.MessageAttributeValue{})
|
|
assert.NoError(t, err)
|
|
|
|
AssertMessageBodyWait(t, ctx, qucfg, "\"body\"")
|
|
}
|
|
|
|
func TestAssertMessageAttrWait(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("Skipping long test in short mode")
|
|
}
|
|
ctx := context.Background()
|
|
|
|
cfg := CreateBaseConfig()
|
|
|
|
cleanup := CreateQueueClient(t, ctx, &CreateQueueConfig{
|
|
Cfg: cfg,
|
|
})
|
|
defer cleanup()
|
|
url := CreateQueue(t, ctx, cfg, "myname")
|
|
|
|
qucfg := &queue.Config{
|
|
URL: url,
|
|
Client: cfg.QueueClient,
|
|
}
|
|
|
|
name := "name"
|
|
value := "value"
|
|
|
|
err := queue.Send(ctx, qucfg, "body", map[string]types.MessageAttributeValue{
|
|
name: {
|
|
DataType: aws.String("String"),
|
|
StringValue: aws.String(value),
|
|
},
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
AssertMessageAttrWait(t, ctx, qucfg, name, value)
|
|
}
|