Create a measurement

datastore/measurements.go

@@ -1,6 +1,10 @@
 package datastore
 
-import "github.com/thermokarst/bactdb/models"
+import (
+	"time"
+
+	"github.com/thermokarst/bactdb/models"
+)
 
 func init() {
 	DB.AddTableWithName(models.Measurement{}, "measurements").SetKeys(true, "Id")
@@ -20,3 +24,13 @@ func (s *measurementsStore) Get(id int64) (*models.Measurement, error) {
 	}
 	return measurement[0], nil
 }
+
+func (s *measurementsStore) Create(measurement *models.Measurement) (bool, error) {
+	currentTime := time.Now()
+	measurement.CreatedAt = currentTime
+	measurement.UpdatedAt = currentTime
+	if err := s.dbh.Insert(measurement); err != nil {
+		return false, err
+	}
+	return true, nil
+}

datastore/measurements_test.go

@@ -55,3 +55,23 @@ func TestMeasurementsStore_Get_db(t *testing.T) {
 		t.Errorf("got measurement %+v, want %+v", measurement, want)
 	}
 }
+
+func TestMeasurementsStore_Create_db(t *testing.T) {
+	tx, _ := DB.Begin()
+	defer tx.Rollback()
+
+	measurement := newMeasurement(t, tx)
+
+	d := NewDatastore(tx)
+
+	created, err := d.Measurements.Create(measurement)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !created {
+		t.Error("!created")
+	}
+	if measurement.Id == 0 {
+		t.Error("want nonzero measurement.Id after submitting")
+	}
+}