Test: pkg/task

pkg/task/worker_test.go

@@ -0,0 +1,80 @@
+package task
+
+import (
+	model "github.com/HFO4/cloudreve/models"
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+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)
+	}
+
+}