Index: go/src/infra/gae/libs/wrapper/memory/memcache_test.go |
diff --git a/go/src/infra/gae/libs/gae/memory/memcache_test.go b/go/src/infra/gae/libs/wrapper/memory/memcache_test.go |
similarity index 59% |
rename from go/src/infra/gae/libs/gae/memory/memcache_test.go |
rename to go/src/infra/gae/libs/wrapper/memory/memcache_test.go |
index 0829aa40185f6da07d4d7b1ec71ddfdcaaa19da0..ab1934c7aea78b8a39f47e86d67d223e6ab8764c 100644 |
--- a/go/src/infra/gae/libs/gae/memory/memcache_test.go |
+++ b/go/src/infra/gae/libs/wrapper/memory/memcache_test.go |
@@ -5,14 +5,16 @@ |
package memory |
import ( |
- "golang.org/x/net/context" |
- "infra/gae/libs/gae" |
+ "infra/gae/libs/wrapper" |
+ "infra/gae/libs/wrapper/unsafe" |
+ "infra/libs/clock/testclock" |
"testing" |
"time" |
- "github.com/luci/luci-go/common/clock/testclock" |
- |
. "github.com/smartystreets/goconvey/convey" |
+ "golang.org/x/net/context" |
+ |
+ "appengine/memcache" |
) |
func TestMemcache(t *testing.T) { |
@@ -22,45 +24,45 @@ func TestMemcache(t *testing.T) { |
now := time.Date(2015, 1, 1, 0, 0, 0, 0, time.UTC) |
c, tc := testclock.UseTime(context.Background(), now) |
c = Use(c) |
- mc := gae.GetMC(c) |
- mci := gae.GetMC(c).(*memcacheImpl) |
+ mc := wrapper.GetMC(c) |
+ mci := wrapper.GetMC(c).(*memcacheImpl) |
So(mc, ShouldNotEqual, mci) // two impls with the same memcacheData |
Convey("implements MCSingleReadWriter", func() { |
Convey("Add", func() { |
- itm := &mcItem{ |
- key: "sup", |
- value: []byte("cool"), |
- expiration: time.Second, |
+ itm := &memcache.Item{ |
+ Key: "sup", |
+ Value: []byte("cool"), |
+ Expiration: time.Second, |
} |
err := mc.Add(itm) |
So(err, ShouldBeNil) |
Convey("which rejects objects already there", func() { |
err := mc.Add(itm) |
- So(err, ShouldEqual, gae.ErrMCNotStored) |
+ So(err, ShouldEqual, memcache.ErrNotStored) |
}) |
Convey("which can be broken intentionally", func() { |
mci.BreakFeatures(nil, "Add") |
err := mc.Add(itm) |
- So(err, ShouldEqual, gae.ErrMCServerError) |
+ So(err, ShouldEqual, memcache.ErrServerError) |
}) |
}) |
Convey("Get", func() { |
- itm := &mcItem{ |
- key: "sup", |
- value: []byte("cool"), |
- expiration: time.Second, |
+ itm := &memcache.Item{ |
+ Key: "sup", |
+ Value: []byte("cool"), |
+ Expiration: time.Second, |
} |
err := mc.Add(itm) |
So(err, ShouldBeNil) |
- testItem := &mcItem{ |
- key: "sup", |
- value: []byte("cool"), |
- CasID: 1, |
+ testItem := &memcache.Item{ |
+ Key: "sup", |
+ Value: []byte("cool"), |
} |
+ unsafe.MCSetCasID(testItem, 1) |
i, err := mc.Get("sup") |
So(err, ShouldBeNil) |
So(i, ShouldResemble, testItem) |
@@ -68,23 +70,23 @@ func TestMemcache(t *testing.T) { |
Convey("which can expire", func() { |
tc.Add(time.Second * 4) |
i, err := mc.Get("sup") |
- So(err, ShouldEqual, gae.ErrMCCacheMiss) |
+ So(err, ShouldEqual, memcache.ErrCacheMiss) |
So(i, ShouldBeNil) |
}) |
Convey("which can be broken intentionally", func() { |
mci.BreakFeatures(nil, "Get") |
_, err := mc.Get("sup") |
- So(err, ShouldEqual, gae.ErrMCServerError) |
+ So(err, ShouldEqual, memcache.ErrServerError) |
}) |
}) |
Convey("Delete", func() { |
Convey("works if it's there", func() { |
- itm := &mcItem{ |
- key: "sup", |
- value: []byte("cool"), |
- expiration: time.Second, |
+ itm := &memcache.Item{ |
+ Key: "sup", |
+ Value: []byte("cool"), |
+ Expiration: time.Second, |
} |
err := mc.Add(itm) |
So(err, ShouldBeNil) |
@@ -93,40 +95,40 @@ func TestMemcache(t *testing.T) { |
So(err, ShouldBeNil) |
i, err := mc.Get("sup") |
- So(err, ShouldEqual, gae.ErrMCCacheMiss) |
+ So(err, ShouldEqual, memcache.ErrCacheMiss) |
So(i, ShouldBeNil) |
}) |
Convey("but not if it's not there", func() { |
err := mc.Delete("sup") |
- So(err, ShouldEqual, gae.ErrMCCacheMiss) |
+ So(err, ShouldEqual, memcache.ErrCacheMiss) |
}) |
Convey("and can be broken", func() { |
mci.BreakFeatures(nil, "Delete") |
err := mc.Delete("sup") |
- So(err, ShouldEqual, gae.ErrMCServerError) |
+ So(err, ShouldEqual, memcache.ErrServerError) |
}) |
}) |
Convey("Set", func() { |
- itm := &mcItem{ |
- key: "sup", |
- value: []byte("cool"), |
- expiration: time.Second, |
+ itm := &memcache.Item{ |
+ Key: "sup", |
+ Value: []byte("cool"), |
+ Expiration: time.Second, |
} |
err := mc.Add(itm) |
So(err, ShouldBeNil) |
- itm.SetValue([]byte("newp")) |
+ itm.Value = []byte("newp") |
err = mc.Set(itm) |
So(err, ShouldBeNil) |
- testItem := &mcItem{ |
- key: "sup", |
- value: []byte("newp"), |
- CasID: 2, |
+ testItem := &memcache.Item{ |
+ Key: "sup", |
+ Value: []byte("newp"), |
} |
+ unsafe.MCSetCasID(testItem, 2) |
i, err := mc.Get("sup") |
So(err, ShouldBeNil) |
So(i, ShouldResemble, testItem) |
@@ -134,66 +136,66 @@ func TestMemcache(t *testing.T) { |
Convey("and can be broken", func() { |
mci.BreakFeatures(nil, "Set") |
err := mc.Set(itm) |
- So(err, ShouldEqual, gae.ErrMCServerError) |
+ So(err, ShouldEqual, memcache.ErrServerError) |
}) |
}) |
Convey("CompareAndSwap", func() { |
- itm := gae.MCItem(&mcItem{ |
- key: "sup", |
- value: []byte("cool"), |
- expiration: time.Second * 2, |
- }) |
+ itm := &memcache.Item{ |
+ Key: "sup", |
+ Value: []byte("cool"), |
+ Expiration: time.Second * 2, |
+ } |
err := mc.Add(itm) |
So(err, ShouldBeNil) |
Convey("works after a Get", func() { |
itm, err = mc.Get("sup") |
So(err, ShouldBeNil) |
- So(itm.(*mcItem).CasID, ShouldEqual, 1) |
+ So(unsafe.MCGetCasID(itm), ShouldEqual, 1) |
- itm.SetValue([]byte("newp")) |
+ itm.Value = []byte("newp") |
err = mc.CompareAndSwap(itm) |
So(err, ShouldBeNil) |
}) |
Convey("but fails if you don't", func() { |
- itm.SetValue([]byte("newp")) |
+ itm.Value = []byte("newp") |
err = mc.CompareAndSwap(itm) |
- So(err, ShouldEqual, gae.ErrMCCASConflict) |
+ So(err, ShouldEqual, memcache.ErrCASConflict) |
}) |
Convey("and fails if the item is expired/gone", func() { |
tc.Add(3 * time.Second) |
- itm.SetValue([]byte("newp")) |
+ itm.Value = []byte("newp") |
err = mc.CompareAndSwap(itm) |
- So(err, ShouldEqual, gae.ErrMCNotStored) |
+ So(err, ShouldEqual, memcache.ErrNotStored) |
}) |
Convey("and can be broken", func() { |
mci.BreakFeatures(nil, "CompareAndSwap") |
err = mc.CompareAndSwap(itm) |
- So(err, ShouldEqual, gae.ErrMCServerError) |
+ So(err, ShouldEqual, memcache.ErrServerError) |
}) |
}) |
}) |
Convey("check that the internal implementation is sane", func() { |
curTime := now |
- err := mc.Add(&mcItem{ |
- key: "sup", |
- value: []byte("cool"), |
- expiration: time.Second * 2, |
+ err := mc.Add(&memcache.Item{ |
+ Key: "sup", |
+ Value: []byte("cool"), |
+ Expiration: time.Second * 2, |
}) |
So(err, ShouldBeNil) |
So(len(mci.data.items), ShouldEqual, 1) |
So(mci.data.casID, ShouldEqual, 1) |
- So(mci.data.items["sup"], ShouldResemble, &mcItem{ |
- key: "sup", |
- value: []byte("cool"), |
- expiration: time.Duration(curTime.Add(time.Second * 2).UnixNano()), |
+ So(mci.data.items["sup"], ShouldResemble, &unsafe.Item{ |
+ Key: "sup", |
+ Value: []byte("cool"), |
CasID: 1, |
+ Expiration: time.Duration(curTime.Add(time.Second * 2).UnixNano()), |
}) |
el, err := mc.Get("sup") |
@@ -201,11 +203,11 @@ func TestMemcache(t *testing.T) { |
So(len(mci.data.items), ShouldEqual, 1) |
So(mci.data.casID, ShouldEqual, 1) |
- testItem := &mcItem{ |
- key: "sup", |
- value: []byte("cool"), |
- CasID: 1, |
+ testItem := &memcache.Item{ |
+ Key: "sup", |
+ Value: []byte("cool"), |
} |
+ unsafe.MCSetCasID(testItem, 1) |
So(el, ShouldResemble, testItem) |
}) |