scheduler WIP: add GetAllJobs api.

scheduler/appengine/presentation/state_test.go

Index: scheduler/appengine/presentation/state_test.go
diff --git a/scheduler/appengine/presentation/state_test.go b/scheduler/appengine/presentation/state_test.go
new file mode 100644
index 0000000000000000000000000000000000000000..1f211bde2ea94cef2ba439107ee3f21f4387bdf2
--- /dev/null
+++ b/scheduler/appengine/presentation/state_test.go
@@ -0,0 +1,83 @@
+// Copyright 2017 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 presentation
+
+import (
+	"testing"
+
+	"golang.org/x/net/context"
+
+	"github.com/golang/protobuf/proto"
+	"github.com/luci/luci-go/scheduler/appengine/catalog"
+	"github.com/luci/luci-go/scheduler/appengine/engine"
+	"github.com/luci/luci-go/scheduler/appengine/messages"
+	"github.com/luci/luci-go/scheduler/appengine/task"
+	"github.com/luci/luci-go/scheduler/appengine/task/urlfetch"
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+func TestGetPublicStateKind(t *testing.T) {
+	t.Parallel()
+
+	Convey("works", t, func() {
+		So(GetPublicStateKind(&engine.Job{
+			State: engine.JobState{State: engine.JobStateOverrun},
+		}, task.Traits{}), ShouldEqual, PublicStateOverrun)
+
+		So(GetPublicStateKind(&engine.Job{
+			State: engine.JobState{State: engine.JobStateSlowQueue},
+		}, task.Traits{}), ShouldEqual, PublicStateStarting)
+
+		So(GetPublicStateKind(&engine.Job{
+			State: engine.JobState{State: engine.JobStateSlowQueue, InvocationRetryCount: 1},
+		}, task.Traits{}), ShouldEqual, PublicStateRetrying)
+
+		So(GetPublicStateKind(&engine.Job{
+			Paused: true,
+			State:  engine.JobState{State: engine.JobStateSuspended},
+		}, task.Traits{}), ShouldEqual, PublicStatePaused)
+
+		So(GetPublicStateKind(&engine.Job{
+			State: engine.JobState{State: engine.JobStateQueued, InvocationID: 1},
+		}, task.Traits{Multistage: true}), ShouldEqual, PublicStateStarting)
+
+		So(GetPublicStateKind(&engine.Job{
+			State: engine.JobState{State: engine.JobStateQueued, InvocationID: 1},
+		}, task.Traits{Multistage: false}), ShouldEqual, PublicStateRunning)
+	})
+}
+
+func TestGetJobTraits(t *testing.T) {
+	t.Parallel()
+	Convey("works", t, func() {
+		cat := catalog.New("scheduler.cfg")
+		So(cat.RegisterTaskManager(&urlfetch.TaskManager{}), ShouldBeNil)
+		ctx := context.Background()
+
+		Convey("bad task", func() {
+			taskBlob, err := proto.Marshal(&messages.TaskDefWrapper{
+				Noop: &messages.NoopTask{},
+			})
+			So(err, ShouldBeNil)
+			traits, err := GetJobTraits(ctx, cat, &engine.Job{
+				Task: taskBlob,
+			})
+			So(err, ShouldNotBeNil)
+			So(traits, ShouldResemble, task.Traits{})
+		})
+
+		Convey("OK task", func() {
+			taskBlob, err := proto.Marshal(&messages.TaskDefWrapper{
+				UrlFetch: &messages.UrlFetchTask{Url: "http://example.com/path"},
+			})
+			So(err, ShouldBeNil)
+			traits, err := GetJobTraits(ctx, cat, &engine.Job{
+				Task: taskBlob,
+			})
+			So(err, ShouldBeNil)
+			So(traits, ShouldResemble, task.Traits{})
+		})
+	})
+}