package models

import (
	"runtime"
	"testing"

	"github.com/TeaOSLab/EdgeAPI/internal/utils"
	_ "github.com/go-sql-driver/mysql"
	"github.com/iwind/TeaGo/dbs"
)

func TestAPINodeDAO_FindEnabledAPINodeIdWithAddr(t *testing.T) {
	dao := NewAPINodeDAO()
	var tx *dbs.Tx
	{
		apiNodeId, err := dao.FindEnabledAPINodeIdWithAddr(tx, "http", "127.0.0.1", 123)
		if err != nil {
			t.Fatal(err)
		}
		t.Log("apiNodeId:", apiNodeId)
	}

	{
		apiNodeId, err := dao.FindEnabledAPINodeIdWithAddr(tx, "http", "127.0.0.1", 8003)
		if err != nil {
			t.Fatal(err)
		}
		t.Log("apiNodeId:", apiNodeId)
	}
}

func TestAPINodeDAO_FindAllEnabledAPIAccessIPs(t *testing.T) {
	var cacheMap = utils.NewCacheMap()
	t.Log(NewAPINodeDAO().FindAllEnabledAPIAccessIPs(nil, cacheMap))
	t.Log(NewAPINodeDAO().FindAllEnabledAPIAccessIPs(nil, cacheMap))
}

func TestAPINodeDAO_CheckAPINodeIsPrimary(t *testing.T) {
	var dao = NewAPINodeDAO()
	t.Log(dao.CheckAPINodeIsPrimary(nil))
}

func TestAPINodeDAO_ResetPrimaryAPINode(t *testing.T) {
	var dao = NewAPINodeDAO()
	t.Log(dao.ResetPrimaryAPINode(nil))
}

func BenchmarkAPINodeDAO_New(b *testing.B) {
	runtime.GOMAXPROCS(1)
	for i := 0; i < b.N; i++ {
		_ = NewAPINodeDAO()
	}
}