summaryrefslogtreecommitdiffstats
path: root/api4/job_test.go
diff options
context:
space:
mode:
authorHarrison Healey <harrisonmhealey@gmail.com>2017-05-18 15:05:57 -0400
committerGitHub <noreply@github.com>2017-05-18 15:05:57 -0400
commit577ed27f1bb060080d311342047e31943a02ccbb (patch)
treead57fa69b1daf143e914ea2480a475e5450cc236 /api4/job_test.go
parent920bc0d8712a50691b1f698779f60132536eb214 (diff)
downloadchat-577ed27f1bb060080d311342047e31943a02ccbb.tar.gz
chat-577ed27f1bb060080d311342047e31943a02ccbb.tar.bz2
chat-577ed27f1bb060080d311342047e31943a02ccbb.zip
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
Diffstat (limited to 'api4/job_test.go')
-rw-r--r--api4/job_test.go103
1 files changed, 103 insertions, 0 deletions
diff --git a/api4/job_test.go b/api4/job_test.go
new file mode 100644
index 000000000..0f39fc306
--- /dev/null
+++ b/api4/job_test.go
@@ -0,0 +1,103 @@
+// 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)
+}