Merge pull request #4301 from Karunakar/refactor_tests

Refactor tests

1 files changed, 7 insertions, 4 deletions

diff --git a/activerecord/test/cases/locking_test.rb b/activerecord/test/cases/locking_test.rb
index 65cd9f9755..0c458d5318 100644
--- a/activerecord/test/cases/locking_test.rb
+++ b/activerecord/test/cases/locking_test.rb
@@ -230,10 +230,13 @@ class OptimisticLockingTest < ActiveRecord::TestCase
 
   def test_polymorphic_destroy_with_dependencies_and_lock_version
     car = Car.create!
-    car.wheels << Wheel.create!
-    assert_equal 1, car.wheels.count
-    assert car.destroy
-    assert_equal 0, car.wheels.count
+    
+    assert_difference 'car.wheels.count'  do
+    	car.wheels << Wheel.create!
+    end 
+    assert_difference 'car.wheels.count', -1  do
+      car.destroy
+    end
     assert car.destroyed?
   end
 end