summaryrefslogtreecommitdiff
path: root/test/ruby/test_lambda.rb
diff options
context:
space:
mode:
Diffstat (limited to 'test/ruby/test_lambda.rb')
-rw-r--r--test/ruby/test_lambda.rb298
1 files changed, 279 insertions, 19 deletions
diff --git a/test/ruby/test_lambda.rb b/test/ruby/test_lambda.rb
index bb0861ab53..c1858a36dd 100644
--- a/test/ruby/test_lambda.rb
+++ b/test/ruby/test_lambda.rb
@@ -1,3 +1,4 @@
+# frozen_string_literal: false
require 'test/unit'
class TestLambdaParameters < Test::Unit::TestCase
@@ -9,46 +10,51 @@ class TestLambdaParameters < Test::Unit::TestCase
def test_call_simple
assert_equal(1, lambda{|a| a}.call(1))
assert_equal([1,2], lambda{|a, b| [a,b]}.call(1,2))
- assert_raises(ArgumentError) { lambda{|a|}.call(1,2) }
- assert_raises(ArgumentError) { lambda{|a|}.call() }
- assert_raises(ArgumentError) { lambda{}.call(1) }
- assert_raises(ArgumentError) { lambda{|a, b|}.call(1,2,3) }
+ assert_raise(ArgumentError) { lambda{|a|}.call(1,2) }
+ assert_raise(ArgumentError) { lambda{|a|}.call() }
+ assert_raise(ArgumentError) { lambda{}.call(1) }
+ assert_raise(ArgumentError) { lambda{|a, b|}.call(1,2,3) }
assert_equal(1, ->(a){ a }.call(1))
assert_equal([1,2], ->(a,b){ [a,b] }.call(1,2))
- assert_raises(ArgumentError) { ->(a){ }.call(1,2) }
- assert_raises(ArgumentError) { ->(a){ }.call() }
- assert_raises(ArgumentError) { ->(){ }.call(1) }
- assert_raises(ArgumentError) { ->(a,b){ }.call(1,2,3) }
+ assert_raise(ArgumentError) { ->(a){ }.call(1,2) }
+ assert_raise(ArgumentError) { ->(a){ }.call() }
+ assert_raise(ArgumentError) { ->(){ }.call(1) }
+ assert_raise(ArgumentError) { ->(a,b){ }.call(1,2,3) }
end
-end
-
-__END__
def test_lambda_as_iterator
a = 0
2.times(&->(_){ a += 1 })
- assert_equal(a, 2)
+ assert_equal(2, a)
+ assert_raise(ArgumentError) {1.times(&->(){ a += 1 })}
+ bug9605 = '[ruby-core:61468] [Bug #9605]'
+ assert_nothing_raised(ArgumentError, bug9605) {1.times(&->(n){ a += 1 })}
+ assert_equal(3, a, bug9605)
+ assert_nothing_raised(ArgumentError, bug9605) {
+ a = %w(Hi there how are you).each_with_index.detect(&->(w, i) {w.length == 3})
+ }
+ assert_equal(["how", 2], a, bug9605)
end
def test_call_rest_args
assert_equal([1,2], ->(*a){ a }.call(1,2))
assert_equal([1,2,[]], ->(a,b,*c){ [a,b,c] }.call(1,2))
- assert_raises(ArgumentError){ ->(a,*b){ }.call() }
+ assert_raise(ArgumentError){ ->(a,*b){ }.call() }
end
def test_call_opt_args
assert_equal([1,2,3,4], ->(a,b,c=3,d=4){ [a,b,c,d] }.call(1,2))
assert_equal([1,2,3,4], ->(a,b,c=0,d=4){ [a,b,c,d] }.call(1,2,3))
- assert_raises(ArgumentError){ ->(a,b=1){ }.call() }
- assert_raises(ArgumentError){ ->(a,b=1){ }.call(1,2,3) }
+ assert_raise(ArgumentError){ ->(a,b=1){ }.call() }
+ assert_raise(ArgumentError){ ->(a,b=1){ }.call(1,2,3) }
end
def test_call_rest_and_opt
assert_equal([1,2,3,[]], ->(a,b=2,c=3,*d){ [a,b,c,d] }.call(1))
assert_equal([1,2,3,[]], ->(a,b=0,c=3,*d){ [a,b,c,d] }.call(1,2))
assert_equal([1,2,3,[4,5,6]], ->(a,b=0,c=0,*d){ [a,b,c,d] }.call(1,2,3,4,5,6))
- assert_raises(ArgumentError){ ->(a,b=1,*c){ }.call() }
+ assert_raise(ArgumentError){ ->(a,b=1,*c){ }.call() }
end
def test_call_with_block
@@ -57,12 +63,266 @@ __END__
assert_equal(nil, ->(&b){ b }.call)
foo { puts "bogus block " }
assert_equal(1, ->(&b){ b.call }.call { 1 })
- b = nil
- assert_equal(1, ->(&b){ b.call }.call { 1 })
- assert_nil(b)
+ _b = nil
+ assert_equal(1, ->(&_b){ _b.call }.call { 1 })
+ assert_nil(_b)
+ end
+
+ def test_call_block_from_lambda
+ bug9605 = '[ruby-core:61470] [Bug #9605]'
+ plus = ->(x,y) {x+y}
+ assert_raise(ArgumentError, bug9605) {proc(&plus).call [1,2]}
+ end
+
+ def test_proc_inside_lambda_inside_method_return_inside_lambda_inside_method
+ def self.a
+ -> do
+ p = Proc.new{return :a}
+ p.call
+ end.call
+ end
+ assert_equal(:a, a)
+
+ def self.b
+ lambda do
+ p = Proc.new{return :b}
+ p.call
+ end.call
+ end
+ assert_equal(:b, b)
+ end
+
+ def test_proc_inside_lambda_inside_method_return_inside_lambda_outside_method
+ def self.a
+ -> do
+ p = Proc.new{return :a}
+ p.call
+ end
+ end
+ assert_equal(:a, a.call)
+
+ def self.b
+ lambda do
+ p = Proc.new{return :b}
+ p.call
+ end
+ end
+ assert_equal(:b, b.call)
+ end
+
+ def test_proc_inside_lambda_inside_method_return_outside_lambda_inside_method
+ def self.a
+ -> do
+ Proc.new{return :a}
+ end.call.call
+ end
+ assert_raise(LocalJumpError) {a}
+
+ def self.b
+ lambda do
+ Proc.new{return :b}
+ end.call.call
+ end
+ assert_raise(LocalJumpError) {b}
+ end
+
+ def test_proc_inside_lambda_inside_method_return_outside_lambda_outside_method
+ def self.a
+ -> do
+ Proc.new{return :a}
+ end
+ end
+ assert_raise(LocalJumpError) {a.call.call}
+
+ def self.b
+ lambda do
+ Proc.new{return :b}
+ end
+ end
+ assert_raise(LocalJumpError) {b.call.call}
+ end
+
+ def test_proc_inside_lambda2_inside_method_return_outside_lambda1_inside_method
+ def self.a
+ -> do
+ -> do
+ Proc.new{return :a}
+ end.call.call
+ end.call
+ end
+ assert_raise(LocalJumpError) {a}
+
+ def self.b
+ lambda do
+ lambda do
+ Proc.new{return :a}
+ end.call.call
+ end.call
+ end
+ assert_raise(LocalJumpError) {b}
+ end
+
+ def test_proc_inside_lambda_toplevel
+ assert_ruby_status [], <<~RUBY
+ lambda{
+ $g = proc{ return :pr }
+ }.call
+ begin
+ $g.call
+ rescue LocalJumpError
+ # OK!
+ else
+ raise
+ end
+ RUBY
+ end
+
+ def test_instance_exec
+ bug12568 = '[ruby-core:76300] [Bug #12568]'
+ assert_nothing_raised(ArgumentError, bug12568) do
+ instance_exec([1,2,3], &->(a=[]){ a })
+ end
+ end
+
+ def test_instance_eval_return
+ bug13090 = '[ruby-core:78917] [Bug #13090] cannot return in lambdas'
+ x = :ng
+ assert_nothing_raised(LocalJumpError) do
+ x = instance_eval(&->(_){return :ok})
+ end
+ ensure
+ assert_equal(:ok, x, bug13090)
+ end
+
+ def test_instance_exec_return
+ bug13090 = '[ruby-core:78917] [Bug #13090] cannot return in lambdas'
+ x = :ng
+ assert_nothing_raised(LocalJumpError) do
+ x = instance_exec(&->(){return :ok})
+ end
+ ensure
+ assert_equal(:ok, x, bug13090)
+ end
+
+ def test_arity_error
+ assert_raise(ArgumentError, '[Bug #12705]') do
+ [1, 2].tap(&lambda {|a, b|})
+ end
end
def foo
assert_equal(nil, ->(&b){ b }.call)
end
+
+ def test_in_basic_object
+ bug5966 = '[ruby-core:42349]'
+ called = false
+ BasicObject.new.instance_eval {->() {called = true}.()}
+ assert_equal(true, called, bug5966)
+ end
+
+ def test_location_on_error
+ bug6151 = '[ruby-core:43314]'
+ called = 0
+ line, f = __LINE__, lambda do
+ called += 1
+ true
+ end
+ e = assert_raise(ArgumentError) do
+ f.call(42)
+ end
+ assert_send([e.backtrace.first, :start_with?, "#{__FILE__}:#{line}:"], bug6151)
+ assert_equal(0, called)
+ e = assert_raise(ArgumentError) do
+ 42.times(&f)
+ end
+ assert_send([e.backtrace.first, :start_with?, "#{__FILE__}:#{line}:"], bug6151)
+ assert_equal(0, called)
+ end
+
+ def return_in_current(val)
+ 1.tap(&->(*) {return 0})
+ val
+ end
+
+ def yield_block
+ yield
+ end
+
+ def return_in_callee(val)
+ yield_block(&->(*) {return 0})
+ val
+ end
+
+ def test_return
+ feature8693 = '[ruby-core:56193] [Feature #8693]'
+ assert_equal(42, return_in_current(42), feature8693)
+ assert_equal(42, return_in_callee(42), feature8693)
+ end
+
+ def break_in_current(val)
+ 1.tap(&->(*) {break 0})
+ val
+ end
+
+ def break_in_callee(val)
+ yield_block(&->(*) {break 0})
+ val
+ end
+
+ def test_break
+ assert_equal(42, break_in_current(42))
+ assert_equal(42, break_in_callee(42))
+ end
+
+ def test_do_lambda_source_location
+ exp = [__LINE__ + 1, 10, __LINE__ + 5, 7]
+ lmd = ->(x,
+ y,
+ z) do
+ #
+ end
+ file, *loc = lmd.source_location
+ assert_match(/^#{ Regexp.quote(__FILE__) }$/, file)
+ assert_equal(exp, loc)
+ end
+
+ def test_brace_lambda_source_location
+ exp = [__LINE__ + 1, 10, __LINE__ + 5, 5]
+ lmd = ->(x,
+ y,
+ z) {
+ #
+ }
+ file, *loc = lmd.source_location
+ assert_match(/^#{ Regexp.quote(__FILE__) }$/, file)
+ assert_equal(exp, loc)
+ end
+
+ def test_not_orphan_return
+ assert_equal(42, Module.new { extend self
+ def m1(&b) b.call end; def m2(); m1(&-> { return 42 }) end }.m2)
+ assert_equal(42, Module.new { extend self
+ def m1(&b) b end; def m2(); m1(&-> { return 42 }).call end }.m2)
+ assert_equal(42, Module.new { extend self
+ def m1(&b) b end; def m2(); m1(&-> { return 42 }) end }.m2.call)
+ end
+
+ def test_not_orphan_break
+ assert_equal(42, Module.new { extend self
+ def m1(&b) b.call end; def m2(); m1(&-> { break 42 }) end }.m2)
+ assert_equal(42, Module.new { extend self
+ def m1(&b) b end; def m2(); m1(&-> { break 42 }).call end }.m2)
+ assert_equal(42, Module.new { extend self
+ def m1(&b) b end; def m2(); m1(&-> { break 42 }) end }.m2.call)
+ end
+
+ def test_not_orphan_next
+ assert_equal(42, Module.new { extend self
+ def m1(&b) b.call end; def m2(); m1(&-> { next 42 }) end }.m2)
+ assert_equal(42, Module.new { extend self
+ def m1(&b) b end; def m2(); m1(&-> { next 42 }).call end }.m2)
+ assert_equal(42, Module.new { extend self
+ def m1(&b) b end; def m2(); m1(&-> { next 42 }) end }.m2.call)
+ end
end
generated by cgit v1.2.3 (git 2.25.1) at 2026-01-18 20:01:15 +0000