pomerium/internal/controlplane/grpc_accesslog_test.go
Caleb Doxsey c47055bece
upgrade to go v1.24 (#5562)
* upgrade to go v1.24

* add a macOS-specific //nolint comment too

---------

Co-authored-by: Kenneth Jenkins <51246568+kenjenkins@users.noreply.github.com>
2025-04-02 15:53:09 -06:00

81 lines
2.7 KiB
Go

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{
DownstreamRemoteAddress: &envoy_config_core_v3.Address{
Address: &envoy_config_core_v3.Address_SocketAddress{
SocketAddress: &envoy_config_core_v3.SocketAddress{
Address: "127.0.0.1",
},
},
},
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.AccessLogFieldIP, `{"ip":"127.0.0.1"}`},
{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"}`},
} {
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()))
})
}
}