From f956759f8ab656484add2d5bc6e8f6f4add79469 Mon Sep 17 00:00:00 2001
From: Les Fletcher <les.fletcher@gmail.com>
Date: Thu, 21 Jul 2011 01:36:02 -0700
Subject: fix after_initialize edge case (close #2074 and close #2175)

fix behavior when after_initialize is defined and a block is passed to Base.create
---
 activerecord/test/cases/base_test.rb          | 23 +++++++++++++++++++++++
 activerecord/test/models/wholesale_product.rb | 13 +++++++++++++
 activerecord/test/schema/schema.rb            |  5 +++++
 3 files changed, 41 insertions(+)
 create mode 100644 activerecord/test/models/wholesale_product.rb

(limited to 'activerecord/test')

diff --git a/activerecord/test/cases/base_test.rb b/activerecord/test/cases/base_test.rb
index 8b4e7dd799..934f0c8f74 100644
--- a/activerecord/test/cases/base_test.rb
+++ b/activerecord/test/cases/base_test.rb
@@ -21,6 +21,7 @@ require 'models/parrot'
 require 'models/person'
 require 'models/edge'
 require 'models/joke'
+require 'models/wholesale_product'
 require 'rexml/document'
 require 'active_support/core_ext/exception'
 
@@ -260,6 +261,28 @@ class BasicsTest < ActiveRecord::TestCase
     end
   end
 
+  def test_create_with_after_initialize
+    wp1 = WholesaleProduct.create(:msrp => 10)
+    assert_equal(10, wp1.msrp)
+    assert_equal(5, wp1.wholesale)
+
+    wp2 = WholesaleProduct.create(:wholesale => 10)
+    assert_equal(20, wp2.msrp)
+    assert_equal(10, wp2.wholesale)
+
+    wp3 = WholesaleProduct.create do |wp|
+      wp.msrp = 10
+    end
+    assert_equal(10, wp3.msrp)
+    assert_equal(5, wp3.wholesale)
+
+    wp4 = WholesaleProduct.create do |wp|
+      wp.wholesale = 10
+    end
+    assert_equal(20, wp4.msrp)
+    assert_equal(10, wp4.wholesale)
+  end
+
   def test_load
     topics = Topic.find(:all, :order => 'id')
     assert_equal(4, topics.size)
diff --git a/activerecord/test/models/wholesale_product.rb b/activerecord/test/models/wholesale_product.rb
new file mode 100644
index 0000000000..e1a1a4c8b1
--- /dev/null
+++ b/activerecord/test/models/wholesale_product.rb
@@ -0,0 +1,13 @@
+class WholesaleProduct < ActiveRecord::Base
+
+  after_initialize :set_prices
+
+  def set_prices
+    if msrp.nil? && !wholesale.nil?
+      self.msrp = 2 * wholesale
+    elsif !msrp.nil? && wholesale.nil?
+      self.wholesale = msrp / 2
+    end
+  end
+
+end
\ No newline at end of file
diff --git a/activerecord/test/schema/schema.rb b/activerecord/test/schema/schema.rb
index 64e0452100..d2a41ed87a 100644
--- a/activerecord/test/schema/schema.rb
+++ b/activerecord/test/schema/schema.rb
@@ -678,6 +678,11 @@ ActiveRecord::Schema.define do
     t.references :wheelable, :polymorphic => true
   end
 
+  create_table :wholesale_products, :force => true do |t|
+    t.integer :msrp
+    t.integer :wholesale
+  end
+
   create_table :zines, :force => true do |t|
     t.string :title
   end
-- 
cgit v1.2.3