mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-01 14:47:38 +00:00
95c67d8309
* allowed using dashes and dots in environment variables names for templating config files with envtemplate (#3999) Signed-off-by: Alexander Marshalov <_@marshalov.org> * Apply suggestions from code review --------- Signed-off-by: Alexander Marshalov <_@marshalov.org> Co-authored-by: Aliaksandr Valialkin <valyala@victoriametrics.com>
96 lines
2.5 KiB
Go
96 lines
2.5 KiB
Go
package envtemplate
|
|
|
|
import (
|
|
"reflect"
|
|
"sort"
|
|
"testing"
|
|
)
|
|
|
|
func TestExpandTemplates(t *testing.T) {
|
|
f := func(envs, resultExpected []string) {
|
|
t.Helper()
|
|
m := parseEnvVars(envs)
|
|
mExpanded := expandTemplates(m)
|
|
result := make([]string, 0, len(mExpanded))
|
|
for k, v := range mExpanded {
|
|
result = append(result, k+"="+v)
|
|
}
|
|
sort.Strings(result)
|
|
if !reflect.DeepEqual(result, resultExpected) {
|
|
t.Fatalf("unexpected result;\ngot\n%q\nwant\n%q", result, resultExpected)
|
|
}
|
|
}
|
|
f(nil, []string{})
|
|
f([]string{"foo=%{bar}", "bar=x"}, []string{"bar=x", "foo=x"})
|
|
f([]string{"a=x%{b}", "b=y%{c}z%{d}", "c=123", "d=qwe"}, []string{"a=xy123zqwe", "b=y123zqwe", "c=123", "d=qwe"})
|
|
f([]string{"a=x%{b}y", "b=z%{a}q", "c"}, []string{"a=xzxzxzxz%{a}qyqyqyqy", "b=zxzxzxzx%{b}yqyqyqyq", "c="})
|
|
f([]string{"a=%{x.y}", "x.y=test"}, []string{"a=test", "x.y=test"})
|
|
f([]string{"a=%{x y}"}, []string{"a=%{x y}"})
|
|
f([]string{"a=%{123}"}, []string{"a=%{123}"})
|
|
}
|
|
|
|
func TestLookupEnv(t *testing.T) {
|
|
envVars = map[string]string{
|
|
"foo": "bar",
|
|
}
|
|
result, ok := LookupEnv("foo")
|
|
if result != "bar" {
|
|
t.Fatalf("unexpected result; got %q; want %q", result, "bar")
|
|
}
|
|
if !ok {
|
|
t.Fatalf("unexpected ok=false")
|
|
}
|
|
result, ok = LookupEnv("bar")
|
|
if result != "" {
|
|
t.Fatalf("unexpected non-empty result: %q", result)
|
|
}
|
|
if ok {
|
|
t.Fatalf("unexpected ok=true")
|
|
}
|
|
}
|
|
|
|
func TestReplaceSuccess(t *testing.T) {
|
|
envVars = map[string]string{
|
|
"foo": "bar",
|
|
"foo.bar_1": "baz",
|
|
"foo-bar_2": "test",
|
|
}
|
|
f := func(s, resultExpected string) {
|
|
t.Helper()
|
|
result, err := ReplaceBytes([]byte(s))
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %s", err)
|
|
}
|
|
if string(result) != resultExpected {
|
|
t.Fatalf("unexpected result for ReplaceBytes(%q);\ngot\n%q\nwant\n%q", s, result, resultExpected)
|
|
}
|
|
resultS, err := ReplaceString(s)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %s", err)
|
|
}
|
|
if resultS != resultExpected {
|
|
t.Fatalf("unexpected result for ReplaceString(%q);\ngot\n%q\nwant\n%q", s, result, resultExpected)
|
|
}
|
|
}
|
|
f("", "")
|
|
f("foo", "foo")
|
|
f("a %{foo}-x", "a bar-x")
|
|
f("%{foo.bar_1}", "baz")
|
|
f("qq.%{foo-bar_2}.ww", "qq.test.ww")
|
|
}
|
|
|
|
func TestReplaceFailure(t *testing.T) {
|
|
f := func(s string) {
|
|
t.Helper()
|
|
if _, err := ReplaceBytes([]byte(s)); err == nil {
|
|
t.Fatalf("expecting non-nil error for ReplaceBytes(%q)", s)
|
|
}
|
|
if _, err := ReplaceString(s); err == nil {
|
|
t.Fatalf("expecting non-nil error for ReplaceString(%q)", s)
|
|
}
|
|
}
|
|
f("foo %{bar} %{baz}")
|
|
f("%{Foo_Foo_1}")
|
|
f("%{Foo-Bar-2}")
|
|
f("%{Foo.Baz.3}")
|
|
}
|