Function TestEngine_Switch

pkg/engine/engine_test.go:64–103

(t *testing.T)

}

func TestEngine_Switch(t *testing.T) {
	ctx := context.Background()
	reg := prometheus.NewRegistry()

	now := time.Now()
	start := time.Now().Add(-time.Minute * 5)
	step := time.Minute
	queryable := promqltest.LoadedStorage(t, "")
	opts := promql.EngineOpts{
		Logger: utillog.GoKitLogToSlog(log.NewNopLogger()),
		Reg:    reg,
	}
	queryEngine := New(opts, ThanosEngineConfig{Enabled: true}, reg)

	// Query Prometheus engine
	r := &http.Request{Header: http.Header{}}
	r.Header.Set(TypeHeader, string(Prometheus))
	ctx = AddEngineTypeToContext(ctx, r)
	_, _ = queryEngine.NewInstantQuery(ctx, queryable, nil, "foo", now)
	_, _ = queryEngine.NewRangeQuery(ctx, queryable, nil, "foo", start, now, step)

	require.NoError(t, testutil.GatherAndCompare(reg, bytes.NewBufferString(`
		# HELP cortex_engine_switch_queries_total Total number of queries where engine_type is set explicitly
		# TYPE cortex_engine_switch_queries_total counter
		cortex_engine_switch_queries_total{engine_type="prometheus"} 2
	`), "cortex_engine_switch_queries_total"))

	// Query Thanos engine
	r.Header.Set(TypeHeader, string(Thanos))
	ctx = AddEngineTypeToContext(ctx, r)
	_, _ = queryEngine.NewInstantQuery(ctx, queryable, nil, "foo", now)
	_, _ = queryEngine.NewRangeQuery(ctx, queryable, nil, "foo", start, now, step)

	require.NoError(t, testutil.GatherAndCompare(reg, bytes.NewBufferString(`
		# HELP cortex_engine_switch_queries_total Total number of queries where engine_type is set explicitly
		# TYPE cortex_engine_switch_queries_total counter
		cortex_engine_switch_queries_total{engine_type="prometheus"} 2
		cortex_engine_switch_queries_total{engine_type="thanos"} 2
	`), "cortex_engine_switch_queries_total"))
}

func TestEngine_XFunctions(t *testing.T) {
	ctx := context.Background()