Initial distributor implementation

appengine/cmd/dm/mutate/ensure_attempt_test.go

 // Copyright 2015 The LUCI Authors. All rights reserved.
 // Use of this source code is governed under the Apache License, Version 2.0
 // that can be found in the LICENSE file.
 
 package mutate
 
 import (
         "testing"
 
         "github.com/luci/gae/filter/featureBreaker"
@@ skipping 18 matching lines @@
                 })
 
                 Convey("RollForward", func() {
                         a := &model.Attempt{ID: *ea.ID}
 
                         Convey("Good", func() {
                                 So(datastore.Get(c).Get(a), ShouldEqual, datastore.ErrNoSuchEntity)
 
                                 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)
                                         So(err, ShouldBeNil)
                                         So(muts, ShouldBeEmpty)
 
                                         So(ds.Get(a), ShouldEqual, nil)
                                         So(a.State, ShouldEqual, dm.Attempt_EXECUTING)
                                 })
                         })
 
                         Convey("Bad", func() {
                                 c, fb := featureBreaker.FilterRDS(c, nil)
                                 fb.BreakFeatures(nil, "GetMulti")
 
                                 muts, err := ea.RollForward(c)
                                 So(err, ShouldErrLike, `feature "GetMulti" is broken`)
                                 So(muts, ShouldBeEmpty)
 
                                 fb.UnbreakFeatures("GetMulti")
 
                                 So(datastore.Get(c).Get(a), ShouldEqual, datastore.ErrNoSuchEntity)
                         })
                 })
         })
 }