parser: add some frontmatter test cases

parser/frontmatter.go

@@ -128,8 +128,6 @@ func FormatToLeadRune(kind string) rune {
 	switch FormatSanitize(kind) {
 	case "yaml":
 		return rune([]byte(YAML_LEAD)[0])
-	case "toml":
-		return rune([]byte(TOML_LEAD)[0])
 	case "json":
 		return rune([]byte(JSON_LEAD)[0])
 	default:

parser/frontmatter_test.go

@@ -0,0 +1,25 @@
+package parser
+
+import (
+	"testing"
+)
+
+func TestFormatToLeadRune(t *testing.T) {
+	for i, this := range []struct {
+		kind   string
+		expect rune
+	}{
+		{"yaml", '-'},
+		{"yml", '-'},
+		{"toml", '+'},
+		{"json", '{'},
+		{"js", '{'},
+		{"unknown", '+'},
+	} {
+		result := FormatToLeadRune(this.kind)
+
+		if result != this.expect {
+			t.Errorf("[%d] Got %q but expected %q", i, result, this.expect)
+		}
+	}
+}