datastore: Update AllocateIDs to take keys.

filter/txnBuf/txnbuf_test.go

Index: filter/txnBuf/txnbuf_test.go
diff --git a/filter/txnBuf/txnbuf_test.go b/filter/txnBuf/txnbuf_test.go
index aabdc5ed839bbfd29a2361eb9de507968b20eb52..936598bb246b7607c8a877e828280387e0550eb4 100644
--- a/filter/txnBuf/txnbuf_test.go
+++ b/filter/txnBuf/txnbuf_test.go
@@ -130,7 +130,9 @@ func init() {
 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)
+
+	dataKey := ds.KeyForObj(data[0])
+	_, err := ds.AllocateIDs(ds.NewIncompleteKeys(100, dataKey.Kind(), dataKey.Parent())...)
 	if err != nil {
 		panic(err)
 	}