Function TestRulerEvalWithQueryFrontend

integration/ruler_test.go:1716–1811

(t *testing.T)

}

func TestRulerEvalWithQueryFrontend(t *testing.T) {
	s, err := e2e.NewScenario(networkName)
	require.NoError(t, err)
	defer s.Close()

	// Start dependencies.
	consul := e2edb.NewConsul()
	minio := e2edb.NewMinio(9000, bucketName, rulestoreBucketName)
	require.NoError(t, s.StartAndWaitReady(consul, minio))

	// Configure the ruler.
	flags := mergeFlags(
		BlocksStorageFlags(),
		RulerFlags(),
		map[string]string{
			// Evaluate rules often, so that we don't need to wait for metrics to show up.
			"-ruler.evaluation-interval": "2s",
			// We run single ingester only, no replication.
			"-distributor.replication-factor": "1",
		},
	)

	const namespace = "test"
	const user = "user"

	distributor := e2ecortex.NewDistributor("distributor", e2ecortex.RingStoreConsul, consul.NetworkHTTPEndpoint(), flags, "")
	ingester := e2ecortex.NewIngester("ingester", e2ecortex.RingStoreConsul, consul.NetworkHTTPEndpoint(), flags, "")
	require.NoError(t, s.StartAndWaitReady(distributor, ingester))
	for _, format := range []string{"protobuf", "json"} {
		t.Run(fmt.Sprintf("format:%s", format), func(t *testing.T) {
			queryFrontendFlag := mergeFlags(flags, map[string]string{
				"-ruler.query-response-format":  format,
				"-frontend.query-stats-enabled": "true",
			})
			queryFrontend := e2ecortex.NewQueryFrontend("query-frontend", queryFrontendFlag, "")
			require.NoError(t, s.Start(queryFrontend))

			querier := e2ecortex.NewQuerier("querier", e2ecortex.RingStoreConsul, consul.NetworkHTTPEndpoint(), mergeFlags(queryFrontendFlag, map[string]string{
				"-querier.frontend-address": queryFrontend.NetworkGRPCEndpoint(),
			}), "")
			require.NoError(t, s.StartAndWaitReady(querier))

			rulerFlag := mergeFlags(queryFrontendFlag, map[string]string{
				"-ruler.frontend-address": queryFrontend.NetworkGRPCEndpoint(),
			})
			ruler := e2ecortex.NewRuler("ruler", consul.NetworkHTTPEndpoint(), rulerFlag, "")
			require.NoError(t, s.StartAndWaitReady(ruler))

			t.Cleanup(func() {
				_ = s.Stop(ruler)
				_ = s.Stop(queryFrontend)
				_ = s.Stop(querier)
			})

			c, err := e2ecortex.NewClient("", "", "", ruler.HTTPEndpoint(), user)
			require.NoError(t, err)

			expression := "metric" // vector