mirror of https://github.com/fatedier/frp
Use text/template instead of html/template for config pre-processing (#4656)
parent
450b8393bc
commit
b8d3ace113
|
@ -18,10 +18,10 @@ import (
|
|||
"bytes"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"html/template"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"text/template"
|
||||
|
||||
toml "github.com/pelletier/go-toml/v2"
|
||||
"github.com/samber/lo"
|
||||
|
|
|
@ -112,6 +112,29 @@ func TestLoadServerConfigStrictMode(t *testing.T) {
|
|||
}
|
||||
}
|
||||
|
||||
func TestRenderWithTemplate(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
content string
|
||||
want string
|
||||
}{
|
||||
{"toml", tomlServerContent, tomlServerContent},
|
||||
{"yaml", yamlServerContent, yamlServerContent},
|
||||
{"json", jsonServerContent, jsonServerContent},
|
||||
{"template numeric", `key = {{ 123 }}`, "key = 123"},
|
||||
{"template string", `key = {{ "xyz" }}`, "key = xyz"},
|
||||
{"template quote", `key = {{ printf "%q" "with space" }}`, `key = "with space"`},
|
||||
}
|
||||
for _, test := range tests {
|
||||
t.Run(test.name, func(t *testing.T) {
|
||||
require := require.New(t)
|
||||
got, err := RenderWithTemplate([]byte(test.content), nil)
|
||||
require.NoError(err)
|
||||
require.EqualValues(test.want, string(got))
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestCustomStructStrictMode(t *testing.T) {
|
||||
require := require.New(t)
|
||||
|
||||
|
|
Loading…
Reference in New Issue