blob: e5293a532a60062342f7a9e82dc69e9e7ea6dd7f (
plain)
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
|
###
### $Release:$
### $Copyright$
###
require 'yaml'
require 'test/unit/testcase'
def ruby18? # :nodoc:
RUBY_VERSION =~ /\A1.8/
end
def ruby19? # :nodoc:
RUBY_VERSION =~ /\A1.9/
end
def rubinius? # :nodoc:
defined?(RUBY_ENGINE) && RUBY_ENGINE == "rbx"
end
class Test::Unit::TestCase
def self.load_yaml_datafile(filename, options={}, &block) # :nodoc:
# read datafile
s = File.read(filename)
if filename =~ /\.rb$/
s =~ /^__END__$/ or raise "*** error: __END__ is not found in '#{filename}'."
s = $'
end
# untabify
s = _untabify(s) unless options[:tabify] == false
# load yaml document
testdata_list = []
YAML.load_documents(s) do |ydoc|
if ydoc.is_a?(Hash)
testdata_list << ydoc
elsif ydoc.is_a?(Array)
ydoc.each do |hash|
raise "testdata should be a mapping." unless hash.is_a?(Hash)
testdata_list << hash
end
else
raise "testdata should be a mapping."
end
end
# data check
identkey = options[:identkey] || 'name'
table = {}
testdata_list.each do |hash|
ident = hash[identkey]
ident or raise "*** key '#{identkey}' is required but not found."
table[ident] and raise "*** #{identkey} '#{ident}' is duplicated."
table[ident] = hash
yield(hash) if block
end
#
return testdata_list
end
def self.define_testmethods(testdata_list, options={}, &block)
identkey = options[:identkey] || 'name'
testmethod = options[:testmethod] || '_test'
testdata_list.each do |hash|
yield(hash) if block
ident = hash[identkey]
s = "def test_#{ident}\n"
hash.each do |key, val|
s << " @#{key} = #{val.inspect}\n"
end
s << " #{testmethod}\n"
s << "end\n"
$stderr.puts "*** load_yaml_testdata(): eval_str=<<'END'\n#{s}END" if $DEBUG
self.module_eval s
end
end
def self.post_definition
if ENV['TEST']
target = "test_#{ENV['TEST']}"
self.instance_methods.each do |method_name|
m = method_name.to_s
private m if m =~ /\Atest_/ && m != target
end
end
end
def self._untabify(str, width=8)
return str if str.nil?
list = str.split(/\t/, -1) # if 2nd arg is negative then split() doesn't remove tailing empty strings
last = list.pop
sb = ''
list.each do |s|
column = (n = s.rindex(?\n)) ? s.length - n - 1 : s.length
n = width - (column % width)
sb << s << (' ' * n)
end
sb << last if last
return sb
end
end
|