use strings.Cut, add unit tests

internal/controlplane/grpc_accesslog_test.go

@@ -0,0 +1,74 @@
+package controlplane
+
+import (
+	"bytes"
+	"strings"
+	"testing"
+	"time"
+
+	envoy_config_core_v3 "github.com/envoyproxy/go-control-plane/envoy/config/core/v3"
+	envoy_data_accesslog_v3 "github.com/envoyproxy/go-control-plane/envoy/data/accesslog/v3"
+	"github.com/rs/zerolog"
+	"github.com/stretchr/testify/assert"
+	"google.golang.org/protobuf/types/known/durationpb"
+	"google.golang.org/protobuf/types/known/wrapperspb"
+
+	"github.com/pomerium/pomerium/internal/log"
+)
+
+func Test_populateLogEvent(t *testing.T) {
+	t.Parallel()
+
+	entry := &envoy_data_accesslog_v3.HTTPAccessLogEntry{
+		CommonProperties: &envoy_data_accesslog_v3.AccessLogCommon{
+			TimeToLastDownstreamTxByte: durationpb.New(time.Second * 3),
+			UpstreamCluster:            "UPSTREAM-CLUSTER",
+		},
+		Request: &envoy_data_accesslog_v3.HTTPRequestProperties{
+			Authority:     "AUTHORITY",
+			ForwardedFor:  "FORWARDED-FOR",
+			Path:          "https://www.example.com/some/path?a=b",
+			Referer:       "https://www.example.com/referer?a=b",
+			RequestId:     "REQUEST-ID",
+			RequestMethod: envoy_config_core_v3.RequestMethod_GET,
+			UserAgent:     "USER-AGENT",
+		},
+		Response: &envoy_data_accesslog_v3.HTTPResponseProperties{
+			ResponseBodyBytes:   1234,
+			ResponseCode:        wrapperspb.UInt32(200),
+			ResponseCodeDetails: "RESPONSE-CODE-DETAILS",
+		},
+	}
+
+	for _, tc := range []struct {
+		field  log.AccessLogField
+		expect string
+	}{
+		{log.AccessLogFieldAuthority, `{"authority":"AUTHORITY"}`},
+		{log.AccessLogFieldDuration, `{"duration":3000}`},
+		{log.AccessLogFieldForwardedFor, `{"forwarded-for":"FORWARDED-FOR"}`},
+		{log.AccessLogFieldMethod, `{"method":"GET"}`},
+		{log.AccessLogFieldPath, `{"path":"https://www.example.com/some/path"}`},
+		{log.AccessLogFieldQuery, `{"query":"a=b"}`},
+		{log.AccessLogFieldReferer, `{"referer":"https://www.example.com/referer"}`},
+		{log.AccessLogFieldRequestID, `{"request-id":"REQUEST-ID"}`},
+		{log.AccessLogFieldResponseCode, `{"response-code":200}`},
+		{log.AccessLogFieldResponseCodeDetails, `{"response-code-details":"RESPONSE-CODE-DETAILS"}`},
+		{log.AccessLogFieldSize, `{"size":1234}`},
+		{log.AccessLogFieldUpstreamCluster, `{"upstream-cluster":"UPSTREAM-CLUSTER"}`},
+		{log.AccessLogFieldUserAgent, `{"user-agent":"USER-AGENT"}`},
+	} {
+		tc := tc
+		t.Run(string(tc.field), func(t *testing.T) {
+			t.Parallel()
+
+			var buf bytes.Buffer
+			log := zerolog.New(&buf)
+			evt := log.Log()
+			evt = populateLogEvent(tc.field, evt, entry)
+			evt.Send()
+
+			assert.Equal(t, tc.expect, strings.TrimSpace(buf.String()))
+		})
+	}
+}