func DeleteModel(ctx appengine.Context, model models.Model) error {
key, err := model.Key()
if err != nil {
return err
}
return datastore.Delete(ctx, key)
}
// If the key has already been added to the model, we can just generate
// the key argument from the model.
func GetModelWithKey(ctx appengine.Context, model models.Model) error {
key, err := model.Key()
if err != nil {
return err
}
return GetModel(ctx, key, model)
}
// Store a Model, and call SetKey after generating its key (if it doesn't
// have one already).
func PutModel(ctx appengine.Context, model models.Model) error {
modelKey, err := model.Key()
if err == models.NoKeyLoaded {
modelKey = datastore.NewIncompleteKey(ctx, model.Kind(), nil)
} else if err != nil {
panic(err)
}
key, err := datastore.Put(ctx, modelKey, model)
if err == nil {
model.SetKey(key)
}
return err
}
func (suite *datastoreTestSuite) TestModelIterator() {
loadTestData(suite.GAEContext)
query := NewModelQuery(kindForTest())
count, err := query.Count(suite.GAEContext)
assert.Nil(suite.T, err, "Query should be countable.")
assert.NotEqual(suite.T, count, 0, "Data should exist in the database by now.")
iterator := query.Run(suite.GAEContext)
var (
entry models.Model = new(testModel)
key *datastore.Key
)
for _, err := iterator.Next(entry); err != datastore.Done; _, err = iterator.Next(entry) {
if err != nil {
panic(err)
}
key, err = entry.Key()
assert.NotNil(suite.T, key, "Query iterators should populate entry keys.")
}
}
