Ejemplo n.º 1
0
// Tests if jobs are starting before starting the scheduler.
func TestStart_NoStart(test *testing.T) {
	sh := &Scheduler{}
	var wg sync.WaitGroup

	wg.Add(1)
	job1 := &counterJob{}
	job2 := &counterJob{}
	sh.Schedule("hello", job1, &t.When{Every: t.Every(200).Milliseconds()})
	sh.Schedule("hi", job2, &t.When{Every: t.Every(100).Milliseconds()})
	time.AfterFunc(300*time.Millisecond, func() {
		defer wg.Done()
		if job1.Count+job2.Count > 1 {
			test.Fatalf("scheduler not started but jobs have run")
		}
	})
	wg.Wait()
}
Ejemplo n.º 2
0
// Tests if repeating job is running on time.
func TestSchedule_OntimeRepeating(test *testing.T) {
	sh := &Scheduler{}
	var wg sync.WaitGroup

	wg.Add(1)
	job := &counterJob{}
	sh.Schedule("hi", job, &t.When{Every: t.Every(100).Milliseconds()})
	time.AfterFunc(300*time.Millisecond, func() {
		defer wg.Done()
		sh.Cancel("hi")
		if job.Count < 2 {
			test.Fatalf("scheduler worked for %v, expected to run 3 times", job.Count)
		}
	})
	go sh.Start()
	wg.Wait()
}
Ejemplo n.º 3
0
// Tests if jobs are being cancelled.
func TestCancel(test *testing.T) {
	sh := &Scheduler{}
	var wg sync.WaitGroup

	wg.Add(1)
	job := &counterJob{}
	sh.Schedule("hi", job, &t.When{Every: t.Every(100).Milliseconds()})
	time.AfterFunc(100*time.Millisecond, func() {
		defer wg.Done()
		sh.Cancel("hi")
		if job.Count > 2 {
			test.Fatalf("scheduler cancelled but job worked more than expected")
		}
	})
	sh.Start()
	wg.Wait()
}
Ejemplo n.º 4
0
// Tests if a job is being scheduled if it's registered after the start.
func TestStart_AfterStart(test *testing.T) {
	sh := &Scheduler{}
	var wg sync.WaitGroup

	wg.Add(1)
	job := &counterJob{}
	sh.Start()
	sh.Schedule("hi", job, &t.When{Every: t.Every(100).Milliseconds()})
	time.AfterFunc(300*time.Millisecond, func() {
		defer wg.Done()
		sh.Cancel("hi")
		if job.Count == 0 {
			test.Fatalf("job is expected to run even though it's scheduled after Start, but it didn't")
		}
	})
	wg.Wait()
}
Ejemplo n.º 5
0
// Tests if job is retried if it fails.
func TestRetryCount(test *testing.T) {
	sh := &Scheduler{}
	var wg sync.WaitGroup

	wg.Add(1)
	job := &errorJob{errorAfter: 2}
	sh.ScheduleWithOpts("hi", job, &t.Opts{
		RetryCount: 2,
		When:       &t.When{Every: t.Every(100).Milliseconds()},
	})
	time.AfterFunc(200*time.Millisecond, func() {
		defer wg.Done()
		sh.Cancel("hi")
		if job.count < 2 {
			test.Fatalf("expected to retry for 2 times, ram %v times", job.count)
		}
	})
	sh.Start()
	wg.Wait()
}