pages-server/server/handler_fasthttp_test.go

//go:build fasthttp

package server

import (
	"fmt"
	"testing"
	"time"

	"github.com/valyala/fasthttp"

	"codeberg.org/codeberg/pages/server/cache"
	"codeberg.org/codeberg/pages/server/gitea"
)

func TestHandlerPerformance(t *testing.T) {
	giteaRoot := "https://codeberg.org"
	giteaClient, _ := gitea.NewClient(giteaRoot, "", cache.NewKeyValueCache(), false, false)
	testHandler := Handler(
		"codeberg.page", "raw.codeberg.org",
		giteaClient,
		giteaRoot, "https://docs.codeberg.org/pages/raw-content/",
		[]string{"/.well-known/acme-challenge/"},
		[]string{"raw.codeberg.org", "fonts.codeberg.org", "design.codeberg.org"},
		cache.NewKeyValueCache(),
		cache.NewKeyValueCache(),
	)

	testCase := func(uri string, status int) {
		ctx := &fasthttp.RequestCtx{
			Request:  *fasthttp.AcquireRequest(),
			Response: *fasthttp.AcquireResponse(),
		}
		ctx.Request.SetRequestURI(uri)
		fmt.Printf("Start: %v\n", time.Now())
		start := time.Now()
		testHandler(ctx)
		end := time.Now()
		fmt.Printf("Done: %v\n", time.Now())
		if ctx.Response.StatusCode() != status {
			t.Errorf("request failed with status code %d", ctx.Response.StatusCode())
		} else {
			t.Logf("request took %d milliseconds", end.Sub(start).Milliseconds())
		}
	}

	testCase("https://mondstern.codeberg.page/", 424) // TODO: expect 200
	testCase("https://mondstern.codeberg.page/", 424) // TODO: expect 200
	testCase("https://example.momar.xyz/", 424)       // TODO: expect 200
	testCase("https://codeberg.page/", 424)           // TODO: expect 200
}