Create a measurement

models/measurements.go

@@ -3,6 +3,7 @@ package models
 import (
 	"database/sql"
 	"errors"
+	"net/http"
 	"strconv"
 	"time"
 
@@ -36,6 +37,9 @@ func NewMeasurement() *Measurement {
 type MeasurementsService interface {
 	// Get a measurement
 	Get(id int64) (*Measurement, error)
+
+	// Create a measurement
+	Create(measurement *Measurement) (bool, error)
 }
 
 var (
@@ -68,8 +72,28 @@ func (s *measurementsService) Get(id int64) (*Measurement, error) {
 	return measurement, nil
 }
 
+func (s *measurementsService) Create(measurement *Measurement) (bool, error) {
+	url, err := s.client.url(router.CreateMeasurement, nil, nil)
+	if err != nil {
+		return false, err
+	}
+
+	req, err := s.client.NewRequest("POST", url.String(), measurement)
+	if err != nil {
+		return false, err
+	}
+
+	resp, err := s.client.Do(req, &measurement)
+	if err != nil {
+		return false, err
+	}
+
+	return resp.StatusCode == http.StatusCreated, nil
+}
+
 type MockMeasurementsService struct {
-	Get_ func(id int64) (*Measurement, error)
+	Get_    func(id int64) (*Measurement, error)
+	Create_ func(measurement *Measurement) (bool, error)
 }
 
 var _ MeasurementsService = &MockMeasurementsService{}
@@ -80,3 +104,10 @@ func (s *MockMeasurementsService) Get(id int64) (*Measurement, error) {
 	}
 	return s.Get_(id)
 }
+
+func (s *MockMeasurementsService) Create(measurement *Measurement) (bool, error) {
+	if s.Create_ == nil {
+		return false, nil
+	}
+	return s.Create_(measurement)
+}

models/measurements_test.go

@@ -1,6 +1,7 @@
 package models
 
 import (
+	"database/sql"
 	"net/http"
 	"reflect"
 	"testing"
@@ -11,6 +12,9 @@ import (
 func newMeasurement() *Measurement {
 	measurement := NewMeasurement()
 	measurement.Id = 1
+	measurement.StrainId = 1
+	measurement.ObservationId = 1
+	measurement.UnitTypeId = sql.NullInt64{Int64: 1, Valid: true}
 	return measurement
 }
 
@@ -43,3 +47,39 @@ func TestMeasurementService_Get(t *testing.T) {
 		t.Errorf("Measurements.Get return %+v, want %+v", measurement, want)
 	}
 }
+
+func TestMeasurementService_Create(t *testing.T) {
+	setup()
+	defer teardown()
+
+	want := newMeasurement()
+
+	var called bool
+	mux.HandleFunc(urlPath(t, router.CreateMeasurement, nil), func(w http.ResponseWriter, r *http.Request) {
+		called = true
+		testMethod(t, r, "POST")
+		testBody(t, r, `{"id":1,"strainId":1,"observationId":1,"textMeasurementTypeId":{"Int64":0,"Valid":false},"measurementValue":{"Float64":1.23,"Valid":true},"confidenceInterval":{"Float64":0,"Valid":false},"unitTypeId":{"Int64":1,"Valid":true},"createdAt":"0001-01-01T00:00:00Z","updatedAt":"0001-01-01T00:00:00Z","deletedAt":{"Time":"0001-01-01T00:00:00Z","Valid":false}}`+"\n")
+
+		w.WriteHeader(http.StatusCreated)
+		writeJSON(w, want)
+	})
+
+	measurement := newMeasurement()
+	created, err := client.Measurements.Create(measurement)
+	if err != nil {
+		t.Errorf("Measurements.Create returned error: %v", err)
+	}
+
+	if !created {
+		t.Error("!created")
+	}
+
+	if !called {
+		t.Fatal("!called")
+	}
+
+	normalizeTime(&want.CreatedAt, &want.UpdatedAt, &want.DeletedAt)
+	if !reflect.DeepEqual(measurement, want) {
+		t.Errorf("Measurements.Create returned %+v, want %+v", measurement, want)
+	}
+}