add with limit

go-co-op · Sep 29, 2023 · 865b396 · 865b396
1 parent 1ebbdd8
commit 865b396
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/scheduler.go b/scheduler.go
@@ -37,6 +37,19 @@ func WithFakeClock(clock clockwork.Clock) SchedulerOption {
 	}
 }
 
+type LimitMode int
+
+const (
+	LimitModeReschedule = 1
+	LimitModeWait       = 2
+)
+
+func WithLimit(limit int, mode LimitMode) SchedulerOption {
+	return func(s *scheduler) error {
+		return nil
+	}
+}
+
 func WithLocation(location *time.Location) SchedulerOption {
 	return func(s *scheduler) error {
 		if location == nil {

diff --git a/scheduler_test.go b/scheduler_test.go
@@ -8,7 +8,7 @@ import (
 	"github.com/stretchr/testify/require"
 )
 
-func TestScheduler_Start(t *testing.T) {
+func TestScheduler_Cron_Start_Stop(t *testing.T) {
 	s, err := NewScheduler()
 	require.NoError(t, err)