func main() {
headers := make(map[string][]string)
headers["first"] = []string{"first-value"}
headers["second"] = []string{"second-value", "second-second-value"}
router := mux.NewRouter()
router.Handle("/", headerhandler.New(&simpleHandler{}, headers))
err := http.ListenAndServe("localhost:8080", router)
if err != nil {
log.Fatal(err)
}
}
"net/http"
"net/http/httptest"
)
var _ = Describe("HeaderHandler", func() {
It("adds headers then allows its inner handler to continue processing", func() {
headers := make(map[string][]string)
headers["first"] = []string{"first-value"}
headers["second"] = []string{"second-value", "second-second-value"}
innerHandler := mockHandler{
responses: make(chan http.ResponseWriter, 10),
requests: make(chan *http.Request, 10),
}
handler := headerhandler.New(&innerHandler, headers)
response := httptest.NewRecorder()
request := &http.Request{
RequestURI: "/index.html",
}
handler.ServeHTTP(response, request)
Expect(innerHandler.responses).To(Receive(Equal(response)))
Expect(innerHandler.requests).To(Receive(Equal(request)))
Expect(response.Header().Get("first")).To(Equal("first-value"))
Expect(response.Header().Get("second")).To(Equal("second-value"))
})
})