Get a text measurement type

datastore/datastore.go

@@ -9,13 +9,14 @@ import (
 
 // A datastore access point (in PostgreSQL)
 type Datastore struct {
-	Users            models.UsersService
-	Genera           models.GeneraService
-	Species          models.SpeciesService
-	Strains          models.StrainsService
-	ObservationTypes models.ObservationTypesService
-	Observations     models.ObservationsService
-	dbh              modl.SqlExecutor
+	Users                models.UsersService
+	Genera               models.GeneraService
+	Species              models.SpeciesService
+	Strains              models.StrainsService
+	ObservationTypes     models.ObservationTypesService
+	Observations         models.ObservationsService
+	TextMeasurementTypes models.TextMeasurementTypesService
+	dbh                  modl.SqlExecutor
 }
 
 var (
@@ -37,16 +38,18 @@ func NewDatastore(dbh modl.SqlExecutor) *Datastore {
 	d.Strains = &strainsStore{d}
 	d.ObservationTypes = &observationTypesStore{d}
 	d.Observations = &observationsStore{d}
+	d.TextMeasurementTypes = &textMeasurementTypesStore{d}
 	return d
 }
 
 func NewMockDatastore() *Datastore {
 	return &Datastore{
-		Users:            &models.MockUsersService{},
-		Genera:           &models.MockGeneraService{},
-		Species:          &models.MockSpeciesService{},
-		Strains:          &models.MockStrainsService{},
-		ObservationTypes: &models.MockObservationTypesService{},
-		Observations:     &models.MockObservationsService{},
+		Users:                &models.MockUsersService{},
+		Genera:               &models.MockGeneraService{},
+		Species:              &models.MockSpeciesService{},
+		Strains:              &models.MockStrainsService{},
+		ObservationTypes:     &models.MockObservationTypesService{},
+		Observations:         &models.MockObservationsService{},
+		TextMeasurementTypes: &models.MockTextMeasurementTypesService{},
 	}
 }

datastore/text_measurement_types.go

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

datastore/text_measurement_types_test.go

@@ -0,0 +1,44 @@
+package datastore
+
+import (
+	"reflect"
+	"testing"
+
+	"github.com/jmoiron/modl"
+	"github.com/thermokarst/bactdb/models"
+)
+
+func insertTextMeasurementType(t *testing.T, tx *modl.Transaction) *models.TextMeasurementType {
+	// clean up our target table
+	tx.Exec(`DELETE FROM text_measurement_types;`)
+	text_measurement_type := newTextMeasurementType(t, tx)
+	if err := tx.Insert(text_measurement_type); err != nil {
+		t.Fatal(err)
+	}
+	return text_measurement_type
+}
+
+func newTextMeasurementType(t *testing.T, tx *modl.Transaction) *models.TextMeasurementType {
+	return &models.TextMeasurementType{TextMeasurementName: "Test Text Measurement Type"}
+}
+
+func TestTextMeasurementTypesStore_Get_db(t *testing.T) {
+	tx, _ := DB.Begin()
+	defer tx.Rollback()
+
+	want := insertTextMeasurementType(t, tx)
+
+	d := NewDatastore(tx)
+
+	text_measurement_type, err := d.TextMeasurementTypes.Get(want.Id)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	normalizeTime(&want.CreatedAt, &want.UpdatedAt, &want.DeletedAt)
+	normalizeTime(&text_measurement_type.CreatedAt, &text_measurement_type.UpdatedAt, &text_measurement_type.DeletedAt)
+
+	if !reflect.DeepEqual(text_measurement_type, want) {
+		t.Errorf("got text_measurement_type %+v, want %+v", text_measurement_type, want)
+	}
+}