Added atoi function to template funcs and test

jjneely · Feb 15, 2018 · d84d890 · d84d890
1 parent 9671ca0
commit d84d890
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/resource/template/template_funcs.go b/resource/template/template_funcs.go
@@ -9,6 +9,7 @@ import (
 	"os"
 	"path"
 	"sort"
+	"strconv"
 	"strings"
 	"time"
 
@@ -45,6 +46,7 @@ func newFuncMap() map[string]interface{} {
 	m["mod"] = func(a, b int) int { return a % b }
 	m["mul"] = func(a, b int) int { return a * b }
 	m["seq"] = Seq
+	m["atoi"] = strconv.Atoi
 	return m
 }
 

diff --git a/resource/template/template_test.go b/resource/template/template_test.go
@@ -698,6 +698,25 @@ dest = "./tmp/test.conf"
 [1 2 3]
 `,
 		updateStore: func(tr *TemplateResource) {},
+	}, templateTest{
+		desc: "atoi test",
+		toml: `
+[template]
+src = "test.conf.tmpl"
+dest = "./tmp/test.conf"
+keys = [
+    "/test/count/",
+]
+`,
+		tmpl: `
+{{ seq 1 (atoi (getv "/test/count")) }}
+`,
+		expected: `
+[1 2 3]
+`,
+		updateStore: func(tr *TemplateResource) {
+			tr.store.Set("/test/count", "3")
+		},
 	},
 }