update

scheduler/scheduler_test.go

@@ -0,0 +1,86 @@
+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))
+	})
+})