healthchecks_test.go

package main

import (
	"crypto/tls"
	"io"
	"log"
	"net"
	"net/http"
	"net/http/httptest"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
)

func TestHttpsGetCheck(t *testing.T) {
	h := NewHealthCheck(8888,
		S3Config{url: "http://localhost:8080", readypath: "/"},
		BrokerConfig{host: "localhost", port: "8080"},
		new(tls.Config))

	assert.NoError(t, h.httpsGetCheck("https://www.nbis.se", 10*time.Second)())
	assert.Error(t, h.httpsGetCheck("https://www.nbis.se/nonexistent", 5*time.Second)(), "404 should fail")
}

func TestHealthchecks(t *testing.T) {
	l, err := net.Listen("tcp", "127.0.0.1:8080")
	if err != nil {
		log.Fatal(err)
	}
	foo := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {})
	ts := httptest.NewUnstartedServer(foo)
	ts.Listener.Close()
	ts.Listener = l
	ts.Start()

	h := NewHealthCheck(8888,
		S3Config{url: "http://localhost:8080", readypath: "/"},
		BrokerConfig{host: "localhost", port: "8080"},
		new(tls.Config))

	go h.RunHealthChecks()

	time.Sleep(100 * time.Millisecond)

	res, err := http.Get("http://localhost:8888/ready?full=1")
	if err != nil {
		log.Fatal(err)
	}
	_, err = io.ReadAll(res.Body)
	res.Body.Close()
	if err != nil {
		log.Fatal(err)
	}

	if res.StatusCode != http.StatusOK {
		t.Errorf("Response code was %v; want 200", res.StatusCode)
	}

	ts.Close()
}