Index: go/src/infra/gae/libs/memlock/memlock_test.go |
diff --git a/go/src/infra/gae/libs/memlock/memlock_test.go b/go/src/infra/gae/libs/memlock/memlock_test.go |
index 6d86249507ea8ca3316bb2f45f7af8e1362e5b83..ab4fecac94b38a8135e614c80362bef8e7af8b6b 100644 |
--- a/go/src/infra/gae/libs/memlock/memlock_test.go |
+++ b/go/src/infra/gae/libs/memlock/memlock_test.go |
@@ -6,18 +6,17 @@ package memlock |
import ( |
"fmt" |
- "infra/gae/libs/wrapper" |
- "infra/gae/libs/wrapper/memory" |
- "infra/libs/clock" |
- "infra/libs/clock/testclock" |
"runtime" |
"testing" |
"time" |
+ "infra/gae/libs/gae" |
+ "infra/gae/libs/gae/memory" |
+ |
+ "github.com/luci/luci-go/common/clock" |
+ "github.com/luci/luci-go/common/clock/testclock" |
. "github.com/smartystreets/goconvey/convey" |
"golang.org/x/net/context" |
- |
- "appengine/memcache" |
) |
func init() { |
@@ -42,9 +41,9 @@ func TestSimple(t *testing.T) { |
} |
}) |
ctx = memory.Use(ctx) |
- mc := wrapper.GetMC(ctx).(interface { |
- wrapper.Testable |
- wrapper.MCSingleReadWriter |
+ mc := gae.GetMC(ctx).(interface { |
+ gae.Testable |
+ gae.Memcache |
}) |
Convey("fails to acquire when memcache is down", func() { |
@@ -105,7 +104,7 @@ func TestSimple(t *testing.T) { |
}) |
Convey("or because it was stolen", func() { |
- mc.Set(&memcache.Item{Key: key, Value: []byte("wat")}) |
+ mc.Set(mc.NewItem(key).SetValue([]byte("wat"))) |
waitFalse() |
}) |