130 lines
2.5 KiB
Go
130 lines
2.5 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/micro/go-micro/v2/config/source/env"
|
|
"github.com/micro/go-micro/v2/config/source/file"
|
|
)
|
|
|
|
func createFileForIssue18(t *testing.T, content string) *os.File {
|
|
data := []byte(content)
|
|
path := filepath.Join(os.TempDir(), fmt.Sprintf("file.%d", time.Now().UnixNano()))
|
|
fh, err := os.Create(path)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
_, err = fh.Write(data)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
return fh
|
|
}
|
|
|
|
func createFileForTest(t *testing.T) *os.File {
|
|
data := []byte(`{"foo": "bar"}`)
|
|
path := filepath.Join(os.TempDir(), fmt.Sprintf("file.%d", time.Now().UnixNano()))
|
|
fh, err := os.Create(path)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
_, err = fh.Write(data)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
return fh
|
|
}
|
|
|
|
func TestConfigLoadWithGoodFile(t *testing.T) {
|
|
fh := createFileForTest(t)
|
|
path := fh.Name()
|
|
defer func() {
|
|
fh.Close()
|
|
os.Remove(path)
|
|
}()
|
|
|
|
// Create new config
|
|
conf, err := NewConfig()
|
|
if err != nil {
|
|
t.Fatalf("Expected no error but got %v", err)
|
|
}
|
|
// Load file source
|
|
if err := conf.Load(file.NewSource(
|
|
file.WithPath(path),
|
|
)); err != nil {
|
|
t.Fatalf("Expected no error but got %v", err)
|
|
}
|
|
}
|
|
|
|
func TestConfigLoadWithInvalidFile(t *testing.T) {
|
|
fh := createFileForTest(t)
|
|
path := fh.Name()
|
|
defer func() {
|
|
fh.Close()
|
|
os.Remove(path)
|
|
}()
|
|
|
|
// Create new config
|
|
conf, err := NewConfig()
|
|
if err != nil {
|
|
t.Fatalf("Expected no error but got %v", err)
|
|
}
|
|
// Load file source
|
|
err = conf.Load(file.NewSource(
|
|
file.WithPath(path),
|
|
file.WithPath("/i/do/not/exists.json"),
|
|
))
|
|
|
|
if err == nil {
|
|
t.Fatal("Expected error but none !")
|
|
}
|
|
if !strings.Contains(fmt.Sprintf("%v", err), "/i/do/not/exists.json") {
|
|
t.Fatalf("Expected error to contain the unexisting file but got %v", err)
|
|
}
|
|
}
|
|
|
|
func TestConfigMerge(t *testing.T) {
|
|
fh := createFileForIssue18(t, `{
|
|
"amqp": {
|
|
"host": "rabbit.platform",
|
|
"port": 80
|
|
},
|
|
"handler": {
|
|
"exchange": "springCloudBus"
|
|
}
|
|
}`)
|
|
path := fh.Name()
|
|
defer func() {
|
|
fh.Close()
|
|
os.Remove(path)
|
|
}()
|
|
os.Setenv("AMQP_HOST", "rabbit.testing.com")
|
|
|
|
conf, err := NewConfig()
|
|
if err != nil {
|
|
t.Fatalf("Expected no error but got %v", err)
|
|
}
|
|
if err := conf.Load(
|
|
file.NewSource(
|
|
file.WithPath(path),
|
|
),
|
|
env.NewSource(),
|
|
); err != nil {
|
|
t.Fatalf("Expected no error but got %v", err)
|
|
}
|
|
|
|
actualHost := conf.Get("amqp", "host").String("backup")
|
|
if actualHost != "rabbit.testing.com" {
|
|
t.Fatalf("Expected %v but got %v",
|
|
"rabbit.testing.com",
|
|
actualHost)
|
|
}
|
|
}
|