Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with HTTPS or Subversion.

Download ZIP
Fetching contributors…

Cannot retrieve contributors at this time

59 lines (46 sloc) 1.409 kb
package mango
import (
"bytes"
"net/http"
"log"
"testing"
)
var loggerBuffer = &bytes.Buffer{}
func loggerTestServer(env Env) (Status, Headers, Body) {
env.Logger().Println("Never gonna give you up")
return 200, Headers{}, Body("Hello World!")
}
func TestLogger(t *testing.T) {
// Compile the stack
loggerStack := new(Stack)
custom_logger := log.New(loggerBuffer, "prefixed:", 0)
loggerStack.Middleware(Logger(custom_logger))
loggerApp := loggerStack.Compile(loggerTestServer)
// Request against it
request, err := http.NewRequest("GET", "http://localhost:3000/", nil)
status, _, _ := loggerApp(Env{"mango.request": &Request{request}})
if err != nil {
t.Error(err)
}
if status != 200 {
t.Error("Expected status to equal 200, got:", status)
}
expected := "prefixed:Never gonna give you up\n"
if loggerBuffer.String() != expected {
t.Error("Expected logger to print: \"", expected, "\" got: \"", loggerBuffer.String(), "\"")
}
}
func BenchmarkLogger(b *testing.B) {
b.StopTimer()
loggerStack := new(Stack)
custom_logger := log.New(loggerBuffer, "prefixed:", 0)
loggerStack.Middleware(Logger(custom_logger))
loggerApp := loggerStack.Compile(loggerTestServer)
// Request against it
request, _ := http.NewRequest("GET", "http://localhost:3000/", nil)
b.StartTimer()
for i := 0; i < b.N; i++ {
loggerApp(Env{"mango.request": &Request{request}})
}
b.StopTimer()
}
Jump to Line
Something went wrong with that request. Please try again.