test/test_helper.rb in jekyll-toc-0.0.4 vs test/test_helper.rb in jekyll-toc-0.1.0

@@ -1,2 +1,18 @@
 require 'jekyll'
 require 'minitest/autorun'
+
+SIMPLE_HTML = <<EOL
+<h1>Simple H1</h1>
+<h2>Simple H2</h2>
+<h3>Simple H3</h3>
+<h4>Simple H4</h4>
+<h5>Simple H5</h5>
+<h6>Simple H6</h6>
+EOL
+
+module TestHelpers
+  def read_html_and_create_parser
+    @parser = Jekyll::TableOfContents::Parser.new(SIMPLE_HTML)
+    assert_match /Simple H1/, @parser.doc.inner_html
+  end
+end