Merge pull request #19 from slightfoot/master

blaubaer · web-flow · commit ce8aa6e5e77a · 2017-07-23T13:13:58.000+02:00
Fixed header test cases.
diff --git a/responseWriterWrapper.go b/responseWriterWrapper.go
@@ -19,9 +19,9 @@ func newResponseWriterWrapperFor(delegate http.ResponseWriter, beforeFirstWrite
 		maximumBufferSize:   -1,
 		header:              http.Header{},
 	}
-	for key, values := range delegate.header {
+	for key, values := range delegate.Header() {
 		for _, value := range values {
-			w.Header().Add(key, value)
+			wrapper.header.Add(key, value)
 		}
 	}
 	return wrapper
diff --git a/responseWriterWrapper_test.go b/responseWriterWrapper_test.go
@@ -38,8 +38,8 @@ func (s *responseWriterWrapperTest) Test_Header(c *C) {
 
 	wrapper.Header().Del("a")
 	wrapper.Header().Add("c", "3")
-	c.Assert(original.header.Get("a"), Equals, "")
-	c.Assert(original.header.Get("c"), Equals, "3")
+	c.Assert(wrapper.Header().Get("a"), Equals, "")
+	c.Assert(wrapper.Header().Get("c"), Equals, "3")
 }
 
 func (s *responseWriterWrapperTest) Test_WriteHeader(c *C) {

Original file line number	Diff line number	Diff line change
`@@ -19,9 +19,9 @@ func newResponseWriterWrapperFor(delegate http.ResponseWriter, beforeFirstWrite`
`19`	`19`	`maximumBufferSize: -1,`
`20`	`20`	`header: http.Header{},`
`21`	`21`	`}`
`22`		`- for key, values := range delegate.header {`
	`22`	`+ for key, values := range delegate.Header() {`
`23`	`23`	`for _, value := range values {`
`24`		`- w.Header().Add(key, value)`
	`24`	`+ wrapper.header.Add(key, value)`
`25`	`25`	`}`
`26`	`26`	`}`
`27`	`27`	`return wrapper`
Original file line number	Diff line number	Diff line change
`@@ -38,8 +38,8 @@ func (s responseWriterWrapperTest) Test_Header(c C) {`
`38`	`38`
`39`	`39`	`wrapper.Header().Del("a")`
`40`	`40`	`wrapper.Header().Add("c", "3")`
`41`		`- c.Assert(original.header.Get("a"), Equals, "")`
`42`		`- c.Assert(original.header.Get("c"), Equals, "3")`
	`41`	`+ c.Assert(wrapper.Header().Get("a"), Equals, "")`
	`42`	`+ c.Assert(wrapper.Header().Get("c"), Equals, "3")`
`43`	`43`	`}`
`44`	`44`
`45`	`45`	`func (s responseWriterWrapperTest) Test_WriteHeader(c C) {`