Function TestSchedulerMetrics

pkg/scheduler/scheduler_test.go:412–452

(t *testing.T)

}

func TestSchedulerMetrics(t *testing.T) {
	reg := prometheus.NewPedanticRegistry()

	scheduler, frontendClient, _ := setupScheduler(t, reg, false)

	frontendLoop := initFrontendLoop(t, frontendClient, "frontend-12345")
	frontendToScheduler(t, frontendLoop, &schedulerpb.FrontendToScheduler{
		Type:        schedulerpb.ENQUEUE,
		QueryID:     1,
		UserID:      "test",
		HttpRequest: &httpgrpc.HTTPRequest{Method: "GET", Url: "/hello"},
	})
	frontendToScheduler(t, frontendLoop, &schedulerpb.FrontendToScheduler{
		Type:        schedulerpb.ENQUEUE,
		QueryID:     1,
		UserID:      "another",
		HttpRequest: &httpgrpc.HTTPRequest{Method: "GET", Url: "/hello"},
	})

	require.NoError(t, promtest.GatherAndCompare(reg, strings.NewReader(`
		# HELP cortex_query_scheduler_queue_length Number of queries in the queue.
		# TYPE cortex_query_scheduler_queue_length gauge
		cortex_query_scheduler_queue_length{priority="0",type="fifo",user="another"} 1
		cortex_query_scheduler_queue_length{priority="0",type="fifo",user="test"} 1
		# HELP cortex_request_queue_requests_total Total number of query requests going to the request queue.
		# TYPE cortex_request_queue_requests_total counter
		cortex_request_queue_requests_total{priority="0",user="another"} 1
		cortex_request_queue_requests_total{priority="0",user="test"} 1
	`), "cortex_query_scheduler_queue_length", "cortex_request_queue_requests_total"))

	scheduler.cleanupMetricsForInactiveUser("test")

	require.NoError(t, promtest.GatherAndCompare(reg, strings.NewReader(`
		# HELP cortex_query_scheduler_queue_length Number of queries in the queue.
		# TYPE cortex_query_scheduler_queue_length gauge
		cortex_query_scheduler_queue_length{priority="0",type="fifo",user="another"} 1
		# HELP cortex_request_queue_requests_total Total number of query requests going to the request queue.
		# TYPE cortex_request_queue_requests_total counter
		cortex_request_queue_requests_total{priority="0",user="another"} 1
	`), "cortex_query_scheduler_queue_length", "cortex_request_queue_requests_total"))
}

// TestQuerierLoopClient_WithLogicalPlan tests to see if the scheduler enqueues the fragment
// with the expected QueryID, logical plan, and other fragment meta-data