major refactoring of the api

api/createdoorman_test.go

@@ -0,0 +1,57 @@
+package api
+
+import (
+	"bytes"
+	"github.com/didiercrunch/doormanserver/inmemoryconnector"
+	"gopkg.in/mgo.v2/bson"
+	"net/http"
+	"net/http/httptest"
+	"path"
+	"testing"
+)
+
+func GetIdFromLocation(p string) bson.ObjectId {
+	return bson.ObjectIdHex(path.Base(p))
+}
+
+func TestCreateDoorman(t *testing.T) {
+	conn = inmemoryconnector.NewInMemoryDBConnector()
+	payload := `{"name":"name","values":[{"name":"red","probability":0.2},{"name":"blue","probability":0.8}]}`
+	req, err := http.NewRequest(
+		"POST",
+		"http://bigtits.com/api/doormen",
+		bytes.NewBufferString(payload),
+	)
+	if err != nil {
+		t.Error(err)
+		return
+	}
+	w := httptest.NewRecorder()
+	CreateDoorman(w, req)
+	if w.Code != 201 {
+		t.Error("bad status code", w.Code)
+	} else if location := w.Header().Get("location"); location == "" {
+		t.Error("missing location header", location)
+	} else if dor, _ := conn.GetDoorman(GetIdFromLocation(location)); dor == nil {
+		t.Error("dorman not created")
+	}
+}
+
+func TestCreateDoormanWithBadInput(t *testing.T) {
+	conn = inmemoryconnector.NewInMemoryDBConnector()
+	payload := `{"name":"name","values":[{"name":"red","probability":0.2},{"name":"blue","probability":0.2}]}`
+	req, err := http.NewRequest(
+		"POST",
+		"http://bigtits.com/api/doormen",
+		bytes.NewBufferString(payload),
+	)
+	if err != nil {
+		t.Error(err)
+		return
+	}
+	w := httptest.NewRecorder()
+	CreateDoorman(w, req)
+	if w.Code != 400 {
+		t.Error("bad status code", w.Code)
+	}
+}