Files
query-orchestration/test/queryrunner_test.go
T
Michael McGuinness cbd1d5f067 Merged in feature/queuetest (pull request #14)
Add tests for queuing

* startingtest

* queuerunnertest

* addedtestfuncs

* testtests

* passintegration

* unittests
2025-01-10 19:17:20 +00:00

39 lines
853 B
Go

package integration_test
import (
"context"
"encoding/json"
"queryorchestration/internal/document"
"queryorchestration/internal/queue"
"queryorchestration/internal/test"
"testing"
"github.com/aws/aws-sdk-go-v2/service/sqs/types"
"github.com/google/uuid"
"github.com/stretchr/testify/assert"
)
func TestQueryRunner(t *testing.T) {
ctx := context.Background()
qCfg, cleanup := test.CreateQueueWithDependencies(t, ctx, "queryRunner")
defer cleanup()
document := document.Document{
ID: uuid.New(),
JobID: uuid.New(),
Name: "documentname",
CleanVersion: int32(1),
TextVersion: int32(1),
}
docJson, err := json.Marshal(document)
assert.Nil(t, err)
cfg := &queue.Config{
URL: qCfg.URL,
Client: qCfg.Client,
}
queue.Send(ctx, cfg, string(docJson), map[string]types.MessageAttributeValue{})
}