imaginative-go_test.go

package main

import (
	"github.com/julienschmidt/httprouter"
	"net/http"
	"net/http/httptest"
	"os"
	"strings"
	"testing"
)

func TestHome(t *testing.T) {
	if os.Getenv("IGO_ENV") != "prod" {
		return
	}

	req, err := http.NewRequest("GET", "/", nil)
	if err != nil {
		t.Fatal(err)
	}

	rr := newRequestRecorder(req, "GET", "/", Home)
	if rr.Code != 200 {
		t.Error("Expected response code to be 200")
	}

	expected := `Imaginative Go`
	if !strings.Contains(rr.Body.String(), expected) {
		t.Error("Response body does not match")
	}
}

func TestReadContent(t *testing.T) {
	if os.Getenv("IGO_ENV") != "prod" {
		return
	}

	req, err := http.NewRequest("GET", "/content/hello-world", nil)
	if err != nil {
		t.Fatal(err)
	}

	rr := newRequestRecorder(req, "GET", "/content/hello-world", ReadContent)
	if rr.Code != 200 {
		t.Error("Expected response code to be 200")
	}

	expected := `Hello World`
	if !strings.Contains(rr.Body.String(), expected) {
		t.Error("Response body does not match")
	}
}

func newRequestRecorder(req *http.Request, method string, strPath string, fnHandler func(w http.ResponseWriter, r *http.Request, param httprouter.Params)) *httptest.ResponseRecorder {
	router := httprouter.New()
	router.Handle(method, strPath, fnHandler)

	// We create a ResponseRecorder (which satisfies http.ResponseWriter) to record the response.
	rr := httptest.NewRecorder()

	// Our handlers satisfy http.Handler, so we can call their ServeHTTP method
	// directly and pass in our Request and ResponseRecorder.
	router.ServeHTTP(rr, req)

	return rr
}