| Index: scheduler/appengine/apiservers/scheduler_test.go
|
| diff --git a/scheduler/appengine/apiservers/scheduler_test.go b/scheduler/appengine/apiservers/scheduler_test.go
|
| index 682af72a5be22fe6fdc7582b3aa1c2160e1abda7..a76afe20a1d17c818f65c7684ae7bdd1e63777ea 100644
|
| --- a/scheduler/appengine/apiservers/scheduler_test.go
|
| +++ b/scheduler/appengine/apiservers/scheduler_test.go
|
| @@ -159,7 +159,9 @@ func TestGetInvocationsApi(t *testing.T) {
|
| ss := SchedulerServer{fakeEng, catalog}
|
|
|
| Convey("Job not found", func() {
|
| - fakeEng.getVisibleJob = func(JobID string) (*engine.Job, error) { return nil, nil }
|
| + fakeEng.listVisibleInvocations = func(int, string) ([]*engine.Invocation, string, error) {
|
| + return nil, "", engine.ErrNoSuchJob
|
| + }
|
| _, err := ss.GetInvocations(ctx, &scheduler.InvocationsRequest{
|
| JobRef: &scheduler.JobRef{Project: "not", Job: "exists"},
|
| })
|
| @@ -169,7 +171,9 @@ func TestGetInvocationsApi(t *testing.T) {
|
| })
|
|
|
| Convey("DS error", func() {
|
| - fakeEng.getVisibleJob = func(JobID string) (*engine.Job, error) { return nil, fmt.Errorf("ds error") }
|
| + fakeEng.listVisibleInvocations = func(int, string) ([]*engine.Invocation, string, error) {
|
| + return nil, "", fmt.Errorf("ds error")
|
| + }
|
| _, err := ss.GetInvocations(ctx, &scheduler.InvocationsRequest{
|
| JobRef: &scheduler.JobRef{Project: "proj", Job: "job"},
|
| })
|
|
|