acts_as_king 0.0.1 → 0.0.2

data/lib/acts_as_king.rb

@@ -13,6 +13,11 @@ module ActsAsKing
     # association (parent) and has_many association (children). It also defines
     # a <tt>kings</tt> scope for your model to return all top level records.
     #
+    # === Options
+    #
+    # * <tt>:foreign_key</tt> - The foreign key used for the association. Normally this will be 'parent_id'. Default is 'parent_id'.
+    # * <tt>:counter_cache</tt> - Defines a counter_cache for the belongs_to association that is created. Set true for the default column name (ex: Comment model would be 'comments_count'). Otherwise specify the column name you wish to use. Default is false (no counter cache).
+    #
     # === Example
     #
     #   class Comment < ActiveRecord::Base
@@ -33,16 +38,13 @@ module ActsAsKing
     #
     #
     def acts_as_king(options = {})
-      options = { :foreign_key => 'parent_id' }.merge(options)
+      options = { :foreign_key => 'parent_id', :counter_cache => false }.merge(options)
 
-      belongs_to :parent, :class_name => name, :foreign_key => options[:foreign_key]
+      belongs_to :parent, :class_name => name, :foreign_key => options[:foreign_key], :counter_cache => options[:counter_cache]
       has_many :children, :class_name => name, :foreign_key => options[:foreign_key], :dependent => :destroy
 
-      class_eval <<-EOL
-        include ActsAsKing::InstanceMethods
-
-        scope :kings, where("#{options[:foreign_key]} IS NULL")
-      EOL
+      include ActsAsKing::InstanceMethods
+      scope :kings, where("#{options[:foreign_key]} IS NULL")
 
     end
 

data/lib/acts_as_king/version.rb

@@ -1,3 +1,3 @@
 module ActsAsKing
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/test/acts_as_king_test.rb

@@ -39,6 +39,11 @@ class ActsAsKingTest < Test::Unit::TestCase
     should "return multiple ancestors" do
       assert_equal [@sub_comment, @comment], @level_3.ancestors
     end
+
+    should "should use counter caching" do
+      assert_equal 1, @comment.reload.comments_count
+      assert_equal 1, @sub_comment.reload.comments_count
+    end
     
   end
 

data/test/helper.rb

@@ -1,5 +1,6 @@
 $LOAD_PATH << File.join(File.dirname(__FILE__), '..', 'lib')
 
+require 'logger'
 require 'active_record'
 require 'acts_as_king'
 require 'test/unit'
@@ -9,6 +10,8 @@ ActiveRecord::Base.send(:include, ActsAsKing)
 ActiveRecord::Base.establish_connection(:adapter => 'sqlite3', :database => ':memory:')
 ActiveRecord::Migration.verbose = false
 
+ActiveRecord::Base.logger = Logger.new(STDOUT) if ENV['SHOW_SQL'] && ENV['SHOW_SQL'] == 'true'
+
 def setup_db
   ActiveRecord::Schema.define(:version => 1) do
     create_table :posts, :force => true do |t|
@@ -17,6 +20,7 @@ def setup_db
     create_table :comments, :force => true do |t|
       t.integer :post_id, :null => false
       t.integer :parent_id
+      t.integer :comments_count
     end
   end
 end
@@ -33,6 +37,6 @@ end
 
 class Comment < ActiveRecord::Base
   belongs_to :post
-  acts_as_king
+  acts_as_king :counter_cache => true
 end
 

metadata

@@ -2,7 +2,7 @@
 name: acts_as_king
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors: 
 - Drew Tempelmeyer
@@ -10,7 +10,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-05-14 00:00:00 Z
+date: 2011-05-16 00:00:00 Z
 dependencies: 
 - !ruby/object:Gem::Dependency 
   name: rails
@@ -99,7 +99,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 
 rubyforge_project: 
-rubygems_version: 1.8.2
+rubygems_version: 1.7.2
 signing_key: 
 specification_version: 3
 summary: Hierarchical ActiveRecord models. Inspired by acts_as_tree. Trees are shady.