✨ 重新初始化项目

utils/timer/timed_task_test.go

@@ -0,0 +1,61 @@
+package timer
+
+import (
+	"fmt"
+	"github.com/stretchr/testify/assert"
+	"testing"
+	"time"
+)
+
+var job = mockJob{}
+
+type mockJob struct{}
+
+func (job mockJob) Run() {
+	mockFunc()
+}
+func mockFunc() {
+	time.Sleep(time.Second)
+	fmt.Println("1s...")
+}
+func TestNewTimerTask(t *testing.T) {
+	tm := NewTimerTask()
+	_tm := tm.(*timer)
+	{
+		_, err := tm.AddTaskByFunc("func", "@every 1s", mockFunc)
+		assert.Nil(t, err)
+		_, ok := _tm.taskList["func"]
+		if !ok {
+			t.Error("no find func")
+		}
+	}
+	{
+		_, err := tm.AddTaskByJob("job", "@every 1s", job)
+		assert.Nil(t, err)
+		_, ok := _tm.taskList["job"]
+		if !ok {
+			t.Error("no find job")
+		}
+	}
+	{
+		_, ok := tm.FindCron("func")
+		if !ok {
+			t.Error("no find func")
+		}
+		_, ok = tm.FindCron("job")
+		if !ok {
+			t.Error("no find job")
+		}
+		_, ok = tm.FindCron("none")
+		if ok {
+			t.Error("find none")
+		}
+	}
+	{
+		tm.Clear("func")
+		_, ok := tm.FindCron("func")
+		if ok {
+			t.Error("find func")
+		}
+	}
+}