github.com/m3db/m3@v1.5.1-0.20231129193456-75a402aa583b/src/integration/prometheus/prometheus.go

// Copyright (c) 2021  Uber Technologies, Inc.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.

// Package prometheus contains resources for starting a docker-backed
// Prometheus.
package prometheus

import (
	"encoding/json"
	"errors"
	"fmt"
	"net/http"
	"regexp"
	"strconv"
	"strings"
	"testing"
	"time"

	"github.com/m3db/m3/src/cluster/generated/proto/kvpb"
	"github.com/m3db/m3/src/dbnode/generated/thrift/rpc"
	"github.com/m3db/m3/src/dbnode/kvconfig"
	"github.com/m3db/m3/src/integration/resources"
	"github.com/m3db/m3/src/integration/resources/docker/dockerexternal"
	"github.com/m3db/m3/src/query/api/v1/handler/database"
	"github.com/m3db/m3/src/query/api/v1/options"
	"github.com/m3db/m3/src/query/generated/proto/prompb"
	"github.com/m3db/m3/src/query/storage"
	"github.com/m3db/m3/src/x/headers"
	"github.com/m3db/m3/src/x/ident"
	"github.com/m3db/m3/src/x/pool"
	"github.com/m3db/m3/src/x/serialize"
	xtime "github.com/m3db/m3/src/x/time"

	"github.com/prometheus/common/model"
	"github.com/stretchr/testify/require"
	"go.uber.org/zap"
)

const (
	// TestPrometheusDBNodeConfig is the test config for the dbnode.
	TestPrometheusDBNodeConfig = `
db: {}
`

	// TestPrometheusCoordinatorConfig is the test config for the coordinator.
	TestPrometheusCoordinatorConfig = `
limits:
  perQuery:
    maxFetchedSeries: 100

query:
  restrictTags:
    match:
      - name: restricted_metrics_type
        type: NOTEQUAL
        value: hidden
    strip:
    - restricted_metrics_type

lookbackDuration: 10m
`
)

// TODO:
// - Refactor RunTest to be a method on a struct and provide resources to constructor
// - Extract query limit and timeout status code as params to RunTest

// RunTest contains the logic for running the prometheus test.
func RunTest(t *testing.T, m3 resources.M3Resources, prom resources.ExternalResources) {
	logger, err := zap.NewDevelopment()
	require.NoError(t, err)

	logger.Info("running prometheus tests")

	p := prom.(*dockerexternal.Prometheus)

	testPrometheusRemoteRead(t, p, logger)
	testPrometheusRemoteWriteMultiNamespaces(t, p, logger)
	testPrometheusRemoteWriteEmptyLabelNameReturns400(t, m3.Coordinator(), logger)
	testPrometheusRemoteWriteEmptyLabelValueReturns400(t, m3.Coordinator(), logger)
	testPrometheusRemoteWriteDuplicateLabelReturns400(t, m3.Coordinator(), logger)
	testPrometheusRemoteWriteTooOldReturns400(t, m3.Coordinator(), logger)
	testPrometheusRemoteWriteRetrictMetricsType(t, m3.Coordinator(), logger)
	testQueryLookbackApplied(t, m3.Coordinator(), logger)
	testQueryLimitsApplied(t, m3.Coordinator(), logger)
	testQueryRestrictMetricsType(t, m3.Coordinator(), logger)
	testQueryTimeouts(t, m3.Coordinator(), logger)
	testPrometheusQueryNativeTimeout(t, m3.Coordinator(), logger)
	testQueryRestrictTags(t, m3.Coordinator(), logger)
	testPrometheusRemoteWriteMapTags(t, m3.Coordinator(), logger)
	testSeries(t, m3.Coordinator(), logger)
	testLabelQueryLimitsApplied(t, m3.Coordinator(), logger)
	testLabels(t, m3.Coordinator(), logger)
	testQueryLimitsGlobalApplied(t, m3.Coordinator(), logger)
	testGlobalAggregateLimits(t, m3.Coordinator(), logger)

	// Correctness tests
	testParseThreshold(t, m3.Coordinator(), logger)
	testReplace(t, m3.Coordinator(), logger)
	testEmptyMatcher(t, m3.Coordinator(), logger)
	testDebugPromReturnsDuplicates(t, m3, logger)
}

func testPrometheusRemoteRead(t *testing.T, p *dockerexternal.Prometheus, logger *zap.Logger) {
	// Ensure Prometheus can proxy a Prometheus query
	logger.Info("testing prometheus remote read")
	verifyPrometheusQuery(t, p, "prometheus_remote_storage_samples_total", 100)
}

func testPrometheusRemoteWriteMultiNamespaces(
	t *testing.T,
	p *dockerexternal.Prometheus,
	logger *zap.Logger,
) {
	logger.Info("testing remote write to multiple namespaces")

	// Make sure we're proxying writes to the unaggregated namespace
	query := fmt.Sprintf(
		"database_write_tagged_success{namespace=\"%v\"}", resources.UnaggName,
	)
	verifyPrometheusQuery(t, p, query, 0)

	// Make sure we're proxying writes to the aggregated namespace
	query = fmt.Sprintf(
		"database_write_tagged_success{namespace=\"%v\"}", resources.AggName,
	)
	verifyPrometheusQuery(t, p, query, 0)
}

func testPrometheusRemoteWriteEmptyLabelNameReturns400(
	t *testing.T,
	coordinator resources.Coordinator,
	logger *zap.Logger,
) {
	logger.Info("test write empty name for a label returns HTTP 400")
	err := coordinator.WriteProm("foo_metric", map[string]string{
		"non_empty_name": "foo",
		"":               "bar",
	}, []prompb.Sample{
		{
			Value:     42,
			Timestamp: storage.TimeToPromTimestamp(xtime.Now()),
		},
	}, nil)
	require.Error(t, err)
	require.Contains(t, err.Error(), "400")
}

func testPrometheusRemoteWriteEmptyLabelValueReturns400(
	t *testing.T,
	coordinator resources.Coordinator,
	logger *zap.Logger,
) {
	logger.Info("test write empty value for a label returns HTTP 400")
	err := coordinator.WriteProm("foo_metric", map[string]string{
		"foo":            "bar",
		"non_empty_name": "",
	}, []prompb.Sample{
		{
			Value:     42,
			Timestamp: storage.TimeToPromTimestamp(xtime.Now()),
		},
	}, nil)
	require.Error(t, err)
	require.Contains(t, err.Error(), "400")
}

func testPrometheusRemoteWriteDuplicateLabelReturns400(
	t *testing.T,
	coordinator resources.Coordinator,
	logger *zap.Logger,
) {
	logger.Info("test write with duplicate labels returns HTTP 400")
	writeRequest := prompb.WriteRequest{
		Timeseries: []prompb.TimeSeries{
			{
				Labels: []prompb.Label{
					{
						Name:  []byte(model.MetricNameLabel),
						Value: []byte("foo_metric"),
					},
					{
						Name:  []byte("dupe_name"),
						Value: []byte("foo"),
					},
					{
						Name:  []byte("non_dupe_name"),
						Value: []byte("bar"),
					},
					{
						Name:  []byte("dupe_name"),
						Value: []byte("baz"),
					},
				},
				Samples: []prompb.Sample{
					{
						Value:     42,
						Timestamp: storage.TimeToPromTimestamp(xtime.Now()),
					},
				},
			},
		},
	}
	err := coordinator.WritePromWithRequest(writeRequest, nil)
	require.Error(t, err)
	require.Contains(t, err.Error(), "400")
}

