Initial distributor implementation

appengine/cmd/dm/deps/finish_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 deps
 
 import (
         "testing"
 
-        "github.com/luci/gae/impl/memory"
         "github.com/luci/gae/service/datastore"
         "github.com/luci/luci-go/appengine/cmd/dm/model"
         "github.com/luci/luci-go/common/api/dm/service/v1"
         "github.com/luci/luci-go/common/clock/testclock"
         google_pb "github.com/luci/luci-go/common/proto/google"
         . "github.com/luci/luci-go/common/testing/assertions"
         . "github.com/smartystreets/goconvey/convey"
-        "golang.org/x/net/context"
 )
 
 func TestFinishAttempt(t *testing.T) {
         t.Parallel()
 
         Convey("FinishAttempt", t, func() {
-                c := memory.Use(context.Background())
+                _, c, _, s := testSetup()
                 ds := datastore.Get(c)
-                s := newDecoratedDeps()
 
                 So(ds.Put(&model.Quest{ID: "quest"}), ShouldBeNil)
                 a := &model.Attempt{
                         ID:           *dm.NewAttemptID("quest", 1),
                         State:        dm.Attempt_EXECUTING,
                         CurExecution: 1,
                 }
+                e := model.ExecutionFromID(c, dm.NewExecutionID("quest", 1, 1))
+                ar := &model.AttemptResult{Attempt: ds.KeyForObj(a)}
                 So(ds.Put(a), ShouldBeNil)
                 So(ds.Put(&model.Execution{
                         ID: 1, Attempt: ds.KeyForObj(a), Token: []byte("exKey"),
                         State: dm.Execution_RUNNING}), ShouldBeNil)
 
                 req := &dm.FinishAttemptReq{
                         Auth: &dm.Execution_Auth{
                                 Id:    dm.NewExecutionID(a.ID.Quest, a.ID.Id, 1),
                                 Token: []byte("exKey"),
                         },
@@ skipping 12 matching lines @@
                                 req.JsonResult = `i am not valid json`
                                 _, err := s.FinishAttempt(c, req)
                                 So(err, ShouldErrLike, "invalid character 'i'")
                         })
                 })
 
                 Convey("good", func() {
                         _, err := s.FinishAttempt(c, req)
                         So(err, ShouldBeNil)
 
-                        So(ds.Get(a), ShouldBeNil)
-                        So(a.State, ShouldEqual, dm.Attempt_FINISHED)
+                        So(ds.GetMulti([]interface{}{a, ar, e}), ShouldBeNil)
+                        So(a.State, ShouldEqual, dm.Attempt_EXECUTING)
+                        So(e.State, ShouldEqual, dm.Execution_STOPPING)
+
+                        So(a.ResultSize, ShouldEqual, 21)
+                        So(ar.Size, ShouldEqual, 21)
                 })
 
         })
 }