Initial commit (code only without large binaries)

EdgeAPI/internal/tasks/task_send_messages_plus_test.go

@@ -0,0 +1,37 @@
+//go:build plus
+
+package tasks_test
+
+import (
+	"github.com/TeaOSLab/EdgeAPI/internal/tasks"
+	_ "github.com/iwind/TeaGo/bootstrap"
+	"github.com/iwind/TeaGo/dbs"
+	"github.com/iwind/TeaGo/logs"
+	"testing"
+	"time"
+)
+
+func TestSendMessagesTask_Loop(t *testing.T) {
+	dbs.NotifyReady()
+
+	var task = tasks.NewSendMessagesTask(5 * time.Second)
+	err := task.Loop()
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log("ok")
+}
+
+func TestSendMessagesTask_checkRate(t *testing.T) {
+	task := tasks.NewSendMessagesTask(5 * time.Second)
+	t.Log(task.CheckRate(1, 0, 0))
+	t.Log(task.CheckRate(1, 1, 2))
+	t.Log(task.CheckRate(1, 1, 2))
+	t.Log(task.CheckRate(1, 1, 2))
+
+	var statMap = task.StatMap()
+	logs.PrintAsJSON(statMap, t)
+	statMap[1].Timestamp = time.Now().Unix() - 61
+	t.Log(task.CheckRate(1, 1, 2))
+	logs.PrintAsJSON(statMap, t)
+}