From 645fb2b02b1563bd98c07515bdc2d6d8478ba74e Mon Sep 17 00:00:00 2001
From: Titus Wormer <tituswormer@gmail.com>
Date: Thu, 13 Oct 2022 13:32:32 +0200
Subject: Add a bunch more tests for frontmatter

---
 src/construct/frontmatter.rs | 15 +++++-------
 tests/frontmatter.rs         | 54 ++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 60 insertions(+), 9 deletions(-)

diff --git a/src/construct/frontmatter.rs b/src/construct/frontmatter.rs
index 268d91d..f4774d6 100644
--- a/src/construct/frontmatter.rs
+++ b/src/construct/frontmatter.rs
@@ -281,13 +281,10 @@ pub fn content_end(tokenizer: &mut Tokenizer) -> State {
 ///        ^
 /// ```
 pub fn after(tokenizer: &mut Tokenizer) -> State {
-    tokenizer.tokenize_state.marker = 0;
-
-    match tokenizer.current {
-        None | Some(b'\n') => {
-            tokenizer.exit(Name::Frontmatter);
-            State::Ok
-        }
-        _ => State::Nok,
-    }
+    debug_assert!(
+        matches!(tokenizer.current, None | Some(b'\n')),
+        "expected eol/eof after closing fence"
+    );
+    tokenizer.exit(Name::Frontmatter);
+    State::Ok
 }
diff --git a/tests/frontmatter.rs b/tests/frontmatter.rs
index 267b694..b1380b6 100644
--- a/tests/frontmatter.rs
+++ b/tests/frontmatter.rs
@@ -44,12 +44,60 @@ fn frontmatter() -> Result<(), String> {
         "should support empty frontmatter"
     );
 
+    assert_eq!(
+        to_html_with_options("--\n---", &frontmatter)?,
+        "<h2>--</h2>",
+        "should not support 2 markers in an opening fence"
+    );
+
+    assert_eq!(
+        to_html_with_options("----\n---", &frontmatter)?,
+        "<hr />\n<hr />",
+        "should not support 4 markers in an opening fence"
+    );
+
+    assert_eq!(
+        to_html_with_options("---\n--", &frontmatter)?,
+        "<hr />\n<p>--</p>",
+        "should not support 2 markers in a closing fence"
+    );
+
+    assert_eq!(
+        to_html_with_options("---\n----", &frontmatter)?,
+        "<hr />\n<hr />",
+        "should not support 4 markers in a closing fence"
+    );
+
     assert_eq!(
         to_html_with_options("---\n---\n## Neptune", &frontmatter)?,
         "<h2>Neptune</h2>",
         "should support content after frontmatter"
     );
 
+    assert_eq!(
+        to_html_with_options("--- \t\n---", &frontmatter)?,
+        "",
+        "should support spaces and tabs after opening fence"
+    );
+
+    assert_eq!(
+        to_html_with_options("---\n---\t ", &frontmatter)?,
+        "",
+        "should support spaces and tabs after closing fence"
+    );
+
+    assert_eq!(
+        to_html_with_options("--- a\n---", &frontmatter)?,
+        "<h2>--- a</h2>",
+        "should not support content after opening fence"
+    );
+
+    assert_eq!(
+        to_html_with_options("---\n--- b", &frontmatter)?,
+        "<hr />\n<p>--- b</p>",
+        "should not support content after closing fence"
+    );
+
     assert_eq!(
         to_html_with_options("## Neptune\n---\n---", &frontmatter)?,
         "<h2>Neptune</h2>\n<hr />\n<hr />",
@@ -74,6 +122,12 @@ fn frontmatter() -> Result<(), String> {
         "should not support a missing closing fence"
     );
 
+    assert_eq!(
+        to_html_with_options("---\na\n\nb\n \t\nc\n---", &frontmatter)?,
+        "",
+        "should support blank lines in frontmatter"
+    );
+
     assert_eq!(
         to_mdast("---\na: b\n---", &frontmatter.parse)?,
         Node::Root(Root {
-- 
cgit