Bugfix: allow multiple headings per chapters in outline

avdgaag · Apr 21, 2012 · 3c8a87a · 3c8a87a
1 parent 797ab5f
commit 3c8a87a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/lib/rpub/epub/html_toc.rb b/lib/rpub/epub/html_toc.rb
@@ -12,9 +12,11 @@ def render
         xml.div :id => 'toc' do
           xml.h1 'Table of Contents'
           xml.div :class => 'toc' do
-            book.outline.each do |(filename, heading)|
-              xml.div :class => "level-#{heading.level}" do
-                xml.a heading.text, :href => [filename, heading.html_id].join('#')
+            book.outline.each do |(filename, headings)|
+              headings.each do |heading|
+                xml.div :class => "level-#{heading.level}" do
+                  xml.a heading.text, :href => [filename, heading.html_id].join('#')
+                end
               end
             end
           end

diff --git a/spec/rpub/epub/html_toc_spec.rb b/spec/rpub/epub/html_toc_spec.rb
@@ -13,7 +13,7 @@
   end
 
   context 'with heading in the outline' do
-    let(:outline) { [['foo.html', double('heading', :text => 'link', :html_id => 'bar', :level => 1)]] }
+    let(:outline) { [['foo.html', [double('heading', :text => 'link', :html_id => 'bar', :level => 1)]]] }
     it { should have_xpath('/div/div/div[@class="level-1"]/a[@href="foo.html#bar"][text()="link"]') }
   end
 end