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, 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, `{"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, 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, `{"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") 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") } }