EdgeNode/internal/utils/idles/run_test.go

// Copyright 2024 GoEdge CDN goedge.cdn@gmail.com. All rights reserved. Official site: https://goedge.cloud .

package idles_test

import (
	"testing"
	"time"

	"github.com/TeaOSLab/EdgeNode/internal/utils/idles"
	"github.com/TeaOSLab/EdgeNode/internal/utils/testutils"
	_ "github.com/iwind/TeaGo/bootstrap"
	"github.com/iwind/TeaGo/types"
	timeutil "github.com/iwind/TeaGo/utils/time"
)

func TestCheckHourlyLoad(t *testing.T) {
	for i := 0; i < 10; i++ {
		idles.CheckHourlyLoad(5)
		idles.CheckHourlyLoad(1)
		idles.CheckHourlyLoad(3)
		idles.CheckHourlyLoad(2)
		idles.CheckHourlyLoad(4)
	}

	t.Log(idles.TestMinLoadHours())
	for h, v := range idles.TestHourlyLoadMap() {
		t.Log(types.String(h)+":", v.Avg)
	}
}

func TestRun(t *testing.T) {
	idles.TestSetMinLoadHours([]int{0, time.Now().Hour()})
	idles.Run(func() {
		t.Log("run once")
	})
}

func TestRunTicker(t *testing.T) {
	if !testutils.IsSingleTesting() {
		return
	}

	var ticker = time.NewTicker(10 * time.Second)
	idles.RunTicker(ticker, func() {
		t.Log(timeutil.Format("H:i:s"), "run once")
	})
}

func TestIsMinHour(t *testing.T) {
	t.Log(idles.IsMinHour())
}