Update a measurement

models/measurements.go

@@ -43,6 +43,9 @@ type MeasurementsService interface {
 
 	// Create a measurement
 	Create(measurement *Measurement) (bool, error)
+
+	// Update an existing measurement
+	Update(id int64, MeasurementType *Measurement) (bool, error)
 }
 
 var (
@@ -118,10 +121,32 @@ func (s *measurementsService) List(opt *MeasurementListOptions) ([]*Measurement,
 	return measurements, nil
 }
 
+func (s *measurementsService) Update(id int64, measurement *Measurement) (bool, error) {
+	strId := strconv.FormatInt(id, 10)
+
+	url, err := s.client.url(router.UpdateMeasurement, map[string]string{"Id": strId}, nil)
+	if err != nil {
+		return false, err
+	}
+
+	req, err := s.client.NewRequest("PUT", 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.StatusOK, nil
+}
+
 type MockMeasurementsService struct {
 	Get_    func(id int64) (*Measurement, error)
 	List_   func(opt *MeasurementListOptions) ([]*Measurement, error)
 	Create_ func(measurement *Measurement) (bool, error)
+	Update_ func(id int64, measurement *Measurement) (bool, error)
 }
 
 var _ MeasurementsService = &MockMeasurementsService{}
@@ -146,3 +171,10 @@ func (s *MockMeasurementsService) List(opt *MeasurementListOptions) ([]*Measurem
 	}
 	return s.List_(opt)
 }
+
+func (s *MockMeasurementsService) Update(id int64, measurement *Measurement) (bool, error) {
+	if s.Update_ == nil {
+		return false, nil
+	}
+	return s.Update_(id, measurement)
+}

models/measurements_test.go

@@ -116,3 +116,34 @@ func TestMeasurementService_List(t *testing.T) {
 		t.Errorf("Measurements.List return %+v, want %+v", measurements, want)
 	}
 }
+
+func TestMeasurementService_Update(t *testing.T) {
+	setup()
+	defer teardown()
+
+	want := newMeasurement()
+
+	var called bool
+	mux.HandleFunc(urlPath(t, router.UpdateMeasurement, map[string]string{"Id": "1"}), func(w http.ResponseWriter, r *http.Request) {
+		called = true
+		testMethod(t, r, "PUT")
+		testBody(t, r, `{"id":1,"strainId":1,"observationId":1,"textMeasurementTypeId":{"Int64":0,"Valid":false},"measurementValue":{"Float64":4.56,"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.StatusOK)
+		writeJSON(w, want)
+	})
+
+	measurement := newMeasurement()
+	measurement.MeasurementValue = sql.NullFloat64{Float64: 4.56, Valid: true}
+	updated, err := client.Measurements.Update(measurement.Id, measurement)
+	if err != nil {
+		t.Errorf("Measurements.Update returned error: %v", err)
+	}
+
+	if !updated {
+		t.Error("!updated")
+	}
+
+	if !called {
+		t.Fatal("!called")
+	}
+}