func testPrometheusRemoteWriteTooOldReturns400(
	t *testing.T,
	coordinator resources.Coordinator,
	logger *zap.Logger,
) {
	logger.Info("test write into the past returns HTTP 400")
	err := coordinator.WriteProm("foo_metric", nil, []prompb.Sample{
		{
			Value:     3.142,
			Timestamp: storage.TimeToPromTimestamp(xtime.Now().Add(-1 * time.Hour)),
		},
	}, nil)
	require.Error(t, err)
	require.Contains(t, err.Error(), "400")
}

func testPrometheusRemoteWriteRetrictMetricsType(
	t *testing.T,
	coordinator resources.Coordinator,
	logger *zap.Logger,
) {
	logger.Info("test write with unaggregated metrics type works as expected")
	err := coordinator.WriteProm("bar_metric", nil, []prompb.Sample{
		{
			Value:     42.42,
			Timestamp: storage.TimeToPromTimestamp(xtime.Now()),
		},
	}, resources.Headers{
		headers.MetricsTypeHeader: []string{"unaggregated"},
	})
	require.NoError(t, err)

	logger.Info("test write with aggregated metrics type works as expected")
	err = coordinator.WriteProm("bar_metric", nil, []prompb.Sample{
		{
			Value:     84.84,
			Timestamp: storage.TimeToPromTimestamp(xtime.Now()),
		},
	}, resources.Headers{
		headers.MetricsTypeHeader:          []string{"aggregated"},
		headers.MetricsStoragePolicyHeader: []string{"5s:6h"},
	})
	require.NoError(t, err)
}

func testQueryLookbackApplied(
	t *testing.T,
	coordinator resources.Coordinator,
	logger *zap.Logger,
) {
	// NB: this test depends on the config in m3coordinator.yml for this test
	// and the following config value "lookbackDuration: 10m".
	logger.Info("test lookback config respected")

	err := coordinator.WriteProm("lookback_test", nil, []prompb.Sample{
		{
			Value:     42.42,
			Timestamp: storage.TimeToPromTimestamp(xtime.Now().Add(-9 * time.Minute)),
		},
	}, resources.Headers{
		headers.MetricsTypeHeader: []string{"unaggregated"},
	})
	require.NoError(t, err)

	requireRangeQuerySuccess(t,
		coordinator,
		resources.RangeQueryRequest{
			Query: "lookback_test",
			Start: time.Now().Add(-10 * time.Minute),
			End:   time.Now(),
			Step:  15 * time.Second,
		},
		nil,
		func(res model.Matrix) error {
			if len(res) == 0 || len(res[0].Values) == 0 {
				return errors.New("no samples found")
			}

			latestTS := res[0].Values[len(res[0].Values)-1].Timestamp.Time()
			nowMinusTwoSteps := time.Now().Add(-30 * time.Second)
			if latestTS.After(nowMinusTwoSteps) {
				return nil
			}

			return errors.New("latest timestamp is not within two steps from now")
		},
	)
}

