Added AST tests for endless method definitions

1 files changed, 19 insertions, 0 deletions

diff --git a/test/ruby/test_ast.rb b/test/ruby/test_ast.rb
index 0d846b76e4..9c218f3ad8 100644
--- a/test/ruby/test_ast.rb
+++ b/test/ruby/test_ast.rb
@@ -255,6 +255,15 @@ class TestAst < Test::Unit::TestCase
     assert_equal(:SCOPE, defn.type)
   end
 
+  def test_defn_endless
+    node = RubyVM::AbstractSyntaxTree.parse("def a = nil")
+    _, _, body = *node.children
+    assert_equal(:DEFN, body.type)
+    mid, defn = body.children
+    assert_equal(:a, mid)
+    assert_equal(:SCOPE, defn.type)
+  end
+
   def test_defs
     node = RubyVM::AbstractSyntaxTree.parse("def a.b; end")
     _, _, body = *node.children
@@ -265,6 +274,16 @@ class TestAst < Test::Unit::TestCase
     assert_equal(:SCOPE, defn.type)
   end
 
+  def test_defs_endless
+    node = RubyVM::AbstractSyntaxTree.parse("def a.b = nil")
+    _, _, body = *node.children
+    assert_equal(:DEFS, body.type)
+    recv, mid, defn = body.children
+    assert_equal(:VCALL, recv.type)
+    assert_equal(:b, mid)
+    assert_equal(:SCOPE, defn.type)
+  end
+
   def test_dstr
     node = parse('"foo#{1}bar"')
     _, _, body = *node.children