Function TestHandlerSuccessTCP

pkg/probe/handler_test.go:45–78

(t *testing.T)

}

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

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

	var userHandler http.HandlerFunc = func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusOK)
	}
	server := httptest.NewServer(userHandler)

	pb := probe.NewDefaultProbe(server.URL, 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)
	}

	handler(w, r)

	require.Equal(t, http.StatusOK, w.Code)
	require.Equal(t, "healthy", w.Body.String())
}

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