parser: add some frontmatter test cases

2 files changed, 25 insertions, 2 deletions

diff --git a/parser/frontmatter.go b/parser/frontmatter.go
index 209786707..94bb60bb0 100644
--- a/parser/frontmatter.go
+++ b/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:
diff --git a/parser/frontmatter_test.go b/parser/frontmatter_test.go
new file mode 100644
index 000000000..33db0a24a
--- /dev/null
+++ b/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)
+		}
+	}
+}