datastore: Update AllocateIDs to take keys.

service/datastore/checkfilter_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.
 
 // adapted from github.com/golang/appengine/datastore
 
 package datastore
 
 import (
         "testing"
 
         "github.com/luci/gae/service/info"
+        "golang.org/x/net/context"
+
         . "github.com/smartystreets/goconvey/convey"
-        "golang.org/x/net/context"
 )
 
 type fakeRDS struct{ RawInterface }
 
 func TestCheckFilter(t *testing.T) {
         t.Parallel()
 
         Convey("Test checkFilter", t, func() {
                 // Note that the way we have this context set up, any calls which aren't
                 // stopped at the checkFilter will nil-pointer panic. We use this panic
@@ skipping 104 matching lines @@
                                 So(rds.DeleteMulti([]*Key{mkKey("s~aid", "ns", "Kind", 1)}, func(error) error {
                                         hit = true
                                         return nil
                                 }), ShouldBeNil)
                         }, ShouldPanic)
                         So(hit, ShouldBeFalse)
                 })
 
         })
 }