datastore: Update AllocateIDs to take keys.

impl/dummy/dummy.go

 // Copyright 2015 The LUCI Authors. All rights reserved.
 // Use of this source code is governed under the Apache License, Version 2.0
 // that can be found in the LICENSE file.
 
 package dummy
 
 import (
         "fmt"
         "runtime"
         "strings"
@@ skipping 55 matching lines @@
                 }
         }
 
         return fmt.Errorf(niFmtStr, iface, funcName)
 }
 
 /////////////////////////////////// ds ////////////////////////////////////
 
 type ds struct{}
 
-func (ds) AllocateIDs(*datastore.Key, int) (int64, error) {
-        panic(ni())
-}
-func (ds) PutMulti([]*datastore.Key, []datastore.PropertyMap, datastore.PutMultiCB) error {
-        panic(ni())
-}
+func (ds) AllocateIDs([]*datastore.Key, datastore.NewKeyCB) error                       { panic(ni()) }
+func (ds) PutMulti([]*datastore.Key, []datastore.PropertyMap, datastore.NewKeyCB) error { panic(ni()) }
 func (ds) GetMulti([]*datastore.Key, datastore.MultiMetaGetter, datastore.GetMultiCB) error {
         panic(ni())
 }
 func (ds) DeleteMulti([]*datastore.Key, datastore.DeleteMultiCB) error { panic(ni()) }
 func (ds) DecodeCursor(string) (datastore.Cursor, error)               { panic(ni()) }
 func (ds) Count(*datastore.FinalizedQuery) (int64, error)              { panic(ni()) }
 func (ds) Run(*datastore.FinalizedQuery, datastore.RawRunCB) error     { panic(ni()) }
 func (ds) RunInTransaction(func(context.Context) error, *datastore.TransactionOptions) error {
         panic(ni())
 }
@@ skipping 123 matching lines @@
 func (mod) DefaultVersion(module string) (string, error)                { panic(ni()) }
 func (mod) Start(module, version string) error                          { panic(ni()) }
 func (mod) Stop(module, version string) error                           { panic(ni()) }
 
 var dummyModuleInst = mod{}
 
 // Module returns a dummy module.Interface implementation suitable for
 // embedding. Every method panics with a message containing the name of the
 // method which was unimplemented.
 func Module() module.Interface { return dummyModuleInst }