package dns_test

import (
	"testing"
	"time"

	"github.com/TeaOSLab/EdgeAPI/internal/db/models/dns"
	_ "github.com/go-sql-driver/mysql"
	_ "github.com/iwind/TeaGo/bootstrap"
	"github.com/iwind/TeaGo/dbs"
)

func TestDNSTaskDAO_CreateDNSTask(t *testing.T) {
	dbs.NotifyReady()
	err := dns.SharedDNSTaskDAO.CreateDNSTask(nil, 1, 2, 3, 0, "cdn", "taskType")
	if err != nil {
		t.Fatal(err)
	}
	t.Log("ok")
}

func TestDNSTaskDAO_UpdateClusterDNSTasksDone(t *testing.T) {
	var dao = dns.NewDNSTaskDAO()
	var tx *dbs.Tx
	err := dao.UpdateClusterDNSTasksDone(tx, 46, time.Now().UnixNano())
	if err != nil {
		t.Fatal(err)
	}
}