Initial commit (code only without large binaries)

EdgeDNS/internal/nodes/manager_domain_test.go

@@ -0,0 +1,63 @@
+// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+
+package nodes
+
+import (
+	"github.com/TeaOSLab/EdgeDNS/internal/dbs"
+	"github.com/iwind/TeaGo/Tea"
+	_ "github.com/iwind/TeaGo/bootstrap"
+	"github.com/iwind/TeaGo/logs"
+	"testing"
+)
+
+func TestDomainManager_Loop(t *testing.T) {
+	var db = dbs.NewDB(Tea.Root + "/data/data.db")
+	err := db.Init()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	var manager = NewDomainManager(db, 1)
+	for i := 0; i < 10; i++ {
+		_, err := manager.Loop()
+		if err != nil {
+			t.Fatal(err)
+		}
+	}
+	logs.PrintAsJSON(manager.domainMap, t)
+}
+
+func TestDomainManager_Load(t *testing.T) {
+	var db = dbs.NewDB(Tea.Root + "/data/data.db")
+	err := db.Init()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	manager := NewDomainManager(db, 2)
+	err = manager.Load()
+	if err != nil {
+		t.Fatal(err)
+	}
+	logs.PrintAsJSON(manager.domainMap, t)
+	t.Log("version:", manager.version)
+}
+
+func TestDomainManager_FindDomain(t *testing.T) {
+	var db = dbs.NewDB(Tea.Root + "/data/data.db")
+	err := db.Init()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	var manager = NewDomainManager(db, 2)
+	err = manager.Load()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	for _, name := range []string{"hello.com", "teaos.cn"} {
+		domain, ok := manager.FindDomain(name)
+		t.Log(name, ok, domain)
+	}
+}