120 lines
4.4 KiB
Go
120 lines
4.4 KiB
Go
package common
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// TestRunHandlerSafely_PanicBecomesError robfig/cron v3 的 cron.New() 不带 Recover 链，
|
|
// startJob 是裸的 `go j.Run()`——任务 panic 会崩掉整个进程。此处必须兜住并转成
|
|
// 普通 error，才能既保住进程、又让失败落进 sys_cron_jobs 供管理后台查看。
|
|
func TestRunHandlerSafely_PanicBecomesError(t *testing.T) {
|
|
job := &CronJob{
|
|
Name: "panicking_job",
|
|
Handler: func(_ context.Context) error {
|
|
panic("boom")
|
|
},
|
|
}
|
|
|
|
var err error
|
|
assert.NotPanics(t, func() {
|
|
err = runHandlerSafely(context.Background(), job)
|
|
}, "panic 必须被兜住，否则整个进程会崩")
|
|
require.Error(t, err)
|
|
assert.Contains(t, err.Error(), "boom", "错误信息应保留 panic 内容供排查")
|
|
}
|
|
|
|
func TestRunHandlerSafely_PassesThroughReturnValue(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
okJob := &CronJob{Name: "ok", Handler: func(_ context.Context) error { return nil }}
|
|
assert.NoError(t, runHandlerSafely(ctx, okJob))
|
|
|
|
failJob := &CronJob{Name: "fail", Handler: func(_ context.Context) error {
|
|
return assert.AnError
|
|
}}
|
|
assert.ErrorIs(t, runHandlerSafely(ctx, failJob), assert.AnError, "普通错误必须原样返回")
|
|
}
|
|
|
|
// TestReschedule_OnlyUpdatesRecordWhenNotStarted StartBackground 之前没有 cron 条目，
|
|
// 此时重排只改 job.Schedule，供随后的 StartBackground 按新表达式注册。
|
|
func TestReschedule_OnlyUpdatesRecordWhenNotStarted(t *testing.T) {
|
|
cs := NewCronScheduler()
|
|
cs.Register("job_a", "任务A", "*/5 * * * *", func(_ context.Context) error { return nil })
|
|
|
|
require.NoError(t, cs.Reschedule(context.Background(), "job_a", "@every 30m"))
|
|
|
|
jobs := cs.ListJobs()
|
|
require.Len(t, jobs, 1)
|
|
assert.Equal(t, "@every 30m", jobs[0].Schedule)
|
|
}
|
|
|
|
func TestReschedule_UnregisteredTaskErrors(t *testing.T) {
|
|
cs := NewCronScheduler()
|
|
assert.Error(t, cs.Reschedule(context.Background(), "nonexistent", "@every 5m"))
|
|
}
|
|
|
|
// TestReschedule_InvalidExprKeepsExistingSchedule 一次错误配置不能把任务从调度器里彻底摘掉，
|
|
// 因此实现是「先加新条目、成功后再摘旧条目」。
|
|
func TestReschedule_InvalidExprKeepsExistingSchedule(t *testing.T) {
|
|
ctx := context.Background()
|
|
cs := NewCronScheduler()
|
|
cs.Register("job_b", "任务B", "@every 5m", func(_ context.Context) error { return nil })
|
|
cs.StartBackground(ctx)
|
|
defer cs.cron.Stop()
|
|
|
|
before := cs.ListJobs()[0].Schedule
|
|
require.Error(t, cs.Reschedule(ctx, "job_b", "不是合法表达式"))
|
|
assert.Equal(t, before, cs.ListJobs()[0].Schedule, "非法表达式不得改动原调度")
|
|
|
|
// 条目仍在调度器中，未被误摘
|
|
assert.Len(t, cs.cron.Entries(), 1)
|
|
}
|
|
|
|
// TestReschedule_ValidExprReplacesEntry 同时验证 @every 语法确实被 cron.New() 的
|
|
// 默认解析器接受——探测间隔配置正是靠它支持任意分钟数（*/N 在 N>=60 时会静默失真）。
|
|
func TestReschedule_ValidExprReplacesEntry(t *testing.T) {
|
|
ctx := context.Background()
|
|
cs := NewCronScheduler()
|
|
cs.Register("job_c", "任务C", "@every 5m", func(_ context.Context) error { return nil })
|
|
cs.StartBackground(ctx)
|
|
defer cs.cron.Stop()
|
|
|
|
require.Len(t, cs.cron.Entries(), 1)
|
|
oldID := cs.entryIDs["job_c"]
|
|
|
|
require.NoError(t, cs.Reschedule(ctx, "job_c", "@every 90m"))
|
|
|
|
assert.Equal(t, "@every 90m", cs.ListJobs()[0].Schedule)
|
|
assert.Len(t, cs.cron.Entries(), 1, "旧条目必须被摘除，否则会双倍触发")
|
|
assert.NotEqual(t, oldID, cs.entryIDs["job_c"], "条目 ID 应已更新")
|
|
}
|
|
|
|
// TestOnSettingsChanged_CallbackPanicDoesNotAffectOthers 回调在订阅 goroutine 内串行执行，
|
|
// 一个回调 panic 不能带崩订阅循环或阻断后续回调。
|
|
func TestOnSettingsChanged_CallbackPanicDoesNotAffectOthers(t *testing.T) {
|
|
settingsHookMu.Lock()
|
|
saved := settingsHooks
|
|
settingsHooks = nil
|
|
settingsHookMu.Unlock()
|
|
defer func() {
|
|
settingsHookMu.Lock()
|
|
settingsHooks = saved
|
|
settingsHookMu.Unlock()
|
|
}()
|
|
|
|
var firstCalled, thirdCalled bool
|
|
OnSettingsChanged(func(_ context.Context, _ string) { firstCalled = true })
|
|
OnSettingsChanged(func(_ context.Context, _ string) { panic("hook boom") })
|
|
OnSettingsChanged(func(_ context.Context, _ string) { thirdCalled = true })
|
|
|
|
assert.NotPanics(t, func() {
|
|
fireSettingsChanged(context.Background(), "some_key")
|
|
})
|
|
assert.True(t, firstCalled)
|
|
assert.True(t, thirdCalled, "前一个回调 panic 不得阻断后续回调")
|
|
}
|