Add test for SetHeader

hoisie · hoisie · commit 4692e21b9315 · 2013-04-20T17:43:52.000-07:00
diff --git a/web_test.go b/web_test.go
@@ -109,6 +109,14 @@ func getTestResponse(method string, path string, body string, headers map[string
     return buildTestResponse(&buf)
 }
 
+func testGet(path string, headers map[string]string) *testResponse {
+    var header http.Header
+    for k,v := range headers {
+        header.Set(k, v)
+    }
+    return getTestResponse("GET", path, "", header, nil)
+}
+
 type Test struct {
     method         string
     path           string
@@ -204,6 +212,11 @@ func init() {
         ctx.SetHeader("Access-Control-Max-Age", "1000", true)
         ctx.WriteHeader(200)
     })
+
+    Get("/dupeheader", func(ctx *Context) string {
+        ctx.SetHeader("Server", "myserver", true)
+        return ""
+    })
 }
 
 var tests = []Test{
@@ -534,3 +547,14 @@ func TestSlug(t *testing.T) {
         }
     }
 }
+
+// tests that we don't duplicate headers
+func TestDuplicateHeader(t *testing.T) {
+    resp := testGet("/dupeheader", nil)
+    if len(resp.headers["Server"]) > 1 {
+        t.Fatalf("Expected only one header, got %#v", resp.headers["Server"])
+    }
+    if resp.headers["Server"][0] != "myserver" {
+        t.Fatalf("Incorrect header, exp 'myserver', got %q", resp.headers["Server"][0])
+    }
+}
