// 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)
	}
}