From 1712d90050bd4b32a9ee93548b5eef05a69491f7 Mon Sep 17 00:00:00 2001
From: Jon Leighton <j@jonathanleighton.com>
Date: Tue, 12 Jul 2011 00:04:01 +0100
Subject: Fix exception if old and new targets are both nil. Fixes #1471.

---
 .../test/cases/associations/belongs_to_associations_test.rb      | 9 +++++++++
 activerecord/test/models/comment.rb                              | 5 ++++-
 activerecord/test/schema/schema.rb                               | 1 +
 3 files changed, 14 insertions(+), 1 deletion(-)

(limited to 'activerecord/test')

diff --git a/activerecord/test/cases/associations/belongs_to_associations_test.rb b/activerecord/test/cases/associations/belongs_to_associations_test.rb
index 4190f6d76d..818902beb5 100644
--- a/activerecord/test/cases/associations/belongs_to_associations_test.rb
+++ b/activerecord/test/cases/associations/belongs_to_associations_test.rb
@@ -681,4 +681,13 @@ class BelongsToAssociationsTest < ActiveRecord::TestCase
     firm = client.create_firm! :name => "baa"
     assert_equal firm.id, client.client_of
   end
+
+  def test_self_referential_belongs_to_with_counter_cache_assigning_nil
+    comment = Comment.create! :post => posts(:thinking), :body => "fuu"
+    comment.parent = nil
+    comment.save!
+
+    assert_equal nil, comment.reload.parent
+    assert_equal 0, comments(:greetings).reload.children_count
+  end
 end
diff --git a/activerecord/test/models/comment.rb b/activerecord/test/models/comment.rb
index 43650c0427..88b139d931 100644
--- a/activerecord/test/models/comment.rb
+++ b/activerecord/test/models/comment.rb
@@ -7,10 +7,13 @@ class Comment < ActiveRecord::Base
               :joins => :post,
               :conditions => { "posts.author_id" => 1 }
   scope :created
-  
+
   belongs_to :post, :counter_cache => true
   has_many :ratings
 
+  has_many :children, :class_name => 'Comment', :foreign_key => :parent_id
+  belongs_to :parent, :class_name => 'Comment', :counter_cache => :children_count
+
   def self.what_are_you
     'a comment...'
   end
diff --git a/activerecord/test/schema/schema.rb b/activerecord/test/schema/schema.rb
index e6cd274586..64e0452100 100644
--- a/activerecord/test/schema/schema.rb
+++ b/activerecord/test/schema/schema.rb
@@ -155,6 +155,7 @@ ActiveRecord::Schema.define do
     end
     t.string  :type
     t.integer :taggings_count, :default => 0
+    t.integer :children_count, :default => 0
   end
 
   create_table :companies, :force => true do |t|
-- 
cgit v1.2.3