Function TestQueryWithRetrySuccess

proxyretry_test.go:122–171

TestQueryWithRetrySuccess function statResponseWriter's statusCode will be StatusOK after executeWithRetry, the query has been proxied The execution will succeeded after retry

(t *testing.T)

// TestQueryWithRetrySuccess function statResponseWriter's statusCode will be StatusOK after executeWithRetry, the query has been proxied
// The execution will succeeded after retry
func TestQueryWithRetrySuccess(t *testing.T) {
	body := "foo query"

	req := newRequest("http://localhost:8080", body)

	mhs := &mockHosts{
		t:  t,
		b:  body,
		hs: []string{"localhost:8080", "localhost:8090"},
	}

	s := newMockScope(mhs.hs)

	srw := mockStatRW(s)

	mrw := &mockResponseWriterWithCode{
		statusCode: 0,
	}

	retryNum := 1

	erroredHost := s.host

	_, err := executeWithRetry(
		context.Background(),
		s,
		retryNum,
		mhs.mockReverseProxy,
		mrw,
		srw,
		req,
		func(f float64) {},
		func(l prometheus.Labels) {},
	)
	if err != nil {
		t.Errorf("The execution with retry failed, %v", err)
	}
	assert.Equal(t, 200, srw.statusCode)
	assert.Equal(t, 1, int(s.host.CurrentConnections()))
	assert.Equal(t, 0, int(s.host.CurrentPenalty()))
	// should be counter + penalty
	assert.Equal(t, 1, int(s.host.CurrentLoad()))

	assert.Equal(t, 0, int(erroredHost.CurrentConnections()))
	assert.Equal(t, topology.DefaultPenaltySize, int(erroredHost.CurrentPenalty()))
	// should be counter + penalty
	assert.Equal(t, topology.DefaultPenaltySize, int(erroredHost.CurrentLoad()))

	assert.Equal(t, mhs.hs, mhs.hst)
}

func (mhs *mockHosts) mockReverseProxy(rw http.ResponseWriter, req *http.Request) {
	if req.URL.Host != "localhost:8090" {