1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
//go:build !windows
// +build !windows
package integration
import (
"testing"
"gotest.tools/v3/assert"
"gotest.tools/v3/fs"
)
func setupPluginsTest(t *testing.T) *fs.Dir {
tmpDir := fs.NewDir(t, "gomplate-inttests",
fs.WithFile("foo.sh", "#!/bin/sh\n\necho $*\n", fs.WithMode(0o755)),
fs.WithFile("foo.ps1", "echo $args\r\n", fs.WithMode(0o755)),
fs.WithFile("bar.sh", "#!/bin/sh\n\neval \"echo $*\"\n", fs.WithMode(0o755)),
fs.WithFile("fail.sh", "#!/bin/sh\n\n>&2 echo $1\nexit $2\n", fs.WithMode(0o755)),
fs.WithFile("fail.ps1", `param (
[Parameter(Position=0)]
[string]$msg,
[Parameter(Position=1)]
[int]$code
)
write-error $msg
exit $code
`, fs.WithMode(0o755)),
fs.WithFile("sleep.sh", "#!/bin/sh\n\nexec sleep $1\n", fs.WithMode(0o755)),
fs.WithFile("replace.sh", `#!/bin/sh
if [ "$#" -eq 2 ]; then
exec tr $1 $2
elif [ "$#" -eq 3 ]; then
printf "=%s" $3 | tr $1 $2
fi
`, fs.WithMode(0o755)),
)
t.Cleanup(tmpDir.Remove)
return tmpDir
}
func TestPlugins(t *testing.T) {
tmpDir := setupPluginsTest(t)
o, e, err := cmd(t, "--plugin", "hi="+tmpDir.Join("foo.sh"),
"-i", `{{ hi "hello world" }}`).run()
assertSuccess(t, o, e, err, "hello world\n")
o, e, err = cmd(t, "--plugin", "echo="+tmpDir.Join("bar.sh"),
"-i", `{{ echo "$HELLO" }}`).
withEnv("HELLO", "hello world").run()
assertSuccess(t, o, e, err, "hello world\n")
}
func TestPlugins_Errors(t *testing.T) {
tmpDir := setupPluginsTest(t)
_, _, err := cmd(t, "--plugin", "f=false",
"-i", `{{ f }}`).run()
assert.ErrorContains(t, err, "exit status 1")
_, _, err = cmd(t, "--plugin", "f="+tmpDir.Join("fail.sh"),
"-i", `{{ f "all is lost" 5 }}`).run()
assert.ErrorContains(t, err, "all is lost")
assert.ErrorContains(t, err, "error calling f: exit status 5")
}
func TestPlugins_Timeout(t *testing.T) {
if testing.Short() {
t.Skip()
}
tmpDir := setupPluginsTest(t)
t.Run("default timeout", func(t *testing.T) {
_, _, err := cmd(t, "--plugin", "sleep="+tmpDir.Join("sleep.sh"),
"-i", `{{ sleep 10 }}`).run()
assert.ErrorContains(t, err, "plugin timed out")
})
t.Run("envvar timeout", func(t *testing.T) {
_, _, err := cmd(t, "--plugin", "sleep="+tmpDir.Join("sleep.sh"),
"-i", `{{ sleep 2 }}`).
withEnv("GOMPLATE_PLUGIN_TIMEOUT", "500ms").run()
assert.ErrorContains(t, err, "plugin timed out")
})
}
func TestPlugins_PipeMode(t *testing.T) {
tmpDir := setupPluginsTest(t)
writeConfig(t, tmpDir, `in: '{{ "hi there" | replace "h" "H" }}'
plugins:
replace:
cmd: `+tmpDir.Join("replace.sh")+`
pipe: true
`)
o, e, err := cmd(t).withDir(tmpDir.Path()).run()
assert.NilError(t, err)
assert.Equal(t, "", e)
assert.Equal(t, "Hi tHere", o)
writeConfig(t, tmpDir, `in: '{{ "hi there" | replace "e" "Z" }}'
plugins:
replace:
cmd: `+tmpDir.Join("replace.sh")+`
`)
o, e, err = cmd(t).withDir(tmpDir.Path()).run()
assert.NilError(t, err)
assert.Equal(t, "", e)
assert.Equal(t, "=hi=thZrZ", o)
}
func TestPlugins_Args(t *testing.T) {
tmpDir := setupPluginsTest(t)
writeConfig(t, tmpDir, `in: '{{ echo "world" }}'
plugins:
echo:
cmd: echo
args: [ oh, hello ]
`)
o, e, err := cmd(t).withDir(tmpDir.Path()).run()
assertSuccess(t, o, e, err, "oh hello world\n")
}
|
