parser: add some frontmatter test cases

1 files changed, 25 insertions, 0 deletions

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)
+		}
+	}
+}