Michael McGuinness
518927c395
Query Update * baselineupdate * baseupdateplusmodelupdates * passtests * somemoresubmittesting * testinnerfunctions * readmeandinstall * cleanerstartup * readmeplusdeps * tidyatrighttime * validatetests * normalizedontvalidate * abitofzenormalizationcleanup * addunitstestforhelperfuns * normalizeactiveversiontestas
126 lines
2.8 KiB
Go
126 lines
2.8 KiB
Go
package queryservice
|
|
|
|
import (
|
|
"queryorchestration/internal/query"
|
|
queryprocessor "queryorchestration/internal/query/processor"
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestParseQueries(t *testing.T) {
|
|
cfg := "hey"
|
|
in := []*query.Query{
|
|
{
|
|
ID: uuid.New(),
|
|
Type: queryprocessor.TypeContextFull,
|
|
ActiveVersion: 1,
|
|
LatestVersion: 2,
|
|
RequiredQueryIDs: &[]uuid.UUID{
|
|
uuid.New(),
|
|
},
|
|
Config: &cfg,
|
|
},
|
|
}
|
|
out, err := parseQueries(in)
|
|
assert.Nil(t, err)
|
|
assert.Len(t, out, len(in))
|
|
assert.ElementsMatch(t, []Query{
|
|
{
|
|
Id: in[0].ID.String(),
|
|
Type: CONTEXTFULL,
|
|
ActiveVersion: 1,
|
|
LatestVersion: 2,
|
|
RequiredQueries: &[]string{
|
|
(*in[0].RequiredQueryIDs)[0].String(),
|
|
},
|
|
Config: in[0].Config,
|
|
},
|
|
}, out)
|
|
}
|
|
|
|
func TestParseQuery(t *testing.T) {
|
|
cfg := "hey"
|
|
in := &query.Query{
|
|
ID: uuid.New(),
|
|
Type: queryprocessor.TypeContextFull,
|
|
ActiveVersion: 1,
|
|
LatestVersion: 2,
|
|
RequiredQueryIDs: &[]uuid.UUID{
|
|
uuid.New(),
|
|
},
|
|
Config: &cfg,
|
|
}
|
|
out, err := parseQuery(in)
|
|
assert.Nil(t, err)
|
|
assert.EqualExportedValues(t, Query{
|
|
Id: in.ID.String(),
|
|
Type: CONTEXTFULL,
|
|
ActiveVersion: 1,
|
|
LatestVersion: 2,
|
|
RequiredQueries: &[]string{
|
|
(*in.RequiredQueryIDs)[0].String(),
|
|
},
|
|
Config: in.Config,
|
|
},
|
|
*out)
|
|
}
|
|
|
|
func TestParseQueryMinimal(t *testing.T) {
|
|
in := &query.Query{
|
|
ID: uuid.New(),
|
|
Type: queryprocessor.TypeContextFull,
|
|
ActiveVersion: 1,
|
|
LatestVersion: 2,
|
|
}
|
|
out, err := parseQuery(in)
|
|
assert.Nil(t, err)
|
|
assert.EqualExportedValues(t, Query{
|
|
Id: in.ID.String(),
|
|
Type: CONTEXTFULL,
|
|
ActiveVersion: 1,
|
|
LatestVersion: 2,
|
|
},
|
|
*out)
|
|
}
|
|
|
|
func TestParseQueryType(t *testing.T) {
|
|
qt, err := parseQueryType(queryprocessor.TypeContextFull)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, CONTEXTFULL, qt)
|
|
|
|
qt, err = parseQueryType(queryprocessor.TypeJsonExtractor)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, JSONEXTRACTOR, qt)
|
|
|
|
_, err = parseQueryType(queryprocessor.Type(-1))
|
|
assert.NotNil(t, err)
|
|
}
|
|
|
|
func TestParseSpecQueryType(t *testing.T) {
|
|
qt, err := parseSpecQueryType(CONTEXTFULL)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, queryprocessor.Type(queryprocessor.TypeContextFull), qt)
|
|
|
|
qt, err = parseSpecQueryType(JSONEXTRACTOR)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, queryprocessor.Type(queryprocessor.TypeJsonExtractor), qt)
|
|
|
|
_, err = parseSpecQueryType("invalid")
|
|
assert.NotNil(t, err)
|
|
}
|
|
|
|
func TestParseStringToUUIDArray(t *testing.T) {
|
|
ids := []uuid.UUID{uuid.New()}
|
|
|
|
out, err := parseStringToUUIDArray(&[]string{ids[0].String()})
|
|
assert.Nil(t, err)
|
|
assert.ElementsMatch(t, ids, *out)
|
|
_, err = parseStringToUUIDArray(&[]string{"invalid_uuid"})
|
|
assert.Error(t, err)
|
|
out, err = parseStringToUUIDArray(nil)
|
|
assert.Nil(t, err)
|
|
assert.Nil(t, out)
|
|
}
|