0
@@ -166,12 +166,18 @@ module ActiveRecord
0
         if attrs.is_a?(Array)
0
           attrs.collect { |attr| create(attr) }
0
         else
0
-          create_record(attrs) { |record| record.save }
0
+          create_record(attrs) do |record|
0
+            yield(record) if block_given?
0
+            record.save
0
+          end
0
         end
0
       end
0
 
0
       def create!(attrs = {})
0
-        create_record(attrs) { |record| record.save! }
0
+        create_record(attrs) do |record|
0
+          yield(record) if block_given?
0
+          record.save!
0
+        end
0
       end
0
 
0
       # Returns the size of the collection by executing a SELECT COUNT(*) query if the collection hasn't been loaded and

0
@@ -161,12 +161,15 @@ class AssociationProxyTest < ActiveRecord::TestCase
0
   end
0
 
0
   def test_create_via_association_with_block
0
-    post1 = Post.create(:title => "setting body with a block") {|p| p.body = "will work"}
0
-    assert_equal post1.body, "will work"
0
-    assert_nothing_raised do
0
-      post2 = authors(:david).posts.create(:title => "setting body with a block") {|p| p.body = "won't work"}
0
-    end
0
-    assert_equal post2.body, "won't work"
0
+    post = authors(:david).posts.create(:title => "New on Edge") {|p| p.body = "More cool stuff!"}
0
+    assert_equal post.title, "New on Edge"
0
+    assert_equal post.body, "More cool stuff!"
0
+  end
0
+
0
+  def test_create_with_bang_via_association_with_block
0
+    post = authors(:david).posts.create!(:title => "New on Edge") {|p| p.body = "More cool stuff!"}
0
+    assert_equal post.title, "New on Edge"
0
+    assert_equal post.body, "More cool stuff!"
0
   end
0
 
0
   def test_failed_reload_returns_nil