mirror of
https://github.com/gofiber/fiber.git
synced 2025-02-22 22:03:39 +00:00
b74676704d
* Add disable html support to monitor middleware. * Change field. * Update README.md * Update middleware/monitor/config.go * Fix tests. Co-authored-by: hi019 <65871571+hi019@users.noreply.github.com>
131 lines
3.2 KiB
Go
131 lines
3.2 KiB
Go
package monitor
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/gofiber/fiber/v2/utils"
|
|
"github.com/valyala/fasthttp"
|
|
)
|
|
|
|
func Test_Monitor_405(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
app := fiber.New()
|
|
|
|
app.Use("/", New())
|
|
|
|
resp, err := app.Test(httptest.NewRequest(fiber.MethodPost, "/", nil))
|
|
utils.AssertEqual(t, nil, err)
|
|
utils.AssertEqual(t, 405, resp.StatusCode)
|
|
}
|
|
|
|
func Test_Monitor_Html(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
app := fiber.New()
|
|
|
|
app.Get("/", New())
|
|
|
|
resp, err := app.Test(httptest.NewRequest(fiber.MethodGet, "/", nil))
|
|
utils.AssertEqual(t, nil, err)
|
|
utils.AssertEqual(t, 200, resp.StatusCode)
|
|
utils.AssertEqual(t, fiber.MIMETextHTMLCharsetUTF8, resp.Header.Get(fiber.HeaderContentType))
|
|
|
|
b, err := ioutil.ReadAll(resp.Body)
|
|
utils.AssertEqual(t, nil, err)
|
|
utils.AssertEqual(t, true, bytes.Contains(b, []byte("<title>Fiber Monitor</title>")))
|
|
}
|
|
|
|
// go test -run Test_Monitor_JSON -race
|
|
func Test_Monitor_JSON(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
app := fiber.New()
|
|
|
|
app.Get("/", New())
|
|
|
|
req := httptest.NewRequest(fiber.MethodGet, "/", nil)
|
|
req.Header.Set(fiber.HeaderAccept, fiber.MIMEApplicationJSON)
|
|
resp, err := app.Test(req)
|
|
utils.AssertEqual(t, nil, err)
|
|
utils.AssertEqual(t, 200, resp.StatusCode)
|
|
utils.AssertEqual(t, fiber.MIMEApplicationJSON, resp.Header.Get(fiber.HeaderContentType))
|
|
|
|
b, err := ioutil.ReadAll(resp.Body)
|
|
utils.AssertEqual(t, nil, err)
|
|
utils.AssertEqual(t, true, bytes.Contains(b, []byte("pid")))
|
|
utils.AssertEqual(t, true, bytes.Contains(b, []byte("os")))
|
|
}
|
|
|
|
// go test -v -run=^$ -bench=Benchmark_Monitor -benchmem -count=4
|
|
func Benchmark_Monitor(b *testing.B) {
|
|
app := fiber.New()
|
|
|
|
app.Get("/", New())
|
|
|
|
h := app.Handler()
|
|
|
|
fctx := &fasthttp.RequestCtx{}
|
|
fctx.Request.Header.SetMethod("GET")
|
|
fctx.Request.SetRequestURI("/")
|
|
fctx.Request.Header.Set(fiber.HeaderAccept, fiber.MIMEApplicationJSON)
|
|
|
|
b.ReportAllocs()
|
|
b.ResetTimer()
|
|
|
|
b.RunParallel(func(pb *testing.PB) {
|
|
for pb.Next() {
|
|
h(fctx)
|
|
}
|
|
})
|
|
|
|
utils.AssertEqual(b, 200, fctx.Response.Header.StatusCode())
|
|
utils.AssertEqual(b,
|
|
fiber.MIMEApplicationJSON,
|
|
string(fctx.Response.Header.Peek(fiber.HeaderContentType)))
|
|
}
|
|
|
|
// go test -run Test_Monitor_Next
|
|
func Test_Monitor_Next(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
app := fiber.New()
|
|
|
|
app.Use("/", New(Config{
|
|
Next: func(_ *fiber.Ctx) bool {
|
|
return true
|
|
},
|
|
}))
|
|
|
|
resp, err := app.Test(httptest.NewRequest(fiber.MethodPost, "/", nil))
|
|
utils.AssertEqual(t, nil, err)
|
|
utils.AssertEqual(t, 404, resp.StatusCode)
|
|
}
|
|
|
|
// go test -run Test_Monitor_APIOnly -race
|
|
func Test_Monitor_APIOnly(t *testing.T) {
|
|
//t.Parallel()
|
|
|
|
app := fiber.New()
|
|
|
|
app.Get("/", New(Config{
|
|
APIOnly: true,
|
|
}))
|
|
|
|
req := httptest.NewRequest(fiber.MethodGet, "/", nil)
|
|
req.Header.Set(fiber.HeaderAccept, fiber.MIMEApplicationJSON)
|
|
resp, err := app.Test(req)
|
|
utils.AssertEqual(t, nil, err)
|
|
utils.AssertEqual(t, 200, resp.StatusCode)
|
|
utils.AssertEqual(t, fiber.MIMEApplicationJSON, resp.Header.Get(fiber.HeaderContentType))
|
|
|
|
b, err := ioutil.ReadAll(resp.Body)
|
|
utils.AssertEqual(t, nil, err)
|
|
utils.AssertEqual(t, true, bytes.Contains(b, []byte("pid")))
|
|
utils.AssertEqual(t, true, bytes.Contains(b, []byte("os")))
|
|
}
|