jay brown
87 lines
2.4 KiB
Go
87 lines
2.4 KiB
Go
package docsyncrunner_test
|
|
|
|
import (
|
|
"fmt"
|
|
"regexp"
|
|
"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"
|
|
"queryorchestration/internal/serviceconfig/queue/documentclean"
|
|
"queryorchestration/internal/test"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type DocSyncConfig struct {
|
|
runner.BaseConfig[docsyncrunner.Body]
|
|
documentclean.DocCleanConfig
|
|
queue.QueueConfig
|
|
objectstore.ObjectStoreConfig
|
|
}
|
|
|
|
func TestDocSyncRunner(t *testing.T) {
|
|
cfg := &DocSyncConfig{}
|
|
test.CreateDB(t, cfg)
|
|
acfg := test.CreateAWSContainer(t, cfg)
|
|
|
|
test.SetQueueClient(t, t.Context(), cfg, acfg.ExternalEndpoint)
|
|
cfg.DocumentCleanURL = test.CreateQueue(t, cfg, test.DocCleanRunnerName)
|
|
|
|
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)
|
|
|
|
doc := docsyncrunner.Body{
|
|
DocumentID: docId,
|
|
}
|
|
|
|
assert.True(t, runner.Process(t.Context(), doc))
|
|
|
|
test.AssertMessageBody(t, cfg, cfg.GetDocumentCleanURL(), regexp.MustCompile(fmt.Sprintf("{\"id\":\"%s\"}", docId)))
|
|
}
|