blob: 97c86858994eed5c6bea3f9ff14f73f58ef53466 [file] [log] [blame]
// Copyright 2013 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 or at
// https://developers.google.com/open-source/licenses/bsd.
package header
import (
"net/http"
"testing"
"time"
"github.com/google/go-cmp/cmp"
)
var getHeaderListTests = []struct {
s string
l []string
}{
{s: `a`, l: []string{`a`}},
{s: `a, b , c `, l: []string{`a`, `b`, `c`}},
{s: `a,, b , , c `, l: []string{`a`, `b`, `c`}},
{s: `a,b,c`, l: []string{`a`, `b`, `c`}},
{s: ` a b, c d `, l: []string{`a b`, `c d`}},
{s: `"a, b, c", d `, l: []string{`"a, b, c"`, "d"}},
{s: `","`, l: []string{`","`}},
{s: `"\""`, l: []string{`"\""`}},
{s: `" "`, l: []string{`" "`}},
}
func TestGetHeaderList(t *testing.T) {
for _, tt := range getHeaderListTests {
header := http.Header{"Foo": {tt.s}}
if l := ParseList(header, "foo"); !cmp.Equal(tt.l, l) {
t.Errorf("ParseList for %q = %q, want %q", tt.s, l, tt.l)
}
}
}
var parseValueAndParamsTests = []struct {
s string
value string
params map[string]string
}{
{`text/html`, "text/html", map[string]string{}},
{`text/html `, "text/html", map[string]string{}},
{`text/html ; `, "text/html", map[string]string{}},
{`tExt/htMl`, "text/html", map[string]string{}},
{`tExt/htMl; fOO=";"; hellO=world`, "text/html", map[string]string{
"hello": "world",
"foo": `;`,
}},
{`text/html; foo=bar, hello=world`, "text/html", map[string]string{"foo": "bar"}},
{`text/html ; foo=bar `, "text/html", map[string]string{"foo": "bar"}},
{`text/html ;foo=bar `, "text/html", map[string]string{"foo": "bar"}},
{`text/html; foo="b\ar"`, "text/html", map[string]string{"foo": "bar"}},
{`text/html; foo="bar\"baz\"qux"`, "text/html", map[string]string{"foo": `bar"baz"qux`}},
{`text/html; foo="b,ar"`, "text/html", map[string]string{"foo": "b,ar"}},
{`text/html; foo="b;ar"`, "text/html", map[string]string{"foo": "b;ar"}},
{`text/html; FOO="bar"`, "text/html", map[string]string{"foo": "bar"}},
{`form-data; filename="file.txt"; name=file`, "form-data", map[string]string{"filename": "file.txt", "name": "file"}},
}
func TestParseValueAndParams(t *testing.T) {
for _, tt := range parseValueAndParamsTests {
header := http.Header{"Content-Type": {tt.s}}
value, params := ParseValueAndParams(header, "Content-Type")
if value != tt.value {
t.Errorf("%q, value=%q, want %q", tt.s, value, tt.value)
}
if !cmp.Equal(params, tt.params) {
t.Errorf("%q, param=%#v, want %#v", tt.s, params, tt.params)
}
}
}
var parseTimeValidTests = []string{
"Sun, 06 Nov 1994 08:49:37 GMT",
"Sunday, 06-Nov-94 08:49:37 GMT",
"Sun Nov 6 08:49:37 1994",
}
var parseTimeInvalidTests = []string{
"junk",
}
func TestParseTime(t *testing.T) {
expected := time.Date(1994, 11, 6, 8, 49, 37, 0, time.UTC)
for _, s := range parseTimeValidTests {
header := http.Header{"Date": {s}}
actual := ParseTime(header, "Date")
if actual != expected {
t.Errorf("GetTime(%q)=%v, want %v", s, actual, expected)
}
}
for _, s := range parseTimeInvalidTests {
header := http.Header{"Date": {s}}
actual := ParseTime(header, "Date")
if !actual.IsZero() {
t.Errorf("GetTime(%q) did not return zero", s)
}
}
}
var parseAcceptTests = []struct {
s string
expected []AcceptSpec
}{
{"text/html", []AcceptSpec{{"text/html", 1}}},
{"text/html; q=0", []AcceptSpec{{"text/html", 0}}},
{"text/html; q=0.0", []AcceptSpec{{"text/html", 0}}},
{"text/html; q=1", []AcceptSpec{{"text/html", 1}}},
{"text/html; q=1.0", []AcceptSpec{{"text/html", 1}}},
{"text/html; q=0.1", []AcceptSpec{{"text/html", 0.1}}},
{"text/html;q=0.1", []AcceptSpec{{"text/html", 0.1}}},
{"text/html, text/plain", []AcceptSpec{{"text/html", 1}, {"text/plain", 1}}},
{"text/html; q=0.1, text/plain", []AcceptSpec{{"text/html", 0.1}, {"text/plain", 1}}},
{"iso-8859-5, unicode-1-1;q=0.8,iso-8859-1", []AcceptSpec{{"iso-8859-5", 1}, {"unicode-1-1", 0.8}, {"iso-8859-1", 1}}},
{"iso-8859-1", []AcceptSpec{{"iso-8859-1", 1}}},
{"*", []AcceptSpec{{"*", 1}}},
{"da, en-gb;q=0.8, en;q=0.7", []AcceptSpec{{"da", 1}, {"en-gb", 0.8}, {"en", 0.7}}},
{"da, q, en-gb;q=0.8", []AcceptSpec{{"da", 1}, {"q", 1}, {"en-gb", 0.8}}},
{"image/png, image/*;q=0.5", []AcceptSpec{{"image/png", 1}, {"image/*", 0.5}}},
// bad cases
{"value1; q=0.1.2", []AcceptSpec{{"value1", 0.1}}},
{"da, en-gb;q=foo", []AcceptSpec{{"da", 1}}},
}
func TestParseAccept(t *testing.T) {
for _, tt := range parseAcceptTests {
header := http.Header{"Accept": {tt.s}}
actual := ParseAccept(header, "Accept")
if !cmp.Equal(actual, tt.expected) {
t.Errorf("ParseAccept(h, %q)=%v, want %v", tt.s, actual, tt.expected)
}
}
}