mirror of
https://github.com/StarFleetCPTN/GoMFT.git
synced 2026-09-20 13:30:51 +02:00
StarFleetCPTN
558e81c7e8
- Update .gitignore to include new provider files and ensure proper tracking. - Refactor job and config forms to support multiple configuration selections, improving user experience. - Implement logic to handle the initialization of skipProcessedFiles with a default value. - Add new provider form templates for better organization and management of source and destination configurations. - Enhance tests for provider forms and job configurations to ensure robust functionality. - Introduce nullable handling for skipProcessedFiles in the database schema and update related migrations. - Improve error handling and validation in job creation and editing processes.
91 lines
2.2 KiB
Go
91 lines
2.2 KiB
Go
package scheduler
|
|
|
|
import (
|
|
"github.com/starfleetcptn/gomft/internal/db"
|
|
)
|
|
|
|
// MockScheduler is a mock implementation of a scheduler for testing
|
|
type MockScheduler struct {
|
|
ScheduledJobs map[uint]bool
|
|
UnscheduledJobs map[uint]bool
|
|
RunJobsNow map[uint]bool
|
|
ScheduleJobErr error
|
|
RunJobNowErr error
|
|
UnscheduleJobCalls int
|
|
MultiConfigJobs map[uint][]uint // Track jobs with multiple configs (job ID -> config IDs)
|
|
}
|
|
|
|
// NewMockScheduler creates a new mock scheduler
|
|
func NewMockScheduler() *MockScheduler {
|
|
return &MockScheduler{
|
|
ScheduledJobs: make(map[uint]bool),
|
|
UnscheduledJobs: make(map[uint]bool),
|
|
RunJobsNow: make(map[uint]bool),
|
|
MultiConfigJobs: make(map[uint][]uint),
|
|
}
|
|
}
|
|
|
|
// ScheduleJob mocks scheduling a job
|
|
func (m *MockScheduler) ScheduleJob(job *db.Job) error {
|
|
if m.ScheduleJobErr != nil {
|
|
return m.ScheduleJobErr
|
|
}
|
|
|
|
if job.Enabled {
|
|
m.ScheduledJobs[job.ID] = true
|
|
delete(m.UnscheduledJobs, job.ID)
|
|
} else {
|
|
m.UnscheduledJobs[job.ID] = true
|
|
delete(m.ScheduledJobs, job.ID)
|
|
}
|
|
|
|
// Track jobs with multiple configurations
|
|
if job.ConfigIDs != "" {
|
|
m.MultiConfigJobs[job.ID] = job.GetConfigIDsList()
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// RunJobNow mocks running a job immediately
|
|
func (m *MockScheduler) RunJobNow(jobID uint) error {
|
|
if m.RunJobNowErr != nil {
|
|
return m.RunJobNowErr
|
|
}
|
|
|
|
m.RunJobsNow[jobID] = true
|
|
|
|
// In a real implementation, this would execute the job
|
|
// But for testing, we just record that it was called
|
|
return nil
|
|
}
|
|
|
|
// UnscheduleJob mocks unscheduling a job
|
|
func (m *MockScheduler) UnscheduleJob(jobID uint) {
|
|
m.UnscheduleJobCalls++
|
|
m.UnscheduledJobs[jobID] = true
|
|
delete(m.ScheduledJobs, jobID)
|
|
delete(m.MultiConfigJobs, jobID)
|
|
}
|
|
|
|
// Stop mocks stopping the scheduler
|
|
func (m *MockScheduler) Stop() {
|
|
// Nothing to do
|
|
}
|
|
|
|
// RotateLogs mocks log rotation
|
|
func (m *MockScheduler) RotateLogs() error {
|
|
return nil
|
|
}
|
|
|
|
// IsJobWithMultipleConfigs checks if a job is scheduled with multiple configs
|
|
func (m *MockScheduler) IsJobWithMultipleConfigs(jobID uint) bool {
|
|
configs, exists := m.MultiConfigJobs[jobID]
|
|
return exists && len(configs) > 1
|
|
}
|
|
|
|
// GetConfigsForJob returns the configs for a job
|
|
func (m *MockScheduler) GetConfigsForJob(jobID uint) []uint {
|
|
return m.MultiConfigJobs[jobID]
|
|
}
|