PLT-6595-Server: Job Management APIs. (#6931)

* PLT-6595-Server: Job Management APIs.

* MANAGE_JOBS Permission

* Fix test.

1 files changed, 154 insertions, 28 deletions

diff --git a/api4/job_test.go b/api4/job_test.go
index 8bbea83e1..3dcdbe58b 100644
--- a/api4/job_test.go
+++ b/api4/job_test.go
@@ -12,74 +12,157 @@ import (
 	"github.com/mattermost/platform/store"
 )
 
-func TestGetJobStatus(t *testing.T) {
+func TestCreateJob(t *testing.T) {
 	th := Setup().InitBasic().InitSystemAdmin()
 	defer TearDown()
 
-	status := &model.Job{
+	job := &model.Job{
+		Type: model.JOB_TYPE_DATA_RETENTION,
+		Data: map[string]interface{}{
+			"thing": "stuff",
+		},
+	}
+
+	received, resp := th.SystemAdminClient.CreateJob(job)
+	CheckNoError(t, resp)
+
+	defer app.Srv.Store.Job().Delete(received.Id)
+
+	job = &model.Job{
+		Type: model.NewId(),
+	}
+
+	_, resp = th.SystemAdminClient.CreateJob(job)
+	CheckBadRequestStatus(t, resp)
+
+	_, resp = th.Client.CreateJob(job)
+	CheckForbiddenStatus(t, resp)
+}
+
+func TestGetJob(t *testing.T) {
+	th := Setup().InitBasic().InitSystemAdmin()
+	defer TearDown()
+
+	job := &model.Job{
 		Id:     model.NewId(),
-		Status: model.NewId(),
+		Status: model.JOB_STATUS_PENDING,
 	}
-	if result := <-app.Srv.Store.Job().Save(status); result.Err != nil {
+	if result := <-app.Srv.Store.Job().Save(job); result.Err != nil {
 		t.Fatal(result.Err)
 	}
 
-	defer app.Srv.Store.Job().Delete(status.Id)
+	defer app.Srv.Store.Job().Delete(job.Id)
 
-	received, resp := th.SystemAdminClient.GetJob(status.Id)
+	received, resp := th.SystemAdminClient.GetJob(job.Id)
 	CheckNoError(t, resp)
 
-	if received.Id != status.Id || received.Status != status.Status {
-		t.Fatal("incorrect job status received")
+	if received.Id != job.Id || received.Status != job.Status {
+		t.Fatal("incorrect job received")
 	}
 
 	_, resp = th.SystemAdminClient.GetJob("1234")
 	CheckBadRequestStatus(t, resp)
 
-	_, resp = th.Client.GetJob(status.Id)
+	_, resp = th.Client.GetJob(job.Id)
 	CheckForbiddenStatus(t, resp)
 
 	_, resp = th.SystemAdminClient.GetJob(model.NewId())
 	CheckNotFoundStatus(t, resp)
 }
 
-func TestGetJobStatusesByType(t *testing.T) {
+func TestGetJobs(t *testing.T) {
+	th := Setup().InitBasic().InitSystemAdmin()
+	defer TearDown()
+
+	jobType := model.NewId()
+
+	jobs := []*model.Job{
+		{
+			Id:       model.NewId(),
+			Type:     jobType,
+			CreateAt: model.GetMillis() + 1,
+		},
+		{
+			Id:       model.NewId(),
+			Type:     jobType,
+			CreateAt: model.GetMillis(),
+		},
+		{
+			Id:       model.NewId(),
+			Type:     jobType,
+			CreateAt: model.GetMillis() + 2,
+		},
+	}
+
+	for _, job := range jobs {
+		store.Must(app.Srv.Store.Job().Save(job))
+		defer app.Srv.Store.Job().Delete(job.Id)
+	}
+
+	received, resp := th.SystemAdminClient.GetJobs(0, 2)
+	CheckNoError(t, resp)
+
+	if len(received) != 2 {
+		t.Fatal("received wrong number of jobs")
+	} else if received[0].Id != jobs[2].Id {
+		t.Fatal("should've received newest job first")
+	} else if received[1].Id != jobs[0].Id {
+		t.Fatal("should've received second newest job second")
+	}
+
+	received, resp = th.SystemAdminClient.GetJobs(1, 2)
+	CheckNoError(t, resp)
+
+	if received[0].Id != jobs[1].Id {
+		t.Fatal("should've received oldest job last")
+	}
+
+	_, resp = th.Client.GetJobs(0, 60)
+	CheckForbiddenStatus(t, resp)
+}
+
+func TestGetJobsByType(t *testing.T) {
 	th := Setup().InitBasic().InitSystemAdmin()
 	defer TearDown()
 
 	jobType := model.NewId()
 
-	statuses := []*model.Job{
+	jobs := []*model.Job{
 		{
-			Id:      model.NewId(),
-			Type:    jobType,
-			StartAt: 1000,
+			Id:       model.NewId(),
+			Type:     jobType,
+			CreateAt: 1000,
 		},
 		{
-			Id:      model.NewId(),
-			Type:    jobType,
-			StartAt: 999,
+			Id:       model.NewId(),
+			Type:     jobType,
+			CreateAt: 999,
 		},
 		{
-			Id:      model.NewId(),
-			Type:    jobType,
-			StartAt: 1001,
+			Id:       model.NewId(),
+			Type:     jobType,
+			CreateAt: 1001,
+		},
+		{
+			Id:       model.NewId(),
+			Type:     model.NewId(),
+			CreateAt: 1002,
 		},
 	}
 
-	for _, status := range statuses {
-		store.Must(app.Srv.Store.Job().Save(status))
-		defer app.Srv.Store.Job().Delete(status.Id)
+	for _, job := range jobs {
+		store.Must(app.Srv.Store.Job().Save(job))
+		defer app.Srv.Store.Job().Delete(job.Id)
 	}
 
 	received, resp := th.SystemAdminClient.GetJobsByType(jobType, 0, 2)
 	CheckNoError(t, resp)
 
 	if len(received) != 2 {
-		t.Fatal("received wrong number of statuses")
-	} else if received[0].Id != statuses[1].Id {
+		t.Fatal("received wrong number of jobs")
+	} else if received[0].Id != jobs[2].Id {
 		t.Fatal("should've received newest job first")
-	} else if received[1].Id != statuses[0].Id {
+	} else if received[1].Id != jobs[0].Id {
 		t.Fatal("should've received second newest job second")
 	}
 
@@ -87,8 +170,8 @@ func TestGetJobStatusesByType(t *testing.T) {
 	CheckNoError(t, resp)
 
 	if len(received) != 1 {
-		t.Fatal("received wrong number of statuses")
-	} else if received[0].Id != statuses[2].Id {
+		t.Fatal("received wrong number of jobs")
+	} else if received[0].Id != jobs[1].Id {
 		t.Fatal("should've received oldest job last")
 	}
 
@@ -101,3 +184,46 @@ func TestGetJobStatusesByType(t *testing.T) {
 	_, resp = th.Client.GetJobsByType(jobType, 0, 60)
 	CheckForbiddenStatus(t, resp)
 }
+
+func TestCancelJob(t *testing.T) {
+	th := Setup().InitBasic().InitSystemAdmin()
+	defer TearDown()
+
+	jobs := []*model.Job{
+		{
+			Id:     model.NewId(),
+			Type:   model.NewId(),
+			Status: model.JOB_STATUS_PENDING,
+		},
+		{
+			Id:     model.NewId(),
+			Type:   model.NewId(),
+			Status: model.JOB_STATUS_IN_PROGRESS,
+		},
+		{
+			Id:     model.NewId(),
+			Type:   model.NewId(),
+			Status: model.JOB_STATUS_SUCCESS,
+		},
+	}
+
+	for _, job := range jobs {
+		store.Must(app.Srv.Store.Job().Save(job))
+		defer app.Srv.Store.Job().Delete(job.Id)
+	}
+
+	_, resp := th.Client.CancelJob(jobs[0].Id)
+	CheckForbiddenStatus(t, resp)
+
+	_, resp = th.SystemAdminClient.CancelJob(jobs[0].Id)
+	CheckNoError(t, resp)
+
+	_, resp = th.SystemAdminClient.CancelJob(jobs[1].Id)
+	CheckNoError(t, resp)
+
+	_, resp = th.SystemAdminClient.CancelJob(jobs[2].Id)
+	CheckInternalErrorStatus(t, resp)
+
+	_, resp = th.SystemAdminClient.CancelJob(model.NewId())
+	CheckInternalErrorStatus(t, resp)
+}