mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
82 lines
3.4 KiB
Go
82 lines
3.4 KiB
Go
// 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"},
|
|
}
|
|
|
|
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")
|
|
}
|
|
}
|