mirror of
https://github.com/golang/go
synced 2024-10-06 08:21:21 -06:00
97 lines
2.0 KiB
Go
97 lines
2.0 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"
|
||
|
"reflect"
|
||
|
"testing"
|
||
|
)
|
||
|
|
||
|
|
||
|
var writeSetCookiesTests = []struct {
|
||
|
Cookies []*Cookie
|
||
|
Raw string
|
||
|
}{
|
||
|
{
|
||
|
[]*Cookie{&Cookie{Name: "cookie-1", Value: "v$1", MaxAge: -1}},
|
||
|
"Set-Cookie: Cookie-1=v$1\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
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
var writeCookiesTests = []struct {
|
||
|
Cookies []*Cookie
|
||
|
Raw string
|
||
|
}{
|
||
|
{
|
||
|
[]*Cookie{&Cookie{Name: "cookie-1", Value: "v$1", MaxAge: -1}},
|
||
|
"Cookie: Cookie-1=v$1\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", MaxAge: -1, Raw: "Cookie-1=v$1"}},
|
||
|
},
|
||
|
}
|
||
|
|
||
|
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%#v\nwant\n%#v\n", i, c, tt.Cookies)
|
||
|
continue
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
var readCookiesTests = []struct {
|
||
|
Header Header
|
||
|
Cookies []*Cookie
|
||
|
}{
|
||
|
{
|
||
|
Header{"Cookie": {"Cookie-1=v$1"}},
|
||
|
[]*Cookie{&Cookie{Name: "Cookie-1", Value: "v$1", MaxAge: -1, Raw: "Cookie-1=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%#v\nwant\n%#v\n", i, c, tt.Cookies)
|
||
|
continue
|
||
|
}
|
||
|
}
|
||
|
}
|