41 lines
1.3 KiB
Go
41 lines
1.3 KiB
Go
|
package scheduler
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"sync/atomic"
|
|||
|
|
"testing"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// TestNewSkipsNonPositiveIntervals：time.NewTicker 对非正间隔会 panic，这类任务必须在建表时丢掉。
|
|||
|
|
func TestNewSkipsNonPositiveIntervals(t *testing.T) {
|
|||
|
|
s := New(
|
|||
|
|
Job{Name: "zero", Interval: 0, Run: func(context.Context) {}},
|
|||
|
|
Job{Name: "negative", Interval: -time.Second, Run: func(context.Context) {}},
|
|||
|
|
Job{Name: "ok", Interval: time.Hour, Run: func(context.Context) {}},
|
|||
|
|
)
|
|||
|
|
if len(s.jobs) != 1 || s.jobs[0].Name != "ok" {
|
|||
|
|
t.Fatalf("非正间隔的任务应被跳过，实际保留 %d 个", len(s.jobs))
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// TestRunJobRecoversFromPanic：单个任务 panic 不能带走整个进程，且要能继续下一轮。
|
|||
|
|
func TestRunJobRecoversFromPanic(t *testing.T) {
|
|||
|
|
var calls int32
|
|||
|
|
s := New(Job{Name: "panic", Interval: 5 * time.Millisecond, Run: func(context.Context) {
|
|||
|
|
atomic.AddInt32(&calls, 1)
|
|||
|
|
panic("boom")
|
|||
|
|
}})
|
|||
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|||
|
|
defer cancel()
|
|||
|
|
s.Start(ctx)
|
|||
|
|
|
|||
|
|
deadline := time.Now().Add(2 * time.Second)
|
|||
|
|
for time.Now().Before(deadline) && atomic.LoadInt32(&calls) < 3 {
|
|||
|
|
time.Sleep(5 * time.Millisecond)
|
|||
|
|
}
|
|||
|
|
if got := atomic.LoadInt32(&calls); got < 3 {
|
|||
|
|
t.Fatalf("panic 后应继续后续轮次，实际只执行了 %d 次", got)
|
|||
|
|
}
|
|||
|
|
}
|