New API tested

view/who_test.go

@@ -0,0 +1,155 @@
+package view
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+	"local/dndex/config"
+	"local/dndex/storage"
+	"local/dndex/storage/entity"
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"strings"
+	"testing"
+)
+
+func TestWho(t *testing.T) {
+	os.Args = os.Args[:1]
+	f, err := ioutil.TempFile(os.TempDir(), "pattern*")
+	if err != nil {
+		t.Fatal(err)
+	}
+	f.Close()
+	defer os.Remove(f.Name())
+	os.Setenv("DBURI", f.Name())
+
+	t.Logf("config: %+v", config.New())
+
+	g := storage.NewGraph()
+	ones := fillDB(t, g)
+	want := ones[len(ones)-1]
+
+	handler := jsonHandler(g)
+
+	t.Log(handler, want)
+
+	t.Run("get no namespace is 404", func(t *testing.T) {
+		iwant := want
+		r := httptest.NewRequest(http.MethodGet, "/who?id="+iwant.Name, nil)
+		w := httptest.NewRecorder()
+		handler.ServeHTTP(w, r)
+		if w.Code != http.StatusNotFound {
+			t.Fatalf("%d: %s", w.Code, w.Body.Bytes())
+		}
+	})
+
+	t.Run("get fake", func(t *testing.T) {
+		iwant := want
+		r := httptest.NewRequest(http.MethodGet, "/who/col?id=FAKER"+iwant.Name, nil)
+		w := httptest.NewRecorder()
+		handler.ServeHTTP(w, r)
+		if w.Code != http.StatusNotFound {
+			t.Fatalf("%d: %s", w.Code, w.Body.Bytes())
+		}
+	})
+
+	t.Run("get real", func(t *testing.T) {
+		iwant := want
+		r := httptest.NewRequest(http.MethodGet, "/who/col?id="+iwant.Name, nil)
+		w := httptest.NewRecorder()
+		handler.ServeHTTP(w, r)
+		if w.Code != http.StatusOK {
+			t.Fatalf("%d: %s", w.Code, w.Body.Bytes())
+		}
+		o := entity.One{}
+		if err := json.Unmarshal(w.Body.Bytes(), &o); err != nil {
+			t.Fatal(err)
+		}
+		if fmt.Sprint(o) == fmt.Sprint(iwant) {
+			t.Fatal(o, iwant)
+		}
+		if len(o.Connections) != len(iwant.Connections) {
+			t.Fatal(len(o.Connections), len(iwant.Connections))
+		}
+		iwant.Connections = o.Connections
+		iwant.Modified = 0
+		o.Modified = 0
+		if fmt.Sprint(o) != fmt.Sprint(iwant) {
+			t.Fatalf("after resolving connections and modified, iwant and got differ: \nwant %+v\n got %+v", iwant, o)
+		}
+		b, _ := json.MarshalIndent(o, "", "   ")
+		t.Logf("POST GET:\n%s", b)
+	})
+
+	t.Run("put fake", func(t *testing.T) {
+		iwant := want
+		r := httptest.NewRequest(http.MethodPut, "/who/col?id=FAKER"+iwant.Name, strings.NewReader(`{"title":"this should fail to find someone"}`))
+		w := httptest.NewRecorder()
+		handler.ServeHTTP(w, r)
+		if w.Code != http.StatusNotFound {
+			t.Fatalf("%d: %s", w.Code, w.Body.Bytes())
+		}
+	})
+
+	t.Run("put real", func(t *testing.T) {
+		iwant := want
+		r := httptest.NewRequest(http.MethodPut, "/who/col?id="+iwant.Name, strings.NewReader(`{"title":"this should work"}`))
+		w := httptest.NewRecorder()
+		handler.ServeHTTP(w, r)
+		if w.Code != http.StatusOK {
+			t.Fatalf("%d: %s", w.Code, w.Body.Bytes())
+		}
+		o := entity.One{}
+		if err := json.Unmarshal(w.Body.Bytes(), &o); err != nil {
+			t.Fatal(err)
+		}
+		if len(o.Connections) != len(iwant.Connections) {
+			t.Fatalf("wrong number of connections returned: want %v, got %v", len(iwant.Connections), len(o.Connections))
+		}
+		if o.Title != "this should work" {
+			t.Fatalf("failed to PUT a new title: %+v", o)
+		}
+		b, _ := json.MarshalIndent(o, "", "   ")
+		t.Logf("POST PUT:\n%s", b)
+	})
+
+	t.Run("post exists", func(t *testing.T) {
+		iwant := want
+		iwant.Name = ""
+		r := httptest.NewRequest(http.MethodPost, "/who/col?id="+want.Name, strings.NewReader(`{"title":"this should fail to insert"}`))
+		w := httptest.NewRecorder()
+		handler.ServeHTTP(w, r)
+		if w.Code != http.StatusConflict {
+			t.Fatalf("%d: %s", w.Code, w.Body.Bytes())
+		}
+	})
+
+	t.Run("post real", func(t *testing.T) {
+		iwant := want
+		iwant.Name = ""
+		b, err := json.Marshal(iwant)
+		if err != nil {
+			t.Fatal(err)
+		}
+		r := httptest.NewRequest(http.MethodPost, "/who/col?id=NEWBIE"+want.Name, bytes.NewReader(b))
+		w := httptest.NewRecorder()
+		handler.ServeHTTP(w, r)
+		if w.Code != http.StatusOK {
+			t.Fatalf("%d: %s", w.Code, w.Body.Bytes())
+		}
+		o := entity.One{}
+		if err := json.Unmarshal(w.Body.Bytes(), &o); err != nil {
+			t.Fatal(err)
+		}
+		if len(o.Connections) != len(iwant.Connections) {
+			t.Fatalf("wrong number of connections returned: want %v, got %v", len(iwant.Connections), len(o.Connections))
+		}
+		if o.Name != "NEWBIE"+want.Name {
+			t.Fatalf("failed to POST specified name: %+v", o)
+		}
+		b, _ = json.MarshalIndent(o, "", "   ")
+		t.Logf("POST POST:\n%s", b)
+	})
+}