Function TestSimpleTaskExecutor_Schedule_MultiTasks

executor_test.go:174–215

(t *testing.T)

}

func TestSimpleTaskExecutor_Schedule_MultiTasks(t *testing.T) {
	executor := NewSimpleTaskExecutor(NewDefaultTaskRunner())

	var task1Counter int32
	var task2Counter int32
	var task3Counter int32

	task1, err := executor.ScheduleAtFixedRate(func(ctx context.Context) {
		atomic.AddInt32(&task1Counter, 1)
		<-time.After(500 * time.Millisecond)
	}, 1*time.Second, 200*time.Millisecond)

	assert.Nil(t, err)

	task2, err := executor.ScheduleWithFixedDelay(func(ctx context.Context) {
		atomic.AddInt32(&task2Counter, 1)
		<-time.After(500 * time.Millisecond)
	}, 0, 200*time.Millisecond)

	assert.Nil(t, err)

	task3, err := executor.ScheduleAtFixedRate(func(ctx context.Context) {
		atomic.AddInt32(&task3Counter, 1)
	}, 0, 200*time.Millisecond)

	assert.Nil(t, err)

	<-time.After(2*time.Second - 50*time.Millisecond)

	task1.Cancel()
	task2.Cancel()
	task3.Cancel()

	assert.True(t, task1Counter >= 5 && task1Counter <= 10,
		"number of scheduled task 1 execution must be between 5 and 10, actual: %d", task1Counter)

	assert.True(t, task2Counter >= 1 && task2Counter <= 3,
		"number of scheduled task 2 execution must be between 1 and 3, actual: %d", task2Counter)

	assert.True(t, task3Counter >= 1 && task3Counter <= 10,
		"number of scheduled task execution must be between 5 and 10, actual: %d", task3Counter)
}

func TestSimpleTaskExecutor_ScheduleWithNilTask(t *testing.T) {
	executor := NewSimpleTaskExecutor(NewDefaultTaskRunner())