/
attribute.rb
154 lines (133 loc) · 4.44 KB
/
attribute.rb
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
module Test
module Unit
module Attribute
class StringifyKeyHash < Hash
class << self
def stringify(object)
object.to_s
end
end
def key?(key)
super(self.class.stringify(key))
end
def [](key)
super(self.class.stringify(key))
end
def []=(key, value)
super(self.class.stringify(key), value)
end
end
class << self
def included(base)
base.extend(BaseClassMethods)
base.extend(ClassMethods)
end
end
module BaseClassMethods
def attributes_table
{}
end
end
module ClassMethods
def method_added(name)
super
return unless defined?(@current_attributes)
attributes = {}
kept_attributes = StringifyKeyHash.new
@current_attributes.each do |attribute_name, attribute|
attributes[attribute_name] = attribute[:value]
kept_attributes[attribute_name] = attribute if attribute[:keep]
end
set_attributes(name, attributes)
@current_attributes = kept_attributes
end
def attribute(name, value, options={}, *method_names)
unless options.is_a?(Hash)
method_names << options
options = {}
end
if method_names.empty?
current_attributes[name] = options.merge(:value => value)
else
method_names.each do |method_name|
set_attributes(method_name, {name => value})
end
end
end
def current_attributes
@current_attributes ||= StringifyKeyHash.new
end
def current_attribute(name)
current_attributes[name] || StringifyKeyHash.new
end
def attributes_table
@attributes_table ||= StringifyKeyHash.new
super.merge(@attributes_table)
end
def set_attributes(method_name, new_attributes)
return if new_attributes.empty?
@attributes_table ||= StringifyKeyHash.new
@attributes_table[method_name] ||= StringifyKeyHash.new
current_attributes = @attributes_table[method_name]
new_attributes.each do |key, value|
observers = attribute_observers(key) || []
observers.each do |observer|
observer.call(self,
StringifyKeyHash.stringify(key),
(attributes(method_name) || {})[key],
value,
method_name)
end
current_attributes[key] = value
end
end
def attributes(method_name)
attributes = attributes_table[method_name]
ancestors.each do |ancestor|
next if ancestor == self
if ancestor.is_a?(Class) and ancestor < Test::Unit::Attribute
parent_attributes = ancestor.attributes(method_name)
if attributes
attributes = (parent_attributes || {}).merge(attributes)
else
attributes = parent_attributes
end
break
end
end
attributes || StringifyKeyHash.new
end
def find_attribute(method_name, name)
@attributes_table ||= StringifyKeyHash.new
if @attributes_table.key?(method_name)
attributes = @attributes_table[method_name]
if attributes.key?(name)
return attributes[name]
end
end
return nil if self == TestCase
@cached_parent_test_case ||= ancestors.find do |ancestor|
ancestor != self and
ancestor.is_a?(Class) and
ancestor < Test::Unit::Attribute
end
@cached_parent_test_case.find_attribute(method_name, name)
end
@@attribute_observers = StringifyKeyHash.new
def register_attribute_observer(attribute_name, observer=Proc.new)
@@attribute_observers[attribute_name] ||= []
@@attribute_observers[attribute_name] << observer
end
def attribute_observers(attribute_name)
@@attribute_observers[attribute_name]
end
end
def attributes
self.class.attributes(@method_name) || StringifyKeyHash.new
end
def [](name)
self.class.find_attribute(@method_name, name)
end
end
end
end