Michael McGuinness
39 lines
882 B
Go
39 lines
882 B
Go
package document_test
|
|
|
|
import (
|
|
"context"
|
|
contextfull "queryorchestration/internal/contextFull"
|
|
queryprocessor "queryorchestration/internal/queryProcessor"
|
|
"queryorchestration/internal/result"
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestContextFull(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
extractor := contextfull.NewExtractor()
|
|
|
|
query := &queryprocessor.Query{
|
|
ID: uuid.New(),
|
|
Type: queryprocessor.TypeJsonExtractor,
|
|
RequiredQueryIDs: []uuid.UUID{},
|
|
Version: int32(1),
|
|
}
|
|
|
|
values := []result.Value{}
|
|
|
|
value, err := extractor.Process(ctx, query, values)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "", value)
|
|
|
|
values = []result.Value{
|
|
contextfull.NewResult("example_result"),
|
|
}
|
|
|
|
_, err = extractor.Process(ctx, query, values)
|
|
assert.EqualError(t, err, "no requirements expected")
|
|
}
|