Index: go/src/infra/gae/libs/gae/memory/testing_utils_test.go |
diff --git a/go/src/infra/gae/libs/wrapper/memory/testing_utils_test.go b/go/src/infra/gae/libs/gae/memory/testing_utils_test.go |
similarity index 58% |
rename from go/src/infra/gae/libs/wrapper/memory/testing_utils_test.go |
rename to go/src/infra/gae/libs/gae/memory/testing_utils_test.go |
index 59fc990dbf2174e07c9447c6be1615fbbe7071fe..81fdedbf46d2923625109b743eba0903397c4847 100644 |
--- a/go/src/infra/gae/libs/wrapper/memory/testing_utils_test.go |
+++ b/go/src/infra/gae/libs/gae/memory/testing_utils_test.go |
@@ -7,10 +7,10 @@ package memory |
import ( |
"bytes" |
"fmt" |
- "reflect" |
"time" |
- "appengine/datastore" |
+ "infra/gae/libs/gae" |
+ "infra/gae/libs/gae/helper" |
"github.com/luci/luci-go/common/cmpbin" |
) |
@@ -35,43 +35,30 @@ func indx(kind string, orders ...string) *qIndex { |
return ret |
} |
-func pl(props ...datastore.Property) *propertyList { |
- return (*propertyList)(&props) |
-} |
- |
-func prop(name string, val interface{}, noIndex ...bool) (ret datastore.Property) { |
- ret.Name = name |
- ret.Value = val |
+func prop(val interface{}, noIndex ...bool) (ret gae.DSProperty) { |
+ ni := false |
if len(noIndex) > 0 { |
- ret.NoIndex = noIndex[0] |
+ ni = noIndex[0] |
+ } |
+ if err := ret.SetValue(val, ni); err != nil { |
+ panic(err) |
} |
return |
} |
-func key(kind string, id interface{}, parent ...*datastore.Key) *datastore.Key { |
- stringID := "" |
- intID := int64(0) |
+func key(kind string, id interface{}, parent ...gae.DSKey) gae.DSKey { |
+ p := gae.DSKey(nil) |
+ if len(parent) > 0 { |
+ p = parent[0] |
+ } |
switch x := id.(type) { |
case string: |
- stringID = x |
+ return helper.NewDSKey(globalAppID, "ns", kind, x, 0, p) |
case int: |
- intID = int64(x) |
+ return helper.NewDSKey(globalAppID, "ns", kind, "", int64(x), p) |
default: |
panic(fmt.Errorf("what the %T: %v", id, id)) |
} |
- par := (*datastore.Key)(nil) |
- if len(parent) > 0 { |
- par = parent[0] |
- } |
- return newKey("ns", kind, stringID, intID, par) |
-} |
- |
-func mustLoadLocation(loc string) *time.Location { |
- if z, err := time.LoadLocation(loc); err != nil { |
- panic(err) |
- } else { |
- return z |
- } |
} |
// cat is a convenience method for concatenating anything with an underlying |
@@ -80,22 +67,28 @@ func cat(bytethings ...interface{}) []byte { |
buf := &bytes.Buffer{} |
for _, thing := range bytethings { |
switch x := thing.(type) { |
- case int, int64: |
- cmpbin.WriteInt(buf, reflect.ValueOf(x).Int()) |
- case uint, uint64: |
- cmpbin.WriteUint(buf, reflect.ValueOf(x).Uint()) |
+ case int64: |
+ cmpbin.WriteInt(buf, x) |
+ case int: |
+ cmpbin.WriteInt(buf, int64(x)) |
+ case uint64: |
+ cmpbin.WriteUint(buf, x) |
+ case uint: |
+ cmpbin.WriteUint(buf, uint64(x)) |
case float64: |
- writeFloat64(buf, x) |
- case byte, propValType: |
- buf.WriteByte(byte(reflect.ValueOf(x).Uint())) |
- case []byte, serializedPval: |
- buf.Write(reflect.ValueOf(x).Convert(byteSliceType).Interface().([]byte)) |
+ cmpbin.WriteFloat64(buf, x) |
+ case byte: |
+ buf.WriteByte(x) |
+ case gae.DSPropertyType: |
+ buf.WriteByte(byte(x)) |
case string: |
- writeString(buf, x) |
+ cmpbin.WriteString(buf, x) |
+ case []byte: |
+ buf.Write(x) |
case time.Time: |
- writeTime(buf, x) |
- case *datastore.Key: |
- writeKey(buf, noNS, x) |
+ helper.WriteTime(buf, x) |
+ case gae.DSKey: |
+ helper.WriteDSKey(buf, helper.WithoutContext, x) |
case *qIndex: |
x.WriteBinary(buf) |
default: |