//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)
}