watchdog/checks_test.go

package watchdog_test

import (
	"net"
	"net/http"
	"net/http/httptest"
	"testing"
	"time"

	"code.uint32.ru/tiny/watchdog"
)

func TestGetHTTPSuccess(t *testing.T) {
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("OK"))
	}))

	addr := srv.URL

	fn, err := watchdog.GetHTTP(addr, time.Second)
	if err != nil {
		t.Fatal(err)
	}

	err = fn(t.Context())
	if err != nil {
		t.Fatal(err)
	}
}

func TestGetHTTPError(t *testing.T) {
	addr := "https://127.0.0.1:42014"
	fn, err := watchdog.GetHTTP(addr, time.Second)
	if err != nil {
		t.Fatal(err)
	}

	err = fn(t.Context())
	if err == nil {
		t.Errorf("incorrect status for unavalable host")
	}
}

func TestHeadHTTPSuccess(t *testing.T) {
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusOK)
	}))

	addr := srv.URL

	fn, err := watchdog.HeadHTTP(addr, time.Second)
	if err != nil {
		t.Fatal(err)
	}

	err = fn(t.Context())
	if err != nil {
		t.Fatal(err)
	}
}

func TestHeadHTTPError(t *testing.T) {
	addr := "https://127.0.0.1:42014"
	fn, err := watchdog.HeadHTTP(addr, time.Second)
	if err != nil {
		t.Fatal(err)
	}

	err = fn(t.Context())
	if err == nil {
		t.Errorf("incorrect status for unavalable host")
	}
}

func TestDialTCPSuccess(t *testing.T) {
	addr := "127.0.0.1:42013"
	lis, err := net.Listen("tcp", addr)
	if err != nil {
		t.Fatal(err)
	}

	go func() {
		conn, err := lis.Accept()
		if err != nil {
			t.Error(err)
		}
		<-t.Context().Done()
		conn.Close()
	}()

	fn, err := watchdog.DialTCP(addr, time.Second)
	if err != nil {
		t.Fatal(err)
	}

	err = fn(t.Context())
	if err != nil {
		t.Fatal(err)
	}
}

func TestDialTCPError(t *testing.T) {
	addr := "127.0.0.1:65535"
	// check for non-existent addr
	fn, err := watchdog.DialTCP(addr, time.Second)
	if err != nil {
		t.Fatal(err)
	}

	err = fn(t.Context())
	if err == nil {
		t.Errorf("incorrect status")
	}
}