Michael McGuinness
39 lines
826 B
Go
39 lines
826 B
Go
package document_test
|
|
|
|
import (
|
|
"context"
|
|
contextfull "queryorchestration/internal/contextFull"
|
|
"queryorchestration/internal/query"
|
|
"queryorchestration/internal/result"
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestContextFull(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
extractor := contextfull.New()
|
|
|
|
query := query.QueryRow{
|
|
ID: uuid.New(),
|
|
Type: query.TypeJsonExtractor,
|
|
RequiredQueryID: uuid.Nil,
|
|
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")
|
|
}
|