Adding API tests.

web/web.go

@@ -278,11 +278,9 @@ func (h *defaultHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 		http.StatusTemporaryRedirect)
 }
 
-// ListenAndServe sets up all web routes, binds the port and handles incoming
-// web requests.
-func ListenAndServe(addr string, ctx *context.Context) error {
+// Setup a Mux with all web routes.
+func allRoutes(ctx *context.Context) *http.ServeMux {
 	mux := http.NewServeMux()
-
 	mux.Handle("/", &defaultHandler{ctx})
 	mux.Handle("/api/url/", &apiHandler{ctx})
 	mux.HandleFunc("/edit/", func(w http.ResponseWriter, r *http.Request) {
@@ -291,6 +289,11 @@ func ListenAndServe(addr string, ctx *context.Context) error {
 	mux.HandleFunc("/s/", func(w http.ResponseWriter, r *http.Request) {
 		serveAsset(w, r, r.URL.Path[len("/s/"):])
 	})
+	return mux
+}
 
-	return http.ListenAndServe(addr, mux)
+// ListenAndServe sets up all web routes, binds the port and handles incoming
+// web requests.
+func ListenAndServe(addr string, ctx *context.Context) error {
+	return http.ListenAndServe(addr, allRoutes(ctx))
 }

web/web_test.go

@@ -0,0 +1,110 @@
+package web
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
+	"io/ioutil"
+	"net/http"
+	"os"
+	"path/filepath"
+	"testing"
+
+	"github.com/kellegous/go/context"
+)
+
+type env struct {
+	mux *http.ServeMux
+	dir string
+	ctx *context.Context
+}
+
+func (e *env) destroy() {
+	os.RemoveAll(e.dir)
+}
+
+func (e *env) callAPI(method, name string, body io.Reader) (*msg, error) {
+	req, err := http.NewRequest(method, fmt.Sprintf("/api/url/%s", name), body)
+	if err != nil {
+		return nil, err
+	}
+
+	res := mockResponse{
+		header: map[string][]string{},
+	}
+
+	e.mux.ServeHTTP(&res, req)
+
+	var m msg
+
+	if err := json.NewDecoder(&res).Decode(&m); err != nil {
+		return nil, err
+	}
+
+	return &m, nil
+}
+
+func newEnv() (*env, error) {
+	dir, err := ioutil.TempDir("", "")
+	if err != nil {
+		return nil, err
+	}
+
+	ctx, err := context.Open(filepath.Join(dir, "data"))
+	if err != nil {
+		os.RemoveAll(dir)
+		return nil, err
+	}
+
+	return &env{
+		mux: allRoutes(ctx),
+		dir: dir,
+		ctx: ctx,
+	}, nil
+}
+
+type mockResponse struct {
+	header http.Header
+	bytes.Buffer
+	status int
+}
+
+func (r *mockResponse) Header() http.Header {
+	return r.header
+}
+
+func (r *mockResponse) WriteHeader(status int) {
+	r.status = status
+}
+
+func assertJustOk(t *testing.T, m *msg) {
+	if !m.Ok {
+		t.Fatal("expected OK message, but it's not OK")
+	}
+
+	if m.Error != "" {
+		t.Fatalf("expected no error, but got %s", m.Error)
+	}
+
+	if m.Route != nil {
+		t.Fatalf("expected no route, got %v", m.Route)
+	}
+}
+
+func TestAPIGet(t *testing.T) {
+	e, err := newEnv()
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer e.destroy()
+
+	names := []string{"", "nothing", "nothing/there"}
+	for _, name := range names {
+		m, err := e.callAPI("GET", name, nil)
+		if err != nil {
+			t.Fatal(err)
+		}
+		assertJustOk(t, m)
+	}
+}