gocommon/pkg/net/nacos/discover_test.go

package nacos

import (
	"git.shuncheng.lu/bigthing/gocommon/pkg/conf"
	"git.shuncheng.lu/bigthing/gocommon/pkg/internal/properties"
	"testing"
)

func mockDiscoverConfig(t *testing.T) {
	config := `
nacos-server.host=10.100.101.20:8848,10.100.103.230:8848,10.100.99.14:8848
nacos-server.log_path=/data/log/go-template
nacos-server.log_level=warn
nacos-server.namespace_id=84663e76-f64a-4331-a3a6-efd26ead7bf1
nacos-server.user_name=cityservice_dev_nacos
nacos-server.password=MrHu0qvmdK
nacos-server.timeout=5000

# ebike-factory-api 通过nacos调用
ebike-factory-api.service_name = ebike-factory-api
ebike-factory-api.group_name = DEFAULT_GROUP
ebike-factory-api.clusters = DEFAULT
ebike-factory-api.domain_addr=http://www.baidu.com
`
	pro, err := properties.ReadFromString(config)
	if err != nil {
		t.Fatal(err)
	}
	conf.MustValue = func(section, key string, defaultVal ...string) string {
		return pro.GetString(section+"."+key, defaultVal...)
	}
}

func TestInitNacosDiscoverClient(t *testing.T) {
	mockDiscoverConfig(t)
	err := InitNacosDiscoverClient()
	if err != nil {
		t.Fatal(err)
	}
	host, err := GetHost("ebike-factory-api")
	if err != nil {
		t.Fatal(err)
	}
	t.Logf("get host: %s", host)
}

func BenchmarkGetHost(b *testing.B) {
	mockDiscoverConfig(nil)
	err := InitNacosDiscoverClient()
	if err != nil {
		b.Fatal(err)
	}
	for i := 0; i < b.N; i++ {
		host, err := GetHost("ebike-factory-api")
		if err != nil {
			b.Fatal(err)
		}
		if host == "" {
			b.Fatal("empty")
		}
	}
}

func TestInitGetDomainHost(t *testing.T) {
	mockDiscoverConfig(t)
	err := InitNacosDiscoverClient()
	if err != nil {
		t.Fatal(err)
	}
	host, err := GetDomainHost("ebike-factory-api")
	if err != nil {
		t.Fatal(err)
	}
	t.Logf("get host: %s", host)
}