scheduler/appengine/presentation/state_test.go

// Copyright 2017 The LUCI Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

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{})
		})
	})
}