blob: 8d2577c64a0771e8403d14b8e28faae97aa535ab (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
require 'abstract_unit'
class WraithAttack < StandardError
end
class NuclearExplosion < StandardError
end
class MadRonon < StandardError
end
class Stargate
attr_accessor :result
include ActiveSupport::Rescuable
rescue_from WraithAttack, :with => :sos_first
rescue_from WraithAttack, :with => :sos
rescue_from NuclearExplosion do
@result = 'alldead'
end
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
def sos_first
@result = 'sos_first'
end
end
class RescueableTest < Test::Unit::TestCase
def setup
@stargate = Stargate.new
end
def test_rescue_from_with_method
@stargate.dispatch :attack
assert_equal 'killed', @stargate.result
end
def test_rescue_from_with_block
@stargate.dispatch :nuke
assert_equal 'alldead', @stargate.result
end
def test_rescue_from_with_block_with_args
@stargate.dispatch :ronanize
assert_equal 'dex', @stargate.result
end
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
end
|