func testQueryLimitsApplied(
	t *testing.T,
	coordinator resources.Coordinator,
	logger *zap.Logger,
) {
	logger.Info("test query series limit with coordinator limit header " +
		"(default errors without RequireExhaustive disabled)")
	requireError(t, func() error {
		_, err := coordinator.InstantQuery(resources.QueryRequest{
			Query: "{metrics_storage=\"m3db_remote\"}",
		}, resources.Headers{
			headers.LimitMaxSeriesHeader: []string{"10"},
		})
		return err
	}, "query exceeded limit")

	logger.Info("test query series limit with require-exhaustive headers false")
	requireInstantQuerySuccess(t,
		coordinator,
		resources.QueryRequest{
			Query: "database_write_tagged_success",
		},
		resources.Headers{
			headers.LimitMaxSeriesHeader:         []string{"2"},
			headers.LimitRequireExhaustiveHeader: []string{"false"},
		},
		func(res model.Vector) error {
			if len(res) != 2 {
				return fmt.Errorf("expected two results. received %d", len(res))
			}

			return nil
		})

	logger.Info("test query series limit with require-exhaustive headers true " +
		"(below limit therefore no error)")
	requireInstantQuerySuccess(t,
		coordinator,
		resources.QueryRequest{
			Query: "database_write_tagged_success",
		},
		resources.Headers{
			headers.LimitMaxSeriesHeader:         []string{"4"},
			headers.LimitRequireExhaustiveHeader: []string{"true"},
		},
		func(res model.Vector) error {
			if len(res) != 3 {
				return fmt.Errorf("expected three results. received %d", len(res))
			}

			return nil
		})

	logger.Info("test query series limit with require-exhaustive headers " +
		"true (above limit therefore error)")
	requireError(t, func() error {
		_, err := coordinator.InstantQuery(resources.QueryRequest{
			Query: "database_write_tagged_success",
		}, resources.Headers{
			headers.LimitMaxSeriesHeader:         []string{"3"},
			headers.LimitRequireExhaustiveHeader: []string{"true"},
		})
		return err
	}, "query exceeded limit")

	requireError(t, func() error {
		_, err := coordinator.InstantQuery(resources.QueryRequest{
			Query: "database_write_tagged_success",
		}, resources.Headers{
			headers.LimitMaxSeriesHeader:         []string{"3"},
			headers.LimitRequireExhaustiveHeader: []string{"true"},
		})
		return err
	}, "400")

	logger.Info("test query docs limit with require-exhaustive headers false")
	requireInstantQuerySuccess(t,
		coordinator,
		resources.QueryRequest{
			Query: "database_write_tagged_success",
		},
		resources.Headers{
			headers.LimitMaxDocsHeader:           []string{"1"},
			headers.LimitRequireExhaustiveHeader: []string{"false"},
		},
		func(res model.Vector) error {
			// NB(nate): docs limit is imprecise so will not match exact number of series
			// returned
			if len(res) != 3 {
				return fmt.Errorf("expected three results. received %d", len(res))
			}

			return nil
		})

	logger.Info("test query docs limit with require-exhaustive headers true " +
		"(below limit therefore no error)")
	requireInstantQuerySuccess(t,
		coordinator,
		resources.QueryRequest{
			Query: "database_write_tagged_success",
		},
		resources.Headers{
			headers.LimitMaxDocsHeader:           []string{"4"},
			headers.LimitRequireExhaustiveHeader: []string{"true"},
		},
		func(res model.Vector) error {
			// NB(nate): docs limit is imprecise so will not match exact number of series
			// returned
			if len(res) != 3 {
				return fmt.Errorf("expected three results. received %d", len(res))
			}

			return nil
		})

	logger.Info("test query docs limit with require-exhaustive headers " +
		"true (above limit therefore error)")
	requireError(t, func() error {
		_, err := coordinator.InstantQuery(resources.QueryRequest{
			Query: "database_write_tagged_success",
		}, resources.Headers{
			headers.LimitMaxDocsHeader:           []string{"1"},
			headers.LimitRequireExhaustiveHeader: []string{"true"},
		})
		return err
	}, "query exceeded limit")

	requireError(t, func() error {
		_, err := coordinator.InstantQuery(resources.QueryRequest{
			Query: "database_write_tagged_success",
		}, resources.Headers{
			headers.LimitMaxDocsHeader:           []string{"1"},
			headers.LimitRequireExhaustiveHeader: []string{"true"},
		})
		return err
	}, "400")

	logger.Info("test query returned datapoints limit - zero limit disabled")
	requireRangeQuerySuccess(t,
		coordinator,
		resources.RangeQueryRequest{
			Query: "database_write_tagged_success",
			Start: time.Now().Add(-100 * time.Minute),
			End:   time.Now(),
			Step:  15 * time.Second,
		},
		resources.Headers{
			headers.LimitMaxReturnedDatapointsHeader: []string{"0"},
		},
		func(res model.Matrix) error {
			if len(res) != 3 {
				return fmt.Errorf("expected three results. received %d", len(res))
			}

			return nil
		})

	logger.Info("test query returned series limit - zero limit disabled")
	requireRangeQuerySuccess(t,
		coordinator,
		resources.RangeQueryRequest{
			Query: "database_write_tagged_success",
			Start: time.Now().Add(-100 * time.Minute),
			End:   time.Now(),
			Step:  15 * time.Second,
		},
		resources.Headers{
			headers.LimitMaxReturnedSeriesHeader: []string{"0"},
		},
		func(res model.Matrix) error {
			if len(res) != 3 {
				return fmt.Errorf("expected three results. received %d", len(res))
			}

			return nil
		})

	logger.Info("test query returned series limit - above limit disabled")
	requireRangeQuerySuccess(t,
		coordinator,
		resources.RangeQueryRequest{
			Query: "database_write_tagged_success",
			Start: time.Now().Add(-100 * time.Minute),
			End:   time.Now(),
			Step:  15 * time.Second,
		},
		resources.Headers{
			headers.LimitMaxReturnedSeriesHeader: []string{"4"},
		},
		func(res model.Matrix) error {
			if len(res) != 3 {
				return fmt.Errorf("expected three results. received %d", len(res))
			}

			return nil
		})

	logger.Info("test query returned series limit - at limit")
	requireRangeQuerySuccess(t,
		coordinator,
		resources.RangeQueryRequest{
			Query: "database_write_tagged_success",
			Start: time.Now().Add(-100 * time.Minute),
			End:   time.Now(),
			Step:  15 * time.Second,
		},
		resources.Headers{
			headers.LimitMaxReturnedSeriesHeader: []string{"3"},
		},
		func(res model.Matrix) error {
			if len(res) != 3 {
				return fmt.Errorf("expected three results. received %d", len(res))
			}

			return nil
		})

	logger.Info("test query returned series limit - below limit")
	requireRangeQuerySuccess(t,
		coordinator,
		resources.RangeQueryRequest{
			Query: "database_write_tagged_success",
			Start: time.Now().Add(-100 * time.Minute),
			End:   time.Now(),
			Step:  15 * time.Second,
		},
		resources.Headers{
			headers.LimitMaxReturnedSeriesHeader: []string{"2"},
		},
		func(res model.Matrix) error {
			if len(res) != 2 {
				return fmt.Errorf("expected two results. received %d", len(res))
			}

			return nil
		})

	// Test writes to prep for testing returned series metadata limits
	for i := 0; i < 3; i++ {
		err := coordinator.WriteProm("metadata_test_series", map[string]string{
			"metadata_test_label": fmt.Sprintf("series_label_%d", i),
		}, []prompb.Sample{
			{
				Value:     42.42,
				Timestamp: storage.TimeToPromTimestamp(xtime.Now()),
			},
		}, nil)
		require.NoError(t, err)
	}

	logger.Info("test query returned series metadata limit - zero limit disabled")
	requireLabelValuesSuccess(t,
		coordinator,
		resources.LabelValuesRequest{
			MetadataRequest: resources.MetadataRequest{
				Match: "metadata_test_series",
			},
			LabelName: "metadata_test_label",
		},
		resources.Headers{
			headers.LimitMaxReturnedSeriesMetadataHeader: []string{"0"},
		},
		func(res model.LabelValues) error {
			if len(res) != 3 {
				return fmt.Errorf("expected three results. received %d", len(res))
			}

			return nil
		})

	logger.Info("test query returned series metadata limit - above limit disabled")
	requireLabelValuesSuccess(t,
		coordinator,
		resources.LabelValuesRequest{
			MetadataRequest: resources.MetadataRequest{
				Match: "metadata_test_series",
			},
			LabelName: "metadata_test_label",
		},
		resources.Headers{
			headers.LimitMaxReturnedSeriesMetadataHeader: []string{"4"},
		},
		func(res model.LabelValues) error {
			if len(res) != 3 {
				return fmt.Errorf("expected three results. received %d", len(res))
			}

			return nil
		})

	logger.Info("test query returned series metadata limit - at limit")
	requireLabelValuesSuccess(t,
		coordinator,
		resources.LabelValuesRequest{
			MetadataRequest: resources.MetadataRequest{
				Match: "metadata_test_series",
			},
			LabelName: "metadata_test_label",
		},
		resources.Headers{
			headers.LimitMaxReturnedSeriesMetadataHeader: []string{"3"},
		},
		func(res model.LabelValues) error {
			if len(res) != 3 {
				return fmt.Errorf("expected three results. received %d", len(res))
			}

			return nil
		})

	logger.Info("test query returned series metadata limit - below limit")
	requireLabelValuesSuccess(t,
		coordinator,
		resources.LabelValuesRequest{
			MetadataRequest: resources.MetadataRequest{
				Match: "metadata_test_series",
			},
			LabelName: "metadata_test_label",
		},
		resources.Headers{
			headers.LimitMaxReturnedSeriesMetadataHeader: []string{"2"},
		},
		func(res model.LabelValues) error {
			if len(res) != 2 {
				return fmt.Errorf("expected two results. received %d", len(res))
			}

			return nil
		})

	logger.Info("test query time range limit with coordinator defaults")
	requireRangeQuerySuccess(t,
		coordinator,
		resources.RangeQueryRequest{
			Query: "database_write_tagged_success",
			Start: time.Time{},
			End:   time.Now(),
			Step:  15 * time.Second,
		},
		nil,
		func(res model.Matrix) error {
			if len(res) == 0 {
				return errors.New("expected results to be greater than 0")
			}

			return nil
		})

	logger.Info("test query time range limit with require-exhaustive headers false")
	requireRangeQuerySuccess(t,
		coordinator,
		resources.RangeQueryRequest{
			Query: "database_write_tagged_success",
			Start: time.Unix(0, 0),
			End:   time.Now(),
			Step:  15 * time.Second,
		},
		resources.Headers{
			headers.LimitMaxRangeHeader:          []string{"4h"},
			headers.LimitRequireExhaustiveHeader: []string{"false"},
		},
		func(res model.Matrix) error {
			if len(res) == 0 {
				return errors.New("expected results to be greater than 0")
			}

			return nil
		})

	logger.Info("test query time range limit with require-exhaustive headers true " +
		"(above limit therefore error)")
	requireError(t, func() error {
		_, err := coordinator.RangeQuery(resources.RangeQueryRequest{
			Query: "database_write_tagged_success",
			Start: time.Unix(0, 0),
			End:   time.Now(),
			Step:  15 * time.Second,
		}, resources.Headers{
			headers.LimitMaxRangeHeader:          []string{"4h"},
			headers.LimitRequireExhaustiveHeader: []string{"true"},
		})
		return err
	}, "query exceeded limit")
	requireError(t, func() error {
		_, err := coordinator.RangeQuery(resources.RangeQueryRequest{
			Query: "database_write_tagged_success",
			Start: time.Unix(0, 0),
			End:   time.Now(),
			Step:  15 * time.Second,
		}, resources.Headers{
			headers.LimitMaxRangeHeader:          []string{"4h"},
			headers.LimitRequireExhaustiveHeader: []string{"true"},
		})
		return err
	}, "400")

	logger.Info("test query time range limit with coordinator defaults")
	requireLabelValuesSuccess(t,
		coordinator,
		resources.LabelValuesRequest{
			MetadataRequest: resources.MetadataRequest{
				Match: "metadata_test_series",
			},
			LabelName: "metadata_test_label",
		},
		resources.Headers{
			headers.LimitMaxReturnedSeriesMetadataHeader: []string{"2"},
		},
		func(res model.LabelValues) error {
			if len(res) == 0 {
				return errors.New("expected results to be greater than 0")
			}

			return nil
		})

	logger.Info("test query time range limit with require-exhaustive headers false")
	requireLabelValuesSuccess(t,
		coordinator,
		resources.LabelValuesRequest{
			MetadataRequest: resources.MetadataRequest{
				Match: "metadata_test_series",
			},
			LabelName: "metadata_test_label",
		},
		resources.Headers{
			headers.LimitMaxRangeHeader:          []string{"4h"},
			headers.LimitRequireExhaustiveHeader: []string{"false"},
		},
		func(res model.LabelValues) error {
			if len(res) == 0 {
				return errors.New("expected results to be greater than 0")
			}

			return nil
		})

	logger.Info("test query time range limit with require-exhaustive headers true " +
		"(above limit therefore error)")
	requireError(t, func() error {
		_, err := coordinator.LabelValues(resources.LabelValuesRequest{
			MetadataRequest: resources.MetadataRequest{
				Match: "metadata_test_series",
			},
			LabelName: "metadata_test_label",
		}, resources.Headers{
			headers.LimitMaxRangeHeader:          []string{"4h"},
			headers.LimitRequireExhaustiveHeader: []string{"true"},
		})
		return err
	}, "query exceeded limit")
	requireError(t, func() error {
		_, err := coordinator.LabelValues(resources.LabelValuesRequest{
			MetadataRequest: resources.MetadataRequest{
				Match: "metadata_test_series",
			},
			LabelName: "metadata_test_label",
		}, resources.Headers{
			headers.LimitMaxRangeHeader:          []string{"4h"},
			headers.LimitRequireExhaustiveHeader: []string{"true"},
		})
		return err
	}, "400")
}

