query-orchestration/internal/contextFull/process_test.go

package contextfull_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")
}