diff --git a/server/handler_test.go b/server/handler_test.go index 34aed7f..2ab9ea3 100644 --- a/server/handler_test.go +++ b/server/handler_test.go @@ -25,46 +25,25 @@ func TestHandlerPerformance(t *testing.T) { cache.NewKeyValueCache(), ) - ctx := &fasthttp.RequestCtx{ - Request: *fasthttp.AcquireRequest(), - Response: *fasthttp.AcquireResponse(), - } - ctx.Request.SetRequestURI("http://mondstern.codeberg.page/") - 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() != 200 || len(ctx.Response.Body()) < 2048 { - t.Errorf("request failed with status code %d and body length %d", ctx.Response.StatusCode(), len(ctx.Response.Body())) - } else { - t.Logf("request took %d milliseconds", end.Sub(start).Milliseconds()) + 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()) + } } - ctx.Response.Reset() - ctx.Response.ResetBody() - 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() != 200 || len(ctx.Response.Body()) < 2048 { - t.Errorf("request failed with status code %d and body length %d", ctx.Response.StatusCode(), len(ctx.Response.Body())) - } else { - t.Logf("request took %d milliseconds", end.Sub(start).Milliseconds()) - } - - ctx.Response.Reset() - ctx.Response.ResetBody() - ctx.Request.SetRequestURI("http://example.momar.xyz/") - 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() != 200 || len(ctx.Response.Body()) < 1 { - t.Errorf("request failed with status code %d and body length %d", ctx.Response.StatusCode(), len(ctx.Response.Body())) - } else { - t.Logf("request took %d milliseconds", end.Sub(start).Milliseconds()) - } + testCase("https://mondstern.codeberg.page/", 200) + testCase("https://mondstern.codeberg.page/", 200) + testCase("https://example.momar.xyz/", 200) }