Function TestHandlerSuccessHTTP

pkg/probe/handler_test.go:80–142

(t *testing.T)

}

func TestHandlerSuccessHTTP(t *testing.T) {
	t.Parallel()

	log := newLogger(t)
	defer func() { _ = log.Sync() }()

	headers := []kcore.HTTPHeader{
		{
			Name:  "X-Cortex-Blah",
			Value: "Blah",
		},
	}

	var userHandler http.HandlerFunc = func(w http.ResponseWriter, r *http.Request) {
		require.True(t, probe.IsRequestKubeletProbe(r))
		for _, header := range headers {
			require.Equal(t, header.Value, r.Header.Get(header.Name))
		}

		w.WriteHeader(http.StatusOK)
	}
	server := httptest.NewServer(userHandler)
	targetURL, err := url.Parse(server.URL)
	require.NoError(t, err)

	pb := probe.NewProbe(
		&kcore.Probe{
			Handler: kcore.Handler{
				HTTPGet: &kcore.HTTPGetAction{
					Path:        "/",
					Port:        intstr.FromString(targetURL.Port()),
					Host:        targetURL.Hostname(),
					HTTPHeaders: headers,
				},
			},
			TimeoutSeconds:   3,
			PeriodSeconds:    1,
			SuccessThreshold: 1,
			FailureThreshold: 3,
		}, log,
	)
	handler := generateHandler(pb)

	r := httptest.NewRequest(http.MethodGet, "http://fake.cortex.dev/healthz", nil)
	w := httptest.NewRecorder()

	stopper := pb.StartProbing()
	defer func() {
		stopper <- struct{}{}
	}()

	for {
		if pb.HasRunOnce() {
			break
		}
		time.Sleep(time.Second)
	}