Merged in feature/remove-mocks (pull request #202)

Feature/remove mocks

* remove mocks

* cleanup db migrations
This commit is contained in:
Jay Brown
2026-01-15 20:39:32 +00:00
parent 0ddae4f91e
commit 35d72fccbe
244 changed files with 2573 additions and 50860 deletions
-70
View File
@@ -1,70 +0,0 @@
package documentsync_test
import (
"fmt"
"testing"
"queryorchestration/internal/client"
"queryorchestration/internal/database/repository"
"queryorchestration/internal/document"
documentsync "queryorchestration/internal/document/sync"
queuemock "queryorchestration/mocks/queue"
"github.com/stretchr/testify/require"
"github.com/aws/aws-sdk-go-v2/service/sqs"
"github.com/google/uuid"
"github.com/pashagolub/pgxmock/v3"
"github.com/stretchr/testify/mock"
)
func TestSync(t *testing.T) {
ctx := t.Context()
pool, err := pgxmock.NewPool()
require.NoError(t, err)
cfg := &DocSyncConfig{}
cfg.DBPool = pool
cfg.DBQueries = repository.New(pool)
mockSQS := queuemock.NewMockSQSClient(t)
cfg.QueueClient = mockSQS
cfg.DocumentCleanURL = "/i/am/here"
svc := documentsync.New(cfg, &documentsync.Services{
Document: document.New(cfg),
Client: client.New(cfg),
})
j := client.Client{
ID: "clientid",
CanSync: true,
}
doc := document.DocumentSummary{
ID: uuid.New(),
ClientID: j.ID,
Hash: "example_hash",
}
pool.ExpectQuery("name: GetDocumentSummary :one").WithArgs(doc.ID).
WillReturnRows(
pgxmock.NewRows([]string{"id", "clientId", "hash"}).
AddRow(doc.ID, doc.ClientID, doc.Hash),
)
pool.ExpectQuery("name: GetClient :one").WithArgs(j.ID).WillReturnRows(
pgxmock.NewRows([]string{"id", "name", "canSync"}).
AddRow(j.ID, "client_name", true),
)
mockSQS.EXPECT().
SendMessage(
mock.Anything,
mock.MatchedBy(func(in *sqs.SendMessageInput) bool {
return *in.QueueUrl == cfg.DocumentCleanURL && *in.MessageBody == fmt.Sprintf("{\"id\":\"%s\"}", doc.ID)
}),
mock.Anything,
).
Return(&sqs.SendMessageOutput{}, nil)
err = svc.Sync(ctx, doc.ID)
require.NoError(t, err)
}