Delete an observation type

This commit is contained in:
Matthew Dillon 2014-11-10 15:24:16 -09:00
parent 73f7b580c1
commit 68b98f6d6b
9 changed files with 141 additions and 0 deletions

View file

@ -45,6 +45,7 @@ func Handler() *mux.Router {
m.Get(router.CreateObservationType).Handler(handler(serveCreateObservationType))
m.Get(router.ObservationTypes).Handler(handler(serveObservationTypeList))
m.Get(router.UpdateObservationType).Handler(handler(serveUpdateObservationType))
m.Get(router.DeleteObservationType).Handler(handler(serveDeleteObservationType))
return m
}

View file

@ -76,3 +76,17 @@ func serveUpdateObservationType(w http.ResponseWriter, r *http.Request) error {
return writeJSON(w, observation_type)
}
func serveDeleteObservationType(w http.ResponseWriter, r *http.Request) error {
id, _ := strconv.ParseInt(mux.Vars(r)["Id"], 10, 0)
deleted, err := store.ObservationTypes.Delete(id)
if err != nil {
return err
}
if deleted {
w.WriteHeader(http.StatusOK)
}
return writeJSON(w, &models.ObservationType{})
}

View file

@ -124,3 +124,30 @@ func TestObservationType_Update(t *testing.T) {
t.Error("!success")
}
}
func TestObservationType_Delete(t *testing.T) {
setup()
want := newObservationType()
calledDelete := false
store.ObservationTypes.(*models.MockObservationTypesService).Delete_ = func(id int64) (bool, error) {
if id != want.Id {
t.Errorf("wanted request for observation_type %d but got %d", want.Id, id)
}
calledDelete = true
return true, nil
}
success, err := apiClient.ObservationTypes.Delete(want.Id)
if err != nil {
t.Fatal(err)
}
if !calledDelete {
t.Error("!calledDelete")
}
if !success {
t.Error("!success")
}
}

View file

@ -69,3 +69,19 @@ func (s *observationTypesStore) Update(id int64, observation_type *models.Observ
return true, nil
}
func (s *observationTypesStore) Delete(id int64) (bool, error) {
observation_type, err := s.Get(id)
if err != nil {
return false, err
}
deleted, err := s.dbh.Delete(observation_type)
if err != nil {
return false, err
}
if deleted == 0 {
return false, ErrNoRowsDeleted
}
return true, nil
}

View file

@ -105,3 +105,22 @@ func TestObservationTypesStore_Update_db(t *testing.T) {
t.Error("!updated")
}
}
func TestObservationTypesStore_Delete_db(t *testing.T) {
tx, _ := DB.Begin()
defer tx.Rollback()
observation_type := insertObservationType(t, tx)
d := NewDatastore(tx)
// Delete it
deleted, err := d.ObservationTypes.Delete(observation_type.Id)
if err != nil {
t.Fatal(err)
}
if !deleted {
t.Error("!delete")
}
}

View file

@ -37,6 +37,9 @@ type ObservationTypesService interface {
// Update an existing observation type
Update(id int64, observation_type *ObservationType) (updated bool, err error)
// Delete an existing observation type
Delete(id int64) (deleted bool, err error)
}
var (
@ -133,11 +136,34 @@ func (s *observationTypesService) Update(id int64, observation_type *Observation
return resp.StatusCode == http.StatusOK, nil
}
func (s *observationTypesService) Delete(id int64) (bool, error) {
strId := strconv.FormatInt(id, 10)
url, err := s.client.url(router.DeleteObservationType, map[string]string{"Id": strId}, nil)
if err != nil {
return false, err
}
req, err := s.client.NewRequest("DELETE", url.String(), nil)
if err != nil {
return false, err
}
var observation_type *ObservationType
resp, err := s.client.Do(req, &observation_type)
if err != nil {
return false, err
}
return resp.StatusCode == http.StatusOK, nil
}
type MockObservationTypesService struct {
Get_ func(id int64) (*ObservationType, error)
List_ func(opt *ObservationTypeListOptions) ([]*ObservationType, error)
Create_ func(observation_type *ObservationType) (bool, error)
Update_ func(id int64, observation_type *ObservationType) (bool, error)
Delete_ func(id int64) (bool, error)
}
var _ ObservationTypesService = &MockObservationTypesService{}
@ -169,3 +195,10 @@ func (s *MockObservationTypesService) Update(id int64, observation_type *Observa
}
return s.Update_(id, observation_type)
}
func (s *MockObservationTypesService) Delete(id int64) (bool, error) {
if s.Delete_ == nil {
return false, nil
}
return s.Delete_(id)
}

View file

@ -143,3 +143,32 @@ func TestObservationTypeService_Update(t *testing.T) {
t.Fatal("!called")
}
}
func TestObservationTypeService_Delete(t *testing.T) {
setup()
defer teardown()
want := newObservationType()
var called bool
mux.HandleFunc(urlPath(t, router.DeleteObservationType, 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.ObservationTypes.Delete(want.Id)
if err != nil {
t.Errorf("ObservationTypes.Delete returned error: %v", err)
}
if !deleted {
t.Error("!deleted")
}
if !called {
t.Fatal("!called")
}
}

View file

@ -36,6 +36,7 @@ func API() *mux.Router {
m.Path("/observation_types").Methods("POST").Name(CreateObservationType)
m.Path("/observation_types/{Id:.+}").Methods("GET").Name(ObservationType)
m.Path("/observation_types/{Id:.+}").Methods("PUT").Name(UpdateObservationType)
m.Path("/observation_types/{Id:.+}").Methods("DELETE").Name(DeleteObservationType)
return m
}

View file

@ -27,4 +27,5 @@ const (
CreateObservationType = "observation_type:create"
ObservationTypes = "observation_type:list"
UpdateObservationType = "observation_type:update"
DeleteObservationType = "observation_type:delete"
)