mirror of
https://github.com/golang/go
synced 2024-10-07 11:21:22 -06:00
9ea0bd3986
R=golang-dev, gary.burd, rsc CC=golang-dev https://golang.org/cl/4526062
157 lines
3.6 KiB
Go
157 lines
3.6 KiB
Go
// Copyright 2010 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package http
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"json"
|
|
"os"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
|
|
var writeSetCookiesTests = []struct {
|
|
Cookies []*Cookie
|
|
Raw string
|
|
}{
|
|
{
|
|
[]*Cookie{
|
|
&Cookie{Name: "cookie-1", Value: "v$1"},
|
|
&Cookie{Name: "cookie-2", Value: "two", MaxAge: 3600},
|
|
&Cookie{Name: "cookie-3", Value: "three", Domain: ".example.com"},
|
|
&Cookie{Name: "cookie-4", Value: "four", Path: "/restricted/"},
|
|
},
|
|
"Set-Cookie: cookie-1=v$1\r\n" +
|
|
"Set-Cookie: cookie-2=two; Max-Age=3600\r\n" +
|
|
"Set-Cookie: cookie-3=three; Domain=.example.com\r\n" +
|
|
"Set-Cookie: cookie-4=four; Path=/restricted/\r\n",
|
|
},
|
|
}
|
|
|
|
func TestWriteSetCookies(t *testing.T) {
|
|
for i, tt := range writeSetCookiesTests {
|
|
var w bytes.Buffer
|
|
writeSetCookies(&w, tt.Cookies)
|
|
seen := string(w.Bytes())
|
|
if seen != tt.Raw {
|
|
t.Errorf("Test %d, expecting:\n%s\nGot:\n%s\n", i, tt.Raw, seen)
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
|
|
type headerOnlyResponseWriter Header
|
|
|
|
func (ho headerOnlyResponseWriter) Header() Header {
|
|
return Header(ho)
|
|
}
|
|
|
|
func (ho headerOnlyResponseWriter) Write([]byte) (int, os.Error) {
|
|
panic("NOIMPL")
|
|
}
|
|
|
|
func (ho headerOnlyResponseWriter) WriteHeader(int) {
|
|
panic("NOIMPL")
|
|
}
|
|
|
|
func TestSetCookie(t *testing.T) {
|
|
m := make(Header)
|
|
SetCookie(headerOnlyResponseWriter(m), &Cookie{Name: "cookie-1", Value: "one", Path: "/restricted/"})
|
|
SetCookie(headerOnlyResponseWriter(m), &Cookie{Name: "cookie-2", Value: "two", MaxAge: 3600})
|
|
if l := len(m["Set-Cookie"]); l != 2 {
|
|
t.Fatalf("expected %d cookies, got %d", 2, l)
|
|
}
|
|
if g, e := m["Set-Cookie"][0], "cookie-1=one; Path=/restricted/"; g != e {
|
|
t.Errorf("cookie #1: want %q, got %q", e, g)
|
|
}
|
|
if g, e := m["Set-Cookie"][1], "cookie-2=two; Max-Age=3600"; g != e {
|
|
t.Errorf("cookie #2: want %q, got %q", e, g)
|
|
}
|
|
}
|
|
|
|
var writeCookiesTests = []struct {
|
|
Cookies []*Cookie
|
|
Raw string
|
|
}{
|
|
{
|
|
[]*Cookie{},
|
|
"",
|
|
},
|
|
{
|
|
[]*Cookie{&Cookie{Name: "cookie-1", Value: "v$1"}},
|
|
"Cookie: cookie-1=v$1\r\n",
|
|
},
|
|
{
|
|
[]*Cookie{
|
|
&Cookie{Name: "cookie-1", Value: "v$1"},
|
|
&Cookie{Name: "cookie-2", Value: "v$2"},
|
|
&Cookie{Name: "cookie-3", Value: "v$3"},
|
|
},
|
|
"Cookie: cookie-1=v$1; cookie-2=v$2; cookie-3=v$3\r\n",
|
|
},
|
|
}
|
|
|
|
func TestWriteCookies(t *testing.T) {
|
|
for i, tt := range writeCookiesTests {
|
|
var w bytes.Buffer
|
|
writeCookies(&w, tt.Cookies)
|
|
seen := string(w.Bytes())
|
|
if seen != tt.Raw {
|
|
t.Errorf("Test %d, expecting:\n%s\nGot:\n%s\n", i, tt.Raw, seen)
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
|
|
var readSetCookiesTests = []struct {
|
|
Header Header
|
|
Cookies []*Cookie
|
|
}{
|
|
{
|
|
Header{"Set-Cookie": {"Cookie-1=v$1"}},
|
|
[]*Cookie{&Cookie{Name: "Cookie-1", Value: "v$1", Raw: "Cookie-1=v$1"}},
|
|
},
|
|
}
|
|
|
|
func toJSON(v interface{}) string {
|
|
b, err := json.Marshal(v)
|
|
if err != nil {
|
|
return fmt.Sprintf("%#v", v)
|
|
}
|
|
return string(b)
|
|
}
|
|
|
|
func TestReadSetCookies(t *testing.T) {
|
|
for i, tt := range readSetCookiesTests {
|
|
c := readSetCookies(tt.Header)
|
|
if !reflect.DeepEqual(c, tt.Cookies) {
|
|
t.Errorf("#%d readSetCookies: have\n%s\nwant\n%s\n", i, toJSON(c), toJSON(tt.Cookies))
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
|
|
var readCookiesTests = []struct {
|
|
Header Header
|
|
Cookies []*Cookie
|
|
}{
|
|
{
|
|
Header{"Cookie": {"Cookie-1=v$1"}},
|
|
[]*Cookie{&Cookie{Name: "Cookie-1", Value: "v$1"}},
|
|
},
|
|
}
|
|
|
|
func TestReadCookies(t *testing.T) {
|
|
for i, tt := range readCookiesTests {
|
|
c := readCookies(tt.Header)
|
|
if !reflect.DeepEqual(c, tt.Cookies) {
|
|
t.Errorf("#%d readCookies: have\n%s\nwant\n%s\n", i, toJSON(c), toJSON(tt.Cookies))
|
|
continue
|
|
}
|
|
}
|
|
}
|