添加一些test

cluster_test.go

@@ -1,25 +1,185 @@
-package timerx
+package timerx_test
 
 import (
+	"context"
 	"fmt"
 	"testing"
+	"time"
 
 	"github.com/go-redis/redis/v8"
+	"github.com/yuninks/timerx"
 )
 
-// 示例测试
+func TestCluster_AddEveryMonth(t *testing.T) {
+	ctx := context.Background()
+	redis := redis.NewClient(&redis.Options{
+		Addr: "localhost:6379",
+	})
+	defer redis.Close()
 
-// func exampleDemo(ctx context.Context) bool {
+	cluster := timerx.InitCluster(ctx, redis, "test")
-// 	fmt.Println("fff")
-// 	return false
-// }
 
-// func ExampleB() {
+	taskId := "testTask"
-// 	ctx := context.Background()
+	hour := 2
-// 	timer.InitSingle(ctx)
+	minute := 3
-// 	timer.AddToTimer(1, exampleDemo)
+	second := 4
-// 	// OutPut:
+	callback := func(ctx context.Context, data interface{}) error {
-// }
+		// do something
+		fmt.Println("Task executed:", data)
+		return nil
+	}
+	extendData := "testData"
+
+	err := cluster.AddEveryMonth(ctx, taskId, 1, hour, minute, second, callback, extendData)
+	if err != nil {
+		t.Errorf("AddEveryMonth failed, err: %v", err)
+	}
+
+	// TODO: verify the job is added to the cluster and can be executed at the specified time
+}
+
+func TestCluster_AddEveryWeek(t *testing.T) {
+	ctx := context.Background()
+	redis := redis.NewClient(&redis.Options{
+		Addr: "localhost:6379",
+	})
+	defer redis.Close()
+
+	cluster := timerx.InitCluster(ctx, redis, "test")
+
+	taskId := "testTask"
+	week := time.Sunday
+	hour := 2
+	minute := 3
+	second := 4
+	callback := func(ctx context.Context, data interface{}) error {
+		// do something
+		fmt.Println("Task executed:", data)
+		return nil
+	}
+	extendData := "testData"
+
+	err := cluster.AddEveryWeek(ctx, taskId, week, hour, minute, second, callback, extendData)
+	if err != nil {
+		t.Errorf("AddEveryWeek failed, err: %v", err)
+	}
+
+	// TODO: verify the job is added to the cluster and can be executed at the specified time
+}
+
+func TestCluster_AddEveryDay(t *testing.T) {
+	ctx := context.Background()
+	redis := redis.NewClient(&redis.Options{
+		Addr: "localhost:6379",
+	})
+	defer redis.Close()
+
+	cluster := timerx.InitCluster(ctx, redis, "test")
+
+	taskId := "testTask"
+	hour := 2
+	minute := 3
+	second := 4
+	callback := func(ctx context.Context, data interface{}) error {
+		// do something
+		fmt.Println("Task executed:", data)
+		return nil
+	}
+	extendData := "testData"
+
+	err := cluster.AddEveryDay(ctx, taskId, hour, minute, second, callback, extendData)
+	if err != nil {
+		t.Errorf("AddEveryDay failed, err: %v", err)
+	}
+
+	// TODO: verify the job is added to the cluster and can be executed at the specified time
+}
+
+func TestCluster_AddEveryHour(t *testing.T) {
+	ctx := context.Background()
+	redis := redis.NewClient(&redis.Options{
+		Addr: "localhost:6379",
+	})
+	defer redis.Close()
+
+	cluster := timerx.InitCluster(ctx, redis, "test")
+
+	taskId := "testTask"
+	minute := 3
+	second := 4
+	callback := func(ctx context.Context, data interface{}) error{
+		// do something
+		fmt.Println("Task executed:", data)
+		return nil
+	}
+	extendData := "testData"
+
+	err := cluster.AddEveryHour(ctx, taskId, minute, second, callback, extendData)
+	if err != nil {
+		t.Errorf("AddEveryHour failed, err: %v", err)
+	}
+
+	// TODO: verify the job is added to the cluster and can be executed at the specified time
+}
+
+func TestCluster_AddEveryMinute(t *testing.T) {
+	ctx := context.Background()
+	redis := redis.NewClient(&redis.Options{
+		Addr: "localhost:6379",
+	})
+	defer redis.Close()
+
+	cluster := timerx.InitCluster(ctx, redis, "test")
+
+	taskId := "testTask"
+	second := 4
+	callback := func(ctx context.Context, data interface{}) error{
+		// do something
+		fmt.Println("Task executed:", data)
+		return nil
+	}
+	extendData := "testData"
+
+	err := cluster.AddEveryMinute(ctx, taskId, second, callback, extendData)
+	if err != nil {
+		t.Errorf("AddEveryMinute failed, err: %v", err)
+	}
+
+	// TODO: verify the job is added to the cluster and can be executed at the specified time
+}
+
+func TestCluster_Add(t *testing.T) {
+	fmt.Println("66666")
+	ctx := context.Background()
+	fmt.Println("66666")
+	redis := redis.NewClient(&redis.Options{
+		Addr: "localhost:6379",
+	})
+	defer redis.Close()
+
+	t.Log("6666")
+
+	cluster := timerx.InitCluster(ctx, redis, "test")
+
+	taskId := "testTask"
+	dur := time.Second
+	callback := func(ctx context.Context, data interface{}) error {
+		// do something
+		fmt.Println("Task executed:", data)
+		return nil
+	}
+	extendData := "testData"
+
+	err := cluster.Add(ctx, taskId, dur, callback, extendData)
+	if err != nil {
+		t.Errorf("Add failed, err: %v", err)
+	}
+
+	time.Sleep(time.Second * 20)
+
+
+	// TODO: verify the job is added to the cluster and can be executed after the specified duration
+}
 
 func TestMain(m *testing.M) {
 	client := redis.NewClient(&redis.Options{