Get a measurement

datastore/datastore.go

@@ -17,6 +17,7 @@ type Datastore struct {
 	Observations         models.ObservationsService
 	TextMeasurementTypes models.TextMeasurementTypesService
 	UnitTypes            models.UnitTypesService
+	Measurements         models.MeasurementsService
 	dbh                  modl.SqlExecutor
 }
 
@@ -41,6 +42,7 @@ func NewDatastore(dbh modl.SqlExecutor) *Datastore {
 	d.Observations = &observationsStore{d}
 	d.TextMeasurementTypes = &textMeasurementTypesStore{d}
 	d.UnitTypes = &unitTypesStore{d}
+	d.Measurements = &measurementsStore{d}
 	return d
 }
 
@@ -54,5 +56,6 @@ func NewMockDatastore() *Datastore {
 		Observations:         &models.MockObservationsService{},
 		TextMeasurementTypes: &models.MockTextMeasurementTypesService{},
 		UnitTypes:            &models.MockUnitTypesService{},
+		Measurements:         &models.MockMeasurementsService{},
 	}
 }

datastore/measurements.go

@@ -0,0 +1,22 @@
+package datastore
+
+import "github.com/thermokarst/bactdb/models"
+
+func init() {
+	DB.AddTableWithName(models.Measurement{}, "measurements").SetKeys(true, "Id")
+}
+
+type measurementsStore struct {
+	*Datastore
+}
+
+func (s *measurementsStore) Get(id int64) (*models.Measurement, error) {
+	var measurement []*models.Measurement
+	if err := s.dbh.Select(&measurement, `SELECT * FROM measurements WHERE id=$1;`, id); err != nil {
+		return nil, err
+	}
+	if len(measurement) == 0 {
+		return nil, models.ErrMeasurementNotFound
+	}
+	return measurement[0], nil
+}

datastore/measurements_test.go

@@ -0,0 +1,57 @@
+package datastore
+
+import (
+	"database/sql"
+	"reflect"
+	"testing"
+
+	"github.com/jmoiron/modl"
+	"github.com/thermokarst/bactdb/models"
+)
+
+func insertMeasurement(t *testing.T, tx *modl.Transaction) *models.Measurement {
+	// clean up our target table
+	tx.Exec(`DELETE FROM measurements;`)
+	measurement := newMeasurement(t, tx)
+	if err := tx.Insert(measurement); err != nil {
+		t.Fatal(err)
+	}
+	return measurement
+}
+
+func newMeasurement(t *testing.T, tx *modl.Transaction) *models.Measurement {
+	// we have a few things to take care of first...
+	strain := insertStrain(t, tx)
+	observation := insertObservation(t, tx)
+
+	// we want to create and insert a unit type record, too.
+	unit_type := insertUnitType(t, tx)
+
+	return &models.Measurement{
+		StrainId:         strain.Id,
+		ObservationId:    observation.Id,
+		MeasurementValue: sql.NullFloat64{Float64: 1.23, Valid: true},
+		UnitTypeId:       sql.NullInt64{Int64: unit_type.Id, Valid: true},
+	}
+}
+
+func TestMeasurementsStore_Get_db(t *testing.T) {
+	tx, _ := DB.Begin()
+	defer tx.Rollback()
+
+	want := insertMeasurement(t, tx)
+
+	d := NewDatastore(tx)
+
+	measurement, err := d.Measurements.Get(want.Id)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	normalizeTime(&want.CreatedAt, &want.UpdatedAt, &want.DeletedAt)
+	normalizeTime(&measurement.CreatedAt, &measurement.UpdatedAt, &measurement.DeletedAt)
+
+	if !reflect.DeepEqual(measurement, want) {
+		t.Errorf("got measurement %+v, want %+v", measurement, want)
+	}
+}