Function TestFrontendCheckReady

pkg/frontend/v1/frontend_test.go:119–153

(t *testing.T)

}

func TestFrontendCheckReady(t *testing.T) {
	limits := MockLimits{MockLimits: queue.MockLimits{MaxOutstanding: 100}}
	for _, tt := range []struct {
		name             string
		connectedClients int
		msg              string
		readyForRequests bool
	}{
		{"connected clients are ready", 3, "", true},
		{"no url, no clients is not ready", 0, "not ready: number of queriers connected to query-frontend is 0", false},
	} {
		t.Run(tt.name, func(t *testing.T) {
			f := &Frontend{
				log: log.NewNopLogger(),
				requestQueue: queue.NewRequestQueue(5,
					prometheus.NewGaugeVec(prometheus.GaugeOpts{}, []string{"user"}),
					prometheus.NewCounterVec(prometheus.CounterOpts{}, []string{"user"}),
					limits,
					nil,
				),
			}
			for i := 0; i < tt.connectedClients; i++ {
				f.requestQueue.RegisterQuerierConnection("test")
			}
			err := f.CheckReady(context.Background())
			errMsg := ""

			if err != nil {
				errMsg = err.Error()
			}

			require.Equal(t, tt.msg, errMsg)
		})
	}
}

// TestFrontendCancel ensures that when client requests are cancelled,
// the underlying query is correctly cancelled _and not retried_.