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
|
package gomplate
import (
"bytes"
"context"
"fmt"
"os"
"strings"
"testing"
"text/template"
"time"
"github.com/hairyhenderson/gomplate/v3/internal/config"
"github.com/stretchr/testify/assert"
)
func TestBindPlugins(t *testing.T) {
ctx := context.Background()
fm := template.FuncMap{}
cfg := &config.Config{
Plugins: map[string]config.PluginConfig{},
}
err := bindPlugins(ctx, cfg, fm)
assert.NoError(t, err)
assert.EqualValues(t, template.FuncMap{}, fm)
cfg.Plugins = map[string]config.PluginConfig{"foo": {Cmd: "bar"}}
err = bindPlugins(ctx, cfg, fm)
assert.NoError(t, err)
assert.Contains(t, fm, "foo")
err = bindPlugins(ctx, cfg, fm)
assert.ErrorContains(t, err, "already bound")
}
func TestBuildCommand(t *testing.T) {
ctx := context.Background()
data := []struct {
name, path string
args []string
expected []string
}{
{"foo", "foo", nil, []string{"foo"}},
{"foo", "foo", []string{"bar"}, []string{"foo", "bar"}},
{"foo", "foo.bat", nil, []string{"cmd.exe", "/c", "foo.bat"}},
{"foo", "foo.cmd", []string{"bar"}, []string{"cmd.exe", "/c", "foo.cmd", "bar"}},
{"foo", "foo.ps1", []string{"bar", "baz"}, []string{"pwsh", "-File", "foo.ps1", "bar", "baz"}},
}
for _, d := range data {
p := &plugin{
ctx: ctx,
path: d.path,
}
name, args := p.buildCommand(d.args)
actual := append([]string{name}, args...)
assert.EqualValues(t, d.expected, actual)
}
}
func TestRun(t *testing.T) {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
stderr := &bytes.Buffer{}
p := &plugin{
ctx: ctx,
timeout: 500 * time.Millisecond,
stderr: stderr,
path: "echo",
}
out, err := p.run("foo")
assert.NoError(t, err)
assert.Equal(t, "", stderr.String())
assert.Equal(t, "foo", strings.TrimSpace(out.(string)))
}
func ExamplePluginFunc() {
ctx := context.Background()
// PluginFunc creates a template function that runs an arbitrary command.
f := PluginFunc(ctx, "echo", PluginOpts{})
// The function can be used in a template, but here we'll just run it
// directly. This is equivalent to running 'echo foo bar'
out, err := f("foo", "bar")
if err != nil {
panic(err)
}
fmt.Println(out)
// Output:
// foo bar
}
func ExamplePluginFunc_with_template() {
ctx := context.Background()
f := PluginFunc(ctx, "echo", PluginOpts{})
// PluginFunc is intended for use with gomplate, but can be used in any
// text/template by adding it to the FuncMap.
tmpl := template.New("new").Funcs(template.FuncMap{"echo": f})
tmpl, err := tmpl.Parse(`{{ echo "baz" "qux" }}`)
if err != nil {
panic(err)
}
err = tmpl.Execute(os.Stdout, nil)
if err != nil {
panic(err)
}
// Output:
// baz qux
}
|
