gocommon/pkg/net/nacos/register_test.go

package nacos

import (
	"github.com/stretchr/testify/assert"
	"git.shuncheng.lu/bigthing/gocommon/pkg/conf"
	"git.shuncheng.lu/bigthing/gocommon/pkg/internal/properties"
	"testing"
)

func mockRegisterConfig(config string, t *testing.T) {
	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 TestInitNacosRegisterClient(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

application.port=8080
application.project_name=go-template
#nacos-server.server_port=8080
#nacos-server.server_name=go-template
#nacos-server.server_host=127.0.0.1	 
nacos-server.group_name=DEFAULT_GROUP
nacos-server.cluster_name=DEFAULT
`
	mockRegisterConfig(config, t)
	err := InitNacosRegisterClient()
	if err != nil {
		t.Fatal(err)
	}
	t.Logf("InitNacosRegisterClient success !")
}

func TestInitNacosRegisterClientV2(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

application.port=8080
application.project_name=go-template
nacos-server.server_port=8081
nacos-server.server_name=go-template-1
nacos-server.server_host=127.0.0.1	 
nacos-server.group_name=DEFAULT_GROUP
nacos-server.cluster_name=DEFAULT
`
	mockRegisterConfig(config, t)
	err := InitNacosRegisterClient()
	if err != nil {
		t.Fatal(err)
	}
	t.Logf("InitNacosRegisterClient success !")
	assert.EqualValues(t, registerClientInfo.ServerPort, uint64(8081), "优先级不正确")
	assert.EqualValues(t, registerClientInfo.ProjectName, "go-template-1", "优先级不正确")
}