jacob-mathison/query-orchestration
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Activity

Merged in feature/textextract (pull request #108)

Browse Source 
Start adding Textract + UUID changes

* base

* startclient

* ts

* short

* tests
This commit is contained in:
Michael McGuinness
2025-03-20 11:06:41 +00:00
parent 1587da9d11
commit 53ea7d34e6
196 changed files with 25972 additions and 1456 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/query/sync/sync_test.go
+9 -12
View File
@@ -5,7 +5,6 @@ import (
"fmt"
"testing"
"queryorchestration/internal/database"
"queryorchestration/internal/database/repository"
resultsync "queryorchestration/internal/query/result/sync"
"queryorchestration/internal/serviceconfig"
@@ -44,28 +43,26 @@ func TestSync(t *testing.T) {
t.Run("valid", func(t *testing.T) {
id := uuid.New()
qs := []uuid.UUID{
uuid.New(),
uuid.New(),
}
reqOne := uuid.New()
reqTwo := uuid.New()
pool.ExpectQuery("name: IsDocumentTextExtracted :one").WithArgs(database.MustToDBUUID(id)).
pool.ExpectQuery("name: IsDocumentTextExtracted :one").WithArgs(id).
WillReturnRows(
pgxmock.NewRows([]string{"isextracted"}).
AddRow(true),
)
pool.ExpectQuery("name: ListUnsyncedNoDepsQueriesByDocId :many").WithArgs(database.MustToDBUUID(id)).
pool.ExpectQuery("name: ListUnsyncedNoDepsQueriesByDocId :many").WithArgs(&id).
WillReturnRows(
pgxmock.NewRows([]string{"id"}).
AddRow(database.MustToDBUUID(qs[0])).
AddRow(database.MustToDBUUID(qs[1])),
AddRow(&reqOne).
AddRow(&reqTwo),
)
mockSQS.EXPECT().
SendMessage(
mock.Anything,
mock.MatchedBy(func(in *sqs.SendMessageInput) bool {
return *in.QueueUrl == cfg.QueryURL && *in.MessageBody == fmt.Sprintf("{\"document_id\":\"%s\",\"query_id\":\"%s\"}", id.String(), qs[0].String())
return *in.QueueUrl == cfg.QueryURL && *in.MessageBody == fmt.Sprintf("{\"document_id\":\"%s\",\"query_id\":\"%s\"}", id.String(), reqOne.String())
}),
mock.Anything,
).
@@ -75,7 +72,7 @@ func TestSync(t *testing.T) {
SendMessage(
mock.Anything,
mock.MatchedBy(func(in *sqs.SendMessageInput) bool {
return *in.QueueUrl == cfg.QueryURL && *in.MessageBody == fmt.Sprintf("{\"document_id\":\"%s\",\"query_id\":\"%s\"}", id.String(), qs[1].String())
return *in.QueueUrl == cfg.QueryURL && *in.MessageBody == fmt.Sprintf("{\"document_id\":\"%s\",\"query_id\":\"%s\"}", id.String(), reqTwo.String())
}),
mock.Anything,
).
@@ -86,7 +83,7 @@ func TestSync(t *testing.T) {
t.Run("not extracted", func(t *testing.T) {
id := uuid.New()
pool.ExpectQuery("name: IsDocumentTextExtracted :one").WithArgs(database.MustToDBUUID(id)).
pool.ExpectQuery("name: IsDocumentTextExtracted :one").WithArgs(id).
WillReturnRows(
pgxmock.NewRows([]string{"isextracted"}).
AddRow(false),