386 lines
12 KiB
Ruby
386 lines
12 KiB
Ruby
|
require "cases/helper"
|
||
|
require "models/pirate"
|
||
|
require "models/ship"
|
||
|
require "models/ship_part"
|
||
|
require "models/bird"
|
||
|
require "models/parrot"
|
||
|
require "models/treasure"
|
||
|
|
||
|
class TestAutosaveAssociationsInGeneral < ActiveRecord::TestCase
|
||
|
def test_autosave_should_be_a_valid_option_for_has_one
|
||
|
assert base.valid_keys_for_has_one_association.include?(:autosave)
|
||
|
end
|
||
|
|
||
|
def test_autosave_should_be_a_valid_option_for_belongs_to
|
||
|
assert base.valid_keys_for_belongs_to_association.include?(:autosave)
|
||
|
end
|
||
|
|
||
|
def test_autosave_should_be_a_valid_option_for_has_many
|
||
|
assert base.valid_keys_for_has_many_association.include?(:autosave)
|
||
|
end
|
||
|
|
||
|
def test_autosave_should_be_a_valid_option_for_has_and_belongs_to_many
|
||
|
assert base.valid_keys_for_has_and_belongs_to_many_association.include?(:autosave)
|
||
|
end
|
||
|
|
||
|
private
|
||
|
|
||
|
def base
|
||
|
ActiveRecord::Base
|
||
|
end
|
||
|
end
|
||
|
|
||
|
class TestDestroyAsPartOfAutosaveAssociation < ActiveRecord::TestCase
|
||
|
self.use_transactional_fixtures = false
|
||
|
|
||
|
def setup
|
||
|
@pirate = Pirate.create(:catchphrase => "Don' botharrr talkin' like one, savvy?")
|
||
|
@ship = @pirate.create_ship(:name => 'Nights Dirty Lightning')
|
||
|
end
|
||
|
|
||
|
# reload
|
||
|
def test_a_marked_for_destruction_record_should_not_be_be_marked_after_reload
|
||
|
@pirate.mark_for_destruction
|
||
|
@pirate.ship.mark_for_destruction
|
||
|
|
||
|
assert !@pirate.reload.marked_for_destruction?
|
||
|
assert !@pirate.ship.marked_for_destruction?
|
||
|
end
|
||
|
|
||
|
# has_one
|
||
|
def test_should_destroy_a_child_association_as_part_of_the_save_transaction_if_it_was_marked_for_destroyal
|
||
|
assert !@pirate.ship.marked_for_destruction?
|
||
|
|
||
|
@pirate.ship.mark_for_destruction
|
||
|
id = @pirate.ship.id
|
||
|
|
||
|
assert @pirate.ship.marked_for_destruction?
|
||
|
assert Ship.find_by_id(id)
|
||
|
|
||
|
@pirate.save
|
||
|
assert_nil @pirate.reload.ship
|
||
|
assert_nil Ship.find_by_id(id)
|
||
|
end
|
||
|
|
||
|
def test_should_rollback_destructions_if_an_exception_occurred_while_saving_a_child
|
||
|
# Stub the save method of the @pirate.ship instance to destroy and then raise an exception
|
||
|
class << @pirate.ship
|
||
|
def save(*args)
|
||
|
super
|
||
|
destroy
|
||
|
raise 'Oh noes!'
|
||
|
end
|
||
|
end
|
||
|
|
||
|
assert_raise(RuntimeError) { assert !@pirate.save }
|
||
|
assert_not_nil @pirate.reload.ship
|
||
|
end
|
||
|
|
||
|
# belongs_to
|
||
|
def test_should_destroy_a_parent_association_as_part_of_the_save_transaction_if_it_was_marked_for_destroyal
|
||
|
assert !@ship.pirate.marked_for_destruction?
|
||
|
|
||
|
@ship.pirate.mark_for_destruction
|
||
|
id = @ship.pirate.id
|
||
|
|
||
|
assert @ship.pirate.marked_for_destruction?
|
||
|
assert Pirate.find_by_id(id)
|
||
|
|
||
|
@ship.save
|
||
|
assert_nil @ship.reload.pirate
|
||
|
assert_nil Pirate.find_by_id(id)
|
||
|
end
|
||
|
|
||
|
def test_should_rollback_destructions_if_an_exception_occurred_while_saving_a_parent
|
||
|
# Stub the save method of the @ship.pirate instance to destroy and then raise an exception
|
||
|
class << @ship.pirate
|
||
|
def save(*args)
|
||
|
super
|
||
|
destroy
|
||
|
raise 'Oh noes!'
|
||
|
end
|
||
|
end
|
||
|
|
||
|
assert_raise(RuntimeError) { assert !@ship.save }
|
||
|
assert_not_nil @ship.reload.pirate
|
||
|
end
|
||
|
|
||
|
# has_many & has_and_belongs_to
|
||
|
%w{ parrots birds }.each do |association_name|
|
||
|
define_method("test_should_destroy_#{association_name}_as_part_of_the_save_transaction_if_they_were_marked_for_destroyal") do
|
||
|
2.times { |i| @pirate.send(association_name).create!(:name => "#{association_name}_#{i}") }
|
||
|
|
||
|
assert !@pirate.send(association_name).any? { |child| child.marked_for_destruction? }
|
||
|
|
||
|
@pirate.send(association_name).each { |child| child.mark_for_destruction }
|
||
|
klass = @pirate.send(association_name).first.class
|
||
|
ids = @pirate.send(association_name).map(&:id)
|
||
|
|
||
|
assert @pirate.send(association_name).all? { |child| child.marked_for_destruction? }
|
||
|
ids.each { |id| assert klass.find_by_id(id) }
|
||
|
|
||
|
@pirate.save
|
||
|
assert @pirate.reload.send(association_name).empty?
|
||
|
ids.each { |id| assert_nil klass.find_by_id(id) }
|
||
|
end
|
||
|
|
||
|
define_method("test_should_rollback_destructions_if_an_exception_occurred_while_saving_#{association_name}") do
|
||
|
2.times { |i| @pirate.send(association_name).create!(:name => "#{association_name}_#{i}") }
|
||
|
before = @pirate.send(association_name).map { |c| c }
|
||
|
|
||
|
# Stub the save method of the first child to destroy and the second to raise an exception
|
||
|
class << before.first
|
||
|
def save(*args)
|
||
|
super
|
||
|
destroy
|
||
|
end
|
||
|
end
|
||
|
class << before.last
|
||
|
def save(*args)
|
||
|
super
|
||
|
raise 'Oh noes!'
|
||
|
end
|
||
|
end
|
||
|
|
||
|
assert_raise(RuntimeError) { assert !@pirate.save }
|
||
|
assert_equal before, @pirate.reload.send(association_name)
|
||
|
end
|
||
|
end
|
||
|
end
|
||
|
|
||
|
class TestAutosaveAssociationOnAHasOneAssociation < ActiveRecord::TestCase
|
||
|
self.use_transactional_fixtures = false
|
||
|
|
||
|
def setup
|
||
|
@pirate = Pirate.create(:catchphrase => "Don' botharrr talkin' like one, savvy?")
|
||
|
@ship = @pirate.create_ship(:name => 'Nights Dirty Lightning')
|
||
|
end
|
||
|
|
||
|
def test_should_still_work_without_an_associated_model
|
||
|
@ship.destroy
|
||
|
@pirate.reload.catchphrase = "Arr"
|
||
|
@pirate.save
|
||
|
assert 'Arr', @pirate.reload.catchphrase
|
||
|
end
|
||
|
|
||
|
def test_should_automatically_save_the_associated_model
|
||
|
@pirate.ship.name = 'The Vile Insanity'
|
||
|
@pirate.save
|
||
|
assert_equal 'The Vile Insanity', @pirate.reload.ship.name
|
||
|
end
|
||
|
|
||
|
def test_should_automatically_validate_the_associated_model
|
||
|
@pirate.ship.name = ''
|
||
|
assert !@pirate.valid?
|
||
|
assert !@pirate.errors.on(:ship_name).blank?
|
||
|
end
|
||
|
|
||
|
def test_should_still_allow_to_bypass_validations_on_the_associated_model
|
||
|
@pirate.catchphrase = ''
|
||
|
@pirate.ship.name = ''
|
||
|
@pirate.save(false)
|
||
|
assert_equal ['', ''], [@pirate.reload.catchphrase, @pirate.ship.name]
|
||
|
end
|
||
|
|
||
|
def test_should_allow_to_bypass_validations_on_associated_models_at_any_depth
|
||
|
2.times { |i| @pirate.ship.parts.create!(:name => "part #{i}") }
|
||
|
|
||
|
@pirate.catchphrase = ''
|
||
|
@pirate.ship.name = ''
|
||
|
@pirate.ship.parts.each { |part| part.name = '' }
|
||
|
@pirate.save(false)
|
||
|
|
||
|
values = [@pirate.reload.catchphrase, @pirate.ship.name, *@pirate.ship.parts.map(&:name)]
|
||
|
assert_equal ['', '', '', ''], values
|
||
|
end
|
||
|
|
||
|
def test_should_still_raise_an_ActiveRecordRecord_Invalid_exception_if_we_want_that
|
||
|
@pirate.ship.name = ''
|
||
|
assert_raise(ActiveRecord::RecordInvalid) do
|
||
|
@pirate.save!
|
||
|
end
|
||
|
end
|
||
|
|
||
|
def test_should_rollback_any_changes_if_an_exception_occurred_while_saving
|
||
|
before = [@pirate.catchphrase, @pirate.ship.name]
|
||
|
|
||
|
@pirate.catchphrase = 'Arr'
|
||
|
@pirate.ship.name = 'The Vile Insanity'
|
||
|
|
||
|
# Stub the save method of the @pirate.ship instance to raise an exception
|
||
|
class << @pirate.ship
|
||
|
def save(*args)
|
||
|
super
|
||
|
raise 'Oh noes!'
|
||
|
end
|
||
|
end
|
||
|
|
||
|
assert_raise(RuntimeError) { assert !@pirate.save }
|
||
|
assert_equal before, [@pirate.reload.catchphrase, @pirate.ship.name]
|
||
|
end
|
||
|
|
||
|
def test_should_not_load_the_associated_model
|
||
|
assert_queries(1) { @pirate.catchphrase = 'Arr'; @pirate.save! }
|
||
|
end
|
||
|
end
|
||
|
|
||
|
class TestAutosaveAssociationOnABelongsToAssociation < ActiveRecord::TestCase
|
||
|
self.use_transactional_fixtures = false
|
||
|
|
||
|
def setup
|
||
|
@ship = Ship.create(:name => 'Nights Dirty Lightning')
|
||
|
@pirate = @ship.create_pirate(:catchphrase => "Don' botharrr talkin' like one, savvy?")
|
||
|
end
|
||
|
|
||
|
def test_should_still_work_without_an_associated_model
|
||
|
@pirate.destroy
|
||
|
@ship.reload.name = "The Vile Insanity"
|
||
|
@ship.save
|
||
|
assert 'The Vile Insanity', @ship.reload.name
|
||
|
end
|
||
|
|
||
|
def test_should_automatically_save_the_associated_model
|
||
|
@ship.pirate.catchphrase = 'Arr'
|
||
|
@ship.save
|
||
|
assert_equal 'Arr', @ship.reload.pirate.catchphrase
|
||
|
end
|
||
|
|
||
|
def test_should_automatically_validate_the_associated_model
|
||
|
@ship.pirate.catchphrase = ''
|
||
|
assert !@ship.valid?
|
||
|
assert !@ship.errors.on(:pirate_catchphrase).blank?
|
||
|
end
|
||
|
|
||
|
def test_should_still_allow_to_bypass_validations_on_the_associated_model
|
||
|
@ship.pirate.catchphrase = ''
|
||
|
@ship.name = ''
|
||
|
@ship.save(false)
|
||
|
assert_equal ['', ''], [@ship.reload.name, @ship.pirate.catchphrase]
|
||
|
end
|
||
|
|
||
|
def test_should_still_raise_an_ActiveRecordRecord_Invalid_exception_if_we_want_that
|
||
|
@ship.pirate.catchphrase = ''
|
||
|
assert_raise(ActiveRecord::RecordInvalid) do
|
||
|
@ship.save!
|
||
|
end
|
||
|
end
|
||
|
|
||
|
def test_should_rollback_any_changes_if_an_exception_occurred_while_saving
|
||
|
before = [@ship.pirate.catchphrase, @ship.name]
|
||
|
|
||
|
@ship.pirate.catchphrase = 'Arr'
|
||
|
@ship.name = 'The Vile Insanity'
|
||
|
|
||
|
# Stub the save method of the @ship.pirate instance to raise an exception
|
||
|
class << @ship.pirate
|
||
|
def save(*args)
|
||
|
super
|
||
|
raise 'Oh noes!'
|
||
|
end
|
||
|
end
|
||
|
|
||
|
assert_raise(RuntimeError) { assert !@ship.save }
|
||
|
# TODO: Why does using reload on @ship looses the associated pirate?
|
||
|
assert_equal before, [@ship.pirate.reload.catchphrase, @ship.reload.name]
|
||
|
end
|
||
|
|
||
|
def test_should_not_load_the_associated_model
|
||
|
assert_queries(1) { @ship.name = 'The Vile Insanity'; @ship.save! }
|
||
|
end
|
||
|
end
|
||
|
|
||
|
module AutosaveAssociationOnACollectionAssociationTests
|
||
|
def test_should_automatically_save_the_associated_models
|
||
|
new_names = ['Grace OMalley', 'Privateers Greed']
|
||
|
@pirate.send(@association_name).each_with_index { |child, i| child.name = new_names[i] }
|
||
|
|
||
|
@pirate.save
|
||
|
assert_equal new_names, @pirate.reload.send(@association_name).map(&:name)
|
||
|
end
|
||
|
|
||
|
def test_should_automatically_validate_the_associated_models
|
||
|
@pirate.send(@association_name).each { |child| child.name = '' }
|
||
|
|
||
|
assert !@pirate.valid?
|
||
|
assert_equal "can't be blank", @pirate.errors.on("#{@association_name}_name")
|
||
|
assert @pirate.errors.on(@association_name).blank?
|
||
|
end
|
||
|
|
||
|
def test_should_still_allow_to_bypass_validations_on_the_associated_models
|
||
|
@pirate.catchphrase = ''
|
||
|
@pirate.send(@association_name).each { |child| child.name = '' }
|
||
|
|
||
|
assert @pirate.save(false)
|
||
|
assert_equal ['', '', ''], [
|
||
|
@pirate.reload.catchphrase,
|
||
|
@pirate.send(@association_name).first.name,
|
||
|
@pirate.send(@association_name).last.name
|
||
|
]
|
||
|
end
|
||
|
|
||
|
def test_should_rollback_any_changes_if_an_exception_occurred_while_saving
|
||
|
before = [@pirate.catchphrase, *@pirate.send(@association_name).map(&:name)]
|
||
|
new_names = ['Grace OMalley', 'Privateers Greed']
|
||
|
|
||
|
@pirate.catchphrase = 'Arr'
|
||
|
@pirate.send(@association_name).each_with_index { |child, i| child.name = new_names[i] }
|
||
|
|
||
|
# Stub the save method of the first child instance to raise an exception
|
||
|
class << @pirate.send(@association_name).first
|
||
|
def save(*args)
|
||
|
super
|
||
|
raise 'Oh noes!'
|
||
|
end
|
||
|
end
|
||
|
|
||
|
assert_raise(RuntimeError) { assert !@pirate.save }
|
||
|
assert_equal before, [@pirate.reload.catchphrase, *@pirate.send(@association_name).map(&:name)]
|
||
|
end
|
||
|
|
||
|
def test_should_still_raise_an_ActiveRecordRecord_Invalid_exception_if_we_want_that
|
||
|
@pirate.send(@association_name).each { |child| child.name = '' }
|
||
|
assert_raise(ActiveRecord::RecordInvalid) do
|
||
|
@pirate.save!
|
||
|
end
|
||
|
end
|
||
|
|
||
|
def test_should_not_load_the_associated_models_if_they_were_not_loaded_yet
|
||
|
assert_queries(1) { @pirate.catchphrase = 'Arr'; @pirate.save! }
|
||
|
|
||
|
assert_queries(2) do
|
||
|
@pirate.catchphrase = 'Yarr'
|
||
|
new_names = ['Grace OMalley', 'Privateers Greed']
|
||
|
@pirate.send(@association_name).each_with_index { |child, i| child.name = new_names[i] }
|
||
|
@pirate.save!
|
||
|
end
|
||
|
end
|
||
|
end
|
||
|
|
||
|
class TestAutosaveAssociationOnAHasManyAssociation < ActiveRecord::TestCase
|
||
|
self.use_transactional_fixtures = false
|
||
|
|
||
|
def setup
|
||
|
@association_name = :birds
|
||
|
|
||
|
@pirate = Pirate.create(:catchphrase => "Don' botharrr talkin' like one, savvy?")
|
||
|
@child_1 = @pirate.birds.create(:name => 'Posideons Killer')
|
||
|
@child_2 = @pirate.birds.create(:name => 'Killer bandita Dionne')
|
||
|
end
|
||
|
|
||
|
include AutosaveAssociationOnACollectionAssociationTests
|
||
|
end
|
||
|
|
||
|
class TestAutosaveAssociationOnAHasAndBelongsToManyAssociation < ActiveRecord::TestCase
|
||
|
self.use_transactional_fixtures = false
|
||
|
|
||
|
def setup
|
||
|
@association_name = :parrots
|
||
|
@habtm = true
|
||
|
|
||
|
@pirate = Pirate.create(:catchphrase => "Don' botharrr talkin' like one, savvy?")
|
||
|
@child_1 = @pirate.parrots.create(:name => 'Posideons Killer')
|
||
|
@child_2 = @pirate.parrots.create(:name => 'Killer bandita Dionne')
|
||
|
end
|
||
|
|
||
|
include AutosaveAssociationOnACollectionAssociationTests
|
||
|
end
|