Added some API tests.

web/web_test.go

@@ -24,10 +24,29 @@ func (e *env) destroy() {
 	os.RemoveAll(e.dir)
 }
 
-func (e *env) callAPI(method, name string, body io.Reader) (*msg, error) {
+func (e *env) getAPI(m *msg, name string) error {
+	return e.callAPI(m, "GET", name, nil)
+}
+
+func (e *env) postAPI(m *msg, name, url string) error {
+	r := struct {
+		URL string `json:"url"`
+	}{
+		url,
+	}
+
+	var buf bytes.Buffer
+	if err := json.NewEncoder(&buf).Encode(&r); err != nil {
+		return err
+	}
+
+	return e.callAPI(m, "POST", name, &buf)
+}
+
+func (e *env) callAPI(m *msg, method, name string, body io.Reader) error {
 	req, err := http.NewRequest(method, fmt.Sprintf("/api/url/%s", name), body)
 	if err != nil {
-		return nil, err
+		return err
 	}
 
 	res := mockResponse{
@@ -36,13 +55,11 @@ func (e *env) callAPI(method, name string, body io.Reader) (*msg, error) {
 
 	e.mux.ServeHTTP(&res, req)
 
-	var m msg
-
 	if err := json.NewDecoder(&res).Decode(&m); err != nil {
-		return nil, err
+		return err
 	}
 
-	return &m, nil
+	return nil
 }
 
 func newEnv() (*env, error) {
@@ -64,6 +81,14 @@ func newEnv() (*env, error) {
 	}, nil
 }
 
+func needEnv(t *testing.T) *env {
+	e, err := newEnv()
+	if err != nil {
+		t.Fatal(err)
+	}
+	return e
+}
+
 type mockResponse struct {
 	header http.Header
 	bytes.Buffer
@@ -92,19 +117,104 @@ func assertJustOk(t *testing.T, m *msg) {
 	}
 }
 
-func TestAPIGet(t *testing.T) {
-	e, err := newEnv()
-	if err != nil {
-		t.Fatal(err)
+func assertOkWithRoute(t *testing.T, m *msg, url string) {
+	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("Route is nil, expected one with url of %s", url)
+	}
+
+	if m.Route.URL != url {
+		t.Fatalf("Expected url of %s, got %s", url, m.Route.URL)
+	}
+}
+
+func assertOkWithNamedRoute(t *testing.T, m *msg, name, url string) {
+	assertOkWithRoute(t, m, url)
+	if m.Route.Name != name {
+		t.Fatalf("expected name %s, got %s", name, m.Route.Name)
+	}
+}
+
+func TestAPIGetNotFound(t *testing.T) {
+	e := needEnv(t)
 	defer e.destroy()
 
+	var m msg
 	names := []string{"", "nothing", "nothing/there"}
 	for _, name := range names {
-		m, err := e.callAPI("GET", name, nil)
-		if err != nil {
+		if err := e.getAPI(&m, name); err != nil {
 			t.Fatal(err)
 		}
-		assertJustOk(t, m)
+		assertJustOk(t, &m)
+	}
+}
+
+func TestAPIPutThenGet(t *testing.T) {
+	e := needEnv(t)
+	defer e.destroy()
+
+	var pm msg
+	if err := e.postAPI(&pm, "xxx", "http://ex.com/"); err != nil {
+		t.Fatal(err)
+	}
+	assertOkWithRoute(t, &pm, "http://ex.com/")
+
+	var gm msg
+	if err := e.getAPI(&gm, "xxx"); err != nil {
+		t.Fatal(err)
+	}
+	assertOkWithNamedRoute(t, &gm, "xxx", "http://ex.com/")
+}
+
+func TestAPIDel(t *testing.T) {
+	e := needEnv(t)
+	defer e.destroy()
+
+	var am msg
+	if err := e.postAPI(&am, "yyy", ""); err != nil {
+		t.Fatal(err)
+	}
+	assertJustOk(t, &am)
+
+	var bm msg
+	if err := e.postAPI(&bm, "yyy", "https://a.com/"); err != nil {
+		t.Fatal(err)
+	}
+	assertOkWithNamedRoute(t, &bm, "yyy", "https://a.com/")
+
+	var cm msg
+	if err := e.postAPI(&cm, "yyy", ""); err != nil {
+		t.Fatal(err)
+	}
+	assertJustOk(t, &cm)
+
+	var dm msg
+	if err := e.getAPI(&dm, "yyy"); err != nil {
+		t.Fatal(err)
+	}
+	assertJustOk(t, &dm)
+}
+
+func TestAPIPutThenGetAuto(t *testing.T) {
+	e := needEnv(t)
+	defer e.destroy()
+
+	var am msg
+	if err := e.postAPI(&am, "", "http://b.com/"); err != nil {
+		t.Fatal(err)
+	}
+	assertOkWithRoute(t, &am, "http://b.com/")
+
+	var bm msg
+	if err := e.getAPI(&bm, am.Route.Name); err != nil {
+		t.Fatal(err)
 	}
+	assertOkWithNamedRoute(t, &bm, am.Route.Name, "http://b.com/")
 }