bactdb/models/measurements_test.go

package models

import (
	"database/sql"
	"net/http"
	"reflect"
	"testing"

	"github.com/thermokarst/bactdb/router"
)

func newMeasurement() *Measurement {
	measurement := NewMeasurement()
	measurement.Id = 1
	measurement.StrainId = 1
	measurement.CharacteristicId = 1
	measurement.UnitTypeId = NullInt64{sql.NullInt64{Int64: 1, Valid: true}}
	measurement.Notes = NullString{sql.NullString{String: "a note", Valid: true}}
	return measurement
}

func TestMeasurementService_Get(t *testing.T) {
	setup()
	defer teardown()

	want := newMeasurement()

	var called bool
	mux.HandleFunc(urlPath(t, router.Measurement, map[string]string{"Id": "1"}), func(w http.ResponseWriter, r *http.Request) {
		called = true
		testMethod(t, r, "GET")

		writeJSON(w, MeasurementJSON{Measurement: want})
	})

	measurement, err := client.Measurements.Get(want.Id)
	if err != nil {
		t.Errorf("Measurements.Get returned error: %v", err)
	}

	if !called {
		t.Fatal("!called")
	}

	normalizeTime(&want.CreatedAt, &want.UpdatedAt)

	if !reflect.DeepEqual(measurement, want) {
		t.Errorf("Measurements.Get return %+v, want %+v", measurement, want)
	}
}

func TestMeasurementService_Create(t *testing.T) {
	setup()
	defer teardown()

	want := newMeasurement()

	var called bool
	mux.HandleFunc(urlPath(t, router.CreateMeasurement, nil), func(w http.ResponseWriter, r *http.Request) {
		called = true
		testMethod(t, r, "POST")
		testBody(t, r, `{"measurement":{"id":1,"strainId":1,"characteristicId":1,"textMeasurementTypeId":null,"txtValue":null,"numValue":1.23,"confidenceInterval":null,"unitTypeId":1,"notes":"a note","testMethodId":null,"createdAt":"0001-01-01T00:00:00Z","updatedAt":"0001-01-01T00:00:00Z"}}`+"\n")

		w.WriteHeader(http.StatusCreated)
		writeJSON(w, want)
	})

	measurement := newMeasurement()
	created, err := client.Measurements.Create(measurement)
	if err != nil {
		t.Errorf("Measurements.Create returned error: %v", err)
	}

	if !created {
		t.Error("!created")
	}

	if !called {
		t.Fatal("!called")
	}

	normalizeTime(&want.CreatedAt, &want.UpdatedAt)
	if !reflect.DeepEqual(measurement, want) {
		t.Errorf("Measurements.Create returned %+v, want %+v", measurement, want)
	}
}

func TestMeasurementService_List(t *testing.T) {
	setup()
	defer teardown()

	want := []*Measurement{newMeasurement()}

	var called bool
	mux.HandleFunc(urlPath(t, router.Measurements, nil), func(w http.ResponseWriter, r *http.Request) {
		called = true
		testMethod(t, r, "GET")
		testFormValues(t, r, values{})

		writeJSON(w, MeasurementsJSON{Measurements: want})
	})

	measurements, err := client.Measurements.List(nil)
	if err != nil {
		t.Errorf("Measurements.List returned error: %v", err)
	}

	if !called {
		t.Fatal("!called")
	}

	for _, u := range want {
		normalizeTime(&u.CreatedAt, &u.UpdatedAt)
	}

	if !reflect.DeepEqual(measurements, want) {
		t.Errorf("Measurements.List return %+v, want %+v", measurements, want)
	}
}

func TestMeasurementService_Update(t *testing.T) {
	setup()
	defer teardown()

	want := newMeasurement()

	var called bool
	mux.HandleFunc(urlPath(t, router.UpdateMeasurement, map[string]string{"Id": "1"}), func(w http.ResponseWriter, r *http.Request) {
		called = true
		testMethod(t, r, "PUT")
		testBody(t, r, `{"measurement":{"id":1,"strainId":1,"characteristicId":1,"textMeasurementTypeId":null,"txtValue":null,"numValue":4.56,"confidenceInterval":null,"unitTypeId":1,"notes":"a note","testMethodId":null,"createdAt":"0001-01-01T00:00:00Z","updatedAt":"0001-01-01T00:00:00Z"}}`+"\n")
		w.WriteHeader(http.StatusOK)
		writeJSON(w, want)
	})

	measurement := newMeasurement()
	measurement.NumValue = NullFloat64{sql.NullFloat64{Float64: 4.56, Valid: true}}
	updated, err := client.Measurements.Update(measurement.Id, measurement)
	if err != nil {
		t.Errorf("Measurements.Update returned error: %v", err)
	}

	if !updated {
		t.Error("!updated")
	}

	if !called {
		t.Fatal("!called")
	}
}

func TestMeasurementService_Delete(t *testing.T) {
	setup()
	defer teardown()

	want := newMeasurement()

	var called bool
	mux.HandleFunc(urlPath(t, router.DeleteMeasurement, map[string]string{"Id": "1"}), func(w http.ResponseWriter, r *http.Request) {
		called = true
		testMethod(t, r, "DELETE")
		testBody(t, r, "")

		w.WriteHeader(http.StatusOK)
		writeJSON(w, want)
	})

	deleted, err := client.Measurements.Delete(want.Id)
	if err != nil {
		t.Errorf("Measurements.Delete returned error: %v", err)
	}

	if !deleted {
		t.Error("!deleted")
	}

	if !called {
		t.Fatal("!called")
	}
}