87 lines
1.6 KiB
Go
87 lines
1.6 KiB
Go
|
package scheduler
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/navidrome/navidrome/log"
|
||
|
|
"github.com/navidrome/navidrome/tests"
|
||
|
|
. "github.com/onsi/ginkgo/v2"
|
||
|
|
. "github.com/onsi/gomega"
|
||
|
|
"github.com/robfig/cron/v3"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestScheduler(t *testing.T) {
|
||
|
|
tests.Init(t, false)
|
||
|
|
log.SetLevel(log.LevelFatal)
|
||
|
|
RegisterFailHandler(Fail)
|
||
|
|
RunSpecs(t, "Scheduler Suite")
|
||
|
|
}
|
||
|
|
|
||
|
|
var _ = Describe("Scheduler", func() {
|
||
|
|
var s *scheduler
|
||
|
|
|
||
|
|
BeforeEach(func() {
|
||
|
|
c := cron.New(cron.WithLogger(&logger{}))
|
||
|
|
s = &scheduler{c: c}
|
||
|
|
s.c.Start() // Start the scheduler for tests
|
||
|
|
})
|
||
|
|
|
||
|
|
AfterEach(func() {
|
||
|
|
s.c.Stop() // Stop the scheduler after tests
|
||
|
|
})
|
||
|
|
|
||
|
|
It("adds and executes a job", func() {
|
||
|
|
wg := sync.WaitGroup{}
|
||
|
|
wg.Add(1)
|
||
|
|
|
||
|
|
executed := false
|
||
|
|
id, err := s.Add("@every 100ms", func() {
|
||
|
|
executed = true
|
||
|
|
wg.Done()
|
||
|
|
})
|
||
|
|
|
||
|
|
Expect(err).ToNot(HaveOccurred())
|
||
|
|
Expect(id).ToNot(BeZero())
|
||
|
|
|
||
|
|
wg.Wait()
|
||
|
|
Expect(executed).To(BeTrue())
|
||
|
|
})
|
||
|
|
|
||
|
|
It("removes a job", func() {
|
||
|
|
// Use a WaitGroup to ensure the job executes once
|
||
|
|
wg := sync.WaitGroup{}
|
||
|
|
wg.Add(1)
|
||
|
|
|
||
|
|
counter := 0
|
||
|
|
id, err := s.Add("@every 100ms", func() {
|
||
|
|
counter++
|
||
|
|
if counter == 1 {
|
||
|
|
wg.Done() // Signal that the job has executed once
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
Expect(err).ToNot(HaveOccurred())
|
||
|
|
Expect(id).ToNot(BeZero())
|
||
|
|
|
||
|
|
// Wait for the job to execute at least once
|
||
|
|
wg.Wait()
|
||
|
|
|
||
|
|
// Verify job executed
|
||
|
|
Expect(counter).To(Equal(1))
|
||
|
|
|
||
|
|
// Remove the job
|
||
|
|
s.Remove(id)
|
||
|
|
|
||
|
|
// Store the counter value
|
||
|
|
currentCount := counter
|
||
|
|
|
||
|
|
// Wait some time to ensure job doesn't execute again
|
||
|
|
time.Sleep(200 * time.Millisecond)
|
||
|
|
|
||
|
|
// Verify counter didn't increase
|
||
|
|
Expect(counter).To(Equal(currentCount))
|
||
|
|
})
|
||
|
|
})
|