[scheduler]: ACLs phase 1 - per Job ACL specification and enforcement.

scheduler/appengine/engine/engine_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,
@@ skipping 18 matching lines @@
         "github.com/luci/gae/impl/memory"
         ds "github.com/luci/gae/service/datastore"
         tq "github.com/luci/gae/service/taskqueue"
 
         "github.com/luci/luci-go/common/clock"
         "github.com/luci/luci-go/common/clock/testclock"
         "github.com/luci/luci-go/common/data/rand/mathrand"
         "github.com/luci/luci-go/common/data/stringset"
         "github.com/luci/luci-go/common/errors"
         "github.com/luci/luci-go/common/retry/transient"
+        "github.com/luci/luci-go/server/auth"
+        "github.com/luci/luci-go/server/auth/authtest"
         "github.com/luci/luci-go/server/secrets/testsecrets"
 
+        "github.com/luci/luci-go/scheduler/appengine/acl"
         "github.com/luci/luci-go/scheduler/appengine/catalog"
         "github.com/luci/luci-go/scheduler/appengine/messages"
         "github.com/luci/luci-go/scheduler/appengine/task"
         "github.com/luci/luci-go/scheduler/appengine/task/noop"
 
         . "github.com/luci/luci-go/common/testing/assertions"
         . "github.com/smartystreets/goconvey/convey"
 )
 
 func TestGetAllProjects(t *testing.T) {
@@ skipping 485 matching lines @@
                 So(err, ShouldBeNil)
 
                 So(newer, ShouldBeLessThan, older)
         })
 }
 
 func TestQueries(t *testing.T) {
         Convey("with mock data", t, func() {
                 c := newTestContext(epoch)
                 e, _ := newTestEngine()
+                // TODO(tandrii): remove aclDefault once all Jobs have ACLs.
+                aclDefault := acl.GrantsByRole{}
+                aclSome := acl.GrantsByRole{Readers: []string{"group:some"}}
+                aclOne := acl.GrantsByRole{Owners: []string{"one@example.com"}}
+
+                ctxAnon := auth.WithState(c, &authtest.FakeState{
+                        Identity: "anonymous:anonymous",
+                })
+                ctxOne := auth.WithState(c, &authtest.FakeState{
+                        Identity:       "user:one@example.com",
+                        IdentityGroups: []string{"all"},
+                })
+                ctxSome := auth.WithState(c, &authtest.FakeState{
+                        Identity:       "user:some@example.com",
+                        IdentityGroups: []string{"some"},
+                })
+                ctxAdmin := auth.WithState(c, &authtest.FakeState{
+                        Identity:       "user:admin@example.com",
+                        IdentityGroups: []string{"administrators"},
+                })
 
                 So(ds.Put(c,
-                        &Job{JobID: "abc/1", ProjectID: "abc", Enabled: true},
-                        &Job{JobID: "abc/2", ProjectID: "abc", Enabled: true},
-                        &Job{JobID: "def/1", ProjectID: "def", Enabled: true},
-                        &Job{JobID: "def/2", ProjectID: "def", Enabled: false},
+                        &Job{JobID: "abc/1", ProjectID: "abc", Enabled: true, Acls: aclOne},
+                        &Job{JobID: "abc/2", ProjectID: "abc", Enabled: true, Acls: aclSome},
+                        &Job{JobID: "abc/3", ProjectID: "abc", Enabled: true, Acls: aclDefault},
+                        &Job{JobID: "def/1", ProjectID: "def", Enabled: true, Acls: aclDefault},
+                        &Job{JobID: "def/2", ProjectID: "def", Enabled: false, Acls: aclDefault},
                 ), ShouldBeNil)
 
                 job1 := ds.NewKey(c, "Job", "abc/1", 0, nil)
                 job2 := ds.NewKey(c, "Job", "abc/2", 0, nil)
+                job3 := ds.NewKey(c, "Job", "abc/3", 0, nil)
                 So(ds.Put(c,
                         &Invocation{ID: 1, JobKey: job1, InvocationNonce: 123},
                         &Invocation{ID: 2, JobKey: job1, InvocationNonce: 123},
                         &Invocation{ID: 3, JobKey: job1},
                         &Invocation{ID: 1, JobKey: job2},
                         &Invocation{ID: 2, JobKey: job2},
                         &Invocation{ID: 3, JobKey: job2},
+                        &Invocation{ID: 1, JobKey: job3},
                 ), ShouldBeNil)
 
                 ds.GetTestable(c).CatchupIndexes()
 
-                Convey("GetAllJobs works", func() {
-                        jobs, err := e.GetAllJobs(c)
-                        So(err, ShouldBeNil)
-                        ids := stringset.New(0)
-                        for _, j := range jobs {
-                                ids.Add(j.JobID)
+                Convey("GetAllJobsRA works", func() {
+                        getAllJobsRA := func(ctx context.Context) []string {
+                                jobs, err := e.GetAllJobsRA(ctx)
+                                So(err, ShouldBeNil)
+                                return sortedJobIds(jobs)
                         }
-                        asSlice := ids.ToSlice()
-                        sort.Strings(asSlice)
-                        So(asSlice, ShouldResemble, []string{"abc/1", "abc/2", "def/1"}) // only enabled
+
+                        Convey("Anonymous users see only public jobs", func() {
+                                // Only 3 jobs with default ACLs granting READER access to everyone, but
+                                // def/2 is disabled and so shouldn't be returned.
+                                So(getAllJobsRA(ctxAnon), ShouldResemble, []string{"abc/3", "def/1"})
+                        })
+                        Convey("Owners can see their own jobs + public jobs", func() {
+                                // abc/1 is owned by one@example.com.
+                                So(getAllJobsRA(ctxOne), ShouldResemble, []string{"abc/1", "abc/3", "def/1"})
+                        })
+                        Convey("Explicit readers", func() {
+                                So(getAllJobsRA(ctxSome), ShouldResemble, []string{"abc/2", "abc/3", "def/1"})
+                        })
+                        Convey("Admins have implicit READER access to all jobs", func() {
+                                So(getAllJobsRA(ctxAdmin), ShouldResemble, []string{"abc/1", "abc/2", "abc/3", "def/1"})
+                        })
                 })
 
-                Convey("GetProjectJobs works", func() {
-                        jobs, err := e.GetProjectJobs(c, "def")
-                        So(err, ShouldBeNil)
-                        So(len(jobs), ShouldEqual, 1)
-                        So(jobs[0].JobID, ShouldEqual, "def/1")
+                Convey("GetProjectJobsRA works", func() {
+                        getProjectJobsRA := func(ctx context.Context, project string) []string {
+                                jobs, err := e.GetProjectJobsRA(ctx, project)
+                                So(err, ShouldBeNil)
+                                return sortedJobIds(jobs)
+                        }
+                        Convey("Anonymous can still see public jobs", func() {
+                                So(getProjectJobsRA(ctxAnon, "def"), ShouldResemble, []string{"def/1"})
+                        })
+                        Convey("Admin have implicit READER access to all jobs", func() {
+                                So(getProjectJobsRA(ctxAdmin, "abc"), ShouldResemble, []string{"abc/1", "abc/2", "abc/3"})
+                        })
+                        Convey("Owners can still see their jobs", func() {
+                                So(getProjectJobsRA(ctxOne, "abc"), ShouldResemble, []string{"abc/1", "abc/3"})
+                        })
+                        Convey("Readers can see their jobs", func() {
+                                So(getProjectJobsRA(ctxSome, "abc"), ShouldResemble, []string{"abc/2", "abc/3"})
+                        })
                 })
 
-                Convey("GetJob works", func() {
-                        job, err := e.GetJob(c, "missing/job")
+                Convey("GetJobRA works", func() {
+                        job, err := e.GetJobRA(ctxAdmin, "missing/job")
                         So(job, ShouldBeNil)
                         So(err, ShouldBeNil)
 
-                        job, err = e.GetJob(c, "abc/1")
+                        job, err = e.GetJobRA(ctxAnon, "abc/1") // no READER permission.
+                        So(job, ShouldBeNil)
+                        So(err, ShouldBeNil)
+
+                        job, err = e.GetJobRA(ctxAnon, "def/1") // OK.
+                        So(job, ShouldNotBeNil)
+                        So(err, ShouldBeNil)
+
+                        job, err = e.GetJobRA(ctxAnon, "def/2") // OK, even though not enabled.
                         So(job, ShouldNotBeNil)
                         So(err, ShouldBeNil)
                 })
 
-                Convey("ListInvocations works", func() {
-                        invs, cursor, err := e.ListInvocations(c, "abc/1", 2, "")
+                Convey("ListInvocations works w/o ACL enforcement", func() {
+                        invs, cursor, err := e.ListInvocations(ctxAnon, "abc/1", 2, "")
                         So(err, ShouldBeNil)
                         So(len(invs), ShouldEqual, 2)
                         So(invs[0].ID, ShouldEqual, 1)
                         So(invs[1].ID, ShouldEqual, 2)
                         So(cursor, ShouldNotEqual, "")
 
-                        invs, cursor, err = e.ListInvocations(c, "abc/1", 2, cursor)
+                        invs, cursor, err = e.ListInvocations(ctxAdmin, "abc/1", 2, cursor)
                         So(err, ShouldBeNil)
                         So(len(invs), ShouldEqual, 1)
                         So(invs[0].ID, ShouldEqual, 3)
                         So(cursor, ShouldEqual, "")
                 })
 
                 Convey("GetInvocation works", func() {
                         inv, err := e.GetInvocation(c, "missing/job", 1)
                         So(inv, ShouldBeNil)
                         So(err, ShouldBeNil)
@@ skipping 163 matching lines @@
                                 ctl.State().Status = task.StatusRunning
                                 So(ctl.Save(ctx), ShouldBeNil)
                                 return nil
                         }
                         So(e.startInvocation(c, jobID, invNonce, "", 0), ShouldBeNil)
 
                         // It is alive and the job entity tracks it.
                         inv, err := e.GetInvocation(c, jobID, invID)
                         So(err, ShouldBeNil)
                         So(inv.Status, ShouldEqual, task.StatusRunning)
-                        job, err := e.GetJob(c, jobID)
+                        job, err := e.GetJobRA(c, jobID)
                         So(err, ShouldBeNil)
                         So(job.State.State, ShouldEqual, JobStateRunning)
                         So(job.State.InvocationID, ShouldEqual, invID)
 
                         return invID
                 }
 
                 Convey("AbortInvocation works", func() {
                         // Actually launch the queued invocation.
                         invID := launchInv()
 
                         // Kill it.
                         So(e.AbortInvocation(c, jobID, invID, ""), ShouldBeNil)
 
                         // It is dead.
                         inv, err := e.GetInvocation(c, jobID, invID)
                         So(err, ShouldBeNil)
                         So(inv.Status, ShouldEqual, task.StatusAborted)
 
                         // The job moved on with its life.
-                        job, err := e.GetJob(c, jobID)
+                        job, err := e.GetJobRA(c, jobID)
                         So(err, ShouldBeNil)
                         So(job.State.State, ShouldEqual, JobStateSuspended)
                         So(job.State.InvocationID, ShouldEqual, 0)
                 })
 
                 Convey("AbortJob kills running invocation", func() {
                         // Actually launch the queued invocation.
                         invID := launchInv()
 
                         // Kill it.
                         So(e.AbortJob(c, jobID, ""), ShouldBeNil)
 
                         // It is dead.
                         inv, err := e.GetInvocation(c, jobID, invID)
                         So(err, ShouldBeNil)
                         So(inv.Status, ShouldEqual, task.StatusAborted)
 
                         // The job moved on with its life.
-                        job, err := e.GetJob(c, jobID)
+                        job, err := e.GetJobRA(c, jobID)
                         So(err, ShouldBeNil)
                         So(job.State.State, ShouldEqual, JobStateSuspended)
                         So(job.State.InvocationID, ShouldEqual, 0)
                 })
 
                 Convey("AbortJob kills queued invocation", func() {
                         So(e.AbortJob(c, jobID, ""), ShouldBeNil)
 
                         // The job moved on with its life.
-                        job, err := e.GetJob(c, jobID)
+                        job, err := e.GetJobRA(c, jobID)
                         So(err, ShouldBeNil)
                         So(job.State.State, ShouldEqual, JobStateSuspended)
                         So(job.State.InvocationID, ShouldEqual, 0)
                 })
         })
 }
 
 func TestAddTimer(t *testing.T) {
         Convey("with mock job", t, func() {
                 c := newTestContext(epoch)
                 e, mgr := newTestEngine()
 
                 // A job in "QUEUED" state (about to run an invocation).
                 const jobID = "abc/1"
                 const invNonce = int64(12345)
                 prepareQueuedJob(c, jobID, invNonce)
 
                 Convey("AddTimer works", func() {
                         // Start an invocation that adds a timer.
                         mgr.launchTask = func(ctx context.Context, ctl task.Controller) error {
                                 ctl.AddTimer(ctx, time.Minute, "timer-name", []byte{1, 2, 3})
                                 ctl.State().Status = task.StatusRunning
                                 return nil
                         }
                         So(e.startInvocation(c, jobID, invNonce, "", 0), ShouldBeNil)
 
                         // The job is running.
-                        job, err := e.GetJob(c, jobID)
+                        job, err := e.GetJobRA(c, jobID)
                         So(err, ShouldBeNil)
                         So(job.State.State, ShouldEqual, JobStateRunning)
 
                         // Added a task to the timers task queue.
                         tasks := tq.GetTestable(c).GetScheduledTasks()["timers-q"]
                         So(len(tasks), ShouldEqual, 1)
                         var tqt *tq.Task
                         for _, tqt = range tasks {
                         }
                         So(tqt.ETA, ShouldResemble, clock.Now(c).Add(time.Minute))
@@ skipping 19 matching lines @@
                                 So(name, ShouldEqual, "timer-name")
                                 So(payload, ShouldResemble, []byte{1, 2, 3})
                                 ctl.AddTimer(ctx, time.Minute, "ignored-timer", nil)
                                 ctl.State().Status = task.StatusSucceeded
                                 return nil
                         }
                         clock.Get(c).(testclock.TestClock).Add(time.Minute)
                         So(e.ExecuteSerializedAction(c, tqt.Payload, 0), ShouldBeNil)
 
                         // The job has finished (by timer handler). Moves back to SUSPENDED state.
-                        job, err = e.GetJob(c, jobID)
+                        job, err = e.GetJobRA(c, jobID)
                         So(err, ShouldBeNil)
                         So(job.State.State, ShouldEqual, JobStateSuspended)
 
                         // No new timers added for finished job.
                         tasks = tq.GetTestable(c).GetScheduledTasks()["timers-q"]
                         So(len(tasks), ShouldEqual, 0)
                 })
         })
 }
 
@@ skipping 122 matching lines @@
 func (m *fakeTaskManager) HandleNotification(c context.Context, ctl task.Controller, msg *pubsub.PubsubMessage) error {
         return m.handleNotification(c, msg)
 }
 
 func (m fakeTaskManager) HandleTimer(c context.Context, ctl task.Controller, name string, payload []byte) error {
         return m.handleTimer(c, ctl, name, payload)
 }
 
 ////
 
+func sortedJobIds(jobs []*Job) []string {
+        ids := stringset.New(len(jobs))
+        for _, j := range jobs {
+                ids.Add(j.JobID)
+        }
+        asSlice := ids.ToSlice()
+        sort.Strings(asSlice)
+        return asSlice
+}
+
 // prepareQueuedJob makes datastore entries for a job in QUEUED state.
 func prepareQueuedJob(c context.Context, jobID string, invNonce int64) {
         taskBlob, err := proto.Marshal(&messages.TaskDefWrapper{
                 Noop: &messages.NoopTask{},
         })
         if err != nil {
                 panic(err)
         }
         chunks := strings.Split(jobID, "/")
         err = ds.Put(c, &Job{
@@ skipping 45 matching lines @@
 
 func ensureOneTask(c context.Context, q string) *tq.Task {
         tqt := tq.GetTestable(c)
         tasks := tqt.GetScheduledTasks()[q]
         So(len(tasks), ShouldEqual, 1)
         for _, t := range tasks {
                 return t
         }
         return nil
 }