Merged in feature/splitqueryrunning (pull request #57)

Split Query Running + Debugging Full Flow * completedquerysyncrunner * spliitinglogic * synccomplete * informdependents * only push same collector * deps * livetesting * foundissue * some issues resolved * activeupdate * collectorupdatefixes * fix dbquesries * tests * tests * pollingdebug
2025-02-11 15:22:59 +00:00
parent 24a038ec3d
commit 71f9802e1a
108 changed files with 3013 additions and 1666 deletions
@@ -0,0 +1,86 @@
+package querytest_test
+
+import (
+	"context"
+	"queryorchestration/internal/database"
+	"queryorchestration/internal/database/repository"
+	"queryorchestration/internal/document"
+	"queryorchestration/internal/job/collector"
+	"queryorchestration/internal/query"
+	"queryorchestration/internal/query/result"
+	querytest "queryorchestration/internal/query/test"
+	"queryorchestration/internal/serviceconfig"
+	"testing"
+
+	"github.com/google/uuid"
+	"github.com/jackc/pgx/v5/pgtype"
+	"github.com/pashagolub/pgxmock/v3"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestTest(t *testing.T) {
+	ctx := context.Background()
+
+	pool, err := pgxmock.NewPool()
+	if err != nil {
+		t.Fatalf("failed to open pgxmock database: %v", err)
+	}
+	cfg := &serviceconfig.BaseConfig{}
+	cfg.DBPool = pool
+	cfg.DBQueries = repository.New(pool)
+	docsvc := document.New(cfg)
+	col := collector.New(cfg, &collector.Services{
+		Document: docsvc,
+	})
+	svc := querytest.New(cfg, &querytest.Services{
+		Document:  docsvc,
+		Collector: col,
+		Result: result.New(cfg, &result.Services{
+			Query: query.New(cfg),
+		}),
+	})
+
+	coll := collector.Collector{
+		ID:    uuid.New(),
+		JobID: uuid.New(),
+	}
+	doc := document.Document{
+		ID:    uuid.New(),
+		JobID: coll.JobID,
+		Hash:  "example_hash",
+	}
+	params := &result.Process{
+		QueryID:      uuid.New(),
+		DocumentID:   doc.ID,
+		QueryVersion: int32(1),
+	}
+
+	reqID := database.MustToDBUUID(uuid.New())
+	pool.ExpectQuery("name: GetDocument :one").WithArgs(database.MustToDBUUID(doc.ID)).
+		WillReturnRows(
+			pgxmock.NewRows([]string{"id", "jobId", "hash"}).
+				AddRow(database.MustToDBUUID(doc.ID), database.MustToDBUUID(doc.JobID), doc.Hash),
+		)
+	pool.ExpectQuery("name: GetCollectorByJobID :one").WithArgs(database.MustToDBUUID(doc.JobID)).
+		WillReturnRows(
+			pgxmock.NewRows([]string{"id", "jobId", "minCleanVersion", "minTextVersion", "activeVersion", "latestVersion", "fields"}).
+				AddRow(database.MustToDBUUID(coll.ID), database.MustToDBUUID(doc.JobID), coll.MinCleanVersion, coll.MinTextVersion, int32(1), int32(2), []byte("")),
+		)
+	pool.ExpectQuery("name: GetQueryWithVersion :one").WithArgs(database.MustToDBUUID(params.QueryID), params.QueryVersion).WillReturnRows(
+		pgxmock.NewRows([]string{"id", "type", "activeVersion", "latestVersion", "config", "requiredIds"}).
+			AddRow(database.MustToDBUUID(params.QueryID), repository.QuerytypeJsonExtractor, int32(1), params.QueryVersion+1, []byte("{\"path\":\"oldkey\"}"), []pgtype.UUID{reqID}),
+	)
+	pool.ExpectQuery("name: ListQueryRequirementValues :many").WithArgs(database.MustToDBUUID(params.DocumentID), database.MustToDBUUID(params.QueryID), params.QueryVersion).
+		WillReturnRows(
+			pgxmock.NewRows([]string{"queryId", "type", "value"}).
+				AddRow(reqID, repository.QuerytypeContextFull, "{\"mykey\":\"example_value\",\"oldkey\":\"old_value\"}"),
+		)
+	pool.ExpectQuery("name: GetQueryConfig :one").WithArgs(database.MustToDBUUID(params.QueryID), params.QueryVersion).WillReturnRows(
+		pgxmock.NewRows([]string{"id", "config"}).
+			AddRow(pgtype.UUID{}, []byte("{\"path\":\"oldkey\"}")),
+	)
+
+	result, err := svc.Test(ctx, *params)
+	assert.NoError(t, err)
+	assert.Equal(t, "old_value", result)
+}