Update a measurement
This commit is contained in:
parent
d462935682
commit
bc3c724a1d
9 changed files with 158 additions and 0 deletions
|
@ -46,3 +46,26 @@ func (s *measurementsStore) List(opt *models.MeasurementListOptions) ([]*models.
|
|||
}
|
||||
return measurements, nil
|
||||
}
|
||||
|
||||
func (s *measurementsStore) Update(id int64, measurement *models.Measurement) (bool, error) {
|
||||
_, err := s.Get(id)
|
||||
if err != nil {
|
||||
return false, err
|
||||
}
|
||||
|
||||
if id != measurement.Id {
|
||||
return false, models.ErrMeasurementNotFound
|
||||
}
|
||||
|
||||
measurement.UpdatedAt = time.Now()
|
||||
changed, err := s.dbh.Update(measurement)
|
||||
if err != nil {
|
||||
return false, err
|
||||
}
|
||||
|
||||
if changed == 0 {
|
||||
return false, ErrNoRowsUpdated
|
||||
}
|
||||
|
||||
return true, nil
|
||||
}
|
||||
|
|
|
@ -98,3 +98,23 @@ func TestMeasurementsStore_List_db(t *testing.T) {
|
|||
t.Errorf("got measurements %+v, want %+v", measurements, want)
|
||||
}
|
||||
}
|
||||
|
||||
func TestMeasurementsStore_Update_db(t *testing.T) {
|
||||
tx, _ := DB.Begin()
|
||||
defer tx.Rollback()
|
||||
|
||||
measurement := insertMeasurement(t, tx)
|
||||
|
||||
d := NewDatastore(tx)
|
||||
|
||||
// Tweak it
|
||||
measurement.MeasurementValue = sql.NullFloat64{Float64: 4.56, Valid: true}
|
||||
updated, err := d.Measurements.Update(measurement.Id, measurement)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
if !updated {
|
||||
t.Error("!updated")
|
||||
}
|
||||
}
|
||||
|
|
Reference in a new issue