| // 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 |
| } |
| } |
| } |