unified task package

common/task/periodic_test.go

@@ -0,0 +1,30 @@
+package task_test
+
+import (
+	"testing"
+	"time"
+
+	. "v2ray.com/core/common/task"
+	. "v2ray.com/ext/assert"
+
+	"v2ray.com/core/common"
+)
+
+func TestPeriodicTaskStop(t *testing.T) {
+	assert := With(t)
+
+	value := 0
+	task := &Periodic{
+		Interval: time.Second * 2,
+		Execute: func() error {
+			value++
+			return nil
+		},
+	}
+	common.Must(task.Start())
+	time.Sleep(time.Second * 5)
+	common.Must(task.Close())
+	assert(value, Equals, 3)
+	time.Sleep(time.Second * 4)
+	assert(value, Equals, 3)
+}