Function TestSimpleTaskScheduler_Shutdown

scheduler_test.go:290–311

(t *testing.T)

}

func TestSimpleTaskScheduler_Shutdown(t *testing.T) {
	scheduler := NewSimpleTaskScheduler(NewDefaultTaskExecutor())

	var counter int32

	_, err := scheduler.ScheduleAtFixedRate(func(ctx context.Context) {
		atomic.AddInt32(&counter, 1)
		<-time.After(500 * time.Millisecond)
	}, 1*time.Second)

	assert.Nil(t, err)

	<-time.After(2 * time.Second)
	scheduler.Shutdown()

	expected := counter
	<-time.After(3 * time.Second)

	assert.True(t, scheduler.IsShutdown())
	assert.Equal(t, expected, counter,
		"after shutdown, previously scheduled tasks should not be rescheduled", counter)
}