Files
mattermost/api4/job_test.go
Harrison Healey 577ed27f1b PLT-6408 Framework for job server (#6404)
* Added initial job server

* Added job server to be ran as part of platform

* Added test job to the enterprise repo

* Fixed job server not loading license

* Renamed job package to jobs

* Fixed TE not being buildable

* Added JobStatus table to database

* Changed fields used by JobStatus

* Added APIs to query job status

* Added config change listener to server

* Added option to run job server from Makefile

* Added ability to enable/disable jobs from config

* Commented out placeholder for search indexing job

* Fixed govet

* Removed debug messages and fixed job api init message
2017-05-18 15:05:57 -04:00

104 lines
2.5 KiB
Go

// Copyright (c) 2017-present Mattermost, Inc. All Rights Reserved.
// See License.txt for license information.
package api4
import (
"strings"
"testing"
"github.com/mattermost/platform/app"
"github.com/mattermost/platform/model"
"github.com/mattermost/platform/store"
)
func TestGetJobStatus(t *testing.T) {
th := Setup().InitBasic().InitSystemAdmin()
defer TearDown()
status := &model.JobStatus{
Id: model.NewId(),
Status: model.NewId(),
}
if result := <-app.Srv.Store.JobStatus().SaveOrUpdate(status); result.Err != nil {
t.Fatal(result.Err)
}
defer app.Srv.Store.JobStatus().Delete(status.Id)
received, resp := th.SystemAdminClient.GetJobStatus(status.Id)
CheckNoError(t, resp)
if received.Id != status.Id || received.Status != status.Status {
t.Fatal("incorrect job status received")
}
_, resp = th.SystemAdminClient.GetJobStatus("1234")
CheckBadRequestStatus(t, resp)
_, resp = th.Client.GetJobStatus(status.Id)
CheckForbiddenStatus(t, resp)
_, resp = th.SystemAdminClient.GetJobStatus(model.NewId())
CheckNotFoundStatus(t, resp)
}
func TestGetJobStatusesByType(t *testing.T) {
th := Setup().InitBasic().InitSystemAdmin()
defer TearDown()
jobType := model.NewId()
statuses := []*model.JobStatus{
{
Id: model.NewId(),
Type: jobType,
StartAt: 1000,
},
{
Id: model.NewId(),
Type: jobType,
StartAt: 999,
},
{
Id: model.NewId(),
Type: jobType,
StartAt: 1001,
},
}
for _, status := range statuses {
store.Must(app.Srv.Store.JobStatus().SaveOrUpdate(status))
defer app.Srv.Store.JobStatus().Delete(status.Id)
}
received, resp := th.SystemAdminClient.GetJobStatusesByType(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("should've received newest job first")
} else if received[1].Id != statuses[0].Id {
t.Fatal("should've received second newest job second")
}
received, resp = th.SystemAdminClient.GetJobStatusesByType(jobType, 1, 2)
CheckNoError(t, resp)
if len(received) != 1 {
t.Fatal("received wrong number of statuses")
} else if received[0].Id != statuses[2].Id {
t.Fatal("should've received oldest job last")
}
_, resp = th.SystemAdminClient.GetJobStatusesByType("", 0, 60)
CheckNotFoundStatus(t, resp)
_, resp = th.SystemAdminClient.GetJobStatusesByType(strings.Repeat("a", 33), 0, 60)
CheckBadRequestStatus(t, resp)
_, resp = th.Client.GetJobStatusesByType(jobType, 0, 60)
CheckForbiddenStatus(t, resp)
}