mux_test.go

package security_test

import (
	"context"
	"embed"
	"net/http/httptest"
	"testing"

	"github.com/andygeiss/cloud-native-utils/assert"
	"github.com/andygeiss/cloud-native-utils/security"
)

//go:embed testdata
var efs embed.FS

func TestServeMux_Is_Not_Nil(t *testing.T) {
	ctx := context.Background()
	mux, _ := security.NewServeMux(ctx, efs)
	assert.That(t, "mux must not be nil", mux != nil, true)
}

func TestServeMux_Has_Health_Check(t *testing.T) {
	ctx := context.Background()
	mux, _ := security.NewServeMux(ctx, efs)
	req := httptest.NewRequest("GET", "/liveness", nil)
	w := httptest.NewRecorder()
	mux.ServeHTTP(w, req)
	assert.That(t, "status code must be 200", w.Code, 200)
	assert.That(t, "body must be OK", w.Body.String(), "OK")
}

func TestServeMux_Has_Readiness_Check_When_Context_Active(t *testing.T) {
	ctx := context.Background()
	mux, _ := security.NewServeMux(ctx, efs)
	req := httptest.NewRequest("GET", "/readiness", nil)
	w := httptest.NewRecorder()
	mux.ServeHTTP(w, req)
	assert.That(t, "status code must be 200", w.Code, 200)
	// The body is empty in this example, but you can also check it if needed.
}

func TestServeMux_Has_Readiness_Check_When_Context_Canceled(t *testing.T) {
	ctx, cancel := context.WithCancel(context.Background())
	cancel() // Immediately cancel the context.
	mux, _ := security.NewServeMux(ctx, efs)
	req := httptest.NewRequest("GET", "/readiness", nil)
	w := httptest.NewRecorder()
	mux.ServeHTTP(w, req)
	assert.That(t, "status code must be 503", w.Code, 503)
}

func TestServeMux_Unknown_Route(t *testing.T) {
	ctx := context.Background()
	mux, _ := security.NewServeMux(ctx, efs)
	req := httptest.NewRequest("GET", "/unknown", nil)
	w := httptest.NewRecorder()
	mux.ServeHTTP(w, req)
	assert.That(t, "status code must be 404", w.Code, 404)
}

func TestServeMux_Has_Static_Assets(t *testing.T) {
	ctx := context.Background()
	mux, _ := security.NewServeMux(ctx, efs)
	req := httptest.NewRequest("GET", "/testdata/server.crt", nil)
	w := httptest.NewRecorder()
	mux.ServeHTTP(w, req)
	assert.That(t, "status code must be 200", w.Code, 200)
	assert.That(t, "body length must be correct", len(w.Body.String()), 1598)
}