// Remove removes an item into the items collection and remove any // corresponding quads from the graph database. func Remove(context interface{}, db *db.DB, graph *cayley.Handle, itemID string) error { log.Dev(context, "Remove", "Started : ID[%s]", itemID) // Get the item from the items collection. items, err := item.GetByIDs(context, db, []string{itemID}) if err != nil { log.Error(context, "Remove", err, "Completed") return err } // Prepare the item map data. itmMap := map[string]interface{}{ "item_id": items[0].ID, "type": items[0].Type, "version": items[0].Version, "data": items[0].Data, } // Remove the corresponding relationships from the graph. if err := wire.RemoveFromGraph(context, db, graph, itmMap); err != nil { log.Error(context, "Remove", err, "Completed") return err } // Delete the item. if err := item.Delete(context, db, itemID); err != nil { log.Error(context, "Remove", err, "Completed") return err } log.Dev(context, "Remove", "Completed") return nil }
// TestGetByIDs tests if we can get items from the db. func TestGetByIDs(t *testing.T) { tests.ResetLog() defer tests.DisplayLog() db, err := db.NewMGO(tests.Context, tests.TestSession) if err != nil { t.Fatalf("\t%s\tShould be able to get a Mongo session : %v", tests.Failed, err) } defer db.CloseMGO(tests.Context) defer func() { if err := itemfix.Remove(tests.Context, db, prefix); err != nil { t.Fatalf("\t%s\tShould be able to remove the items : %v", tests.Failed, err) } t.Logf("\t%s\tShould be able to remove the items.", tests.Success) }() t.Log("Given the need to get items in the database by IDs.") { t.Log("\tWhen starting from an empty items collection") { items1, err := itemfix.Get() if err != nil { t.Fatalf("\t%s\tShould be able retrieve item fixture : %s", tests.Failed, err) } var itemIDs []string for _, it := range items1 { if err := item.Upsert(tests.Context, db, &it); err != nil { t.Fatalf("\t%s\tShould be able to upsert items : %s", tests.Failed, err) } itemIDs = append(itemIDs, it.ID) } t.Logf("\t%s\tShould be able to upsert items.", tests.Success) items2, err := item.GetByIDs(tests.Context, db, itemIDs) if err != nil { t.Fatalf("\t%s\tShould be able to get items by IDs : %s", tests.Failed, err) } t.Logf("\t%s\tShould be able to get items by IDs.", tests.Success) if len(items1) != len(items2) { t.Logf("\t%+v", items1) t.Logf("\t%+v", items2) t.Fatalf("\t%s\tShould be able to get back the same items.", tests.Failed) } t.Logf("\t%s\tShould be able to get back the same items.", tests.Success) } } }
// Retrieve returns the items, specified by IDs, from the system. // 200 Success, 400 Bad Request, 404 Not Found, 500 Internal func (itemHandle) Retrieve(c *web.Context) error { var items []item.Item ids := strings.Split(c.Params["id"], ",") items, err := item.GetByIDs(c.SessionID, c.Ctx["DB"].(*db.DB), ids) if err != nil { if err == item.ErrNotFound { err = web.ErrNotFound } return err } c.Respond(items, http.StatusOK) return nil }
// TestUpsertDelete tests if we can add/remove an item to/from the db. func TestUpsertDelete(t *testing.T) { tests.ResetLog() defer tests.DisplayLog() db, err := db.NewMGO(tests.Context, tests.TestSession) if err != nil { t.Fatalf("\t%s\tShould be able to get a Mongo session : %v", tests.Failed, err) } defer db.CloseMGO(tests.Context) defer func() { if err := itemfix.Remove(tests.Context, db, prefix); err != nil { t.Fatalf("\t%s\tShould be able to remove the items : %v", tests.Failed, err) } t.Logf("\t%s\tShould be able to remove the items.", tests.Success) }() t.Log("Given the need to upsert and delete items.") { t.Log("\tWhen starting from an empty items collection") { //---------------------------------------------------------------------- // Get the fixture. items, err := itemfix.Get() if err != nil { t.Fatalf("\t%s\tShould be able retrieve item fixture : %s", tests.Failed, err) } //---------------------------------------------------------------------- // Upsert the item. if err := item.Upsert(tests.Context, db, &items[0]); err != nil { t.Fatalf("\t%s\tShould be able to upsert a item : %s", tests.Failed, err) } t.Logf("\t%s\tShould be able to upsert a item.", tests.Success) //---------------------------------------------------------------------- // Get the item. itemsBack, err := item.GetByIDs(tests.Context, db, []string{items[0].ID}) if err != nil { t.Fatalf("\t%s\tShould be able to get the item by ID : %s", tests.Failed, err) } t.Logf("\t%s\tShould be able to get the item by ID.", tests.Success) //---------------------------------------------------------------------- // Check that we got the item we expected. if items[0].ID != itemsBack[0].ID { t.Logf("\t%+v", items[0]) t.Logf("\t%+v", itemsBack[0]) t.Fatalf("\t%s\tShould be able to get back the same item.", tests.Failed) } t.Logf("\t%s\tShould be able to get back the same item.", tests.Success) //---------------------------------------------------------------------- // Check that CreatedAt and UpdatedAt were set. if itemsBack[0].CreatedAt.IsZero() || itemsBack[0].UpdatedAt.IsZero() { t.Fatalf("\t%s\tShould set CreatedAt and UpdatedAt on upsert.", tests.Failed) } t.Logf("\t%s\tShould set CreatedAt and UpdatedAt on upsert.", tests.Success) //---------------------------------------------------------------------- // Delete the item. if err := item.Delete(tests.Context, db, items[0].ID); err != nil { t.Fatalf("\t%s\tShould be able to delete the item : %s", tests.Failed, err) } t.Logf("\t%s\tShould be able to delete the item.", tests.Success) //---------------------------------------------------------------------- // Get the item. itemsBack, err = item.GetByIDs(tests.Context, db, []string{items[0].ID}) if len(itemsBack) != 0 { t.Fatalf("\t%s\tShould generate an error when getting an item with the deleted ID : %s", tests.Failed, err) } t.Logf("\t%s\tShould generate an error when getting an item with the deleted ID.", tests.Success) } } }