func testQueryRestrictMetricsType(
	t *testing.T,
	coordinator resources.Coordinator,
	logger *zap.Logger,
) {
	logger.Info("test query restrict to unaggregated metrics type (instant)")
	requireNativeInstantQuerySuccess(t,
		coordinator,
		resources.QueryRequest{
			Query: "bar_metric",
		},
		resources.Headers{
			headers.MetricsTypeHeader: []string{"unaggregated"},
		},
		func(res model.Vector) error {
			if len(res) == 0 {
				return errors.New("expected results. received none")
			}

			if res[0].Value != 42.42 {
				return fmt.Errorf("expected 42.42. received %v", res[0].Value)
			}

			return nil
		})

	logger.Info("test query restrict to unaggregated metrics type (range)")
	requireNativeRangeQuerySuccess(t,
		coordinator,
		resources.RangeQueryRequest{
			Query: "bar_metric",
			Start: time.Now().Add(-1 * time.Hour),
			End:   time.Now(),
			Step:  30 * time.Second,
		},
		resources.Headers{
			headers.MetricsTypeHeader: []string{"unaggregated"},
		},
		func(res model.Matrix) error {
			if len(res) == 0 {
				return errors.New("expected results. received none")
			}

			if len(res[0].Values) == 0 {
				return errors.New("expected values for initial result. received none")
			}

			if res[0].Values[0].Value != 42.42 {
				return fmt.Errorf("expected 42.42. received %v", res[0].Values[0].Value)
			}

			return nil
		})

	logger.Info("test query restrict to aggregated metrics type (instant)")
	requireNativeInstantQuerySuccess(t,
		coordinator,
		resources.QueryRequest{
			Query: "bar_metric",
		},
		resources.Headers{
			headers.MetricsTypeHeader:          []string{"aggregated"},
			headers.MetricsStoragePolicyHeader: []string{"5s:6h"},
		},
		func(res model.Vector) error {
			if len(res) == 0 {
				return errors.New("expected results. received none")
			}

			if res[0].Value != 84.84 {
				return fmt.Errorf("expected 84.84. received %v", res[0].Value)
			}

			return nil
		})

	logger.Info("test query restrict to aggregated metrics type (range)")
	requireNativeRangeQuerySuccess(t,
		coordinator,
		resources.RangeQueryRequest{
			Query: "bar_metric",
			Start: time.Now().Add(-1 * time.Hour),
			End:   time.Now(),
			Step:  30 * time.Second,
		},
		resources.Headers{
			headers.MetricsTypeHeader:          []string{"aggregated"},
			headers.MetricsStoragePolicyHeader: []string{"5s:6h"},
		},
		func(res model.Matrix) error {
			if len(res) == 0 {
				return errors.New("expected results. received none")
			}

			if len(res[0].Values) == 0 {
				return errors.New("expected values for initial result. received none")
			}

			if res[0].Values[0].Value != 84.84 {
				return fmt.Errorf("expected 84.84. received %v", res[0].Values[0].Value)
			}

			return nil
		})
}

func testQueryTimeouts(
	t *testing.T,
	coordinator resources.Coordinator,
	logger *zap.Logger,
) {
	tests := func(timeout, message string) {
		logger.Info(message)
		requireError(t, func() error {
			_, err := coordinator.InstantQuery(resources.QueryRequest{
				Query: "database_write_tagged_success",
			}, resources.Headers{
				headers.TimeoutHeader: []string{timeout},
			})
			return err
		}, "504")

		requireError(t, func() error {
			_, err := coordinator.RangeQuery(resources.RangeQueryRequest{
				Query: "database_write_tagged_success",
				Start: time.Unix(0, 0),
				End:   time.Now(),
			}, resources.Headers{
				headers.TimeoutHeader: []string{timeout},
			})
			return err
		}, "504")

		requireError(t, func() error {
			_, err := coordinator.LabelNames(resources.LabelNamesRequest{},
				resources.Headers{
					headers.TimeoutHeader: []string{timeout},
				})
			return err
		}, "504")

		requireError(t, func() error {
			_, err := coordinator.LabelValues(resources.LabelValuesRequest{
				LabelName: "__name__",
			}, resources.Headers{
				headers.TimeoutHeader: []string{timeout},
			})
			return err
		}, "504")
	}

	tests("1ns", "test timeouts at the coordinator layer")
	tests("1ms", "test timeouts at the coordinator -> m3db layer")
}

func testPrometheusQueryNativeTimeout(
	t *testing.T,
	coordinator resources.Coordinator,
	logger *zap.Logger,
) {
	logger.Info("test query gateway timeout (instant)")
	requireError(t, func() error {
		_, err := coordinator.InstantQueryWithEngine(resources.QueryRequest{
			Query: "bar_metric",
		}, options.M3QueryEngine, resources.Headers{
			headers.TimeoutHeader:     []string{"1ms"},
			headers.MetricsTypeHeader: []string{"unaggregated"},
		})
		return err
	}, "504")

	logger.Info("test query gateway timeout (range)")
	requireError(t, func() error {
		_, err := coordinator.RangeQueryWithEngine(resources.RangeQueryRequest{
			Query: "bar_metric",
			Start: time.Now().Add(-1 * time.Hour),
			End:   time.Now(),
			Step:  30 * time.Second,
		}, options.M3QueryEngine, resources.Headers{
			headers.TimeoutHeader:     []string{"1ms"},
			headers.MetricsTypeHeader: []string{"unaggregated"},
		})
		return err
	}, "504")
}

