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 ab4fecac94b38a8135e614c80362bef8e7af8b6b..e6de4fb4b1a56eb81bda9c46d09d5d906df3cee0 100644 |
--- a/go/src/infra/gae/libs/memlock/memlock_test.go |
+++ b/go/src/infra/gae/libs/memlock/memlock_test.go |
@@ -11,6 +11,7 @@ import ( |
"time" |
"infra/gae/libs/gae" |
+ "infra/gae/libs/gae/filters/featureBreaker" |
"infra/gae/libs/gae/memory" |
"github.com/luci/luci-go/common/clock" |
@@ -40,14 +41,11 @@ func TestSimple(t *testing.T) { |
default: |
} |
}) |
- ctx = memory.Use(ctx) |
- mc := gae.GetMC(ctx).(interface { |
- gae.Testable |
- gae.Memcache |
- }) |
+ ctx, fb := featureBreaker.FilterMC(memory.Use(ctx), nil) |
+ mc := gae.GetMC(ctx) |
Convey("fails to acquire when memcache is down", func() { |
- mc.BreakFeatures(nil, "Add") |
+ fb.BreakFeatures(nil, "Add") |
err := TryWithLock(ctx, "testkey", "id", func(check func() bool) error { |
// should never reach here |
So(false, ShouldBeTrue) |
@@ -109,7 +107,7 @@ func TestSimple(t *testing.T) { |
}) |
Convey("or because of service issues", func() { |
- mc.BreakFeatures(nil, "CompareAndSwap") |
+ fb.BreakFeatures(nil, "CompareAndSwap") |
waitFalse() |
}) |
}) |