142 lines
3.6 KiB
Go
142 lines
3.6 KiB
Go
// Copyright 2015 Google Inc. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package gensupport
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"io/ioutil"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestContentSniffing(t *testing.T) {
|
|
type testCase struct {
|
|
data []byte // the data to read from the Reader
|
|
finalErr error // error to return after data has been read
|
|
|
|
wantContentType string
|
|
wantContentTypeResult bool
|
|
}
|
|
|
|
for _, tc := range []testCase{
|
|
{
|
|
data: []byte{0, 0, 0, 0},
|
|
finalErr: nil,
|
|
wantContentType: "application/octet-stream",
|
|
wantContentTypeResult: true,
|
|
},
|
|
{
|
|
data: []byte(""),
|
|
finalErr: nil,
|
|
wantContentType: "text/plain; charset=utf-8",
|
|
wantContentTypeResult: true,
|
|
},
|
|
{
|
|
data: []byte(""),
|
|
finalErr: io.ErrUnexpectedEOF,
|
|
wantContentType: "text/plain; charset=utf-8",
|
|
wantContentTypeResult: false,
|
|
},
|
|
{
|
|
data: []byte("abc"),
|
|
finalErr: nil,
|
|
wantContentType: "text/plain; charset=utf-8",
|
|
wantContentTypeResult: true,
|
|
},
|
|
{
|
|
data: []byte("abc"),
|
|
finalErr: io.ErrUnexpectedEOF,
|
|
wantContentType: "text/plain; charset=utf-8",
|
|
wantContentTypeResult: false,
|
|
},
|
|
// The following examples contain more bytes than are buffered for sniffing.
|
|
{
|
|
data: bytes.Repeat([]byte("a"), 513),
|
|
finalErr: nil,
|
|
wantContentType: "text/plain; charset=utf-8",
|
|
wantContentTypeResult: true,
|
|
},
|
|
{
|
|
data: bytes.Repeat([]byte("a"), 513),
|
|
finalErr: io.ErrUnexpectedEOF,
|
|
wantContentType: "text/plain; charset=utf-8",
|
|
wantContentTypeResult: true, // true because error is after first 512 bytes.
|
|
},
|
|
} {
|
|
er := &errReader{buf: tc.data, err: tc.finalErr}
|
|
|
|
sct := newContentSniffer(er)
|
|
|
|
// Even if was an error during the first 512 bytes, we should still be able to read those bytes.
|
|
buf, err := ioutil.ReadAll(sct)
|
|
|
|
if !reflect.DeepEqual(buf, tc.data) {
|
|
t.Fatalf("Failed reading buffer: got: %q; want:%q", buf, tc.data)
|
|
}
|
|
|
|
if err != tc.finalErr {
|
|
t.Fatalf("Reading buffer error: got: %v; want: %v", err, tc.finalErr)
|
|
}
|
|
|
|
ct, ok := sct.ContentType()
|
|
if ok != tc.wantContentTypeResult {
|
|
t.Fatalf("Content type result got: %v; want: %v", ok, tc.wantContentTypeResult)
|
|
}
|
|
if ok && ct != tc.wantContentType {
|
|
t.Fatalf("Content type got: %q; want: %q", ct, tc.wantContentType)
|
|
}
|
|
}
|
|
}
|
|
|
|
type staticContentTyper struct {
|
|
io.Reader
|
|
}
|
|
|
|
func (sct staticContentTyper) ContentType() string {
|
|
return "static content type"
|
|
}
|
|
|
|
func TestDetermineContentType(t *testing.T) {
|
|
data := []byte("abc")
|
|
rdr := func() io.Reader {
|
|
return bytes.NewBuffer(data)
|
|
}
|
|
|
|
type testCase struct {
|
|
r io.Reader
|
|
explicitConentType string
|
|
wantContentType string
|
|
}
|
|
|
|
for _, tc := range []testCase{
|
|
{
|
|
r: rdr(),
|
|
wantContentType: "text/plain; charset=utf-8",
|
|
},
|
|
{
|
|
r: staticContentTyper{rdr()},
|
|
wantContentType: "static content type",
|
|
},
|
|
{
|
|
r: staticContentTyper{rdr()},
|
|
explicitConentType: "explicit",
|
|
wantContentType: "explicit",
|
|
},
|
|
} {
|
|
r, ctype := DetermineContentType(tc.r, tc.explicitConentType)
|
|
got, err := ioutil.ReadAll(r)
|
|
if err != nil {
|
|
t.Fatalf("Failed reading buffer: %v", err)
|
|
}
|
|
if !reflect.DeepEqual(got, data) {
|
|
t.Fatalf("Failed reading buffer: got: %q; want:%q", got, data)
|
|
}
|
|
|
|
if ctype != tc.wantContentType {
|
|
t.Fatalf("Content type got: %q; want: %q", ctype, tc.wantContentType)
|
|
}
|
|
}
|
|
}
|