mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2025-11-03 15:00:27 +08:00
107 lines
2.5 KiB
Go
107 lines
2.5 KiB
Go
package models_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"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"
|
|
timeutil "github.com/iwind/TeaGo/utils/time"
|
|
)
|
|
|
|
func TestUserBandwidthStatDAO_FindUserPeekBandwidthInMonth(t *testing.T) {
|
|
var dao = models.NewUserBandwidthStatDAO()
|
|
var tx *dbs.Tx
|
|
|
|
// max
|
|
{
|
|
stat, err := dao.FindUserPeekBandwidthInMonth(tx, 1, timeutil.Format("Ym"), false)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
logs.PrintAsJSON(stat, t)
|
|
}
|
|
|
|
// avg
|
|
{
|
|
stat, err := dao.FindUserPeekBandwidthInMonth(tx, 1, timeutil.Format("Ym"), true)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
logs.PrintAsJSON(stat, t)
|
|
}
|
|
}
|
|
|
|
func TestUserBandwidthStatDAO_FindUserPeekBandwidthInDay(t *testing.T) {
|
|
var dao = models.NewUserBandwidthStatDAO()
|
|
var tx *dbs.Tx
|
|
stat, err := dao.FindUserPeekBandwidthInDay(tx, 1, timeutil.Format("Ymd"), false)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
logs.PrintAsJSON(stat, t)
|
|
}
|
|
|
|
func TestUserBandwidthStatDAO_UpdateServerBandwidth(t *testing.T) {
|
|
var dao = models.NewUserBandwidthStatDAO()
|
|
var tx *dbs.Tx
|
|
err := dao.UpdateUserBandwidth(tx, 1, 0, timeutil.Format("Ymd"), timeutil.FormatTime("Hi", time.Now().Unix()/300*300), 1024, 300, 0, 0, 0, 0, 0)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Log("ok")
|
|
}
|
|
|
|
func TestUserBandwidthStatDAO_CleanDays(t *testing.T) {
|
|
var dao = models.NewUserBandwidthStatDAO()
|
|
var tx *dbs.Tx
|
|
err := dao.CleanDays(tx, 100)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Log("ok")
|
|
}
|
|
|
|
func TestUserBandwidthStatDAO_FindBandwidthStatsBetweenDays(t *testing.T) {
|
|
var dao = models.NewUserBandwidthStatDAO()
|
|
var tx *dbs.Tx
|
|
stats, err := dao.FindUserBandwidthStatsBetweenDays(tx, 1, 0, timeutil.Format("Ymd", time.Now().AddDate(0, 0, -2)), timeutil.Format("Ymd"), false)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var dataCount = 0
|
|
for _, stat := range stats {
|
|
t.Log(stat.Day, stat.TimeAt, "bytes:", stat.Bytes, "bits:", stat.Bits)
|
|
if stat.Bytes > 0 {
|
|
dataCount++
|
|
}
|
|
}
|
|
t.Log("data count:", dataCount)
|
|
}
|
|
|
|
func TestUserBandwidthStatDAO_FindPercentileBetweenDays(t *testing.T) {
|
|
var dao = models.NewUserBandwidthStatDAO()
|
|
var tx *dbs.Tx
|
|
stat, err := dao.FindPercentileBetweenDays(tx, 1, 0, timeutil.Format("Ymd"), timeutil.Format("Ymd"), 95, false)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
logs.PrintAsJSON(stat, t)
|
|
}
|
|
|
|
func TestUserBandwidthStatDAO_HasFullData(t *testing.T) {
|
|
var tx *dbs.Tx
|
|
var dao = models.NewUserBandwidthStatDAO()
|
|
|
|
var month = "202304"
|
|
for i := 0; i < 3; i++ {
|
|
t.Log(dao.HasFullData(tx, 1, month))
|
|
}
|
|
}
|