func testQueryRestrictTags(
	t *testing.T,
	coordinator resources.Coordinator,
	logger *zap.Logger,
) {
	// Test the default restrict tags is applied when directly querying
	// coordinator (restrict tags set to hide any restricted_metrics_type="hidden"
	// in m3coordinator.yml)

	// First write some hidden metrics.
	logger.Info("test write with unaggregated metrics type works as expected")
	require.NoError(t, coordinator.WriteProm("some_hidden_metric", map[string]string{
		"restricted_metrics_type": "hidden",
		"foo_tag":                 "foo_tag_value",
	}, []prompb.Sample{
		{
			Value:     42.42,
			Timestamp: storage.TimeToPromTimestamp(xtime.Now()),
		},
	}, nil))

	// Check that we can see them with zero restrictions applied as an
	// override (we do this check first so that when we test that they
	// don't appear by default we know that the metrics are already visible).
	logger.Info("test restrict by tags with header override to remove restrict works")
	requireInstantQuerySuccess(t, coordinator, resources.QueryRequest{
		Query: "{restricted_metrics_type=\"hidden\"}",
	}, resources.Headers{
		headers.RestrictByTagsJSONHeader: []string{"{}"},
	}, func(res model.Vector) error {
		if len(res) != 1 {
			return fmt.Errorf("expected 1 result, got %v", len(res))
		}
		return nil
	})

	// Now test that the defaults will hide the metrics altogether.
	logger.Info("test restrict by tags with coordinator defaults")
	requireInstantQuerySuccess(t, coordinator, resources.QueryRequest{
		Query: "{restricted_metrics_type=\"hidden\"}",
	}, nil, func(res model.Vector) error {
		if len(res) != 0 {
			return fmt.Errorf("expected no results, got %v", len(res))
		}
		return nil
	})
}

func testPrometheusRemoteWriteMapTags(
	t *testing.T,
	coordinator resources.Coordinator,
	logger *zap.Logger,
) {
	logger.Info("test map tags header works as expected")
	require.NoError(t, coordinator.WriteProm("bar_metric", nil, []prompb.Sample{
		{
			Value:     42.42,
			Timestamp: storage.TimeToPromTimestamp(xtime.Now()),
		},
	}, resources.Headers{
		headers.MetricsTypeHeader:   []string{"unaggregated"},
		headers.MapTagsByJSONHeader: []string{`{"tagMappers":[{"write":{"tag":"globaltag","value":"somevalue"}}]}`},
	}))

	requireNativeInstantQuerySuccess(t, coordinator, resources.QueryRequest{
		Query: "bar_metric",
	}, resources.Headers{
		headers.MetricsTypeHeader: []string{"unaggregated"},
	}, func(res model.Vector) error {
		if len(res) == 0 {
			return errors.New("expecting results, got none")
		}

		if val, ok := res[0].Metric["globaltag"]; !ok || val != "somevalue" {
			return fmt.Errorf("expected metric with globaltag=somevalue, got=%+v", res[0].Metric)
		}

		return nil
	})
}

func testSeries(t *testing.T, coordinator resources.Coordinator, logger *zap.Logger) {
	logger.Info("test series match endpoint")
	requireSeriesSuccess(t, coordinator, resources.SeriesRequest{
		MetadataRequest: resources.MetadataRequest{
			Match: "prometheus_remote_storage_samples_total",
			Start: time.Unix(0, 0),
			End:   time.Now().Add(1 * time.Hour),
		},
	}, nil, func(res []model.Metric) error {
		if len(res) != 1 {
			return fmt.Errorf("expected 1 result, got %v", len(res))
		}
		return nil
	})

	requireSeriesSuccess(t, coordinator, resources.SeriesRequest{
		MetadataRequest: resources.MetadataRequest{
			Match: "prometheus_remote_storage_samples_total",
		},
	}, nil, func(res []model.Metric) error {
		if len(res) != 1 {
			return fmt.Errorf("expected 1 result, got %v", len(res))
		}
		return nil
	})

	// NB(nate): Use raw RunQuery method here since we want to use a custom format for start
	// and end
	queryAndParms := "api/v1/series?match[]=prometheus_remote_storage_samples_total&start=" +
		"-292273086-05-16T16:47:06Z&end=292277025-08-18T07:12:54.999999999Z"
	require.NoError(t, coordinator.RunQuery(
		func(status int, headers map[string][]string, resp string, err error) error {
			if status != http.StatusOK {
				return fmt.Errorf("expected 200, got %d. body=%v", status, resp)
			}
			var parsedResp seriesResponse
			if err := json.Unmarshal([]byte(resp), &parsedResp); err != nil {
				return err
			}

			if len(parsedResp.Data) != 1 {
				return fmt.Errorf("expected 1 result, got %d", len(parsedResp.Data))
			}

			return nil
		}, queryAndParms, nil))
}

func testLabelQueryLimitsApplied(
	t *testing.T,
	coordinator resources.Coordinator,
	logger *zap.Logger,
) {
	logger.Info("test label limits with require-exhaustive headers true " +
		"(below limit therefore no error)")
	requireLabelValuesSuccess(t,
		coordinator,
		resources.LabelValuesRequest{
			LabelName: "__name__",
		},
		resources.Headers{
			headers.LimitMaxSeriesHeader:         []string{"10000"},
			headers.LimitRequireExhaustiveHeader: []string{"true"},
		},
		func(res model.LabelValues) error {
			// NB(nate): just checking for a 200 and this method only gets called in that case
			return nil
		})

	logger.Info("test label series limit with coordinator limit header (default " +
		"requires exhaustive so error)")
	requireError(t, func() error {
		_, err := coordinator.LabelValues(resources.LabelValuesRequest{
			LabelName: "__name__",
		}, resources.Headers{
			headers.LimitMaxSeriesHeader: []string{"1"},
		})
		return err
	}, "query exceeded limit")

	logger.Info("test label series limit with require-exhaustive headers false")
	requireLabelValuesSuccess(t,
		coordinator,
		resources.LabelValuesRequest{
			LabelName: "__name__",
		},
		resources.Headers{
			headers.LimitMaxSeriesHeader:         []string{"2"},
			headers.LimitRequireExhaustiveHeader: []string{"false"},
		},
		func(res model.LabelValues) error {
			if len(res) != 1 {
				return fmt.Errorf("expected 1 result, got %d", len(res))
			}
			return nil
		})

	logger.Info("Test label series limit with require-exhaustive headers " +
		"true (above limit therefore error)")
	requireError(t, func() error {
		_, err := coordinator.LabelValues(resources.LabelValuesRequest{
			LabelName: "__name__",
		}, resources.Headers{
			headers.LimitMaxSeriesHeader:         []string{"2"},
			headers.LimitRequireExhaustiveHeader: []string{"true"},
		})
		return err
	}, "query exceeded limit")
	requireError(t, func() error {
		_, err := coordinator.LabelValues(resources.LabelValuesRequest{
			LabelName: "__name__",
		}, resources.Headers{
			headers.LimitMaxSeriesHeader:         []string{"2"},
			headers.LimitRequireExhaustiveHeader: []string{"true"},
		})
		return err
	}, "400")

	logger.Info("test label docs limit with coordinator limit header " +
		"(default requires exhaustive so error)")
	requireError(t, func() error {
		_, err := coordinator.LabelValues(resources.LabelValuesRequest{
			LabelName: "__name__",
		}, resources.Headers{
			headers.LimitMaxDocsHeader: []string{"1"},
		})
		return err
	}, "query exceeded limit")

	logger.Info("test label docs limit with require-exhaustive headers false")
	requireLabelValuesSuccess(t,
		coordinator,
		resources.LabelValuesRequest{
			LabelName: "__name__",
		},
		resources.Headers{
			headers.LimitMaxDocsHeader:           []string{"2"},
			headers.LimitRequireExhaustiveHeader: []string{"false"},
		},
		func(res model.LabelValues) error {
			if len(res) != 1 {
				return fmt.Errorf("expected 1 result, got %d", len(res))
			}
			return nil
		})

	logger.Info("Test label docs limit with require-exhaustive headers " +
		"true (above limit therefore error)")
	requireError(t, func() error {
		_, err := coordinator.LabelValues(resources.LabelValuesRequest{
			LabelName: "__name__",
		}, resources.Headers{
			headers.LimitMaxSeriesHeader:         []string{"1"},
			headers.LimitRequireExhaustiveHeader: []string{"true"},
		})
		return err
	}, "query exceeded limit")
	requireError(t, func() error {
		_, err := coordinator.LabelValues(resources.LabelValuesRequest{
			LabelName: "__name__",
		}, resources.Headers{
			headers.LimitMaxSeriesHeader:         []string{"1"},
			headers.LimitRequireExhaustiveHeader: []string{"true"},
		})
		return err
	}, "400")
}

