diff --git a/test/test_helper.rb b/test/test_helper.rb index 05a0e19..cd8a22a 100644 --- a/test/test_helper.rb +++ b/test/test_helper.rb @@ -15,8 +15,13 @@ end def create_stub(funcs) stub = Class.new - funcs.each_pair do |k,v| - stub.define_singleton_method(k) {|*unused| return v } + if (RUBY_VERSION.split('.')[0].to_i >= 2) + funcs.each_pair {|k,v| stub.define_singleton_method(k) {|unused=nil| return v } } + else + blob = "class << stub\n" + funcs.each_pair {|k,v| blob += "def #{k.to_s}(unused=nil)\n #{v.inspect}\nend\n" } + blob += "end" + eval blob end stub end