contrib/fasthttp
1
0
Fork 0
mirror of https://github.com/valyala/fasthttp.git synced 2026-06-24 17:35:53 +03:00
Code Issues Packages Projects Releases Wiki Activity

Copy cookies in headers' CopyTo() method

Browse Source
This commit is contained in:
Aliaksandr Valialkin
2015-11-10 17:39:28 +02:00
parent d4e9c2ea22
commit 8c92324ef2
2 changed files with 46 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
header.go
+2
View File
@@ -128,6 +128,7 @@ func (h *ResponseHeader) CopyTo(dst *ResponseHeader) {
dst.contentType = append(dst.contentType[:0], h.contentType...)
dst.server = append(dst.server[:0], h.server...)
dst.h = copyArgs(dst.h, h.h)
dst.cookies = copyArgs(dst.cookies, h.cookies)
}
// CopyTo copies all the headers to dst.
@@ -140,6 +141,7 @@ func (h *RequestHeader) CopyTo(dst *RequestHeader) {
dst.contentType = append(dst.contentType[:0], h.contentType...)
dst.userAgent = append(dst.userAgent[:0], h.userAgent...)
dst.h = copyArgs(dst.h, h.h)
dst.cookies = copyArgs(dst.cookies, h.cookies)
}
// VisitAll calls f for each header except Content-Length.
header_test.go
+44
View File
@@ -10,6 +10,50 @@ import (
"testing"
)
func TestResponseHeaderCopyTo(t *testing.T) {
var h ResponseHeader
h.Set("Set-Cookie", "foo=bar")
h.Set("Content-Type", "foobar")
h.Set("AAA-BBB", "aaaa")
var h1 ResponseHeader
h.CopyTo(&h1)
if h1.Get("Set-cookie") != h.Get("Set-Cookie") {
t.Fatalf("unexpected cookie %q. Expected %q", h1.Get("set-cookie"), h.Get("set-cookie"))
}
if h1.Get("Content-Type") != h.Get("Content-Type") {
t.Fatalf("unexpected content-type %q. Expected %q", h1.Get("content-type"), h.Get("content-type"))
}
if h1.Get("aaa-bbb") != h.Get("AAA-BBB") {
t.Fatalf("unexpected aaa-bbb %q. Expected %q", h1.Get("aaa-bbb"), h.Get("aaa-bbb"))
}
}
func TestRequestHeaderCopyTo(t *testing.T) {
var h RequestHeader
h.Set("Cookie", "aa=bb; cc=dd")
h.Set("Content-Type", "foobar")
h.Set("Host", "aaaa")
h.Set("aaaxxx", "123")
var h1 RequestHeader
h.CopyTo(&h1)
if h1.Get("cookie") != h.Get("Cookie") {
t.Fatalf("unexpected cookie after copying: %q. Expected %q", h1.Get("cookie"), h.Get("cookie"))
}
if h1.Get("content-type") != h.Get("Content-Type") {
t.Fatalf("unexpected content-type %q. Expected %q", h1.Get("content-type"), h.Get("content-type"))
}
if h1.Get("host") != h.Get("host") {
t.Fatalf("unexpected host %q. Expected %q", h1.Get("host"), h.Get("host"))
}
if h1.Get("aaaxxx") != h.Get("aaaxxx") {
t.Fatalf("unexpected aaaxxx %q. Expected %q", h1.Get("aaaxxx"), h.Get("aaaxxx"))
}
}
func TestRequestHeaderSetCookie(t *testing.T) {
var h RequestHeader