func testLabels(
	t *testing.T,
	coordinator resources.Coordinator,
	logger *zap.Logger,
) {
	logger.Info("test label APIs")
	require.NoError(t, coordinator.WriteProm("label_metric", map[string]string{
		"name_0": "value_0_1",
		"name_1": "value_1_1",
		"name_2": "value_2_1",
	}, []prompb.Sample{
		{
			Value:     42.42,
			Timestamp: storage.TimeToPromTimestamp(xtime.Now()),
		},
	}, nil))

	require.NoError(t, coordinator.WriteProm("label_metric_2", map[string]string{
		"name_0": "value_0_2",
		"name_1": "value_1_2",
	}, []prompb.Sample{
		{
			Value:     42.42,
			Timestamp: storage.TimeToPromTimestamp(xtime.Now()),
		},
	}, nil))

	requireLabelNamesSuccess(t, coordinator, resources.LabelNamesRequest{}, nil,
		func(res model.LabelNames) error {
			var nameLabels model.LabelNames
			for _, label := range res {
				// nolint:staticcheck
				matched, err := regexp.MatchString("name_[012]", string(label))
				if err != nil {
					return err
				}
				if matched {
					nameLabels = append(nameLabels, label)
				}
			}
			if len(nameLabels) != 3 {
				return fmt.Errorf("expected 3 results, got %d", len(nameLabels))
			}
			return nil
		})

	requireLabelNamesSuccess(t, coordinator, resources.LabelNamesRequest{
		MetadataRequest: resources.MetadataRequest{
			Match: "label_metric",
		},
	}, nil, func(res model.LabelNames) error {
		if len(res) != 4 {
			return fmt.Errorf("expected 4 results, got %d", len(res))
		}
		return nil
	})

	requireLabelNamesSuccess(t, coordinator, resources.LabelNamesRequest{
		MetadataRequest: resources.MetadataRequest{
			Match: "label_metric_2",
		},
	}, nil, func(res model.LabelNames) error {
		if len(res) != 3 {
			return fmt.Errorf("expected 3 results, got %d", len(res))
		}
		return nil
	})

	requireLabelValuesSuccess(t, coordinator, resources.LabelValuesRequest{
		LabelName: "name_1",
	}, nil, func(res model.LabelValues) error {
		if len(res) != 2 {
			return fmt.Errorf("expected 2 results, got %d", len(res))
		}
		return nil
	})

	tests := func(match string, length int, val string) {
		requireLabelValuesSuccess(t, coordinator, resources.LabelValuesRequest{
			MetadataRequest: resources.MetadataRequest{
				Match: match,
			},
			LabelName: "name_1",
		}, nil, func(res model.LabelValues) error {
			if len(res) != length {
				return fmt.Errorf("expected %d results, got %d", length, len(res))
			}
			return nil
		})

		requireLabelValuesSuccess(t, coordinator, resources.LabelValuesRequest{
			MetadataRequest: resources.MetadataRequest{
				Match: match,
			},
			LabelName: "name_1",
		}, nil, func(res model.LabelValues) error {
			if string(res[0]) != val {
				return fmt.Errorf("expected %s, got %s", val, res[0])
			}
			return nil
		})
	}
	tests("label_metric", 1, "value_1_1")
	tests("label_metric_2", 1, "value_1_2")
}

func testQueryLimitsGlobalApplied(
	t *testing.T,
	coordinator resources.Coordinator,
	logger *zap.Logger,
) {
	logger.Info("test global query limits")
	labelVals := []string{"series_label_0", "series_label_1", "series_label_2"}
	for _, val := range labelVals {
		require.NoError(t, coordinator.WriteProm("metadata_test_series", map[string]string{
			"query_global_limit_test": val,
		}, []prompb.Sample{
			{
				Value:     42.42,
				Timestamp: storage.TimeToPromTimestamp(xtime.Now()),
			},
		}, nil))
	}

	// Set global limits
	setSeriesDiskReadLimits(t, coordinator, kvpb.QueryLimits{
		MaxRecentlyQueriedSeriesDiskRead: &kvpb.QueryLimit{Limit: 1, LookbackSeconds: 5},
	})

	// Test that global limits are tripped
	requireError(t, func() error {
		_, err := coordinator.InstantQuery(resources.QueryRequest{
			Query: `{query_global_limit_test!=""}`,
		}, nil)
		return err
	}, "")

	// Force wait for permit
	setSeriesDiskReadLimits(t, coordinator, kvpb.QueryLimits{
		MaxRecentlyQueriedSeriesDiskRead: &kvpb.QueryLimit{
			Limit:           10000,
			LookbackSeconds: 5,
			ForceWaited:     true,
		},
	})

	// Check that waited header is returned. Use RunQuery to get access to headers.
	queryAndParams := `api/v1/query?query={query_global_limit_test!=""}`
	require.NoError(t, coordinator.RunQuery(
		func(status int, h map[string][]string, resp string, err error) error {
			if status != http.StatusOK {
				return fmt.Errorf("expected 200, got %d. body=%v", status, resp)
			}

			if _, ok := h[headers.WaitedHeader]; !ok {
				return fmt.Errorf("expected header %v to be present on response",
					headers.WaitedHeader)
			}

			return nil
		}, queryAndParams, nil))

	// Test that global limits are tripped
	requireError(t, func() error {
		_, err := coordinator.InstantQuery(resources.QueryRequest{
			Query: `{query_global_limit_test!=""}`,
		}, resources.Headers{
			headers.LimitRequireNoWaitHeader: []string{"true"},
		})
		return err
	}, "400")

	// Restore limits
	setSeriesDiskReadLimits(t, coordinator, kvpb.QueryLimits{})
}

