64 lines
1.2 KiB
Go
64 lines
1.2 KiB
Go
// 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)
|
|
}
|
|
}
|