blob: 9f2b783b2ef34debec2d9383e985ef021d714bad (
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
|
require 'abstract_unit'
class WraithAttack < StandardError
end
class NuclearExplosion < StandardError
end
class MadRonon < StandardError
attr_accessor :message
def initialize(message)
@message = message
super()
end
end
class Stargate
attr_accessor :result
include ActiveSupport::Rescuable
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
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
end
|
