Index: appengine/cmd/dm/mutate/ensure_attempt_test.go |
diff --git a/appengine/cmd/dm/mutate/ensure_attempt_test.go b/appengine/cmd/dm/mutate/ensure_attempt_test.go |
index 9dd4cae2350dca190bd36397c299ecc0774011c2..5fe16360ce03903be8449b2cc940009e42e19eb4 100644 |
--- a/appengine/cmd/dm/mutate/ensure_attempt_test.go |
+++ b/appengine/cmd/dm/mutate/ensure_attempt_test.go |
@@ -36,14 +36,14 @@ func TestEnsureAttempt(t *testing.T) { |
muts, err := ea.RollForward(c) |
So(err, ShouldBeNil) |
- So(muts, ShouldBeEmpty) |
+ So(muts, ShouldHaveLength, 1) |
ds := datastore.Get(c) |
So(ds.Get(a), ShouldEqual, nil) |
- So(a.State, ShouldEqual, dm.Attempt_NEEDS_EXECUTION) |
+ So(a.State, ShouldEqual, dm.Attempt_SCHEDULING) |
Convey("replaying the mutation after the state has evolved is a noop", func() { |
- a.State.MustEvolve(dm.Attempt_EXECUTING) |
+ So(a.ModifyState(c, dm.Attempt_EXECUTING), ShouldBeNil) |
So(ds.Put(a), ShouldBeNil) |
muts, err = ea.RollForward(c) |