Create a unit type

api/handler.go

@@ -60,6 +60,7 @@ func Handler() *mux.Router {
 	m.Get(router.DeleteTextMeasurementType).Handler(handler(serveDeleteTextMeasurementType))
 
 	m.Get(router.UnitType).Handler(handler(serveUnitType))
+	m.Get(router.CreateUnitType).Handler(handler(serveCreateUnitType))
 
 	return m
 }

api/unit_types.go

@@ -1,10 +1,12 @@
 package api
 
 import (
+	"encoding/json"
 	"net/http"
 	"strconv"
 
 	"github.com/gorilla/mux"
+	"github.com/thermokarst/bactdb/models"
 )
 
 func serveUnitType(w http.ResponseWriter, r *http.Request) error {
@@ -20,3 +22,21 @@ func serveUnitType(w http.ResponseWriter, r *http.Request) error {
 
 	return writeJSON(w, unit_type)
 }
+
+func serveCreateUnitType(w http.ResponseWriter, r *http.Request) error {
+	var unit_type models.UnitType
+	err := json.NewDecoder(r.Body).Decode(&unit_type)
+	if err != nil {
+		return err
+	}
+
+	created, err := store.UnitTypes.Create(&unit_type)
+	if err != nil {
+		return err
+	}
+	if created {
+		w.WriteHeader(http.StatusCreated)
+	}
+
+	return writeJSON(w, unit_type)
+}

api/unit_types_test.go

@@ -38,3 +38,30 @@ func TestUnitType_Get(t *testing.T) {
 		t.Errorf("got %+v but wanted %+v", got, want)
 	}
 }
+
+func TestUnitType_Create(t *testing.T) {
+	setup()
+
+	want := newUnitType()
+
+	calledPost := false
+	store.UnitTypes.(*models.MockUnitTypesService).Create_ = func(unit_type *models.UnitType) (bool, error) {
+		if !normalizeDeepEqual(want, unit_type) {
+			t.Errorf("wanted request for unit_type %d but got %d", want, unit_type)
+		}
+		calledPost = true
+		return true, nil
+	}
+
+	success, err := apiClient.UnitTypes.Create(want)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if !calledPost {
+		t.Error("!calledPost")
+	}
+	if !success {
+		t.Error("!success")
+	}
+}