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