EdgeAPI/internal/db/models/message_dao_test.go

package models

import (
	"testing"
	"time"

	"github.com/TeaOSLab/EdgeCommon/pkg/nodeconfigs"
	_ "github.com/go-sql-driver/mysql"
	"github.com/iwind/TeaGo/dbs"
)

func TestMessageDAO_CreateClusterMessage(t *testing.T) {
	var tx *dbs.Tx

	dao := NewMessageDAO()
	err := dao.CreateClusterMessage(tx, nodeconfigs.NodeRoleNode, 1, "test", "error", "123", "123", "123", []byte("456"))
	if err != nil {
		t.Fatal(err)
	}
	t.Log("ok")
}

func TestMessageDAO_DeleteMessagesBeforeDay(t *testing.T) {
	var tx *dbs.Tx

	dao := NewMessageDAO()
	err := dao.DeleteMessagesBeforeDay(tx, time.Now())
	if err != nil {
		t.Fatal(err)
	}
	t.Log("ok")
}