Index: dm/appengine/mutate/schedule_execution_test.go |
diff --git a/dm/appengine/mutate/schedule_execution_test.go b/dm/appengine/mutate/schedule_execution_test.go |
index eebfb0263bae3a8ef79a3e7fd40d7b86c0db5c70..ce019322a58a0288129b3ff8a8e847b6f88e6215 100644 |
--- a/dm/appengine/mutate/schedule_execution_test.go |
+++ b/dm/appengine/mutate/schedule_execution_test.go |
@@ -12,6 +12,7 @@ import ( |
ds "github.com/luci/gae/service/datastore" |
"github.com/luci/luci-go/common/errors" |
"github.com/luci/luci-go/common/proto/google" |
+ "github.com/luci/luci-go/common/retry/transient" |
"github.com/luci/luci-go/dm/api/service/v1" |
"github.com/luci/luci-go/dm/appengine/distributor/fake" |
"github.com/luci/luci-go/dm/appengine/model" |
@@ -71,7 +72,7 @@ func TestScheduleExecution(t *testing.T) { |
}) |
Convey("transient", func() { |
- dist.RunError = errors.WrapTransient(errors.New("transient failure")) |
+ dist.RunError = errors.New("transient failure", transient.Tag) |
muts, err := se.RollForward(c) |
So(err, ShouldErrLike, "transient") |