code refactor

2026-05-20 01:34:24 +08:00 · 2024-08-19 00:31:32 +08:00
parent 6fcb0a2b3c
commit e86d4e05ce
73 changed files with 1404 additions and 1344 deletions
@@ -0,0 +1,141 @@
+package cron
+
+import (
+	"testing"
+	"time"
+
+	"github.com/go-co-op/gocron/v2"
+	"github.com/stretchr/testify/assert"
+
+	"github.com/mayswind/ezbookkeeping/pkg/core"
+)
+
+func TestCronJobNextRunTimeWithIntervalPeriod(t *testing.T) {
+	scheduler, err := gocron.NewScheduler(
+		gocron.WithLocation(time.Local),
+	)
+	assert.Nil(t, err)
+
+	job := CronJob{
+		Name:        "TestCronJobWithIntervalPeriod",
+		Description: "The test cron job",
+		Period: CronJobIntervalPeriod{
+			Interval: 2*time.Hour + 34*time.Minute + 56*time.Second,
+		},
+		Run: func(c *core.CronContext) error {
+			return nil
+		},
+	}
+
+	gocronJob, err := scheduler.NewJob(
+		job.Period.ToJobDefinition(),
+		gocron.NewTask(job.doRun),
+		gocron.WithName(job.Name),
+		gocron.WithSingletonMode(gocron.LimitModeReschedule),
+	)
+	assert.Nil(t, err)
+
+	scheduler.Start()
+
+	currentTime := time.Now()
+	nextRunTime, err := gocronJob.NextRun()
+	assert.Nil(t, err)
+
+	expectedNextTime := currentTime.Add(2 * time.Hour).Add(34 * time.Minute).Add(56 * time.Second)
+
+	assert.Equal(t, expectedNextTime.Year(), nextRunTime.Year())
+	assert.Equal(t, expectedNextTime.Month(), nextRunTime.Month())
+	assert.Equal(t, expectedNextTime.Day(), nextRunTime.Day())
+	assert.Equal(t, expectedNextTime.Hour(), nextRunTime.Hour())
+	assert.Equal(t, expectedNextTime.Minute(), nextRunTime.Minute())
+	assert.Equal(t, expectedNextTime.Second(), nextRunTime.Second())
+
+	err = scheduler.Shutdown()
+	assert.Nil(t, err)
+}
+
+func TestCronJobNextRunTimeWithFixedHourPeriod(t *testing.T) {
+	scheduler, err := gocron.NewScheduler(
+		gocron.WithLocation(time.Local),
+	)
+	assert.Nil(t, err)
+
+	job := CronJob{
+		Name:        "TestCronJobWithFixedHourPeriod",
+		Description: "The test cron job",
+		Period: CronJobFixedHourPeriod{
+			Hour: 0,
+		},
+		Run: func(c *core.CronContext) error {
+			return nil
+		},
+	}
+
+	gocronJob, err := scheduler.NewJob(
+		job.Period.ToJobDefinition(),
+		gocron.NewTask(job.doRun),
+		gocron.WithName(job.Name),
+		gocron.WithSingletonMode(gocron.LimitModeReschedule),
+	)
+	assert.Nil(t, err)
+
+	scheduler.Start()
+
+	nextRunTime, err := gocronJob.NextRun()
+	assert.Nil(t, err)
+
+	tomorrow := time.Now().AddDate(0, 0, 1)
+
+	assert.Equal(t, tomorrow.Year(), nextRunTime.Year())
+	assert.Equal(t, tomorrow.Month(), nextRunTime.Month())
+	assert.Equal(t, tomorrow.Day(), nextRunTime.Day())
+	assert.Equal(t, 0, nextRunTime.Hour())
+	assert.Equal(t, 0, nextRunTime.Minute())
+	assert.Equal(t, 0, nextRunTime.Second())
+
+	err = scheduler.Shutdown()
+	assert.Nil(t, err)
+}
+
+func TestCronJobNextRunTimeWithFixedTimePeriod(t *testing.T) {
+	scheduler, err := gocron.NewScheduler(
+		gocron.WithLocation(time.Local),
+	)
+	assert.Nil(t, err)
+
+	expectedTime := time.Now().Add(123456789 * time.Second)
+
+	job := CronJob{
+		Name:        "TestCronJobWithFixedTimePeriod",
+		Description: "The test cron job",
+		Period: CronJobFixedTimePeriod{
+			Time: expectedTime,
+		},
+		Run: func(c *core.CronContext) error {
+			return nil
+		},
+	}
+
+	gocronJob, err := scheduler.NewJob(
+		job.Period.ToJobDefinition(),
+		gocron.NewTask(job.doRun),
+		gocron.WithName(job.Name),
+		gocron.WithSingletonMode(gocron.LimitModeReschedule),
+	)
+	assert.Nil(t, err)
+
+	scheduler.Start()
+
+	nextRunTime, err := gocronJob.NextRun()
+	assert.Nil(t, err)
+
+	assert.Equal(t, expectedTime.Year(), nextRunTime.Year())
+	assert.Equal(t, expectedTime.Month(), nextRunTime.Month())
+	assert.Equal(t, expectedTime.Day(), nextRunTime.Day())
+	assert.Equal(t, expectedTime.Hour(), nextRunTime.Hour())
+	assert.Equal(t, expectedTime.Minute(), nextRunTime.Minute())
+	assert.Equal(t, expectedTime.Second(), nextRunTime.Second())
+
+	err = scheduler.Shutdown()
+	assert.Nil(t, err)
+}