repository_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 storage

import (
	"strings"
	"testing"
	"time"

	"github.com/apolloconfig/agollo/v4/agcache/memory"
	"github.com/apolloconfig/agollo/v4/env/config"
	jsonFile "github.com/apolloconfig/agollo/v4/env/file/json"
	"github.com/apolloconfig/agollo/v4/extension"
	"github.com/apolloconfig/agollo/v4/utils"

	_ "github.com/apolloconfig/agollo/v4/agcache/memory"
	"github.com/apolloconfig/agollo/v4/env"
	_ "github.com/apolloconfig/agollo/v4/env/file/json"
	. "github.com/tevid/gohamcrest"

	_ "github.com/apolloconfig/agollo/v4/utils/parse/normal"
	_ "github.com/apolloconfig/agollo/v4/utils/parse/properties"
)

// init param
func init() {
	extension.SetCacheFactory(&memory.DefaultCacheFactory{})
	extension.SetFileHandler(&jsonFile.FileHandler{})
}

func creatTestApolloConfig(configurations map[string]interface{}, namespace string) *Cache {
	c := CreateNamespaceConfig(namespace)
	appConfig := env.InitFileConfig()
	apolloConfig := &config.ApolloConfig{}
	apolloConfig.NamespaceName = namespace
	apolloConfig.AppID = "test"
	apolloConfig.Cluster = "dev"
	apolloConfig.Configurations = configurations
	c.UpdateApolloConfig(apolloConfig, func() config.AppConfig {
		return *appConfig
	})
	return c
}

func TestUpdateApolloConfigNull(t *testing.T) {
	time.Sleep(1 * time.Second)
	c := CreateNamespaceConfig(defaultNamespace)
	appConfig := env.InitFileConfig()

	configurations := make(map[string]interface{})
	configurations["string"] = "string"
	configurations["int"] = 1
	configurations["float"] = 1.9
	configurations["bool"] = true
	configurations["slice"] = []int{1, 2}

	apolloConfig := &config.ApolloConfig{}
	apolloConfig.NamespaceName = defaultNamespace
	apolloConfig.AppID = "test"
	apolloConfig.Cluster = "dev"
	apolloConfig.Configurations = configurations
	c.UpdateApolloConfig(apolloConfig, func() config.AppConfig {
		return *appConfig
	})

	currentConnApolloConfig := appConfig.GetCurrentApolloConfig().Get()
	config := currentConnApolloConfig[defaultNamespace]

	Assert(t, config, NotNilVal())
	Assert(t, defaultNamespace, Equal(config.NamespaceName))
	Assert(t, apolloConfig.AppID, Equal(config.AppID))
	Assert(t, apolloConfig.Cluster, Equal(config.Cluster))
	Assert(t, "", Equal(config.ReleaseKey))
	Assert(t, len(apolloConfig.Configurations), Equal(5))
}

func TestGetDefaultNamespace(t *testing.T) {
	namespace := GetDefaultNamespace()
	Assert(t, namespace, Equal(defaultNamespace))
}

