mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2025-11-05 01:20:25 +08:00
101 lines
2.6 KiB
Go
101 lines
2.6 KiB
Go
package models_test
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/TeaOSLab/EdgeAPI/internal/db/models"
|
|
_ "github.com/go-sql-driver/mysql"
|
|
_ "github.com/iwind/TeaGo/bootstrap"
|
|
"github.com/iwind/TeaGo/dbs"
|
|
"github.com/iwind/TeaGo/logs"
|
|
"github.com/iwind/TeaGo/rands"
|
|
timeutil "github.com/iwind/TeaGo/utils/time"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestServerBandwidthStatDAO_UpdateServerBandwidth(t *testing.T) {
|
|
var dao = models.NewServerBandwidthStatDAO()
|
|
var tx *dbs.Tx
|
|
err := dao.UpdateServerBandwidth(tx, 1, 1, timeutil.Format("Ymd"), timeutil.Format("Hi"), 1024)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Log("ok")
|
|
}
|
|
|
|
func TestSeverBandwidthStatDAO_InsertManyStats(t *testing.T) {
|
|
var dao = models.NewServerBandwidthStatDAO()
|
|
var tx *dbs.Tx
|
|
var count = 1 // 测试时将此值设为一个比较大的数字
|
|
for i := 0; i < count; i++ {
|
|
var day = timeutil.Format("Ymd", time.Now().AddDate(0, 0, -rands.Int(0, 200)))
|
|
var minute = fmt.Sprintf("%02d%02d", rands.Int(0, 23), rands.Int(0, 59))
|
|
err := dao.UpdateServerBandwidth(tx, 1, 1, day, minute, 1024)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
t.Log("ok")
|
|
}
|
|
|
|
func TestServerBandwidthStatDAO_FindMonthlyPercentile(t *testing.T) {
|
|
var dao = models.NewServerBandwidthStatDAO()
|
|
var tx *dbs.Tx
|
|
t.Log(dao.FindMonthlyPercentile(tx, 23, timeutil.Format("Ym"), 95))
|
|
}
|
|
|
|
func TestServerBandwidthStatDAO_FindAllServerStatsWithMonth(t *testing.T) {
|
|
var dao = models.NewServerBandwidthStatDAO()
|
|
var tx *dbs.Tx
|
|
stats, err := dao.FindAllServerStatsWithMonth(tx, 23, timeutil.Format("Ym"))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for _, stat := range stats {
|
|
t.Logf("%+v", stat)
|
|
}
|
|
}
|
|
|
|
func TestServerBandwidthStatDAO_FindAllServerStatsWithDay(t *testing.T) {
|
|
var dao = models.NewServerBandwidthStatDAO()
|
|
var tx *dbs.Tx
|
|
stats, err := dao.FindAllServerStatsWithDay(tx, 23, timeutil.Format("Ymd"))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for _, stat := range stats {
|
|
t.Logf("%+v", stat)
|
|
}
|
|
}
|
|
|
|
func TestServerBandwidthStatDAO_Clean(t *testing.T) {
|
|
var dao = models.NewServerBandwidthStatDAO()
|
|
var tx *dbs.Tx
|
|
var before = time.Now()
|
|
err := dao.Clean(tx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Log("ok", time.Since(before).Seconds()*1000, "ms")
|
|
}
|
|
|
|
func TestServerBandwidthStatDAO_FindHourlyBandwidthStats(t *testing.T) {
|
|
var dao = models.NewServerBandwidthStatDAO()
|
|
var tx *dbs.Tx
|
|
stats, err := dao.FindHourlyBandwidthStats(tx, 23, 24)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
logs.PrintAsJSON(stats, t)
|
|
}
|
|
|
|
|
|
func TestServerBandwidthStatDAO_FindDailyBandwidthStats(t *testing.T) {
|
|
var dao = models.NewServerBandwidthStatDAO()
|
|
var tx *dbs.Tx
|
|
stats, err := dao.FindDailyBandwidthStats(tx, 23, 14)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
logs.PrintAsJSON(stats, t)
|
|
} |