template - minor cleanup

template.go

@@ -301,7 +301,7 @@ func templateFuncMap(r *rand.Rand, fm *template.FuncMap) *template.FuncMap {
 		return args
 	}
 
-	// Fix merge the user function maps with the template function map
+	// Add passed in function map to the function map
 	if fm != nil {
 		for k, v := range *fm {
 			funcMap[k] = v

template_test.go

@@ -46,12 +46,21 @@ func ExampleFaker_Template() {
 	// Sylvan Mraz
 }
 
-func TestUserFunctionMap(t *testing.T) {
+func TestPassedInFunctionMap(t *testing.T) {
 	f := New(11)
 
 	funcMap := template.FuncMap{
-		"title": strings.Title,
+		"title": func(s string) string {
+			words := strings.Fields(s)
+			for i, word := range words {
+				letters := strings.Split(word, "")
+				letters[0] = strings.ToUpper(letters[0])
+				words[i] = strings.Join(letters, "")
+			}
+			return strings.Join(words, " ")
+		},
 	}
+
 	s := `{{printf "%q" (title .Data)}}`
 
 	value, err := f.Template(s, &TemplateOptions{Funcs: funcMap, Data: "the go programming language"})