datastore: Update AllocateIDs to take keys.

impl/prod/everything_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.
 
 // +build appengine
 
 package prod
 
 import (
         "testing"
@@ skipping 50 matching lines @@
                         // of the test... yeah I know.
                         logging.Debugf(ctx, "SHOULD NOT SEE")
                         logging.Infof(ctx, "SHOULD SEE")
 
                         ctx = logging.SetLevel(ctx, logging.Debug)
                         logging.Debugf(ctx, "SHOULD SEE")
                         logging.Infof(ctx, "SHOULD SEE (2)")
                 })
 
                 Convey("Can probe/change Namespace", func() {
-                        So(inf.GetNamespace(), ShouldEqual, "")
+                        ns, has := inf.GetNamespace()
+                        So(ns, ShouldEqual, "")
+                        So(has, ShouldBeFalse)
+
                         ctx, err = inf.Namespace("wat")
                         So(err, ShouldBeNil)
                         inf = info.Get(ctx)
-                        So(inf.GetNamespace(), ShouldEqual, "wat")
+
+                        ns, has = inf.GetNamespace()
+                        So(ns, ShouldEqual, "wat")
+                        So(has, ShouldBeTrue)
+
                         ds = datastore.Get(ctx)
                         So(ds.MakeKey("Hello", "world").Namespace(), ShouldEqual, "wat")
                 })
 
                 Convey("Can get non-transactional context", func() {
                         ctx, err := inf.Namespace("foo")
                         So(err, ShouldBeNil)
                         ds = datastore.Get(ctx)
                         inf = info.Get(ctx)
 
@@ skipping 170 matching lines @@
 
                 Convey("memcache: Set (nil) is the same as Set ([]byte{})", func() {
                         So(mc.Set(mc.NewItem("bob")), ShouldBeNil) // normally would panic because Value is nil
 
                         bob, err := mc.Get("bob")
                         So(err, ShouldBeNil)
                         So(bob.Value(), ShouldResemble, []byte{})
                 })
         })
 }