Merged in feature/unittests (pull request #9)

Feature/unittests * addedmoretests * easyquerytests * fixotelinit * preppedqueriestotestcontainer * dbqueries * splitintoqueryfiles * covered the bases
2025-01-08 18:14:15 +00:00
parent 61b2354e1b
commit 57764272c3
37 changed files with 1362 additions and 248 deletions
@@ -0,0 +1,90 @@
+package repository_test
+
+import (
+	"context"
+	"log"
+	"queryorchestration/internal/database/repository"
+	"testing"
+
+	"github.com/jackc/pgx/v5/pgtype"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestQueries(t *testing.T) {
+	ctx := context.Background()
+	pool, container := createDB(t, ctx)
+	defer container.Terminate(ctx)
+	queries := repository.New(pool)
+
+	contextQueryID, err := queries.CreateQuery(ctx, repository.Querytype(repository.QuerytypeContextFull))
+	assert.Nil(t, err)
+	assert.True(t, contextQueryID.Valid)
+
+	jsonQueryID, err := queries.CreateQuery(ctx, repository.Querytype(repository.QuerytypeJsonExtractor))
+	assert.Nil(t, err)
+	assert.True(t, jsonQueryID.Valid)
+
+	jsonConfig := []byte("{\"path\": \"example_path\"}")
+
+	err = queries.CreateRequiredQuery(ctx, repository.CreateRequiredQueryParams{
+		Queryid:         jsonQueryID,
+		Requiredqueryid: contextQueryID,
+		Addedversion:    1,
+	})
+	assert.Nil(t, err)
+
+	err = queries.CreateQueryConfig(ctx, repository.CreateQueryConfigParams{
+		Queryid:      jsonQueryID,
+		Config:       jsonConfig,
+		Addedversion: 1,
+	})
+	assert.Nil(t, err)
+
+	jsonQuery, err := queries.GetQuery(ctx, jsonQueryID)
+	assert.Nil(t, err)
+	assert.EqualExportedValues(t, repository.Fullactivequery{
+		ID:            jsonQueryID,
+		Type:          repository.QuerytypeJsonExtractor,
+		Activeversion: 1,
+		Latestversion: 1,
+		Config:        jsonConfig,
+		Requiredids:   []pgtype.UUID{contextQueryID},
+	}, jsonQuery)
+
+	jsonQueryConfig, err := queries.GetQueryConfig(ctx, repository.GetQueryConfigParams{
+		Queryid:      jsonQueryID,
+		Addedversion: jsonQuery.Activeversion,
+	})
+	assert.Nil(t, err)
+	assert.Equal(t, jsonConfig, jsonQueryConfig.Config)
+
+	qs, err := queries.ListQueries(ctx)
+	assert.Nil(t, err)
+	assert.Equal(t, 2, len(qs))
+	log.Print(qs)
+	assert.ElementsMatch(t, []repository.Fullactivequery{
+		{ID: jsonQueryID,
+			Type:          repository.QuerytypeJsonExtractor,
+			Activeversion: 1,
+			Latestversion: 1,
+			Config:        jsonConfig,
+			Requiredids:   []pgtype.UUID{contextQueryID}},
+		{ID: contextQueryID,
+			Type:          repository.QuerytypeContextFull,
+			Activeversion: 1,
+			Latestversion: 1,
+			Config:        nil,
+			Requiredids:   []pgtype.UUID{{}}},
+	}, qs)
+
+	isDeprecated, err := queries.IsQueryDeprecated(ctx, jsonQueryID)
+	assert.Nil(t, err)
+	assert.False(t, isDeprecated)
+
+	err = queries.DeprecateQuery(ctx, jsonQueryID)
+	assert.Nil(t, err)
+
+	isDeprecated, err = queries.IsQueryDeprecated(ctx, jsonQueryID)
+	assert.Nil(t, err)
+	assert.True(t, isDeprecated)
+}