mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2025-11-05 01:20:25 +08:00
33 lines
731 B
Go
33 lines
731 B
Go
package models_test
|
|
|
|
import (
|
|
"github.com/TeaOSLab/EdgeAPI/internal/db/models"
|
|
_ "github.com/go-sql-driver/mysql"
|
|
_ "github.com/iwind/TeaGo/bootstrap"
|
|
"github.com/iwind/TeaGo/dbs"
|
|
"testing"
|
|
)
|
|
|
|
func TestMessageTaskDAO_FindSendingMessageTasks(t *testing.T) {
|
|
dbs.NotifyReady()
|
|
|
|
var tx *dbs.Tx
|
|
tasks, err := models.NewMessageTaskDAO().FindSendingMessageTasks(tx, 100)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Log(len(tasks), "tasks")
|
|
for _, task := range tasks {
|
|
t.Log("task:", task.Id, "recipient:", task.RecipientId)
|
|
}
|
|
}
|
|
|
|
func TestMessageTaskDAO_CleanExpiredMessageTasks(t *testing.T) {
|
|
var dao = models.NewMessageTaskDAO()
|
|
var tx *dbs.Tx
|
|
err := dao.CleanExpiredMessageTasks(tx, 30)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|