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
2025-05-20 12:47:22 +00:00
parent beb221937b
commit 61d12bf8df
69 changed files with 223611 additions and 78482 deletions
@@ -43,64 +43,62 @@ func TestDocInitRunner(t *testing.T) {
 		documentstore.New(cfg),
 	})

-	t.Run("valid", func(t *testing.T) {
-		clientId := "hi"
-		bucketName := "bucketName"
-		location := objectstore.BucketKey{
-			Location:  objectstore.Import,
-			ClientID:  clientId,
-			CreatedAt: time.Now().UTC(),
-		}
-		docinfo := document.DocumentSummary{
-			ID:       uuid.New(),
-			ClientID: clientId,
-			Hash:     "example_hash",
-		}
-		doc := S3EventNotification{
-			Records: []S3EventRecord{
-				{
-					EventName: EventS3ObjectCreatedPut,
-					S3: S3EventRecordDetails{
-						Bucket: S3Bucket{
-							Name: bucketName,
-						},
-						Object: S3Object{
-							Key:  location.String(),
-							ETag: docinfo.Hash,
-						},
+	clientId := "hi"
+	bucketName := "bucketName"
+	location := objectstore.BucketKey{
+		Location:  objectstore.Import,
+		ClientID:  clientId,
+		CreatedAt: time.Now().UTC(),
+	}
+	docinfo := document.DocumentSummary{
+		ID:       uuid.New(),
+		ClientID: clientId,
+		Hash:     "example_hash",
+	}
+	doc := S3EventNotification{
+		Records: []S3EventRecord{
+			{
+				EventName: EventS3ObjectCreatedPut,
+				S3: S3EventRecordDetails{
+					Bucket: S3Bucket{
+						Name: bucketName,
+					},
+					Object: S3Object{
+						Key:  location.String(),
+						ETag: docinfo.Hash,
 					},
 				},
 			},
-		}
+		},
+	}

-		pool.ExpectQuery("name: GetDocumentIDByHash :one").WithArgs(docinfo.Hash, clientId).WillReturnRows(
-			pgxmock.NewRows([]string{"id"}),
+	pool.ExpectQuery("name: GetDocumentIDByHash :one").WithArgs(docinfo.Hash, clientId).WillReturnRows(
+		pgxmock.NewRows([]string{"id"}),
+	)
+	pool.ExpectBegin()
+	pool.ExpectQuery("name: CreateDocument :one").WithArgs(clientId, docinfo.Hash).
+		WillReturnRows(
+			pgxmock.NewRows([]string{"id"}).
+				AddRow(docinfo.ID),
+		)
+	pool.ExpectExec("name: AddDocumentEntry :exec").WithArgs(docinfo.ID, bucketName, location.String()).
+		WillReturnResult(pgxmock.NewResult("", 1))
+	pool.ExpectCommit()
+	pool.ExpectQuery("name: GetDocumentSummary :one").WithArgs(docinfo.ID).
+		WillReturnRows(
+			pgxmock.NewRows([]string{"id", "clientId", "hash"}).
+				AddRow(docinfo.ID, docinfo.ClientID, "example"),
 		)
-		pool.ExpectBegin()
-		pool.ExpectQuery("name: CreateDocument :one").WithArgs(clientId, docinfo.Hash).
-			WillReturnRows(
-				pgxmock.NewRows([]string{"id"}).
-					AddRow(docinfo.ID),
-			)
-		pool.ExpectExec("name: AddDocumentEntry :exec").WithArgs(docinfo.ID, bucketName, location.String()).
-			WillReturnResult(pgxmock.NewResult("", 1))
-		pool.ExpectCommit()
-		pool.ExpectQuery("name: GetDocumentSummary :one").WithArgs(docinfo.ID).
-			WillReturnRows(
-				pgxmock.NewRows([]string{"id", "clientId", "hash"}).
-					AddRow(docinfo.ID, docinfo.ClientID, "example"),
-			)

-		mockSQS.EXPECT().
-			SendMessage(
-				mock.Anything,
-				mock.MatchedBy(func(in *sqs.SendMessageInput) bool {
-					return *in.QueueUrl == cfg.DocInitURL
-				}),
-				mock.Anything,
-			).
-			Return(&sqs.SendMessageOutput{}, nil)
+	mockSQS.EXPECT().
+		SendMessage(
+			mock.Anything,
+			mock.MatchedBy(func(in *sqs.SendMessageInput) bool {
+				return *in.QueueUrl == cfg.DocInitURL
+			}),
+			mock.Anything,
+		).
+		Return(&sqs.SendMessageOutput{}, nil)

-		assert.True(t, runner.Process(ctx, doc))
-	})
+	assert.True(t, runner.Process(ctx, doc))
 }