func TestGetConfig(t *testing.T) {
	configurations := make(map[string]interface{})
	configurations["string"] = "string2"
	configurations["int"] = 2
	configurations["string_int"] = "2"
	configurations["float"] = 1.9
	configurations["string_float"] = "1.9"
	configurations["bool"] = false
	configurations["string_bool"] = "false"
	configurations["sliceString"] = []string{"1", "2", "3"}
	configurations["sliceStringWithSeparator"] = "1,2,3"
	configurations["sliceInt"] = []int{1, 2, 3}
	configurations["sliceIntWithSeparator"] = "1,2,3"
	configurations["sliceInter"] = []interface{}{1, "2", 3}
	c := creatTestApolloConfig(configurations, "test")
	config := c.GetConfig("test")
	Assert(t, config, NotNilVal())

	// string
	s := config.GetStringValue("string", "s")
	Assert(t, s, Equal(configurations["string"]))

	s = config.GetStringValue("s", "s")
	Assert(t, s, Equal("s"))

	// int
	i := config.GetIntValue("int", 3)
	Assert(t, i, Equal(2))
	i = config.GetIntValue("string_int", 3)
	Assert(t, i, Equal(2))
	i = config.GetIntValue("float", 3)
	Assert(t, i, Equal(3))
	i = config.GetIntValue("i", 3)
	Assert(t, i, Equal(3))

	// float
	f := config.GetFloatValue("float", 2)
	Assert(t, f, Equal(1.9))
	f = config.GetFloatValue("string_float", 2)
	Assert(t, f, Equal(1.9))
	f = config.GetFloatValue("int", 2)
	Assert(t, f, Equal(float64(2)))
	f = config.GetFloatValue("f", 2)
	Assert(t, f, Equal(float64(2)))

	// bool
	b := config.GetBoolValue("bool", true)
	Assert(t, b, Equal(false))
	b = config.GetBoolValue("string_bool", true)
	Assert(t, b, Equal(false))
	b = config.GetBoolValue("int", true)
	Assert(t, b, Equal(true))

	b = config.GetBoolValue("b", false)
	Assert(t, b, Equal(false))

	slice := config.GetStringSliceValue("sliceString", ",", []string{})
	Assert(t, slice, Equal([]string{"1", "2", "3"}))
	slice = config.GetStringSliceValue("sliceStringWithSeparator", ",", []string{})
	Assert(t, slice, Equal([]string{"1", "2", "3"}))

	sliceInt := config.GetIntSliceValue("sliceInt", ",", []int{})
	Assert(t, sliceInt, Equal([]int{1, 2, 3}))

	sliceInt = config.GetIntSliceValue("sliceIntWithSeparator", ",", []int{})
	Assert(t, sliceInt, Equal([]int{1, 2, 3}))

	sliceInter := config.GetSliceValue("sliceInter", []interface{}{})
	Assert(t, sliceInter, Equal([]interface{}{1, "2", 3}))

	// content
	content := config.GetContent()
	hasFloat := strings.Contains(content, "float=1")
	Assert(t, hasFloat, Equal(true))

	hasInt := strings.Contains(content, "int=2")
	Assert(t, hasInt, Equal(true))

	hasString := strings.Contains(content, "string=string2")
	Assert(t, hasString, Equal(true))

	hasBool := strings.Contains(content, "bool=false")
	Assert(t, hasBool, Equal(true))

	hasSlice := strings.Contains(content, "sliceString=[1 2 3]")
	Assert(t, hasSlice, Equal(true))
	hasSlice = strings.Contains(content, "sliceInt=[1 2 3]")
	Assert(t, hasSlice, Equal(true))
}

func createChangeEvent() *ChangeEvent {
	addConfig := createAddConfigChange("new")
	deleteConfig := createDeletedConfigChange("old")
	modifyConfig := createModifyConfigChange("old", "new")
	changes := make(map[string]*ConfigChange)
	changes["add"] = addConfig
	changes["adx"] = addConfig
	changes["delete"] = deleteConfig
	changes["modify"] = modifyConfig
	cEvent := &ChangeEvent{
		Changes: changes,
	}
	cEvent.Namespace = "a"
	return cEvent
}

func TestRegDispatchInRepository(t *testing.T) {
	dispatch := UseEventDispatch()
	err := dispatch.RegisterListener(nil, "ad.*")
	Assert(t, err, NotNilVal())
	l := &CustomListener{
		Keys: make(map[string]interface{}, 0),
	}
	err = dispatch.RegisterListener(l, "ad.*")
	Assert(t, err, NilVal())
	cEvent := createChangeEvent()
	cache := CreateNamespaceConfig("abc")
	cache.AddChangeListener(dispatch)
	cache.pushChangeEvent(cEvent)
	time.Sleep(1 * time.Second)
	Assert(t, l.Len(), Equal(2))
	v, ok := l.Value("add")
	Assert(t, v, Equal("new"))
	Assert(t, ok, Equal(true))
	v, ok = l.Value("adx")
	Assert(t, v, Equal("new"))
	Assert(t, ok, Equal(true))
}

