2022-05-14 09:38:44 +00:00
|
|
|
package templates
|
|
|
|
|
|
|
|
import (
|
|
|
|
"strings"
|
|
|
|
"testing"
|
|
|
|
textTpl "text/template"
|
|
|
|
)
|
|
|
|
|
app/vmalert/templates: properly escape all the special chars in `quotesEscape` function
Previously the `quotesEscape` function was escaping only double quotes.
This wasn't enough, since the input string could contain other special chars,
which must be escaped when put inside JSON string. For example, carriage return and line feed chars (\n\r),
backslash char, etc. This led to the following issues, which were improperly fixed:
- https://github.com/VictoriaMetrics/VictoriaMetrics/issues/890 - this issue
was "fixed" by introducing the `crlfEscape` function, which led to unnecessary
complications in user templates, while not fixing various corner cases
such as backslash chars in the input string.
See https://github.com/VictoriaMetrics/VictoriaMetrics/commit/1de15ad490dbde84ad2a657f3b65a6311991f372
- https://github.com/VictoriaMetrics/VictoriaMetrics/issues/3139 - this issue
was "fixed" by urlencoding the whole string passed to -external.alert.source
command-line flag. This led to invalid urls, which couldn't be parsed by Grafana.
See https://github.com/VictoriaMetrics/VictoriaMetrics/commit/00c838353d1246495fd7c7546f3d71095e855eab
and https://github.com/VictoriaMetrics/VictoriaMetrics/commit/4bd024459931a0671dee4abae4bc3556795ee398
This commit properly encodes the input string passed to `quotesEscape`, so it can be safely embedded inside JSON strings.
This commit deprecates crlfEscape template function and adds the following new template functions:
- strvalue and stripDomain - these functions are supported by Prometheus, so they were added
for compatibility purposes.
- jsonEscape and htmlEscape for converting the input string to valid quoted JSON string
and for html-escaping the input string, so it could be safely embedded as a plaintext
into html.
This commit also documents all supported template functions at https://docs.victoriametrics.com/vmalert.html#template-functions
The deprecated crlfEscape function isn't documented on purpose, since its usefulness is negative in general case.
2022-10-27 20:38:19 +00:00
|
|
|
func TestTemplateFuncs(t *testing.T) {
|
|
|
|
funcs := templateFuncs()
|
|
|
|
f := func(funcName, s, resultExpected string) {
|
|
|
|
t.Helper()
|
|
|
|
v := funcs[funcName]
|
|
|
|
fLocal := v.(func(s string) string)
|
|
|
|
result := fLocal(s)
|
|
|
|
if result != resultExpected {
|
|
|
|
t.Fatalf("unexpected result for %s(%q); got\n%s\nwant\n%s", funcName, s, result, resultExpected)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
f("title", "foo bar", "Foo Bar")
|
|
|
|
f("toUpper", "foo", "FOO")
|
|
|
|
f("toLower", "FOO", "foo")
|
|
|
|
f("pathEscape", "foo/bar\n+baz", "foo%2Fbar%0A+baz")
|
|
|
|
f("queryEscape", "foo+bar\n+baz", "foo%2Bbar%0A%2Bbaz")
|
|
|
|
f("jsonEscape", `foo{bar="baz"}`+"\n + 1", `"foo{bar=\"baz\"}\n + 1"`)
|
|
|
|
f("quotesEscape", `foo{bar="baz"}`+"\n + 1", `foo{bar=\"baz\"}\n + 1`)
|
|
|
|
f("htmlEscape", "foo < 10\nabc", "foo < 10\nabc")
|
|
|
|
f("crlfEscape", "foo\nbar\rx", `foo\nbar\rx`)
|
|
|
|
f("stripPort", "foo", "foo")
|
|
|
|
f("stripPort", "foo:1234", "foo")
|
|
|
|
f("stripDomain", "foo.bar.baz", "foo")
|
|
|
|
f("stripDomain", "foo.bar:123", "foo:123")
|
|
|
|
|
|
|
|
// check "match" func
|
|
|
|
matchFunc := funcs["match"].(func(pattern, s string) (bool, error))
|
|
|
|
if _, err := matchFunc("invalid[regexp", "abc"); err == nil {
|
|
|
|
t.Fatalf("expecting non-nil error on invalid regexp")
|
|
|
|
}
|
|
|
|
ok, err := matchFunc("abc", "def")
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unexpected error")
|
|
|
|
}
|
|
|
|
if ok {
|
|
|
|
t.Fatalf("unexpected match")
|
|
|
|
}
|
|
|
|
ok, err = matchFunc("a.+b", "acsdb")
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unexpected error")
|
|
|
|
}
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("unexpected mismatch")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-05-14 09:38:44 +00:00
|
|
|
func mkTemplate(current, replacement interface{}) textTemplate {
|
|
|
|
tmpl := textTemplate{}
|
|
|
|
if current != nil {
|
|
|
|
switch val := current.(type) {
|
|
|
|
case string:
|
|
|
|
tmpl.current = textTpl.Must(newTemplate().Parse(val))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if replacement != nil {
|
|
|
|
switch val := replacement.(type) {
|
|
|
|
case string:
|
|
|
|
tmpl.replacement = textTpl.Must(newTemplate().Parse(val))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return tmpl
|
|
|
|
}
|
|
|
|
|
2022-05-17 13:38:54 +00:00
|
|
|
func equalTemplates(tmpls ...*textTpl.Template) bool {
|
2022-05-14 09:38:44 +00:00
|
|
|
var cmp *textTpl.Template
|
|
|
|
for i, tmpl := range tmpls {
|
|
|
|
if i == 0 {
|
|
|
|
cmp = tmpl
|
|
|
|
} else {
|
|
|
|
if cmp == nil || tmpl == nil {
|
|
|
|
if cmp != tmpl {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
if len(tmpl.Templates()) != len(cmp.Templates()) {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
for _, t := range tmpl.Templates() {
|
|
|
|
tp := cmp.Lookup(t.Name())
|
|
|
|
if tp == nil {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
if tp.Root.String() != t.Root.String() {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestTemplates_Load(t *testing.T) {
|
|
|
|
testCases := []struct {
|
|
|
|
name string
|
|
|
|
initialTemplate textTemplate
|
|
|
|
pathPatterns []string
|
|
|
|
overwrite bool
|
|
|
|
expectedTemplate textTemplate
|
|
|
|
expErr string
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
"non existing path undefined template override",
|
|
|
|
mkTemplate(nil, nil),
|
|
|
|
[]string{
|
|
|
|
"templates/non-existing/good-*.tpl",
|
|
|
|
"templates/absent/good-*.tpl",
|
|
|
|
},
|
|
|
|
true,
|
|
|
|
mkTemplate(``, nil),
|
|
|
|
"",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"non existing path defined template override",
|
|
|
|
mkTemplate(`
|
|
|
|
{{- define "test.1" -}}
|
|
|
|
{{- printf "value" -}}
|
|
|
|
{{- end -}}
|
|
|
|
`, nil),
|
|
|
|
[]string{
|
|
|
|
"templates/non-existing/good-*.tpl",
|
|
|
|
"templates/absent/good-*.tpl",
|
|
|
|
},
|
|
|
|
true,
|
|
|
|
mkTemplate(``, nil),
|
|
|
|
"",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"existing path undefined template override",
|
|
|
|
mkTemplate(nil, nil),
|
|
|
|
[]string{
|
|
|
|
"templates/other/nested/good0-*.tpl",
|
|
|
|
"templates/test/good0-*.tpl",
|
|
|
|
},
|
|
|
|
false,
|
|
|
|
mkTemplate(`
|
|
|
|
{{- define "good0-test.tpl" -}}{{- end -}}
|
|
|
|
{{- define "test.0" -}}
|
|
|
|
{{ printf "Hello %s!" externalURL }}
|
|
|
|
{{- end -}}
|
|
|
|
{{- define "test.1" -}}
|
|
|
|
{{ printf "Hello %s!" externalURL }}
|
|
|
|
{{- end -}}
|
|
|
|
{{- define "test.2" -}}
|
|
|
|
{{ printf "Hello %s!" externalURL }}
|
|
|
|
{{- end -}}
|
|
|
|
{{- define "test.3" -}}
|
|
|
|
{{ printf "Hello %s!" externalURL }}
|
|
|
|
{{- end -}}
|
|
|
|
`, nil),
|
|
|
|
"",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"existing path defined template override",
|
|
|
|
mkTemplate(`
|
|
|
|
{{- define "test.1" -}}
|
|
|
|
{{ printf "Hello %s!" "world" }}
|
|
|
|
{{- end -}}
|
|
|
|
`, nil),
|
|
|
|
[]string{
|
|
|
|
"templates/other/nested/good0-*.tpl",
|
|
|
|
"templates/test/good0-*.tpl",
|
|
|
|
},
|
|
|
|
false,
|
|
|
|
mkTemplate(`
|
|
|
|
{{- define "good0-test.tpl" -}}{{- end -}}
|
|
|
|
{{- define "test.0" -}}
|
|
|
|
{{ printf "Hello %s!" externalURL }}
|
|
|
|
{{- end -}}
|
|
|
|
{{- define "test.1" -}}
|
|
|
|
{{ printf "Hello %s!" "world" }}
|
|
|
|
{{- end -}}
|
|
|
|
{{- define "test.2" -}}
|
|
|
|
{{ printf "Hello %s!" externalURL }}
|
|
|
|
{{- end -}}
|
|
|
|
{{- define "test.3" -}}
|
|
|
|
{{ printf "Hello %s!" externalURL }}
|
|
|
|
{{- end -}}
|
|
|
|
`, `
|
|
|
|
{{- define "good0-test.tpl" -}}{{- end -}}
|
|
|
|
{{- define "test.0" -}}
|
|
|
|
{{ printf "Hello %s!" externalURL }}
|
|
|
|
{{- end -}}
|
|
|
|
{{- define "test.1" -}}
|
|
|
|
{{ printf "Hello %s!" externalURL }}
|
|
|
|
{{- end -}}
|
|
|
|
{{- define "test.2" -}}
|
|
|
|
{{ printf "Hello %s!" externalURL }}
|
|
|
|
{{- end -}}
|
|
|
|
{{- define "test.3" -}}
|
|
|
|
{{ printf "Hello %s!" externalURL }}
|
|
|
|
{{- end -}}
|
|
|
|
`),
|
|
|
|
"",
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"load template with syntax error",
|
|
|
|
mkTemplate(`
|
|
|
|
{{- define "test.1" -}}
|
|
|
|
{{ printf "Hello %s!" "world" }}
|
|
|
|
{{- end -}}
|
|
|
|
`, nil),
|
|
|
|
[]string{
|
|
|
|
"templates/other/nested/bad0-*.tpl",
|
|
|
|
"templates/test/good0-*.tpl",
|
|
|
|
},
|
|
|
|
false,
|
|
|
|
mkTemplate(`
|
|
|
|
{{- define "test.1" -}}
|
|
|
|
{{ printf "Hello %s!" "world" }}
|
|
|
|
{{- end -}}
|
|
|
|
`, nil),
|
|
|
|
"failed to parse template glob",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, tc := range testCases {
|
|
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
|
|
masterTmpl = tc.initialTemplate
|
|
|
|
err := Load(tc.pathPatterns, tc.overwrite)
|
|
|
|
if tc.expErr == "" && err != nil {
|
|
|
|
t.Error("happened error that wasn't expected: %w", err)
|
|
|
|
}
|
|
|
|
if tc.expErr != "" && err == nil {
|
|
|
|
t.Error("%+w", err)
|
|
|
|
t.Error("expected error that didn't happend")
|
|
|
|
}
|
|
|
|
if err != nil && !strings.Contains(err.Error(), tc.expErr) {
|
|
|
|
t.Error("%+w", err)
|
|
|
|
t.Error("expected string doesn't exist in error message")
|
|
|
|
}
|
2022-05-17 13:38:54 +00:00
|
|
|
if !equalTemplates(masterTmpl.replacement, tc.expectedTemplate.replacement) {
|
2022-05-14 09:38:44 +00:00
|
|
|
t.Fatalf("replacement template is not as expected")
|
|
|
|
}
|
2022-05-17 13:38:54 +00:00
|
|
|
if !equalTemplates(masterTmpl.current, tc.expectedTemplate.current) {
|
2022-05-14 09:38:44 +00:00
|
|
|
t.Fatalf("current template is not as expected")
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestTemplates_Reload(t *testing.T) {
|
|
|
|
testCases := []struct {
|
|
|
|
name string
|
|
|
|
initialTemplate textTemplate
|
|
|
|
expectedTemplate textTemplate
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
"empty current and replacement templates",
|
|
|
|
mkTemplate(nil, nil),
|
|
|
|
mkTemplate(nil, nil),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"empty current template only",
|
|
|
|
mkTemplate(`
|
|
|
|
{{- define "test.1" -}}
|
|
|
|
{{- printf "value" -}}
|
|
|
|
{{- end -}}
|
|
|
|
`, nil),
|
|
|
|
mkTemplate(`
|
|
|
|
{{- define "test.1" -}}
|
|
|
|
{{- printf "value" -}}
|
|
|
|
{{- end -}}
|
|
|
|
`, nil),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"empty replacement template only",
|
|
|
|
mkTemplate(nil, `
|
|
|
|
{{- define "test.1" -}}
|
|
|
|
{{- printf "value" -}}
|
|
|
|
{{- end -}}
|
|
|
|
`),
|
|
|
|
mkTemplate(`
|
|
|
|
{{- define "test.1" -}}
|
|
|
|
{{- printf "value" -}}
|
|
|
|
{{- end -}}
|
|
|
|
`, nil),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"defined both templates",
|
|
|
|
mkTemplate(`
|
|
|
|
{{- define "test.0" -}}
|
|
|
|
{{- printf "value" -}}
|
|
|
|
{{- end -}}
|
|
|
|
{{- define "test.1" -}}
|
|
|
|
{{- printf "before" -}}
|
|
|
|
{{- end -}}
|
|
|
|
`, `
|
|
|
|
{{- define "test.1" -}}
|
|
|
|
{{- printf "after" -}}
|
|
|
|
{{- end -}}
|
|
|
|
`),
|
|
|
|
mkTemplate(`
|
|
|
|
{{- define "test.1" -}}
|
|
|
|
{{- printf "after" -}}
|
|
|
|
{{- end -}}
|
|
|
|
`, nil),
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, tc := range testCases {
|
|
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
|
|
masterTmpl = tc.initialTemplate
|
|
|
|
Reload()
|
2022-05-17 13:38:54 +00:00
|
|
|
if !equalTemplates(masterTmpl.replacement, tc.expectedTemplate.replacement) {
|
2022-05-14 09:38:44 +00:00
|
|
|
t.Fatalf("replacement template is not as expected")
|
|
|
|
}
|
2022-05-17 13:38:54 +00:00
|
|
|
if !equalTemplates(masterTmpl.current, tc.expectedTemplate.current) {
|
2022-05-14 09:38:44 +00:00
|
|
|
t.Fatalf("current template is not as expected")
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|