server/config/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_test

import (
	"io"
	"os"
	"path/filepath"
	"testing"

	"github.com/apache/servicecomb-service-center/pkg/util"
	"github.com/apache/servicecomb-service-center/server/config"
	"github.com/go-chassis/go-archaius"
	"github.com/stretchr/testify/assert"
)

func TestInit(t *testing.T) {
	defer archaius.Clean()

	dir := filepath.Join(util.GetAppRoot(), "conf")
	os.Mkdir(dir, 0750)
	defer os.Remove(dir)

	govFile := filepath.Join(dir, "app.yaml")
	f1, err := os.Create(govFile)
	assert.NoError(t, err)
	defer os.Remove(govFile)
	_, err = io.WriteString(f1, `
gov:
  kieSever1:
    type: kie
    endpoint: 127.0.0.1:30110
`)
	assert.NoError(t, err)

	policyFile := filepath.Join(dir, "grc.json")
	f2, err := os.Create(policyFile)
	assert.NoError(t, err)
	defer os.Remove(policyFile)
	_, err = io.WriteString(f2, `
{
  "gov": {
    "match-group": {
      "validationSpec": {
        "type": "object",
        "properties": {
          "alias": {
            "type": "string",
            "minLength": 1
          }
        }
      }
    },
    "policies": {
      "retry": {
        "validationSpec": {
          "type": "object",
          "properties": {
            "retryOnSame": {
              "type": "integer",
              "minimum": 0
            }
          }
        }
      }
    }
  }
}`)
	assert.NoError(t, err)

	config.Init()
	assert.NoError(t, err)
	assert.Equal(t, "kie", config.GetGov().DistMap["kieSever1"].Type)
	assert.Equal(t, "127.0.0.1:30110", config.GetGov().DistMap["kieSever1"].Endpoint)

	assert.NotNil(t, config.GetGov().Policies["match-group"])
	assert.NotNil(t, config.GetGov().Policies["policies"])
}