autoneg_test.go

// Copyright 2011 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 gold

import (
	"net/http"
	"testing"
)

var chrome = "application/xml,application/xhtml+xml,text/html;q=0.9,text/plain;q=0.8,image/png,*/*;q=0.5"

func mockAccept(accept string) (al AcceptList, err error) {
	req := &http.Request{}
	req.Header = make(http.Header)
	req.Header["Accept"] = []string{accept}
	myreq := &httpRequest{req, nil, "", ""}
	al, err = myreq.Accept()
	return
}

func TestNegotiatePicturesOfWebPages(t *testing.T) {
	al, err := mockAccept(chrome)
	if err != nil {
		t.Fatal(err)
	}

	contentType, err := al.Negotiate("text/html", "image/png")
	if err != nil {
		t.Fatal(err)
	}

	if contentType != "image/png" {
		t.Errorf("got %s expected image/png", contentType)
	}
}

func TestNegotiateFirstMatch(t *testing.T) {
	al, err := mockAccept(chrome)
	if err != nil {
		t.Fatal(err)
	}

	contentType, err := al.Negotiate("text/html", "text/plain", "text/n3")
	if err != nil {
		t.Fatal(err)
	}

	if contentType != "text/html" {
		t.Errorf("got %s expected text/html", contentType)
	}
}

func TestNegotiateSecondMatch(t *testing.T) {
	al, err := mockAccept(chrome)
	if err != nil {
		t.Fatal(err)
	}

	contentType, err := al.Negotiate("text/n3", "text/plain")
	if err != nil {
		t.Fatal(err)
	}

	if contentType != "text/plain" {
		t.Errorf("got %s expected text/plain", contentType)
	}
}

func TestNegotiateWildcardMatch(t *testing.T) {
	al, err := mockAccept(chrome)
	if err != nil {
		t.Fatal(err)
	}

	contentType, err := al.Negotiate("text/n3", "application/rdf+xml")
	if err != nil {
		t.Fatal(err)
	}

	if contentType != "text/n3" {
		t.Errorf("got %s expected text/n3", contentType)
	}
}

func TestNegotiateInvalidMediaRange(t *testing.T) {
	_, err := mockAccept("something/valid, rubbish, other/valid")
	if err == nil {
		t.Fatal("expected error on obviously invalid media range")
	}
}

func TestNegotiateInvalidParam(t *testing.T) {
	_, err := mockAccept("text/plain; foo")
	if err == nil {
		t.Fatal("expected error on ill-formed params")
	}
}

func TestNegotiateEmptyAccept(t *testing.T) {
	al, err := mockAccept("")
	if err != nil {
		t.Fatal(err)
	}

	_, err = al.Negotiate("text/plain")
	if err == nil {
		t.Error("expected error with empty but present accept header")
	}
}

func TestNegotiateStarAccept(t *testing.T) {
	al, err := mockAccept("*")
	if err != nil {
		t.Fatal(err)
	}
	if al[0].Type+"/"+al[0].SubType != "*/*" {
		t.Error("expected subtype * for single * accept header")
	}
}

func TestNegotiateNoAlternative(t *testing.T) {
	al, err := mockAccept(chrome)
	if err != nil {
		t.Fatal(err)
	}

	_, err = al.Negotiate()
	if err == nil {
		t.Error("expected error with no alternative")
	}
}