homebox/backend/pkgs/server/server_test.go

package server

import (
	"net/http"
	"sync/atomic"
	"testing"
	"time"

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

func testServer(t *testing.T, r http.Handler) *Server {
	svr := NewServer(WithHost("127.0.0.1"), WithPort("19245"))

	if r != nil {
		svr.mux.Mount("/", r)
	}
	go func() {
		err := svr.Start()
		assert.NoError(t, err)
	}()

	ping := func() error {
		_, err := http.Get("http://127.0.0.1:19245")
		return err
	}

	for {
		if err := ping(); err == nil {
			break
		}
		time.Sleep(time.Millisecond * 100)
	}

	return svr
}

func Test_ServerShutdown_Error(t *testing.T) {
	svr := NewServer(WithHost("127.0.0.1"), WithPort("19245"))

	err := svr.Shutdown("test")
	assert.ErrorIs(t, err, ErrServerNotStarted)
}

func Test_ServerStarts_Error(t *testing.T) {
	svr := testServer(t, nil)

	err := svr.Start()
	assert.ErrorIs(t, err, ErrServerAlreadyStarted)

	err = svr.Shutdown("test")
	assert.NoError(t, err)
}

func Test_ServerStarts(t *testing.T) {
	svr := testServer(t, nil)
	err := svr.Shutdown("test")
	assert.NoError(t, err)
}

func Test_GracefulServerShutdownWithWorkers(t *testing.T) {
	isFinished := false

	svr := testServer(t, nil)

	svr.Background(func() {
		time.Sleep(time.Second * 4)
		isFinished = true
	})

	err := svr.Shutdown("test")

	assert.NoError(t, err)
	assert.True(t, isFinished)
}

func Test_GracefulServerShutdownWithRequests(t *testing.T) {
	var isFinished atomic.Bool

	router := http.NewServeMux()

	// add long running handler func
	router.HandleFunc("/test", func(rw http.ResponseWriter, r *http.Request) {
		time.Sleep(time.Second * 3)
		isFinished.Store(true)
	})

	svr := testServer(t, router)

	// Make request to "/test"
	go func() {
		_, _ = http.Get("http://127.0.0.1:19245/test") // This is probably bad?
	}()

	time.Sleep(time.Second) // Hack to wait for the request to be made

	err := svr.Shutdown("test")
	assert.NoError(t, err)

	assert.True(t, isFinished.Load())
}