Michael McGuinness
61d12bf8df
Checkbox Primary Edge Cases * hascheckboxlogic * gen * preppinggen * exploringcheckbox * removeunselected * tests * failingtest * failintests * nomockqueryapi * db * name * nocontainer * deleterow * cnc * extradocsandupdatetextracts * moretables * appndedtables * baseversion
97 lines
2.5 KiB
Go
97 lines
2.5 KiB
Go
package docsyncrunner_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
docsyncrunner "queryorchestration/api/docSyncRunner"
|
|
"queryorchestration/internal/client"
|
|
"queryorchestration/internal/database/repository"
|
|
"queryorchestration/internal/document"
|
|
documentsync "queryorchestration/internal/document/sync"
|
|
"queryorchestration/internal/server/runner"
|
|
"queryorchestration/internal/serviceconfig/objectstore"
|
|
"queryorchestration/internal/serviceconfig/queue/documentclean"
|
|
"queryorchestration/internal/test"
|
|
queuemock "queryorchestration/mocks/queue"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/service/sqs"
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/mock"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type DocSyncConfig struct {
|
|
runner.BaseConfig[docsyncrunner.Body]
|
|
documentclean.DocCleanConfig
|
|
}
|
|
|
|
func TestDocSyncRunner(t *testing.T) {
|
|
t.Parallel()
|
|
cfg := &DocSyncConfig{}
|
|
net := test.GetNetwork(t)
|
|
test.CreateDB(t, cfg, net, &test.CreateDatabaseConfig{})
|
|
ctx := context.Background()
|
|
|
|
mockSQS := queuemock.NewMockSQSClient(t)
|
|
cfg.QueueClient = mockSQS
|
|
|
|
runner := docsyncrunner.New(&docsyncrunner.Services{
|
|
Document: documentsync.New(cfg, &documentsync.Services{
|
|
Document: document.New(cfg),
|
|
Client: client.New(cfg),
|
|
}),
|
|
})
|
|
|
|
err := cfg.GetDBQueries().CreateClient(t.Context(), &repository.CreateClientParams{
|
|
Clientid: "clientid",
|
|
Name: "client_name",
|
|
})
|
|
require.NoError(t, err)
|
|
err = cfg.GetDBQueries().AddClientCanSync(t.Context(), &repository.AddClientCanSyncParams{
|
|
Clientid: "clientid",
|
|
Cansync: true,
|
|
})
|
|
require.NoError(t, err)
|
|
docId, err := cfg.GetDBQueries().CreateDocument(t.Context(), &repository.CreateDocumentParams{
|
|
Clientid: "clientid",
|
|
Hash: "hash",
|
|
})
|
|
require.NoError(t, err)
|
|
part := uint16(1)
|
|
filetype := "pdf"
|
|
key := objectstore.BucketKey{
|
|
CreatedAt: time.Now().UTC(),
|
|
ClientID: "clientid",
|
|
EntityID: uuid.New(),
|
|
Location: objectstore.Import,
|
|
Part: &part,
|
|
FileType: &filetype,
|
|
}
|
|
err = cfg.GetDBQueries().AddDocumentEntry(t.Context(), &repository.AddDocumentEntryParams{
|
|
Documentid: docId,
|
|
Bucket: "bucket",
|
|
Key: key.String(),
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
mockSQS.EXPECT().
|
|
SendMessage(
|
|
mock.Anything,
|
|
mock.MatchedBy(func(in *sqs.SendMessageInput) bool {
|
|
return *in.QueueUrl == cfg.DocumentCleanURL && *in.MessageBody == fmt.Sprintf("{\"id\":\"%s\"}", docId)
|
|
}),
|
|
mock.Anything,
|
|
).
|
|
Return(&sqs.SendMessageOutput{}, nil)
|
|
|
|
doc := docsyncrunner.Body{
|
|
DocumentID: docId,
|
|
}
|
|
|
|
assert.True(t, runner.Process(ctx, doc))
|
|
}
|