-
Notifications
You must be signed in to change notification settings - Fork 13
/
clockwork_internal_test.go
83 lines (62 loc) · 2.58 KB
/
clockwork_internal_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
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())
}