blob: 92aafccb2011acbe2e4db04aef7e0743940a1630 (
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
|
###
### $Release:$
### $Copyright$
###
require 'yaml'
require 'test/unit/testcase'
def ruby18? # :nodoc:
RUBY_VERSION =~ /\A1.8/
end
def ruby19? # :nodoc:
RUBY_VERSION =~ /\A1.9/
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
unless options[:tabify] == false
s = s.split(/^/).inject('') do |sb, line|
sb << line.gsub(/([^\t]{8})|([^\t]*)\t/n) { [$+].pack("A8") }
end
end
# 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
end
|