git.iko.soy/typeable/oauth2-proxy
on 3155ada help sign in
typeable/oauth2-proxy/logging_handler_test.go
52 lines 1.2 kB history raw
1package main
3import (
4 "bytes"
5 "fmt"
6 "net/http"
7 "net/http/httptest"
8 "strings"
9 "testing"
10 "time"
12 "github.com/pusher/oauth2_proxy/logger"
15func TestLoggingHandler_ServeHTTP(t *testing.T) {
16 ts := time.Now()
18 tests := []struct {
19 Format,
20 ExpectedLogMessage string
21 }{
22 {logger.DefaultRequestLoggingFormat, fmt.Sprintf("127.0.0.1 - - [%s] test-server GET - \"/foo/bar\" HTTP/1.1 \"\" 200 4 0.000\n", logger.FormatTimestamp(ts))},
23 {"{{.RequestMethod}}", "GET\n"},
24 }
26 for _, test := range tests {
27 buf := bytes.NewBuffer(nil)
28 handler := func(w http.ResponseWriter, req *http.Request) {
29 _, ok := w.(http.Hijacker)
30 if !ok {
31 t.Error("http.Hijacker is not available")
32 }
34 w.Write([]byte("test"))
35 }
37 logger.SetOutput(buf)
38 logger.SetReqTemplate(test.Format)
39 h := LoggingHandler(http.HandlerFunc(handler))
41 r, _ := http.NewRequest("GET", "/foo/bar", nil)
42 r.RemoteAddr = "127.0.0.1"
43 r.Host = "test-server"
45 h.ServeHTTP(httptest.NewRecorder(), r)
47 actual := buf.String()
48 if !strings.Contains(actual, test.ExpectedLogMessage) {
49 t.Errorf("Log message was\n%s\ninstead of matching \n%s", actual, test.ExpectedLogMessage)
50 }
51 }