Function TestTask_Cancel

pkg/task/task_test.go:38–65

(t *testing.T)

}

func TestTask_Cancel(t *testing.T) {
	var next atomic.Uint64
	tm := NewTaskManager(3, func(id *uint64) {
		*id = next.Add(1)
	})
	id := tm.Submit(WithCancelCtx(&Task[uint64]{
		Name: "test",
		Func: func(task *Task[uint64]) error {
			for {
				if utils.IsCanceled(task.Ctx) {
					return nil
				} else {
					t.Logf("task is running")
				}
			}
		},
	}))
	task, ok := tm.Get(id)
	if !ok {
		t.Fatal("task not found")
	}
	time.Sleep(time.Microsecond * 50)
	task.Cancel()
	time.Sleep(time.Millisecond)
	if task.state != CANCELED {
		t.Errorf("task status not canceled: %s", task.state)
	}
}

func TestTask_Retry(t *testing.T) {
	var next atomic.Uint64