mirror of
https://github.com/genuinetools/reg.git
synced 2024-09-12 06:11:03 -04:00
88 lines
2 KiB
Go
88 lines
2 KiB
Go
package templates
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
// GitHub #32120
|
|
func TestParseJSONFunctions(t *testing.T) {
|
|
tm, err := Parse(`{{json .Ports}}`)
|
|
assert.NoError(t, err)
|
|
|
|
var b bytes.Buffer
|
|
assert.NoError(t, tm.Execute(&b, map[string]string{"Ports": "0.0.0.0:2->8/udp"}))
|
|
want := "\"0.0.0.0:2->8/udp\""
|
|
assert.Equal(t, want, b.String())
|
|
}
|
|
|
|
func TestParseStringFunctions(t *testing.T) {
|
|
tm, err := Parse(`{{join (split . ":") "/"}}`)
|
|
assert.NoError(t, err)
|
|
|
|
var b bytes.Buffer
|
|
assert.NoError(t, tm.Execute(&b, "text:with:colon"))
|
|
want := "text/with/colon"
|
|
assert.Equal(t, want, b.String())
|
|
}
|
|
|
|
func TestNewParse(t *testing.T) {
|
|
tm, err := NewParse("foo", "this is a {{ . }}")
|
|
assert.NoError(t, err)
|
|
|
|
var b bytes.Buffer
|
|
assert.NoError(t, tm.Execute(&b, "string"))
|
|
want := "this is a string"
|
|
assert.Equal(t, want, b.String())
|
|
}
|
|
|
|
func TestParseTruncateFunction(t *testing.T) {
|
|
source := "tupx5xzf6hvsrhnruz5cr8gwp"
|
|
|
|
testCases := []struct {
|
|
template string
|
|
expected string
|
|
}{
|
|
{
|
|
template: `{{truncate . 5}}`,
|
|
expected: "tupx5",
|
|
},
|
|
{
|
|
template: `{{truncate . 25}}`,
|
|
expected: "tupx5xzf6hvsrhnruz5cr8gwp",
|
|
},
|
|
{
|
|
template: `{{truncate . 30}}`,
|
|
expected: "tupx5xzf6hvsrhnruz5cr8gwp",
|
|
},
|
|
{
|
|
template: `{{pad . 3 3}}`,
|
|
expected: " tupx5xzf6hvsrhnruz5cr8gwp ",
|
|
},
|
|
}
|
|
|
|
for _, testCase := range testCases {
|
|
tm, err := Parse(testCase.template)
|
|
assert.NoError(t, err)
|
|
|
|
t.Run("Non Empty Source Test with template: "+testCase.template, func(t *testing.T) {
|
|
var b bytes.Buffer
|
|
assert.NoError(t, tm.Execute(&b, source))
|
|
assert.Equal(t, testCase.expected, b.String())
|
|
})
|
|
|
|
t.Run("Empty Source Test with template: "+testCase.template, func(t *testing.T) {
|
|
var c bytes.Buffer
|
|
assert.NoError(t, tm.Execute(&c, ""))
|
|
assert.Equal(t, "", c.String())
|
|
})
|
|
|
|
t.Run("Nil Source Test with template: "+testCase.template, func(t *testing.T) {
|
|
var c bytes.Buffer
|
|
assert.Error(t, tm.Execute(&c, nil))
|
|
assert.Equal(t, "", c.String())
|
|
})
|
|
}
|
|
}
|