dm/appengine/deps/activate_execution_test.go

// Copyright 2015 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 deps

import (
	"testing"

	. "github.com/smartystreets/goconvey/convey"
	. "go.chromium.org/luci/common/testing/assertions"
	dm "go.chromium.org/luci/dm/api/service/v1"
	"go.chromium.org/luci/dm/appengine/distributor/fake"
)

func TestActivateExecution(t *testing.T) {
	t.Parallel()

	Convey("Test ActivateExecution", t, func() {
		ttest, c, dist, s := testSetup()

		qid := s.ensureQuest(c, "foo", 1)
		ttest.Drain(c)

		eid := dm.NewExecutionID(qid, 1, 1)
		dist.RunTask(c, eid, func(tsk *fake.Task) error {
			req := &dm.ActivateExecutionReq{
				Auth:           &dm.Execution_Auth{Id: eid},
				ExecutionToken: []byte("sufficiently long new 'random' token"),
			}

			Convey("bad", func() {
				Convey("wrong token", func() {
					_, err := s.ActivateExecution(c, req)
					So(err, ShouldBeRPCPermissionDenied, "failed to activate")
				})

				Convey("wrong token (already activated)", func() {
					req.Auth.Token = tsk.Auth.Token
					_, err := s.ActivateExecution(c, req)
					So(err, ShouldBeNil)

					req.Auth.Token = []byte("bad sekret")
					req.ExecutionToken = []byte("random other super duper long token")
					_, err = s.ActivateExecution(c, req)
					So(err, ShouldBeRPCPermissionDenied, "failed to activate")
				})

				Convey("concurrent activation", func() {
					req.Auth.Token = tsk.Auth.Token
					_, err := s.ActivateExecution(c, req)
					So(err, ShouldBeNil)

					req.ExecutionToken = append(req.ExecutionToken, []byte(" (but incorrect)")...)
					_, err = s.ActivateExecution(c, req)
					So(err, ShouldBeRPCPermissionDenied, "failed to activate")
				})
			})

			Convey("good", func() {
				req.Auth.Token = tsk.Auth.Token

				Convey("normal activation", func() {
					_, err := s.ActivateExecution(c, req)
					So(err, ShouldBeNil)
				})

				Convey("repeated activation", func() {
					_, err := s.ActivateExecution(c, req)
					So(err, ShouldBeNil)
					_, err = s.ActivateExecution(c, req)
					So(err, ShouldBeNil)
				})
			})
			return nil
		})
	})
}