mirror of
https://github.com/pomerium/pomerium.git
synced 2025-04-29 18:36:30 +02:00
49 lines
1.9 KiB
Go
49 lines
1.9 KiB
Go
package envoy
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"regexp"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/rs/zerolog"
|
|
)
|
|
|
|
func TestServer_handleLogs(t *testing.T) {
|
|
logFormatRE := regexp.MustCompile(`^[[]LOG_FORMAT[]](.*?)--(.*?)--(.*?)$`)
|
|
line := "[LOG_FORMAT]debug--filter--[external/envoy/source/extensions/filters/listener/tls_inspector/tls_inspector.cc:78] tls inspector: new connection accepted"
|
|
old := func(s string) string {
|
|
msg := s
|
|
parts := logFormatRE.FindStringSubmatch(s)
|
|
if len(parts) == 4 {
|
|
msg = parts[3]
|
|
}
|
|
return msg
|
|
}
|
|
srv := &Server{}
|
|
expectedMsg := old(line)
|
|
name, logLevel, gotMsg := srv.parseLog(line)
|
|
if name != "filter" {
|
|
t.Errorf("unexpected name, want filter, got: %s", name)
|
|
}
|
|
if logLevel != "debug" {
|
|
t.Errorf("unexpected log level, want debug, got: %s", logLevel)
|
|
}
|
|
if gotMsg != expectedMsg {
|
|
t.Errorf("unexpected msg, want %s, got: %s", expectedMsg, gotMsg)
|
|
}
|
|
}
|
|
|
|
func Benchmark_handleLogs(b *testing.B) {
|
|
line := `[LOG_FORMAT]debug--http--[external/envoy/source/common/http/conn_manager_impl.cc:781] [C25][S14758077654018620250] request headers complete (end_stream=false):\\n\\':authority\\', \\'enabled-ws-echo.localhost.pomerium.io\\'\\n\\':path\\', \\'/\\'\\n\\':method\\', \\'GET\\'\\n\\'upgrade\\', \\'websocket\\'\\n\\'connection\\', \\'upgrade\\'\\n\\'x-request-id\\', \\'30ac7726e0b9e00a9c9ab2bf66d692ac\\'\\n\\'x-real-ip\\', \\'172.17.0.1\\'\\n\\'x-forwarded-for\\', \\'172.17.0.1\\'\\n\\'x-forwarded-host\\', \\'enabled-ws-echo.localhost.pomerium.io\\'\\n\\'x-forwarded-port\\', \\'443\\'\\n\\'x-forwarded-proto\\', \\'https\\'\\n\\'x-scheme\\', \\'https\\'\\n\\'user-agent\\', \\'Go-http-client/1.1\\'\\n\\'sec-websocket-key\\', \\'4bh7+YFVzrJiblaSu/CVfg==\\'\\n\\'sec-websocket-version\\', \\'13\\'`
|
|
rc := io.NopCloser(strings.NewReader(line))
|
|
srv := &Server{}
|
|
zerolog.SetGlobalLevel(zerolog.InfoLevel)
|
|
|
|
b.ResetTimer()
|
|
b.ReportAllocs()
|
|
for i := 0; i < b.N; i++ {
|
|
srv.handleLogs(context.Background(), rc)
|
|
}
|
|
}
|