impl/memory: Replace gkvlite with "treapstore".

impl/memory/datastore_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 memory
 
 import (
         "errors"
         "fmt"
         "testing"
@@ skipping 647 matching lines @@
 }
 
 func TestCompoundIndexes(t *testing.T) {
         t.Parallel()
 
         idxKey := func(def ds.IndexDefinition) string {
                 So(def, ShouldNotBeNil)
                 return "idx::" + string(serialize.ToBytes(*def.PrepForIdxTable()))
         }
 
-        numItms := func(c memCollection) uint64 {
-                ret, _ := c.GetTotals()
-                return ret
-        }
-
         Convey("Test Compound indexes", t, func() {
                 type Model struct {
                         ID int64 `gae:"$id"`
 
                         Field1 []string
                         Field2 []int64
                 }
 
                 c := Use(context.Background())
                 t := ds.GetTestable(c).(*dsImpl)
                 head := t.data.head
 
                 So(ds.Put(c, &Model{1, []string{"hello", "world"}, []int64{10, 11}}), ShouldBeNil)
 
                 idx := ds.IndexDefinition{
                         Kind: "Model",
                         SortBy: []ds.IndexColumn{
                                 {Property: "Field2"},
                         },
                 }
 
-                coll := head.GetCollection(idxKey(idx))
+                coll := head.Snapshot().GetCollection(idxKey(idx))
                 So(coll, ShouldNotBeNil)
-                So(numItms(coll), ShouldEqual, 2)
+                So(countItems(coll), ShouldEqual, 2)
 
                 idx.SortBy[0].Property = "Field1"
-                coll = head.GetCollection(idxKey(idx))
+                coll = head.Snapshot().GetCollection(idxKey(idx))
                 So(coll, ShouldNotBeNil)
-                So(numItms(coll), ShouldEqual, 2)
+                So(countItems(coll), ShouldEqual, 2)
 
                 idx.SortBy = append(idx.SortBy, ds.IndexColumn{Property: "Field1"})
                 So(head.GetCollection(idxKey(idx)), ShouldBeNil)
 
                 t.AddIndexes(&idx)
-                coll = head.GetCollection(idxKey(idx))
+                coll = head.Snapshot().GetCollection(idxKey(idx))
                 So(coll, ShouldNotBeNil)
-                So(numItms(coll), ShouldEqual, 4)
+                So(countItems(coll), ShouldEqual, 4)
         })
 }
 
 // High level test for regression in how zero time is stored,
 // see https://codereview.chromium.org/1334043003/
 func TestDefaultTimeField(t *testing.T) {
         t.Parallel()
 
         Convey("Default time.Time{} can be stored", t, func() {
                 type Model struct {
@@ skipping 88 matching lines @@
                         // Add "foos" to a new namespace, then confirm that it gets indexed.
                         So(ds.Put(infoS.MustNamespace(ctx, "qux"), foos), ShouldBeNil)
                         ds.GetTestable(ctx).CatchupIndexes()
 
                         results = nil
                         So(ds.GetAll(infoS.MustNamespace(ctx, "qux"), q, &results), ShouldBeNil)
                         So(len(results), ShouldEqual, 2)
                 })
         })
 }