(template): Add & Update Tests; Rewrite Helper funcs; Updates #45

integration/help.golden

@@ -1,7 +1,8 @@
 Usage: fakedata [option ...] field...
 
-  -f, --format string   generators rows in f format. Available formats: csv|tab|sql
-  -g, --generators      lists available generators
-  -l, --limit int       limits rows up to n (default 10)
-  -t, --table string    table name of the sql format (default "TABLE")
-  -v, --version         shows version information
+  -f, --format string     generators rows in f format. Available formats: csv|tab|sql
+  -g, --generators        lists available generators
+  -l, --limit int         limits rows up to n (default 10)
+  -t, --table string      table name of the sql format (default "TABLE")
+      --template string   Use template as input
+  -v, --version           shows version information

main.go

@@ -18,7 +18,7 @@ var limitFlag = flag.IntP("limit", "l", 10, "limits rows up to n")
 var formatFlag = flag.StringP("format", "f", "", "generators rows in f format. Available formats: csv|tab|sql")
 var versionFlag = flag.BoolP("version", "v", false, "shows version information")
 var tableFlag = flag.StringP("table", "t", "TABLE", "table name of the sql format")
-var templateFlag = flag.StringP("template", "", "", "Use template")
+var templateFlag = flag.StringP("template", "", "", "Use template as input")
 
 func getFormatter(format string) (f fakedata.Formatter) {
 	switch format {

pkg/fakedata/template.go

@@ -50,11 +50,19 @@ func createConstraints(params []string) string {
 }
 
 func getTemplateNameFromPath(name string) string {
-	ts := strings.Split(name, "/")
+	ts := strings.FieldsFunc(name, splitPathName)
 	tn := ts[len(ts)-1]
 	return tn
 }
 
+// this custom split function is used with strings.FieldsFunc to split the path
+// by `/` (Unix, MacOS) or `\` (Windows) for absolute and relative paths to template files
+func splitPathName(r rune) bool {
+	return r == '/' || r == '\\'
+}
+
+// ParseTemplate takes a path to a template file as argument. It parses the template file and executes it on
+// os.Stdout.
 func ParseTemplate(path string) {
 	tn := getTemplateNameFromPath(path)
 	tmp, err := template.New(tn).Funcs(generatorFunctions).ParseFiles(path)

pkg/fakedata/template_test.go

@@ -0,0 +1,48 @@
+package fakedata
+
+import (
+	"testing"
+)
+
+func TestCreateConstraints(t *testing.T) {
+	tests := []struct {
+		name  string
+		input []string
+		want  string
+	}{
+		{"two integers", []string{"1", "100"}, "1..100"},
+		{"three word enum constraints", []string{"feat", "issue", "docs"}, "feat..issue..docs"},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := createConstraints(tt.input); got != tt.want {
+				t.Errorf("template.createConstraints = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func TestGetTemplateNameFromPath(t *testing.T) {
+	tests := []struct {
+		name  string
+		input string
+		want  string
+	}{
+		{"one word template name", "template.tmpl", "template.tmpl"},
+		{"absolute template path (unix)", "/var/www/home.tmpl", "home.tmpl"},
+		{"absolute template path (Windows)", "C:\\Data\\Templates\\test.tmpl", "test.tmpl"},
+		{"relative template path (Unix)", "./views/user/account.tmpl", "account.tmpl"},
+		{"relative template path (Windows)", "views\\user\\account.tmpl", "account.tmpl"},
+		{"spaced template path (Unix)", "./views/user account/account.tmpl", "account.tmpl"},
+		{"spaced template path (Windows)", "views\\user account\\account.tmpl", "account.tmpl"},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := getTemplateNameFromPath(tt.input); got != tt.want {
+				t.Errorf("template.getTemplateNameFromPath = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}