func testGlobalAggregateLimits(
	t *testing.T,
	coordinator resources.Coordinator,
	logger *zap.Logger,
) {
	logger.Info("test global aggregate limits")
	metricName := fmt.Sprintf("aggregate_limits_%v", xtime.Now().Seconds())
	numMetrics := 60
	for i := 0; i < numMetrics; i++ {
		require.NoError(t, coordinator.WriteProm(metricName, map[string]string{
			metricName: strconv.Itoa(i),
		}, []prompb.Sample{
			{
				Value:     float64(i),
				Timestamp: storage.TimeToPromTimestamp(xtime.Now()),
			},
		}, nil))
	}

	// Set global limits
	setSeriesDiskReadLimits(t, coordinator, kvpb.QueryLimits{
		MaxRecentlyQueriedMetadataRead: &kvpb.QueryLimit{Limit: 150, LookbackSeconds: 5},
	})

	checkLabels := func() {
		requireLabelValuesSuccess(t, coordinator, resources.LabelValuesRequest{
			LabelName: metricName,
		}, nil, func(res model.LabelValues) error {
			if len(res) != numMetrics {
				return fmt.Errorf("expected %d results, got %v", numMetrics, len(res))
			}
			return nil
		})
	}

	// Make sure any existing limit has expired before continuing.
	checkLabels()
	checkLabels()
	requireError(t, func() error {
		_, err := coordinator.LabelValues(resources.LabelValuesRequest{
			LabelName: metricName,
		}, nil)
		return err
	}, "error")

	// Make sure that a query is unaffected by the the metadata limits.
	ts := time.Now()
	requireInstantQuerySuccess(t, coordinator, resources.QueryRequest{
		Query: fmt.Sprintf("sum(%v)", metricName),
		Time:  &ts,
	}, nil, func(res model.Vector) error {
		// NB(nate): just want a 200 here which a call to this method indicates
		return nil
	})

	checkLabels()

	// Restore limits
	setSeriesDiskReadLimits(t, coordinator, kvpb.QueryLimits{})
}

func testParseThreshold(
	t *testing.T,
	coordinator resources.Coordinator,
	logger *zap.Logger,
) {
	logger.Info("test parse and threshold API endpoints")
	require.NoError(t, coordinator.RunQuery(
		func(status int, h map[string][]string, resp string, err error) error {
			if status != http.StatusOK {
				return fmt.Errorf("expected 200, got %d. body=%v", status, resp)
			}

			var parsedResp parseResponse
			if err := json.Unmarshal([]byte(resp), &parsedResp); err != nil {
				return err
			}

			if parsedResp.Name != "fetch" {
				return fmt.Errorf("expected 'fetch', got '%s'", parsedResp.Name)
			}

			return nil
		}, "api/v1/parse?query=up", nil))

	thresholdTest := func(path string, comparator string) {
		require.NoError(t, coordinator.RunQuery(
			func(status int, h map[string][]string, resp string, err error) error {
				if status != http.StatusOK {
					return fmt.Errorf("expected 200, got %d. body=%v", status, resp)
				}

				var parsedResp thresholdResponse
				if err := json.Unmarshal([]byte(resp), &parsedResp); err != nil {
					return err
				}

				if parsedResp.Query.Name != "fetch" {
					return fmt.Errorf("expected 'fetch', got '%s'", parsedResp.Query.Name)
				}

				if parsedResp.Threshold.Comparator != comparator {
					return fmt.Errorf("expected '>', got '%s'", parsedResp.Threshold.Comparator)
				}

				if parsedResp.Threshold.Value != 1 {
					return fmt.Errorf("expected 1, got %d", parsedResp.Threshold.Value)
				}

				return nil
			}, path, nil))
	}
	thresholdTest("api/v1/threshold?query=up>1", ">")
	thresholdTest("api/v1/threshold?query=up<1", "<")
}

func testReplace(
	t *testing.T,
	coordinator resources.Coordinator,
	logger *zap.Logger,
) {
	logger.Info("test label replace")
	var (
		numMetrics = 5
		metricName = fmt.Sprintf("quail_%v", xtime.Now().Seconds())
	)
	writeCorrectnessTestMetrics(t, coordinator, metricName, numMetrics, "extra")

	replaceTest := func(query string) {
		requireInstantQuerySuccess(t, coordinator, resources.QueryRequest{
			Query: query,
		}, nil, func(res model.Vector) error {
			if len(res) != numMetrics {
				return fmt.Errorf("expected %d results, got %d", numMetrics, len(res))
			}

			for i := 0; i < numMetrics; i++ {
				expectedVal := fmt.Sprintf("bar_%d", i)
				if val, ok := res[i].Metric["foo"]; !ok || string(val) != expectedVal {
					return fmt.Errorf("expected %s, got %s", expectedVal, val)
				}
			}

			return nil
		})
	}
	replaceTest(`label_replace(` + metricName + `,"foo","bar_$1","val","(.*)")`)
	replaceTest(`label_replace(` + metricName + `,"foo","bar_$1","val","(.*)")-0`)
}

func testEmptyMatcher(
	t *testing.T,
	coordinator resources.Coordinator,
	logger *zap.Logger,
) {
	logger.Info("test empty matcher")
	var (
		numMetrics = 5
		metricName = fmt.Sprintf("foo_%v", xtime.Now().Seconds())
		tagName1   = "exists"
		tagName2   = "not_exists"
	)
	writeCorrectnessTestMetrics(t, coordinator, metricName, numMetrics, tagName1)
	writeCorrectnessTestMetrics(t, coordinator, metricName, numMetrics, tagName2)

	emptyMatcherTests := func(query, matchedTag, notMatchedTag string) {
		requireInstantQuerySuccess(t, coordinator, resources.QueryRequest{
			Query: query,
		}, nil, func(res model.Vector) error {
			if len(res) != numMetrics {
				return fmt.Errorf("expected %d results, got %d", numMetrics, len(res))
			}

			metricsCount := make(map[string]int)
			for i := 0; i < numMetrics; i++ {
				if _, ok := res[i].Metric[model.LabelName(matchedTag)]; ok {
					metricsCount[matchedTag]++
				}
				if _, ok := res[i].Metric[model.LabelName(notMatchedTag)]; ok {
					metricsCount[notMatchedTag]++
				}
			}

			if metricsCount[matchedTag] != numMetrics {
				return fmt.Errorf("expected %d results, got %d", numMetrics, metricsCount[matchedTag])
			}

			if metricsCount[notMatchedTag] != 0 {
				return fmt.Errorf("expected 0 results, got %d", metricsCount[notMatchedTag])
			}

			return nil
		})
	}
	emptyMatcherTests(fmt.Sprintf("%s{%s%s\"\"}", metricName, tagName1, "="), tagName2, tagName1)
	emptyMatcherTests(fmt.Sprintf("%s{%s%s\"\"}", metricName, tagName1, "!="), tagName1, tagName2)
	emptyMatcherTests(fmt.Sprintf("%s{%s%s\"\"}", metricName, tagName2, "="), tagName1, tagName2)
	emptyMatcherTests(fmt.Sprintf("%s{%s%s\"\"}", metricName, tagName2, "!="), tagName2, tagName1)
}

