1.4.5.2

EdgeAPI/internal/db/models/api_node_dao_test.go

@@ -0,0 +1,52 @@
+package models
+
+import (
+	"github.com/TeaOSLab/EdgeAPI/internal/utils"
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/iwind/TeaGo/dbs"
+	"runtime"
+	"testing"
+)
+
+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()
+	}
+}