parser: add some frontmatter test cases

diff --git a/parser/frontmatter.go b/parser/frontmatter.go
index 209786707a6fb0875e7817cfad78081313e2bbcd..94bb60bb044131a0ae8c3101a7643c33de4a4889 100644 (file)
--- 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 (file)
index 0000000..33db0a2
--- /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)
+               }
+       }
+}