Michael McGuinness
5ca36b0502
Feature/listtypes * started * cleanuplist * someunittestfixes * removemostflakiness
120 lines
3.2 KiB
Go
120 lines
3.2 KiB
Go
package repository_test
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path"
|
|
"queryorchestration/internal/database"
|
|
"queryorchestration/internal/database/repository"
|
|
"queryorchestration/internal/test"
|
|
"testing"
|
|
|
|
"github.com/jackc/pgx/v5/pgtype"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestQueries(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
db, cleanup := test.CreateDB(t, ctx, &test.CreateDatabaseConfig{
|
|
Migrations: &database.MigrationConfig{
|
|
BasePath: path.Join(os.Getenv("PWD"), "../../.."),
|
|
}})
|
|
defer cleanup()
|
|
|
|
queries := repository.New(db.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)
|
|
|
|
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)
|
|
}
|
|
|
|
func TestQueriesList(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
db, cleanup := test.CreateDB(t, ctx, &test.CreateDatabaseConfig{
|
|
Migrations: &database.MigrationConfig{
|
|
BasePath: "../../..",
|
|
}})
|
|
defer cleanup()
|
|
|
|
queries := repository.New(db.Pool)
|
|
|
|
contextQueryID, err := queries.CreateQuery(ctx, repository.Querytype(repository.QuerytypeContextFull))
|
|
assert.Nil(t, err)
|
|
|
|
jsonQueryID, err := queries.CreateQuery(ctx, repository.Querytype(repository.QuerytypeJsonExtractor))
|
|
assert.Nil(t, err)
|
|
|
|
qs, err := queries.ListQueries(ctx)
|
|
assert.Nil(t, err)
|
|
assert.Len(t, qs, 2)
|
|
assert.ElementsMatch(t, []*repository.Fullactivequery{
|
|
{
|
|
ID: jsonQueryID,
|
|
Type: repository.QuerytypeJsonExtractor,
|
|
Activeversion: 1,
|
|
Latestversion: 1,
|
|
Config: nil,
|
|
Requiredids: []pgtype.UUID{{}},
|
|
},
|
|
{
|
|
ID: contextQueryID,
|
|
Type: repository.QuerytypeContextFull,
|
|
Activeversion: 1,
|
|
Latestversion: 1,
|
|
Config: nil,
|
|
Requiredids: []pgtype.UUID{{}},
|
|
},
|
|
}, qs)
|
|
}
|