162 lines
4.2 KiB
Go
162 lines
4.2 KiB
Go
package accesslog
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestCommonLogFormatter_Format(t *testing.T) {
|
|
clf := CommonLogFormatter{}
|
|
|
|
testCases := []struct {
|
|
name string
|
|
data map[string]interface{}
|
|
expectedLog string
|
|
}{
|
|
{
|
|
name: "DownstreamStatus & DownstreamContentSize are nil",
|
|
data: map[string]interface{}{
|
|
StartUTC: time.Date(2009, time.November, 10, 23, 0, 0, 0, time.UTC),
|
|
Duration: 123 * time.Second,
|
|
ClientHost: "10.0.0.1",
|
|
ClientUsername: "Client",
|
|
RequestMethod: http.MethodGet,
|
|
RequestPath: "/foo",
|
|
RequestProtocol: "http",
|
|
DownstreamStatus: nil,
|
|
DownstreamContentSize: nil,
|
|
RequestRefererHeader: "",
|
|
RequestUserAgentHeader: "",
|
|
RequestCount: 0,
|
|
RouterName: "",
|
|
ServiceURL: "",
|
|
},
|
|
expectedLog: `10.0.0.1 - Client [10/Nov/2009:23:00:00 +0000] "GET /foo http" - - "-" "-" 0 "-" "-" 123000ms
|
|
`,
|
|
},
|
|
{
|
|
name: "all data",
|
|
data: map[string]interface{}{
|
|
StartUTC: time.Date(2009, time.November, 10, 23, 0, 0, 0, time.UTC),
|
|
Duration: 123 * time.Second,
|
|
ClientHost: "10.0.0.1",
|
|
ClientUsername: "Client",
|
|
RequestMethod: http.MethodGet,
|
|
RequestPath: "/foo",
|
|
RequestProtocol: "http",
|
|
DownstreamStatus: 123,
|
|
DownstreamContentSize: 132,
|
|
RequestRefererHeader: "referer",
|
|
RequestUserAgentHeader: "agent",
|
|
RequestCount: nil,
|
|
RouterName: "foo",
|
|
ServiceURL: "http://10.0.0.2/toto",
|
|
},
|
|
expectedLog: `10.0.0.1 - Client [10/Nov/2009:23:00:00 +0000] "GET /foo http" 123 132 "referer" "agent" - "foo" "http://10.0.0.2/toto" 123000ms
|
|
`,
|
|
},
|
|
{
|
|
name: "all data with local time",
|
|
data: map[string]interface{}{
|
|
StartLocal: time.Date(2009, time.November, 10, 23, 0, 0, 0, time.UTC),
|
|
Duration: 123 * time.Second,
|
|
ClientHost: "10.0.0.1",
|
|
ClientUsername: "Client",
|
|
RequestMethod: http.MethodGet,
|
|
RequestPath: "/foo",
|
|
RequestProtocol: "http",
|
|
DownstreamStatus: 123,
|
|
DownstreamContentSize: 132,
|
|
RequestRefererHeader: "referer",
|
|
RequestUserAgentHeader: "agent",
|
|
RequestCount: nil,
|
|
RouterName: "foo",
|
|
ServiceURL: "http://10.0.0.2/toto",
|
|
},
|
|
expectedLog: `10.0.0.1 - Client [10/Nov/2009:14:00:00 -0900] "GET /foo http" 123 132 "referer" "agent" - "foo" "http://10.0.0.2/toto" 123000ms
|
|
`,
|
|
},
|
|
}
|
|
|
|
// Set timezone to Etc/GMT+9 to have a constant behavior
|
|
t.Setenv("TZ", "Etc/GMT+9")
|
|
|
|
for _, test := range testCases {
|
|
test := test
|
|
t.Run(test.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
entry := &logrus.Entry{Data: test.data}
|
|
|
|
raw, err := clf.Format(entry)
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, test.expectedLog, string(raw))
|
|
})
|
|
}
|
|
}
|
|
|
|
func Test_toLog(t *testing.T) {
|
|
testCases := []struct {
|
|
desc string
|
|
fields logrus.Fields
|
|
fieldName string
|
|
defaultValue string
|
|
quoted bool
|
|
expectedLog interface{}
|
|
}{
|
|
{
|
|
desc: "Should return int 1",
|
|
fields: logrus.Fields{
|
|
"Powpow": 1,
|
|
},
|
|
fieldName: "Powpow",
|
|
defaultValue: defaultValue,
|
|
quoted: false,
|
|
expectedLog: 1,
|
|
},
|
|
{
|
|
desc: "Should return string foo",
|
|
fields: logrus.Fields{
|
|
"Powpow": "foo",
|
|
},
|
|
fieldName: "Powpow",
|
|
defaultValue: defaultValue,
|
|
quoted: true,
|
|
expectedLog: `"foo"`,
|
|
},
|
|
{
|
|
desc: "Should return defaultValue if fieldName does not exist",
|
|
fields: logrus.Fields{
|
|
"Powpow": "foo",
|
|
},
|
|
fieldName: "",
|
|
defaultValue: defaultValue,
|
|
quoted: false,
|
|
expectedLog: "-",
|
|
},
|
|
{
|
|
desc: "Should return defaultValue if fields is nil",
|
|
fields: nil,
|
|
fieldName: "",
|
|
defaultValue: defaultValue,
|
|
quoted: false,
|
|
expectedLog: "-",
|
|
},
|
|
}
|
|
|
|
for _, test := range testCases {
|
|
test := test
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
lg := toLog(test.fields, test.fieldName, defaultValue, test.quoted)
|
|
|
|
assert.Equal(t, test.expectedLog, lg)
|
|
})
|
|
}
|
|
}
|