func TestDispatchInRepository(t *testing.T) {
	dispatch := UseEventDispatch()
	l := &CustomListener{
		Keys: make(map[string]interface{}, 0),
	}
	err := dispatch.RegisterListener(l, "add", "delete")
	Assert(t, err, NilVal())
	Assert(t, len(dispatch.listeners), Equal(2))
	cEvent := createChangeEvent()
	cache := CreateNamespaceConfig("abc")
	cache.AddChangeListener(dispatch)
	cache.pushChangeEvent(cEvent)
	time.Sleep(1 * time.Second)
	Assert(t, l.Len(), Equal(2))
	v, ok := l.Value("add")
	Assert(t, v, Equal("new"))
	Assert(t, ok, Equal(true))
	v, ok = l.Value("delete")
	Assert(t, ok, Equal(true))
	Assert(t, v, Equal("old"))
	_, ok = l.Value("modify")
	Assert(t, ok, Equal(false))
}

func TestGetValueImmediately(t *testing.T) {
	c := initConfig("namespace", extension.GetCacheFactory())

	res := c.GetValueImmediately("namespace")
	Assert(t, res, Equal(utils.Empty))

	c.isInit.Store(true)
	res = c.GetValueImmediately("namespace")
	Assert(t, res, Equal(utils.Empty))

	res = c.GetValueImmediately("namespace1")
	Assert(t, res, Equal(utils.Empty))

	c.cache.Set("namespace", 1, 3)
	res = c.GetValueImmediately("namespace")
	Assert(t, res, Equal(utils.Empty))

	c.cache.Set("namespace", "config", 3)
	res = c.GetValueImmediately("namespace")
	Assert(t, res, Equal("config"))
}

func TestGetConfigImmediately(t *testing.T) {
	configurations := make(map[string]interface{})
	configurations["string"] = "string2"
	configurations["int"] = 2
	configurations["string_int"] = "2"
	configurations["float"] = 1.9
	configurations["string_float"] = "1.9"
	configurations["bool"] = false
	configurations["string_bool"] = "false"
	configurations["sliceString"] = []string{"1", "2", "3"}
	configurations["sliceInt"] = []int{1, 2, 3}
	configurations["sliceInter"] = []interface{}{1, "2", 3}
	c := creatTestApolloConfig(configurations, "test")
	config := c.GetConfig("test")
	Assert(t, config, NotNilVal())

	// string
	s := config.GetStringValueImmediately("string", "s")
	Assert(t, s, Equal(configurations["string"]))

	s = config.GetStringValueImmediately("s", "s")
	Assert(t, s, Equal("s"))

	// int
	i := config.GetIntValueImmediately("int", 3)
	Assert(t, i, Equal(2))
	i = config.GetIntValueImmediately("string_int", 3)
	Assert(t, i, Equal(2))
	i = config.GetIntValueImmediately("float", 3)
	Assert(t, i, Equal(3))
	i = config.GetIntValueImmediately("i", 3)
	Assert(t, i, Equal(3))

	// float
	f := config.GetFloatValueImmediately("float", 2)
	Assert(t, f, Equal(1.9))
	f = config.GetFloatValueImmediately("string_float", 2)
	Assert(t, f, Equal(1.9))
	f = config.GetFloatValueImmediately("f", 2)
	Assert(t, f, Equal(float64(2)))
	f = config.GetFloatValueImmediately("int", 2)
	Assert(t, f, Equal(float64(2)))

	// bool
	b := config.GetBoolValueImmediately("bool", true)
	Assert(t, b, Equal(false))
	b = config.GetBoolValueImmediately("string_bool", true)
	Assert(t, b, Equal(false))
	b = config.GetBoolValueImmediately("int", false)
	Assert(t, b, Equal(false))
	b = config.GetBoolValueImmediately("b", false)
	Assert(t, b, Equal(false))

	slice := config.GetStringSliceValueImmediately("sliceString", []string{})
	Assert(t, slice, Equal([]string{"1", "2", "3"}))

	sliceInt := config.GetIntSliceValueImmediately("sliceInt", []int{})
	Assert(t, sliceInt, Equal([]int{1, 2, 3}))

	sliceInter := config.GetSliceValueImmediately("sliceInter", []interface{}{})
	Assert(t, sliceInter, Equal([]interface{}{1, "2", 3}))
}