datastore: Update AllocateIDs to take keys.

filter/txnBuf/txnbuf_test.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 txnBuf
 
 import (
         "bytes"
         "fmt"
         "math/rand"
@@ skipping 112 matching lines @@
         }
 
         for i := range hugeData {
                 hugeData[i] = &Foo{ID: int64(i + 1), ValueNI: hugeField}
         }
 }
 
 func mkds(data []*Foo) (under, over *count.DSCounter, ds datastore.Interface) {
         c := memory.UseWithAppID(context.Background(), "something~else")
         ds = datastore.Get(c)
-        _, err := ds.AllocateIDs(ds.KeyForObj(data[0]), 100)
-        if err != nil {
+
+        dataKey := ds.KeyForObj(data[0])
+        if err := ds.AllocateIDs(ds.NewIncompleteKeys(100, dataKey.Kind(), dataKey.Parent())); err != nil {
                 panic(err)
         }
         if err := ds.PutMulti(data); err != nil {
                 panic(err)
         }
 
         c, under = count.FilterRDS(c)
         c = FilterRDS(c)
         c, over = count.FilterRDS(c)
         ds = datastore.Get(c)
@@ skipping 756 matching lines @@
                                         val := pm["Value"]
                                         So(val[len(val)-1].Value(), ShouldResemble, "wat")
                                 }), ShouldBeNil)
                         })
 
                 })
 
         })
 
 }