Skip to content
This repository
branch: master
Fetching contributors…

Cannot retrieve contributors at this time

file 58 lines (46 sloc) 1.409 kb
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58
package mango

import (
"bytes"
"log"
"net/http"
"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()
}
Something went wrong with that request. Please try again.