From bac34fab3a27a12b35add1ecab409b19941e29a6 Mon Sep 17 00:00:00 2001
From: Josh Dady <josh@indecisive.com>
Date: Thu, 25 Dec 2008 13:49:56 -0500
Subject: [PATCH] Test and fix the site.categories hash

---
 lib/jekyll/site.rb                                    | 11 ++++-------
 .../foo/_posts/bar/2008-12-12-topical-post.textile    |  8 ++++++++
 test/test_post.rb                                     |  6 ++++++
 test/test_site.rb                                     |  7 +++++--
 4 files changed, 23 insertions(+), 9 deletions(-)
 create mode 100644 test/source/foo/_posts/bar/2008-12-12-topical-post.textile

diff --git a/lib/jekyll/site.rb b/lib/jekyll/site.rb
index 9c205431..8238b803 100644
--- a/lib/jekyll/site.rb
+++ b/lib/jekyll/site.rb
@@ -2,7 +2,7 @@ module Jekyll
   
   class Site
     attr_accessor :source, :dest
-    attr_accessor :layouts, :posts
+    attr_accessor :layouts, :posts, :categories
     
     # Initialize the site
     #   +source+ is String path to the source directory containing
@@ -16,6 +16,7 @@ module Jekyll
       self.dest = dest
       self.layouts = {}
       self.posts = []
+      self.categories = Hash.new { |hash, key| hash[key] = Array.new }
     end
     
     # Do the actual work of processing the site and generating the
@@ -63,6 +64,7 @@ module Jekyll
         if Post.valid?(f)
           post = Post.new(self.source, dir, f)
           self.posts << post
+          post.categories.each { |c| self.categories[c] << post }
         end
       end
       
@@ -72,6 +74,7 @@ module Jekyll
       end
       
       self.posts.sort!
+      self.categories.values.map { |cats| cats.sort! { |a, b| b <=> a} }
     rescue Errno::ENOENT => e
       # ignore missing layout dir
     end
@@ -133,12 +136,6 @@ module Jekyll
     #
     # Returns {"site" => {"time" => <Time>, "posts" => [<Post>]}}
     def site_payload
-      # Build the category hash map of category ( names => arrays of posts )
-      # then sort each array in reverse order
-      categories = Hash.new { |hash, key| hash[key] = Array.new }
-      self.posts.each { |p| p.categories.each { |c| categories[c] << p } }
-      categories.values.map { |cats| cats.sort! { |a, b| b <=> a} }
-      
       {"site" => {
         "time" => Time.now, 
         "posts" => self.posts.sort { |a,b| b <=> a },
diff --git a/test/source/foo/_posts/bar/2008-12-12-topical-post.textile b/test/source/foo/_posts/bar/2008-12-12-topical-post.textile
new file mode 100644
index 00000000..813db112
--- /dev/null
+++ b/test/source/foo/_posts/bar/2008-12-12-topical-post.textile
@@ -0,0 +1,8 @@
+---
+layout: default
+title: Topical Post
+---
+
+h1. {{ page.title }}
+
+This post has a topic.
diff --git a/test/test_post.rb b/test/test_post.rb
index 01e1dd8b..46a259e7 100644
--- a/test/test_post.rb
+++ b/test/test_post.rb
@@ -88,6 +88,12 @@ class TestPost < Test::Unit::TestCase
     assert_equal "<<< <p>url: /2008/11/21/complex.html<br />\ndate: #{Time.parse("2008-11-21")}<br />\nid: /2008/11/21/complex</p> >>>", p.output
   end
   
+  def test_categories_and_topics
+    p = Post.new(File.join(File.dirname(__FILE__), *%w[source]), 'foo', 'bar/2008-12-12-topical-post.textile')
+    assert_equal ['foo'], p.categories
+    assert_equal ['bar'], p.topics
+  end    
+  
   def test_include
     Jekyll.source = File.join(File.dirname(__FILE__), *%w[source])
     p = Post.new(File.join(File.dirname(__FILE__), *%w[source]), '', "2008-12-13-include.markdown")
diff --git a/test/test_site.rb b/test/test_site.rb
index 52c1ca83..fa256177 100644
--- a/test/test_site.rb
+++ b/test/test_site.rb
@@ -22,9 +22,12 @@ class TestSite < Test::Unit::TestCase
     assert_equal 4, @s.posts.size
   end
   
-  def test_write_posts
+  def test_site_payload
     clear_dest
-    
     @s.process
+    
+    assert_equal 5, @s.posts.length
+    assert_equal ['foo'], @s.categories.keys
+    assert_equal 1, @s.categories['foo'].length
   end
 end