38 lines
810 B
Go
38 lines
810 B
Go
//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)
|
|
}
|