BackupX/server/internal/scheduler/service_test.go

package scheduler

import (
	"backupx/server/internal/repository"
	servicepkg "backupx/server/internal/service"
	"context"
	"testing"
	"time"

	"backupx/server/internal/model"
)

type fakeTaskRepository struct {
	items []model.BackupTask
}

func (r *fakeTaskRepository) List(context.Context, repository.BackupTaskListOptions) ([]model.BackupTask, error) {
	return nil, nil
}
func (r *fakeTaskRepository) FindByID(context.Context, uint) (*model.BackupTask, error) {
	return nil, nil
}
func (r *fakeTaskRepository) FindByName(context.Context, string) (*model.BackupTask, error) {
	return nil, nil
}
func (r *fakeTaskRepository) ListSchedulable(context.Context) ([]model.BackupTask, error) {
	return r.items, nil
}
func (r *fakeTaskRepository) Count(context.Context) (int64, error)        { return 0, nil }
func (r *fakeTaskRepository) CountEnabled(context.Context) (int64, error) { return 0, nil }
func (r *fakeTaskRepository) CountByStorageTargetID(context.Context, uint) (int64, error) {
	return 0, nil
}
func (r *fakeTaskRepository) Create(context.Context, *model.BackupTask) error { return nil }
func (r *fakeTaskRepository) Update(context.Context, *model.BackupTask) error { return nil }
func (r *fakeTaskRepository) Delete(context.Context, uint) error              { return nil }

type fakeRunner struct{ taskIDs []uint }

func (r *fakeRunner) RunTaskByID(_ context.Context, id uint) (*servicepkg.BackupRecordDetail, error) {
	r.taskIDs = append(r.taskIDs, id)
	return nil, nil
}

func TestServiceSyncTaskAndTrigger(t *testing.T) {
	repo := &fakeTaskRepository{}
	runner := &fakeRunner{}
	service := NewService(repo, runner, nil)
	if err := service.SyncTask(context.Background(), &model.BackupTask{ID: 1, Enabled: true, CronExpr: "*/1 * * * * *"}); err != nil {
		t.Fatalf("SyncTask returned error: %v", err)
	}
	service.cron.Start()
	defer service.cron.Stop()
	time.Sleep(1100 * time.Millisecond)
	if len(runner.taskIDs) == 0 {
		t.Fatalf("expected scheduled runner to be triggered")
	}
}