1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
// 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 middleware
import (
"net/http"
"testing"
)
var negotiateContentEncodingTests = []struct {
s string
offers []string
expect string
}{
{"", []string{"identity", "gzip"}, "identity"},
{"*;q=0", []string{"identity", "gzip"}, ""},
{"gzip", []string{"identity", "gzip"}, "gzip"},
}
func TestNegotiateContentEnoding(t *testing.T) {
for _, tt := range negotiateContentEncodingTests {
r := &http.Request{Header: http.Header{"Accept-Encoding": {tt.s}}}
actual := NegotiateContentEncoding(r, tt.offers)
if actual != tt.expect {
t.Errorf("NegotiateContentEncoding(%q, %#v)=%q, want %q", tt.s, tt.offers, actual, tt.expect)
}
}
}
var negotiateContentTypeTests = []struct {
s string
offers []string
defaultOffer string
expect string
}{
{"text/html, */*;q=0", []string{"x/y"}, "", ""},
{"text/html, */*", []string{"x/y"}, "", "x/y"},
{"text/html, image/png", []string{"text/html", "image/png"}, "", "text/html"},
{"text/html, image/png", []string{"image/png", "text/html"}, "", "image/png"},
{"text/html, image/png; q=0.5", []string{"image/png"}, "", "image/png"},
{"text/html, image/png; q=0.5", []string{"text/html"}, "", "text/html"},
{"text/html, image/png; q=0.5", []string{"foo/bar"}, "", ""},
{"text/html, image/png; q=0.5", []string{"image/png", "text/html"}, "", "text/html"},
{"text/html, image/png; q=0.5", []string{"text/html", "image/png"}, "", "text/html"},
{"text/html;q=0.5, image/png", []string{"image/png"}, "", "image/png"},
{"text/html;q=0.5, image/png", []string{"text/html"}, "", "text/html"},
{"text/html;q=0.5, image/png", []string{"image/png", "text/html"}, "", "image/png"},
{"text/html;q=0.5, image/png", []string{"text/html", "image/png"}, "", "image/png"},
{"text/html;q=0.5, image/png", []string{"text/html", "image/png"}, "", "image/png"},
{"image/png, image/*;q=0.5", []string{"image/jpg", "image/png"}, "", "image/png"},
{"image/png, image/*;q=0.5", []string{"image/jpg"}, "", "image/jpg"},
{"image/png, image/*;q=0.5", []string{"image/jpg", "image/gif"}, "", "image/jpg"},
{"image/png, image/*", []string{"image/jpg", "image/gif"}, "", "image/jpg"},
{"image/png, image/*", []string{"image/gif", "image/jpg"}, "", "image/gif"},
{"image/png, image/*", []string{"image/gif", "image/png"}, "", "image/png"},
{"image/png, image/*", []string{"image/png", "image/gif"}, "", "image/png"},
{"application/vnd.google.protobuf;proto=io.prometheus.client.MetricFamily;encoding=delimited;q=0.7,text/plain;version=0.0.4;q=0.3", []string{"text/plain"}, "", "text/plain"},
{"application/json", []string{"application/json; charset=utf-8", "image/png"}, "", "application/json; charset=utf-8"},
{"application/json; charset=utf-8", []string{"application/json; charset=utf-8", "image/png"}, "", "application/json; charset=utf-8"},
{"application/json", []string{"application/vnd.cia.v1+json"}, "", ""},
// Default header of java clients
{"text/html, image/gif, image/jpeg, *; q=.2, */*; q=.2", []string{"application/json"}, "", "application/json"},
}
func TestNegotiateContentType(t *testing.T) {
for _, tt := range negotiateContentTypeTests {
r := &http.Request{Header: http.Header{"Accept": {tt.s}}}
actual := NegotiateContentType(r, tt.offers, tt.defaultOffer)
if actual != tt.expect {
t.Errorf("NegotiateContentType(%q, %#v, %q)=%q, want %q", tt.s, tt.offers, tt.defaultOffer, actual, tt.expect)
}
}
}
func TestNegotiateContentTypeNoAcceptHeader(t *testing.T) {
r := &http.Request{Header: http.Header{}}
offers := []string{"application/json", "text/xml"}
actual := NegotiateContentType(r, offers, "")
if actual != "application/json" {
t.Errorf("NegotiateContentType(empty, %#v, empty)=%q, want %q", offers, actual, "application/json")
}
}
|
