2008-10-04 16:48:18 -04:00
|
|
|
require 'abstract_unit'
|
|
|
|
|
|
|
|
class WraithAttack < StandardError
|
|
|
|
end
|
|
|
|
|
|
|
|
class NuclearExplosion < StandardError
|
|
|
|
end
|
|
|
|
|
|
|
|
class MadRonon < StandardError
|
|
|
|
end
|
|
|
|
|
2010-07-26 09:41:53 -04:00
|
|
|
class CoolError < StandardError
|
|
|
|
end
|
|
|
|
|
2008-10-04 16:48:18 -04:00
|
|
|
class Stargate
|
|
|
|
attr_accessor :result
|
|
|
|
|
|
|
|
include ActiveSupport::Rescuable
|
|
|
|
|
2010-07-23 15:51:46 -04:00
|
|
|
rescue_from WraithAttack, :with => :sos_first
|
|
|
|
|
2008-10-04 16:48:18 -04:00
|
|
|
rescue_from WraithAttack, :with => :sos
|
|
|
|
|
2013-01-22 19:01:30 -05:00
|
|
|
rescue_from 'NuclearExplosion' do
|
2008-10-04 16:48:18 -04:00
|
|
|
@result = 'alldead'
|
|
|
|
end
|
2010-08-14 01:13:00 -04:00
|
|
|
|
2008-10-04 16:48:18 -04:00
|
|
|
rescue_from MadRonon do |e|
|
|
|
|
@result = e.message
|
|
|
|
end
|
|
|
|
|
|
|
|
def dispatch(method)
|
|
|
|
send(method)
|
|
|
|
rescue Exception => e
|
|
|
|
rescue_with_handler(e)
|
|
|
|
end
|
|
|
|
|
|
|
|
def attack
|
|
|
|
raise WraithAttack
|
|
|
|
end
|
|
|
|
|
|
|
|
def nuke
|
|
|
|
raise NuclearExplosion
|
|
|
|
end
|
|
|
|
|
|
|
|
def ronanize
|
|
|
|
raise MadRonon.new("dex")
|
|
|
|
end
|
|
|
|
|
|
|
|
def sos
|
|
|
|
@result = 'killed'
|
|
|
|
end
|
2010-07-23 15:51:46 -04:00
|
|
|
|
|
|
|
def sos_first
|
|
|
|
@result = 'sos_first'
|
|
|
|
end
|
|
|
|
|
2008-10-04 16:48:18 -04:00
|
|
|
end
|
|
|
|
|
2010-07-26 09:41:53 -04:00
|
|
|
class CoolStargate < Stargate
|
|
|
|
attr_accessor :result
|
|
|
|
|
|
|
|
include ActiveSupport::Rescuable
|
|
|
|
|
|
|
|
rescue_from CoolError, :with => :sos_cool_error
|
|
|
|
|
|
|
|
def sos_cool_error
|
|
|
|
@result = 'sos_cool_error'
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
|
2012-01-05 20:12:46 -05:00
|
|
|
class RescuableTest < ActiveSupport::TestCase
|
2008-10-04 16:48:18 -04:00
|
|
|
def setup
|
|
|
|
@stargate = Stargate.new
|
2010-07-26 09:41:53 -04:00
|
|
|
@cool_stargate = CoolStargate.new
|
2008-10-04 16:48:18 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def test_rescue_from_with_method
|
|
|
|
@stargate.dispatch :attack
|
|
|
|
assert_equal 'killed', @stargate.result
|
|
|
|
end
|
2010-08-14 01:13:00 -04:00
|
|
|
|
2008-10-04 16:48:18 -04:00
|
|
|
def test_rescue_from_with_block
|
|
|
|
@stargate.dispatch :nuke
|
|
|
|
assert_equal 'alldead', @stargate.result
|
|
|
|
end
|
2010-08-14 01:13:00 -04:00
|
|
|
|
2008-10-04 16:48:18 -04:00
|
|
|
def test_rescue_from_with_block_with_args
|
|
|
|
@stargate.dispatch :ronanize
|
|
|
|
assert_equal 'dex', @stargate.result
|
2010-08-14 01:13:00 -04:00
|
|
|
end
|
2010-07-23 15:51:46 -04:00
|
|
|
|
|
|
|
def test_rescues_defined_later_are_added_at_end_of_the_rescue_handlers_array
|
|
|
|
expected = ["WraithAttack", "WraithAttack", "NuclearExplosion", "MadRonon"]
|
|
|
|
result = @stargate.send(:rescue_handlers).collect {|e| e.first}
|
|
|
|
assert_equal expected, result
|
|
|
|
end
|
|
|
|
|
2013-05-26 06:57:05 -04:00
|
|
|
def test_children_should_inherit_rescue_definitions_from_parents_and_child_rescue_should_be_appended
|
2010-07-26 09:41:53 -04:00
|
|
|
expected = ["WraithAttack", "WraithAttack", "NuclearExplosion", "MadRonon", "CoolError"]
|
|
|
|
result = @cool_stargate.send(:rescue_handlers).collect {|e| e.first}
|
|
|
|
assert_equal expected, result
|
|
|
|
end
|
2008-10-04 16:48:18 -04:00
|
|
|
end
|