From e4478bf22e07c46ab6f9faa3e1d91fda559a2a1b Mon Sep 17 00:00:00 2001
From: Aliaksandr Valialkin <valyala@gmail.com>
Date: Sun, 22 Nov 2015 02:41:27 +0200
Subject: [PATCH] Added tests for Header.Write

---
 header_timing_test.go | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/header_timing_test.go b/header_timing_test.go
index b49a678..9bc6c5c 100644
--- a/header_timing_test.go
+++ b/header_timing_test.go
@@ -58,6 +58,41 @@ func BenchmarkResponseHeaderRead(b *testing.B) {
 	})
 }
 
+func BenchmarkRequestHeaderWrite(b *testing.B) {
+	b.RunParallel(func (pb *testing.PB) {
+		var h RequestHeader
+		h.SetRequestURI("/foo/bar")
+		h.SetHost("foobar.com")
+		h.SetUserAgent("aaa.bbb")
+		h.SetReferer("http://google.com/aaa/bbb")
+		var w bytes.Buffer
+		for pb.Next() {
+			if _, err := h.WriteTo(&w); err != nil {
+				b.Fatalf("unexpected error when writing header: %s", err)
+			}
+			w.Reset()
+		}
+	})
+}
+
+func BenchmarkResponseHeaderWrite(b *testing.B) {
+	b.RunParallel(func (pb *testing.PB) {
+		var h ResponseHeader
+		h.SetStatusCode(200)
+		h.SetContentType("text/html")
+		h.SetContentLength(1256)
+		h.SetServer("aaa 1/2.3")
+		h.Set("Test", "1.2.3")
+		var w bytes.Buffer
+		for pb.Next() {
+			if _, err := h.WriteTo(&w); err != nil {
+				b.Fatalf("unexpected error when writing header: %s", err)
+			}
+			w.Reset()
+		}
+	})
+}
+
 func BenchmarkRequestHeaderPeekBytesCanonical(b *testing.B) {
 	b.RunParallel(func(pb *testing.PB) {
 		var h RequestHeader