-
Notifications
You must be signed in to change notification settings - Fork 0
/
tree_test.go
32 lines (28 loc) · 648 Bytes
/
tree_test.go
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
package mktree
import (
"io/ioutil"
"os"
"testing"
)
func TestExecTemplateFile(t *testing.T) {
f, err := ioutil.TempFile("", "mktree.tmpl")
if err != nil {
t.Fatal(err)
}
f.Close()
defer os.RemoveAll(f.Name())
template := "Hello {{ CustomFunction }}"
if err := ioutil.WriteFile(f.Name(), []byte(template), os.FileMode(0777)); err != nil {
t.Fatal(err)
}
funcMap := map[string]interface{}{
"CustomFunction": func() string { return "Tester" },
}
content, err := execTemplateFile(f.Name(), funcMap)
if err != nil {
t.Fatal(err)
}
if content != "Hello Tester" {
t.Fatalf("wanted 'Hello Tester' but got %q", content)
}
}