rendering TOC errors on undefined 'sections'

nricciar · Oct 18, 2011 · 86cfda9 · 86cfda9
1 parent 3027538
commit 86cfda9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/lib/wikicloth/wiki_link_handler.rb b/lib/wikicloth/wiki_link_handler.rb
@@ -46,7 +46,7 @@ def section_list(root=nil)
   def toc(sections)
     ret = "<table id=\"toc\" class=\"toc\" summary=\"Contents\"><tr><td><div style=\"font-weight:bold\">Table of Contents</div>"
     previous_depth = 1
-    section_list.each do |section|
+    section_list(sections).each do |section|
       if section.depth > previous_depth
         c = section.depth - previous_depth
         c.times { ret += "<ul>" }

diff --git a/test/wiki_cloth_test.rb b/test/wiki_cloth_test.rb
@@ -241,4 +241,10 @@ class WikiClothTest < ActiveSupport::TestCase
       assert_equal data, "\n<p><h1><span id=\"Hallo\" class=\"mw-headline\"><a name=\"Hallo\">Hallo</a></span></h1></p>"
     end
   end
+
+  test "render toc" do
+    wiki = WikiCloth::WikiCloth.new({:data => "=A=\n=B=\n=C=\n=D="})
+    data = wiki.render
+    assert data =~ /A/
+  end
 end