clockwork_internal_test.go

package clockwork

import (
	"os"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
)

func something() {}

func TestScheduleNextRun(t *testing.T) {
	sched := NewScheduler()
	sched.activateTestMode()

	sched.Schedule().Every(10).Seconds().Do(something)
	s := sched.jobs[0].nextScheduledRun
	assert.Equal(t, s, time.Date(1, 1, 1, 1, 1, 10, 0, time.Local))

	sched.Schedule().Every(3).Minutes().Do(something)
	s = sched.jobs[1].nextScheduledRun
	assert.Equal(t, s, time.Date(1, 1, 1, 1, 4, 0, 0, time.Local))

	sched.Schedule().Every(4).Hours().Do(something)
	s = sched.jobs[2].nextScheduledRun
	assert.Equal(t, s, time.Date(1, 1, 1, 5, 1, 0, 0, time.Local))

	sched.Schedule().Every(2).Days().At("12:32").Do(something)
	s = sched.jobs[3].nextScheduledRun
	assert.Equal(t, s, time.Date(1, 1, 3, 12, 32, 0, 0, time.Local))

	sched.Schedule().Every(12).Weeks().Do(something)
	s = sched.jobs[4].nextScheduledRun
	assert.Equal(t, s, time.Date(1, 3, 26, 1, 1, 0, 0, time.Local))

	sched.Schedule().EverySingle().Second().Do(something) // EverySingle is "shorthand" for Every(1)
	s = sched.jobs[5].nextScheduledRun
	assert.Equal(t, s, time.Date(1, 1, 1, 1, 1, 1, 0, time.Local))

	sched.Schedule().EverySingle().Monday().At("9:10").Do(something)
	s = sched.jobs[6].nextScheduledRun
	assert.Equal(t, s, time.Date(1, 1, 8, 9, 10, 0, 0, time.Local))

	sched.Schedule().Every().Saturday().At("8:00").Do(something)
	s = sched.jobs[7].nextScheduledRun
	assert.Equal(t, s, time.Date(1, 1, 13, 8, 0, 0, 0, time.Local))

	sched.Schedule().Every().Tuesday().At("9:10").Do(something)
	s = sched.jobs[8].nextScheduledRun
	assert.Equal(t, s, time.Date(1, 1, 9, 9, 10, 0, 0, time.Local))

	sched.Schedule().Every().Wednesday().At("19:10").Do(something)
	s = sched.jobs[9].nextScheduledRun
	assert.Equal(t, s, time.Date(1, 1, 10, 19, 10, 0, 0, time.Local))

	sched.Schedule().Every().Thursday().At("19:10").Do(something)
	s = sched.jobs[10].nextScheduledRun
	assert.Equal(t, s, time.Date(1, 1, 11, 19, 10, 0, 0, time.Local))

	sched.Schedule().Every().Friday().At("19:10").Do(something)
	s = sched.jobs[11].nextScheduledRun
	assert.Equal(t, s, time.Date(1, 1, 12, 19, 10, 0, 0, time.Local))

	sched.Schedule().Every().Sunday().At("8:00").Do(something)
	s = sched.jobs[12].nextScheduledRun
	assert.Equal(t, s, time.Date(1, 1, 7, 8, 0, 0, 0, time.Local))
}

func TestSetPollingInterval(t *testing.T) {
	sched := NewScheduler()
	sched.activateTestMode()

	assert.Equal(t, sched.polling_interval, 333)

	sched.SetPollingInterval(500)
	assert.Equal(t, sched.polling_interval, 500)
}

// Main test entrypoint
func TestMain(m *testing.M) {
	os.Exit(m.Run())
}