passedTestingThreshhold

test/unit/internal/queryQueue/queue_test.go

@@ -2,6 +2,7 @@ package document_test
 
 import (
 	"context"
+	"errors"
 	"fmt"
 	"queryorchestration/internal/collector"
 	"queryorchestration/internal/database"
@@ -165,3 +166,41 @@ func TestQueue(t *testing.T) {
 	err = q.Execute(ctx)
 	assert.Nil(t, err)
 }
+
+func TestQueueFail(t *testing.T) {
+	ctx := context.Background()
+
+	db, err := pgxmock.NewConn()
+	if err != nil {
+		t.Fatalf("failed to open pgxmock database: %v", err)
+	}
+	defer db.Close(ctx)
+
+	queries := repository.New(db)
+	jobID := uuid.New()
+	dbJobID := database.MustToDBUUID(jobID)
+	collectorID := uuid.New()
+	dbCollectorID := database.MustToDBUUID(collectorID)
+
+	db.ExpectQuery("name: GetCollectorFromJobID :one").WithArgs(dbJobID).
+		WillReturnRows(
+			pgxmock.NewRows([]string{"id", "jobId", "minCleanVersion", "minTextVersion"}).
+				AddRow(dbCollectorID, dbJobID, int32(1), int32(1)),
+		)
+
+	coll, err := collector.NewByJobId(ctx, queries, jobID)
+	assert.Nil(t, err)
+
+	errr := "database failure"
+	db.ExpectQuery("name: GetCollectorQueries :many").WithArgs(dbCollectorID).
+		WillReturnError(errors.New(errr))
+
+	results := []result.Result{}
+
+	docID := uuid.New()
+	cleanVersion := int32(1)
+	textVersion := int32(1)
+
+	_, err = queryQueue.New(ctx, queries, coll, &results, docID, cleanVersion, textVersion)
+	assert.EqualError(t, err, errr)
+}