Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
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.