feature: accesslogger

internal/net/http/accesslog/fields_test.go

@@ -0,0 +1,72 @@
+package accesslog_test
+
+import (
+	"testing"
+
+	. "github.com/yusing/go-proxy/internal/net/http/accesslog"
+	. "github.com/yusing/go-proxy/internal/utils/testing"
+)
+
+// Cookie header should be removed,
+// stored in JSONLogEntry.Cookies instead.
+func TestAccessLoggerJSONKeepHeaders(t *testing.T) {
+	config := DefaultConfig
+	config.Fields.Headers.DefaultMode = FieldModeKeep
+	entry := getJSONEntry(t, &config)
+	ExpectDeepEqual(t, len(entry.Headers["Cookie"]), 0)
+	for k, v := range req.Header {
+		if k != "Cookie" {
+			ExpectDeepEqual(t, entry.Headers[k], v)
+		}
+	}
+}
+
+func TestAccessLoggerJSONRedactHeaders(t *testing.T) {
+	config := DefaultConfig
+	config.Fields.Headers.DefaultMode = FieldModeRedact
+	entry := getJSONEntry(t, &config)
+	ExpectDeepEqual(t, len(entry.Headers["Cookie"]), 0)
+	for k := range req.Header {
+		if k != "Cookie" {
+			ExpectDeepEqual(t, entry.Headers[k], []string{RedactedValue})
+		}
+	}
+}
+
+func TestAccessLoggerJSONKeepCookies(t *testing.T) {
+	config := DefaultConfig
+	config.Fields.Headers.DefaultMode = FieldModeKeep
+	config.Fields.Cookies.DefaultMode = FieldModeKeep
+	entry := getJSONEntry(t, &config)
+	ExpectDeepEqual(t, len(entry.Headers["Cookie"]), 0)
+	for _, cookie := range req.Cookies() {
+		ExpectEqual(t, entry.Cookies[cookie.Name], cookie.Value)
+	}
+}
+
+func TestAccessLoggerJSONRedactCookies(t *testing.T) {
+	config := DefaultConfig
+	config.Fields.Headers.DefaultMode = FieldModeKeep
+	config.Fields.Cookies.DefaultMode = FieldModeRedact
+	entry := getJSONEntry(t, &config)
+	ExpectDeepEqual(t, len(entry.Headers["Cookie"]), 0)
+	for _, cookie := range req.Cookies() {
+		ExpectEqual(t, entry.Cookies[cookie.Name], RedactedValue)
+	}
+}
+
+func TestAccessLoggerJSONDropQuery(t *testing.T) {
+	config := DefaultConfig
+	config.Fields.Query.DefaultMode = FieldModeDrop
+	entry := getJSONEntry(t, &config)
+	ExpectDeepEqual(t, entry.Query["foo"], nil)
+	ExpectDeepEqual(t, entry.Query["bar"], nil)
+}
+
+func TestAccessLoggerJSONRedactQuery(t *testing.T) {
+	config := DefaultConfig
+	config.Fields.Query.DefaultMode = FieldModeRedact
+	entry := getJSONEntry(t, &config)
+	ExpectDeepEqual(t, entry.Query["foo"], []string{RedactedValue})
+	ExpectDeepEqual(t, entry.Query["bar"], []string{RedactedValue})
+}