Merge pull request #115 from gomes/master

Add common string functions

Author: Jing Dong

services/template/template.go

@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"text/template"
 	"github.com/QubitProducts/bamboo/services/service"
+	"strings"
 )
 
 func hasKey(data map[string]service.Service, appId string) bool {
@@ -20,7 +21,14 @@ func getService(data map[string]service.Service, appId string) service.Service {
 	Returns string content of a rendered template
 */
 func RenderTemplate(templateName string, templateContent string, data interface{}) (string, error) {
-	funcMap := template.FuncMap{ "hasKey": hasKey,  "getService": getService }
+	funcMap := template.FuncMap{ 
+		"hasKey": hasKey,  
+		"getService": getService,
+		"Split": strings.Split,
+		"Join": strings.Join,
+		"Replace": strings.Replace,
+		"ToUpper": strings.ToUpper,
+		"ToLower": strings.ToLower}
 
 	tpl := template.Must(template.New(templateName).Funcs(funcMap).Parse(templateContent))
 

services/template/template_test.go

@@ -17,3 +17,69 @@ func TestTemplateWriter(t *testing.T) {
 		})
 	})
 }
+
+func TestTemplateSplitFunction(t *testing.T) {
+	Convey("#RenderTemplate", t, func() {
+		templateName := "templateName"
+		params := map[string]string{"id": "app", "domain": "example.com"}
+
+		templateContent := "{{.id}}{{range Split .domain \".\"}} {{.}}{{end}}"
+		Convey("should split domain as two different strings", func() {
+			content, _ := RenderTemplate(templateName, templateContent, params)
+			So(content, ShouldEqual, "app example com")
+		})
+	})
+}
+
+func TestTemplateJoinFunction(t *testing.T) {
+	Convey("#RenderTemplate", t, func() {
+		templateName := "templateName"
+		domains := []string{"example.com", "example.net"}
+		params := map[string][]string{"domains": domains}
+
+		templateContent := "{{Join .domains \", \"}}"
+		Convey("should create a list of two domains separated by comma", func() {
+			content, _ := RenderTemplate(templateName, templateContent, params)
+			So(content, ShouldEqual, "example.com, example.net")
+		})
+	})
+}
+
+func TestTemplateReplaceFunction(t *testing.T) {
+	Convey("#RenderTemplate", t, func() {
+		templateName := "templateName"
+		params := map[string]string{"domain": "example.com"}
+
+		templateContent := "{{Replace .domain \"com\" \"net\" 1}}"
+		Convey("should replace example.com into example.net", func() {
+			content, _ := RenderTemplate(templateName, templateContent, params)
+			So(content, ShouldEqual, "example.net")
+		})
+	})
+}
+
+func TestTemplateToUpperFunction(t *testing.T) {
+	Convey("#RenderTemplate", t, func() {
+		templateName := "templateName"
+		params := map[string]string{"id": "app", "domain": "example.com"}
+
+		templateContent := "{{.id}} {{ToUpper .domain}}"
+		Convey("should transform example.com upper case", func() {
+			content, _ := RenderTemplate(templateName, templateContent, params)
+			So(content, ShouldEqual, "app EXAMPLE.COM")
+		})
+	})
+}
+
+func TestTemplateToLowerFunction(t *testing.T) {
+	Convey("#RenderTemplate", t, func() {
+		templateName := "templateName"
+		params := map[string]string{"domain": "EXAMPLE.COM"}
+
+		templateContent := "{{ToLower .domain}}"
+		Convey("should transform example.com lower case", func() {
+			content, _ := RenderTemplate(templateName, templateContent, params)
+			So(content, ShouldEqual, "example.com")
+		})
+	})
+}