Update a unit type

datastore/unit_types.go

@@ -46,3 +46,26 @@ func (s *unitTypesStore) List(opt *models.UnitTypeListOptions) ([]*models.UnitTy
 	}
 	return unit_types, nil
 }
+
+func (s *unitTypesStore) Update(id int64, unit_type *models.UnitType) (bool, error) {
+	_, err := s.Get(id)
+	if err != nil {
+		return false, err
+	}
+
+	if id != unit_type.Id {
+		return false, models.ErrUnitTypeNotFound
+	}
+
+	unit_type.UpdatedAt = time.Now()
+	changed, err := s.dbh.Update(unit_type)
+	if err != nil {
+		return false, err
+	}
+
+	if changed == 0 {
+		return false, ErrNoRowsUpdated
+	}
+
+	return true, nil
+}

datastore/unit_types_test.go

@@ -85,3 +85,23 @@ func TestUnitTypesStore_List_db(t *testing.T) {
 		t.Errorf("got unit_types %+v, want %+v", unit_types, want)
 	}
 }
+
+func TestUnitTypesStore_Update_db(t *testing.T) {
+	tx, _ := DB.Begin()
+	defer tx.Rollback()
+
+	unit_type := insertUnitType(t, tx)
+
+	d := NewDatastore(tx)
+
+	// Tweak it
+	unit_type.Name = "Updated Unit Type"
+	updated, err := d.UnitTypes.Update(unit_type.Id, unit_type)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if !updated {
+		t.Error("!updated")
+	}
+}