Merged in feature/checkbox (pull request #145)

Checkbox Primary Edge Cases

* hascheckboxlogic

* gen

* preppinggen

* exploringcheckbox

* removeunselected

* tests

* failingtest

* failintests

* nomockqueryapi

* db

* name

* nocontainer

* deleterow

* cnc

* extradocsandupdatetextracts

* moretables

* appndedtables

* baseversion

api/docSyncRunner/runner_test.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"testing"
+	"time"
 
 	docsyncrunner "queryorchestration/api/docSyncRunner"
 	"queryorchestration/internal/client"
@@ -11,12 +12,13 @@ import (
 	"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/pashagolub/pgxmock/v3"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/mock"
 	"github.com/stretchr/testify/require"
@@ -27,15 +29,13 @@ type DocSyncConfig struct {
 	documentclean.DocCleanConfig
 }
 
-func TestDocInitRunner(t *testing.T) {
+func TestDocSyncRunner(t *testing.T) {
+	t.Parallel()
+	cfg := &DocSyncConfig{}
+	net := test.GetNetwork(t)
+	test.CreateDB(t, cfg, net, &test.CreateDatabaseConfig{})
 	ctx := context.Background()
 
-	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
 
@@ -46,39 +46,51 @@ func TestDocInitRunner(t *testing.T) {
 		}),
 	})
 
-	t.Run("valid", func(t *testing.T) {
-		j := client.Client{
-			ID: "hello",
-		}
-		docinfo := document.DocumentSummary{
-			ID:       uuid.New(),
-			ClientID: j.ID,
-		}
-		doc := docsyncrunner.Body{
-			DocumentID: docinfo.ID,
-		}
-
-		pool.ExpectQuery("name: GetDocumentSummary :one").WithArgs(docinfo.ID).
-			WillReturnRows(
-				pgxmock.NewRows([]string{"id", "clientId", "hash"}).
-					AddRow(docinfo.ID, docinfo.ClientID, "example"),
-			)
-		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.DocumentID)
-				}),
-				mock.Anything,
-			).
-			Return(&sqs.SendMessageOutput{}, nil)
-
-		assert.True(t, runner.Process(ctx, doc))
+	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))
 }