mirror of
https://github.com/hanzoai/team.git
synced 2026-09-30 01:57:35 +00:00
d1f860cf11
* ClaimJob now returns newly claimed job * internal code affected by change * test changes required * two branches: for mysql, use transaction; for postgres, use returning * two branches: for mysql, use transaction; for postgres, use returning * use same millis value for LastActivityAt and StartAt * blank commit --------- Co-authored-by: Mattermost Build <[email protected]>
43 lines
1.2 KiB
Go
43 lines
1.2 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package jobs
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/mattermost/mattermost/server/public/shared/mlog"
|
|
"github.com/stretchr/testify/mock"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestSimpleWorkerPanic(t *testing.T) {
|
|
jobServer, mockStore, mockMetrics := makeJobServer(t)
|
|
|
|
job := &model.Job{
|
|
Id: "job_id",
|
|
Type: "job_type",
|
|
}
|
|
|
|
exec := func(_ mlog.LoggerIFace, _ *model.Job) error {
|
|
return nil
|
|
}
|
|
|
|
isEnabled := func(_ *model.Config) bool {
|
|
return true
|
|
}
|
|
|
|
mockStore.JobStore.On("UpdateStatusOptimistically", "job_id", model.JobStatusPending, model.JobStatusInProgress).Return(&model.Job{Id: "job_id", Type: "job_type"}, nil)
|
|
mockStore.JobStore.On("UpdateOptimistically", mock.AnythingOfType("*model.Job"), model.JobStatusInProgress).Return(true, nil)
|
|
mockStore.JobStore.On("UpdateStatus", "job_id", "success").Return(nil, errors.New("test"))
|
|
mockMetrics.On("IncrementJobActive", "job_type")
|
|
mockMetrics.On("DecrementJobActive", "job_type")
|
|
sWorker := NewSimpleWorker("test", jobServer, exec, isEnabled)
|
|
|
|
require.NotPanics(t, func() {
|
|
sWorker.DoJob(job)
|
|
})
|
|
}
|