task_test.go

package eva

import "testing"

func TestCustomTaskCancel(t *testing.T) {
	task := NewCustomTask(func(i ...interface{}) (interface{}, error) {
		return nil, nil
	})

	task.Cancel()

	if !task.IsCancelled() {
		t.Fatalf("invalid task status; expect: is canceled")
	}

	task.Get()
	_ = task.Error()
	task.Panic()
}

func TestCustomTaskDone(t *testing.T) {
	task := NewCustomTask(func(i ...interface{}) (interface{}, error) {
		return nil, nil
	})

	task.Run()

	if !task.IsDone() {
		t.Fatalf("invalid task status; want: is done")
	}
}

func TestCustomTaskCancelDone(t *testing.T) {
	task := NewCustomTask(func(i ...interface{}) (interface{}, error) {
		return 1, nil
	})

	task.Cancel()

	task.Run()

	if task.IsDone() {
		t.Fatalf("invalid task status; want: is not done")
	}

	task.Get()
	_ = task.Error()
	task.Panic()
}