Function RunQuery

fuzz/fuzz.go:92–132

(t testing.TB, sctx *super.Context, readers []sio.Reader, querySource string, useDemand func(demandIn demand.Demand))

}

func RunQuery(t testing.TB, sctx *super.Context, readers []sio.Reader, querySource string, useDemand func(demandIn demand.Demand)) []super.Value {
	// Compile query
	engine := mock.NewMockEngine(gomock.NewController(t))
	comp := compiler.NewCompiler(engine)
	ast, err := parser.ParseText(querySource)
	if err != nil {
		t.Skipf("%v", err)
	}
	query, err := runtime.CompileQuery(t.Context(), sctx, comp, ast, readers)
	if err != nil {
		t.Skipf("%v", err)
	}
	defer query.Pull(true)

	// Infer demand
	// TODO This is a hack and should be replaced by a cleaner interface in CompileQuery.
	env := exec.NewEnvironment(engine, nil)
	main, err := semantic.Analyze(t.Context(), ast, env, true)
	if err != nil {
		t.Skipf("%v", err)
	}
	if len(main.Body) > 0 {
		demand := demand.Union(optimizer.DemandForSeq(main.Body, demand.All())...)
		useDemand(demand)
	}

	// Run query
	var valuesOut []super.Value
	for {
		vec, err := query.Pull(false)
		require.NoError(t, err)
		if vec == nil {
			break
		}
		for _, value := range sbuf.Materialize(vec).Values() {
			valuesOut = append(valuesOut, value.Copy())
		}
	}

	return valuesOut
}

func CompareValues(t testing.TB, valuesExpected []super.Value, valuesActual []super.Value) {
	t.Logf("comparing: len(expected)=%v vs len(actual)=%v", len(valuesExpected), len(valuesActual))