mirror of
https://github.com/akvorado/akvorado.git
synced 2025-12-11 22:14:02 +01:00
cmd: add even more tests against configuration
This commit is contained in:
@@ -5,9 +5,17 @@ package cmd
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"io/ioutil"
|
||||
"path/filepath"
|
||||
"reflect"
|
||||
"strconv"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"akvorado/common/helpers"
|
||||
"akvorado/common/reporter"
|
||||
|
||||
"gopkg.in/yaml.v2"
|
||||
)
|
||||
|
||||
func TestOrchestratorStart(t *testing.T) {
|
||||
@@ -19,13 +27,72 @@ func TestOrchestratorStart(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestOrchestratorFullConfig(t *testing.T) {
|
||||
root := RootCmd
|
||||
buf := new(bytes.Buffer)
|
||||
root.SetOut(buf)
|
||||
root.SetArgs([]string{"orchestrator", "--dump", "--check", "../akvorado.yaml"})
|
||||
err := root.Execute()
|
||||
func TestOrchestratorConfig(t *testing.T) {
|
||||
tests, err := ioutil.ReadDir("testdata/configurations")
|
||||
if err != nil {
|
||||
t.Errorf("`orchestrator` command error:\n%+v", err)
|
||||
t.Fatalf("ReadDir(%q) error:\n%+v", "testdata/configurations", err)
|
||||
}
|
||||
for _, test := range tests {
|
||||
if !test.IsDir() {
|
||||
continue
|
||||
}
|
||||
t.Run(test.Name(), func(t *testing.T) {
|
||||
expected, err := ioutil.ReadFile(
|
||||
filepath.Join("testdata/configurations", test.Name(), "expected.yaml"))
|
||||
if err != nil {
|
||||
t.Fatalf("ReadFile() error:\n%+v", err)
|
||||
}
|
||||
var expectedYAML struct {
|
||||
Paths map[string]interface{} `yaml:"paths"`
|
||||
}
|
||||
if err := yaml.Unmarshal(expected, &expectedYAML); err != nil {
|
||||
t.Fatalf("yaml.Unmarshal(expected) error:\n%+v", err)
|
||||
}
|
||||
root := RootCmd
|
||||
buf := new(bytes.Buffer)
|
||||
root.SetOut(buf)
|
||||
root.SetArgs([]string{"orchestrator", "--dump", "--check",
|
||||
filepath.Join("testdata/configurations", test.Name(), "in.yaml")})
|
||||
if err := root.Execute(); err != nil {
|
||||
t.Fatalf("`orchestrator` command error:\n%+v", err)
|
||||
}
|
||||
var gotYAML map[string]interface{}
|
||||
if err := yaml.Unmarshal(buf.Bytes(), &gotYAML); err != nil {
|
||||
t.Fatalf("yaml.Unmarshal(output) error:\n%+v", err)
|
||||
}
|
||||
for path, expected := range expectedYAML.Paths {
|
||||
var got interface{}
|
||||
got = gotYAML
|
||||
i := 0
|
||||
for _, component := range strings.Split(path, ".") {
|
||||
var ok bool
|
||||
i++
|
||||
switch gotConcrete := got.(type) {
|
||||
case []interface{}:
|
||||
index, err := strconv.Atoi(component)
|
||||
if err != nil {
|
||||
t.Fatalf("key %q at level %d should be an int", path, i)
|
||||
}
|
||||
got = gotConcrete[index]
|
||||
case map[interface{}]interface{}:
|
||||
got, ok = gotConcrete[component]
|
||||
if !ok {
|
||||
t.Fatalf("key %q does not exist in result", path)
|
||||
}
|
||||
case map[string]interface{}:
|
||||
got, ok = gotConcrete[component]
|
||||
if !ok {
|
||||
t.Fatalf("key %q does not exist in result", path)
|
||||
}
|
||||
default:
|
||||
t.Fatalf("key %q lead to unexpected type %v at level %d",
|
||||
path, reflect.TypeOf(got), i)
|
||||
}
|
||||
}
|
||||
if diff := helpers.Diff(got, expected); diff != "" {
|
||||
t.Fatalf("`orchestrator` --dump, key %q (-got, +want):\n%s", path, diff)
|
||||
}
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user