func testDebugPromReturnsDuplicates(
	t *testing.T,
	m3 resources.M3Resources,
	logger *zap.Logger,
) {
	logger.Info("test debug prom returns duplicates")
	var (
		metricName  = fmt.Sprintf("duplicate_%v", xtime.Now().Seconds())
		coordinator = m3.Coordinator()
		dbnode      = m3.Nodes()[0]
		encoderPool = newTagEncoderPool()
		encodedTags = encodeTags(t, encoderPool,
			ident.StringTag("__name__", metricName),
			ident.StringTag("val", "extra"),
			ident.StringTag("val", "0"))
	)

	require.NoError(t, dbnode.WriteTaggedBatchRaw(&rpc.WriteTaggedBatchRawRequest{
		NameSpace: []byte(resources.UnaggName),
		Elements: []*rpc.WriteTaggedBatchRawRequestElement{
			{
				ID:          []byte(metricName),
				EncodedTags: encodedTags,
				Datapoint: &rpc.Datapoint{
					Timestamp:         time.Now().UnixNano(),
					TimestampTimeType: rpc.TimeType_UNIX_NANOSECONDS,
					Value:             1,
				},
			},
		},
	}))

	queryAndParams := fmt.Sprintf("api/v1/prom/remote/read?query=%s&start=%v&end=%v"+
		"&format=json", metricName, time.Now().Add(-100*time.Second).Unix(),
		time.Now().Add(100*time.Second).Unix())
	expectedTags := `tags":[["__name__","` + metricName + `"],["val","extra"],["val","0"]]`
	require.NoError(t, coordinator.RunQuery(
		func(status int, h map[string][]string, resp string, err error) error {
			if status != http.StatusOK {
				return fmt.Errorf("expected 200, got %d. body=%v", status, resp)
			}

			if !strings.Contains(resp, expectedTags) {
				return fmt.Errorf("expected %v in response but not found. response=%v",
					expectedTags, resp)
			}

			return nil
		}, queryAndParams, nil))
}

func writeCorrectnessTestMetrics(
	t *testing.T,
	coordinator resources.Coordinator,
	metricName string,
	numMetrics int,
	extraTag string,
) {
	ts := xtime.Now()
	if extraTag == "" {
		extraTag = "default"
	}
	for i := 0; i < numMetrics; i++ {
		err := coordinator.WriteProm(metricName, map[string]string{
			extraTag: "extra",
			"val":    strconv.Itoa(i),
		}, []prompb.Sample{
			{
				Timestamp: storage.TimeToPromTimestamp(ts),
				Value:     float64(i),
			},
		}, nil)
		require.NoError(t, err)
	}
}

type parseResponse struct {
	Name string
}

type thresholdResponse struct {
	Query     parseResponse
	Threshold threshold
}

type threshold struct {
	Comparator string
	Value      int
}

func setSeriesDiskReadLimits(
	t *testing.T,
	coordinator resources.Coordinator,
	limits kvpb.QueryLimits,
) {
	limitBytes, err := json.Marshal(limits)
	require.NoError(t, err)

	update := &database.KeyValueUpdate{
		Key:    kvconfig.QueryLimits,
		Commit: true,
		Value:  limitBytes,
	}
	updateBytes, err := json.Marshal(update)
	require.NoError(t, err)

	require.NoError(t, coordinator.ApplyKVUpdate(string(updateBytes)))
}

type seriesResponse struct {
	Status string
	Data   []map[string]string
}

func requireError(t *testing.T, query func() error, errorMsg string) {
	require.NoError(t, resources.Retry(func() error {
		if err := query(); err != nil {
			if errorMsg == "" || strings.Contains(err.Error(), errorMsg) {
				return nil
			}
		}

		err := errors.New("expected read request to fail with error")
		if errorMsg == "" {
			err = fmt.Errorf("expected read request to fail with error containing: %s", errorMsg)
		}

		return err
	}))
}

func requireInstantQuerySuccess(
	t *testing.T,
	coordinator resources.Coordinator,
	request resources.QueryRequest,
	headers resources.Headers,
	successCond func(res model.Vector) error,
) {
	require.NoError(t, resources.Retry(func() error {
		res, err := coordinator.InstantQuery(request, headers)
		if err != nil {
			return err
		}

		return successCond(res)
	}))
}

func requireNativeInstantQuerySuccess(
	t *testing.T,
	coordinator resources.Coordinator,
	request resources.QueryRequest,
	headers resources.Headers,
	successCond func(res model.Vector) error,
) {
	require.NoError(t, resources.Retry(func() error {
		res, err := coordinator.InstantQueryWithEngine(request, options.M3QueryEngine, headers)
		if err != nil {
			return err
		}

		return successCond(res)
	}))
}

func requireRangeQuerySuccess(
	t *testing.T,
	coordinator resources.Coordinator,
	request resources.RangeQueryRequest,
	headers resources.Headers,
	successCond func(res model.Matrix) error,
) {
	require.NoError(t, resources.Retry(func() error {
		res, err := coordinator.RangeQuery(request, headers)
		if err != nil {
			return err
		}

		return successCond(res)
	}))
}

func requireNativeRangeQuerySuccess(
	t *testing.T,
	coordinator resources.Coordinator,
	request resources.RangeQueryRequest,
	headers resources.Headers,
	successCond func(res model.Matrix) error,
) {
	require.NoError(t, resources.Retry(func() error {
		res, err := coordinator.RangeQueryWithEngine(request, options.M3QueryEngine, headers)
		if err != nil {
			return err
		}

		return successCond(res)
	}))
}

func requireLabelValuesSuccess(
	t *testing.T,
	coordinator resources.Coordinator,
	request resources.LabelValuesRequest,
	headers resources.Headers,
	successCond func(res model.LabelValues) error,
) {
	require.NoError(t, resources.Retry(func() error {
		res, err := coordinator.LabelValues(request, headers)
		if err != nil {
			return err
		}

		return successCond(res)
	}))
}

func requireLabelNamesSuccess(
	t *testing.T,
	coordinator resources.Coordinator,
	request resources.LabelNamesRequest,
	headers resources.Headers,
	successCond func(res model.LabelNames) error,
) {
	require.NoError(t, resources.Retry(func() error {
		res, err := coordinator.LabelNames(request, headers)
		if err != nil {
			return err
		}

		return successCond(res)
	}))
}

func requireSeriesSuccess(
	t *testing.T,
	coordinator resources.Coordinator,
	request resources.SeriesRequest,
	headers resources.Headers,
	successCond func(res []model.Metric) error,
) {
	require.NoError(t, resources.Retry(func() error {
		res, err := coordinator.Series(request, headers)
		if err != nil {
			return err
		}

		return successCond(res)
	}))
}

func verifyPrometheusQuery(t *testing.T, p *dockerexternal.Prometheus, query string, threshold float64) {
	require.NoError(t, resources.Retry(func() error {
		res, err := p.Query(dockerexternal.PrometheusQueryRequest{
			Query: query,
		})
		if err != nil {
			return err
		}
		if len(res) == 0 {
			return errors.New("no samples returned for query")
		}
		if res[0].Value > model.SampleValue(threshold) {
			return nil
		}

		return errors.New("value not greater than threshold")
	}))
}

func newTagEncoderPool() serialize.TagEncoderPool {
	encoderPool := serialize.
		NewTagEncoderPool(serialize.NewTagEncoderOptions(),
			pool.NewObjectPoolOptions().SetSize(1))
	encoderPool.Init()
	return encoderPool
}

func encodeTags(
	t *testing.T,
	pool serialize.TagEncoderPool,
	tags ...ident.Tag,
) []byte {
	encoder := pool.Get()

	seriesTags := ident.NewTags(tags...)
	err := encoder.Encode(ident.NewTagsIterator(seriesTags))
	require.NoError(t, err)

	data, ok := encoder.Data()
	require.True(t, ok)

	return data.Bytes()
}