config/registry_config_test.go

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package config

import (
	"testing"
)

import (
	"github.com/stretchr/testify/assert"
)

import (
	"dubbo.apache.org/dubbo-go/v3/common"
	"dubbo.apache.org/dubbo-go/v3/common/constant"
)

func TestLoadRegistries(t *testing.T) {
	target := []string{"shanghai1"}
	regs := map[string]*RegistryConfig{

		"shanghai1": {
			Protocol: "mock",
			Timeout:  "2s",
			Group:    "shanghai_idc",
			Address:  "127.0.0.2:2181,128.0.0.1:2181",
			Username: "user1",
			Password: "pwd1",
		},
	}
	urls := loadRegistries(target, regs, common.CONSUMER)
	t.Logf("loadRegistries() = urls:%v", urls)
	assert.Equal(t, "127.0.0.2:2181,128.0.0.1:2181", urls[0].Location)
}

func TestLoadRegistries1(t *testing.T) {
	target := []string{"shanghai1"}
	regs := map[string]*RegistryConfig{

		"shanghai1": {
			Protocol: "mock",
			Timeout:  "2s",
			Group:    "shanghai_idc",
			Address:  "127.0.0.2:2181",
			Username: "user1",
			Password: "pwd1",
		},
	}
	urls := loadRegistries(target, regs, common.CONSUMER)
	t.Logf("loadRegistries() = urls:%v", urls)
	assert.Equal(t, "127.0.0.2:2181", urls[0].Location)
}

func TestTranslateRegistryAddress(t *testing.T) {
	reg := new(RegistryConfig)
	reg.Address = "nacos://127.0.0.1:8848"

	reg.translateRegistryAddress()

	assert.Equal(t, "nacos", reg.Protocol)
	assert.Equal(t, "127.0.0.1:8848", reg.Address)
}

func TestNewRegistryConfigBuilder(t *testing.T) {

	config := NewRegistryConfigBuilder().
		SetProtocol("nacos").
		SetTimeout("10s").
		SetGroup("group").
		SetNamespace("public").
		SetTTL("10s").
		SetAddress("127.0.0.1:8848").
		SetUsername("nacos").
		SetPassword("123456").
		SetSimplified(true).
		SetPreferred(true).
		SetZone("zone").
		SetWeight(100).
		SetParams(map[string]string{"timeout": "3s"}).
		AddParam("timeout", "15s").
		SetRegistryType("local").
		Build()

	config.DynamicUpdateProperties(config)

	assert.Equal(t, config.Prefix(), constant.RegistryConfigPrefix)

	values := config.getUrlMap(common.PROVIDER)
	assert.Equal(t, values.Get("timeout"), "15s")

	url, err := config.toMetadataReportUrl()
	assert.NoError(t, err)
	assert.Equal(t, url.GetParam("timeout", "3s"), "10s")

	url, err = config.toURL(common.PROVIDER)
	assert.NoError(t, err)
	assert.Equal(t, url.GetParam("timeout", "3s"), "15s")

	address := config.translateRegistryAddress()
	assert.Equal(t, address, "127.0.0.1:8848")
}

func TestNewRegistryConfig(t *testing.T) {
	config := NewRegistryConfig(
		WithRegistryProtocol("nacos"),
		WithRegistryAddress("127.0.0.1:8848"),
		WithRegistryTimeOut("10s"),
		WithRegistryGroup("group"),
		WithRegistryTTL("10s"),
		WithRegistryUserName("nacos"),
		WithRegistryPassword("123456"),
		WithRegistrySimplified(true),
		WithRegistryPreferred(true),
		WithRegistryWeight(100),
		WithRegistryParams(map[string]string{"timeout": "3s"}))

	assert.Equal(t, config.Timeout, "10s")
}