rest_test.go

package rest

import (
	"encoding/json"
	"errors"
	"net/http"
	"net/http/httptest"
	"testing"
)

func TestGone(t *testing.T) {
	t.Parallel()
	w := httptest.NewRecorder()
	req := httptest.NewRequest("GET", "/", nil)
	Gone(w, req)
	if w.Code != 410 {
		t.Errorf("expected code to be 410, got %d", w.Code)
	}
	var e Error
	err := json.NewDecoder(w.Body).Decode(&e)
	if err != nil {
		t.Fatal(err)
	}
	if e.Title != gone.Title {
		t.Errorf("expected Title to be %s, got %s", serverError.Title, e.Title)
	}
	if e.ID != gone.ID {
		t.Errorf("expected ID to be %s, got %s", serverError.ID, e.ID)
	}
	if e.Status != 410 {
		t.Errorf("expected code to be 410, got %d", e.Status)
	}
}

func TestServerError(t *testing.T) {
	t.Parallel()
	w := httptest.NewRecorder()
	req := httptest.NewRequest("GET", "/", nil)
	ServerError(w, req, errors.New("foo bar"))
	if w.Code != 500 {
		t.Errorf("expected code to be 500, got %d", w.Code)
	}
	var e Error
	err := json.NewDecoder(w.Body).Decode(&e)
	if err != nil {
		t.Fatal(err)
	}
	if e.Title == "" {
		t.Errorf("expected Title to be %s, got the empty string", serverError.Title)
	}
	if e.Title != serverError.Title {
		t.Errorf("expected Title to be %s, got %s", serverError.Title, e.Title)
	}
	if e.ID != serverError.ID {
		t.Errorf("expected ID to be %s, got %s", serverError.ID, e.ID)
	}
	if e.Status != 500 {
		t.Errorf("expected code to be 500, got %d", e.Status)
	}
}

func TestBadRequest(t *testing.T) {
	t.Parallel()
	w := httptest.NewRecorder()
	req := httptest.NewRequest("GET", "/", nil)
	BadRequest(w, req, &Error{
		Title: "Please provide a widget",
		ID:    "missing_widget",
	})
	if w.Code != 400 {
		t.Errorf("expected code to be 400, got %d", w.Code)
	}
	var e Error
	err := json.NewDecoder(w.Body).Decode(&e)
	if err != nil {
		t.Fatal(err)
	}
	if e.Title == "" {
		t.Errorf("expected Title to be %s, got the empty string", "Please provide a widget")
	}
	if e.Title != "Please provide a widget" {
		t.Errorf("expected Title to be %s, got %s", "Please provide a widget", e.Title)
	}
	if e.Status != 400 {
		t.Errorf("expected code to be 400, got %d", e.Status)
	}
	if e.ID != "missing_widget" {
		t.Errorf("expected ID to be %s, got %s", "missing_widget", e.ID)
	}
}

func TestNoContent(t *testing.T) {
	t.Parallel()
	w := httptest.NewRecorder()
	w.Header().Set("Content-Type", "application/json")
	NoContent(w)
	if w.Code != 204 {
		t.Errorf("expected Code to be 204, got %d", w.Code)
	}
	if hdr := w.Header().Get("Content-Type"); hdr != "" {
		t.Errorf("expected Content-Type to be empty, got %s", hdr)
	}
}

func TestUnauthorized(t *testing.T) {
	t.Parallel()
	w := httptest.NewRecorder()
	req := httptest.NewRequest("GET", "/", nil)
	Unauthorized(w, req, "foo")
	if w.Code != 401 {
		t.Errorf("expected Code to be 401, got %d", w.Code)
	}
	expected := `Basic realm="foo"`
	if hdr := w.Header().Get("WWW-Authenticate"); hdr != expected {
		t.Errorf("expected WWW-Authenticate header to be %s, got %s", expected, hdr)
	}
}

func TestRegisterNilHandlerDeletes(t *testing.T) {
	RegisterHandler(500, nil)
	w := httptest.NewRecorder()
	req := httptest.NewRequest("GET", "/", nil)
	ServerError(w, req, errors.New("bad"))
	if ctype := w.Header().Get("Content-Type"); ctype != jsonContentType {
		t.Errorf("expected default JSON content-type, got %s", ctype)
	}
}

func TestRegister401CallsIt(t *testing.T) {
	RegisterHandler(401, http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Custom-Handler", "true")
		w.WriteHeader(401)
	}))
	defer RegisterHandler(401, nil)
	w := httptest.NewRecorder()
	req := httptest.NewRequest("GET", "/", nil)
	Unauthorized(w, req, "domain")
	if hdr := w.Header().Get("Custom-Handler"); hdr != "true" {
		t.Errorf("expected custom handler to be called, got %v", hdr)
	}
}