From 275d7f1622bf1610dfacb98ff42f354fcaad69d2 Mon Sep 17 00:00:00 2001 From: GoEdgeLab Date: Tue, 6 Dec 2022 14:53:38 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E6=B5=8B=E8=AF=95=E7=94=A8?= =?UTF-8?q?=E4=BE=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/db/models/message_task_dao_test.go | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/internal/db/models/message_task_dao_test.go b/internal/db/models/message_task_dao_test.go index 4a394676..dd614842 100644 --- a/internal/db/models/message_task_dao_test.go +++ b/internal/db/models/message_task_dao_test.go @@ -1,6 +1,7 @@ -package models +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" @@ -11,7 +12,7 @@ func TestMessageTaskDAO_FindSendingMessageTasks(t *testing.T) { dbs.NotifyReady() var tx *dbs.Tx - tasks, err := NewMessageTaskDAO().FindSendingMessageTasks(tx, 100) + tasks, err := models.NewMessageTaskDAO().FindSendingMessageTasks(tx, 100) if err != nil { t.Fatal(err) } @@ -20,3 +21,12 @@ func TestMessageTaskDAO_FindSendingMessageTasks(t *testing.T) { 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) + } +}