You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
cloudreve/pkg/task/worker_test.go

82 lines
1.2 KiB

5 years ago
package task
import (
"testing"
model "github.com/cloudreve/Cloudreve/v3/models"
"github.com/stretchr/testify/assert"
5 years ago
)
type MockJob struct {
Err *JobError
Status int
DoFunc func()
}
func (job *MockJob) Type() int {
panic("implement me")
}
func (job *MockJob) Creator() uint {
panic("implement me")
}
func (job *MockJob) Props() string {
panic("implement me")
}
func (job *MockJob) Model() *model.Task {
panic("implement me")
}
func (job *MockJob) SetStatus(status int) {
job.Status = status
}
func (job *MockJob) Do() {
job.DoFunc()
}
func (job *MockJob) SetError(*JobError) {
}
func (job *MockJob) GetError() *JobError {
return job.Err
}
func TestGeneralWorker_Do(t *testing.T) {
asserts := assert.New(t)
worker := &GeneralWorker{}
job := &MockJob{}
// 正常
{
job.DoFunc = func() {
}
worker.Do(job)
asserts.Equal(Complete, job.Status)
}
// 有错误
{
job.DoFunc = func() {
}
job.Status = Queued
job.Err = &JobError{Msg: "error"}
worker.Do(job)
asserts.Equal(Error, job.Status)
}
// 有致命错误
{
job.DoFunc = func() {
panic("mock fatal error")
}
job.Status = Queued
job.Err = nil
worker.Do(job)
asserts.Equal(Error, job.Status)
}
}