mirror of
https://github.com/kataras/iris.git
synced 2025-01-23 18:51:03 +01:00
56 lines
1.4 KiB
Go
56 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"strings"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/kataras/golog"
|
|
"github.com/kataras/iris/v12"
|
|
"github.com/kataras/iris/v12/httptest"
|
|
)
|
|
|
|
func TestJSONLogger(t *testing.T) {
|
|
iters := 500
|
|
|
|
out := new(bytes.Buffer)
|
|
|
|
app := iris.New()
|
|
app.Logger().SetTimeFormat("") // disable timestamps.
|
|
app.Logger().SetStacktraceLimit(1) // limit debug stacktrace to 1, show only the first caller.
|
|
app.Logger().SetOutput(out)
|
|
|
|
app.Logger().Handle(func(l *golog.Log) bool {
|
|
enc := json.NewEncoder(l.Logger.Printer) // you can change the output to a file as well.
|
|
err := enc.Encode(l)
|
|
return err == nil
|
|
})
|
|
|
|
app.Get("/ping", ping)
|
|
|
|
const expectedLogStr = `{"level":"debug","message":"Request path: /ping","fields":{"request_id":null},"stacktrace":[{"function":"json-logger/ping","source":"C:/mygopath/src/github.com/kataras/iris/_examples/logging/json-logger/main.go:82"}]}`
|
|
e := httptest.New(t, app, httptest.LogLevel("debug"))
|
|
wg := new(sync.WaitGroup)
|
|
wg.Add(iters)
|
|
for i := 0; i < iters; i++ {
|
|
go func() {
|
|
e.GET("/ping").Expect().Status(httptest.StatusOK).Body().Equal("pong")
|
|
wg.Done()
|
|
}()
|
|
}
|
|
|
|
wg.Wait()
|
|
expected := ""
|
|
for i := 0; i < iters; i++ {
|
|
expected += expectedLogStr + "\n"
|
|
}
|
|
|
|
got := out.String()
|
|
got = got[strings.Index(got, "{"):] // take only the json we care and after.
|
|
if expected != got {
|
|
t.Fatalf("expected:\n%s\nbut got:\n%s", expected, got)
|
|
}
|
|
}
|