From 91cb0b057e5ff92449e7fa15e10d4128b930e439 Mon Sep 17 00:00:00 2001
From: rfyiamcool <rfyiamcool@163.com>
Date: Sun, 10 Mar 2024 08:40:53 +0800
Subject: [PATCH] feat: add stop for tasker

Signed-off-by: rfyiamcool <rfyiamcool@163.com>
---
 pkg/tasker/tasker_test.go | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/pkg/tasker/tasker_test.go b/pkg/tasker/tasker_test.go
index 09cdd3a..06bc753 100644
--- a/pkg/tasker/tasker_test.go
+++ b/pkg/tasker/tasker_test.go
@@ -200,3 +200,26 @@ func TestConcurrency(t *testing.T) {
 		}
 	})
 }
+
+func TestStopTasker(t *testing.T) {
+	t.Run("Run", func(t *testing.T) {
+		taskr := New(Option{Verbose: true, Out: "../../test/tasker.out"})
+
+		var incr int
+		taskr.Task("* * * * * *", func(ctx context.Context) (int, error) {
+			incr++
+			return 0, nil
+		}, false)
+
+		go func() {
+			time.Sleep(2 * time.Second)
+			taskr.Stop()
+		}()
+		taskr.Run()
+		fmt.Println(incr)
+
+		if incr != 1 {
+			t.Errorf("the task should run 1x, not %dx", incr)
+		}
+	})
+}