From dc6d7cc58e78903e8309ff94c9e7112d661646ee Mon Sep 17 00:00:00 2001
From: nobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Sun, 14 May 2017 11:45:00 +0000
Subject: [PATCH] object.c: use a sized enumerator with #yield_self

* object.c (rb_obj_size): The #yield_self Enumerator instance
  always has a #count of `1`.  This provides a lazy #size of `1`
  to match the count instead of `nil`.  [Fix GH-1615]

Author:    Shannon Skipper <shannonskipper@gmail.com>

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@58714 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 object.c                 | 8 +++++++-
 test/ruby/test_object.rb | 4 +++-
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/object.c b/object.c
index 6c0aa040b9..15c6f960db 100644
--- a/object.c
+++ b/object.c
@@ -497,6 +497,12 @@ rb_obj_itself(VALUE obj)
     return obj;
 }
 
+static VALUE
+rb_obj_size(VALUE self, VALUE args, VALUE obj)
+{
+    return LONG2FIX(1);
+}
+
 /*
  *  call-seq:
  *     obj.yield_self {|_obj|...} -> an_object
@@ -510,7 +516,7 @@ rb_obj_itself(VALUE obj)
 static VALUE
 rb_obj_yield_self(VALUE obj)
 {
-    RETURN_ENUMERATOR(obj, 0, 0);
+    RETURN_SIZED_ENUMERATOR(obj, 0, 0, rb_obj_size);
     return rb_yield_values2(1, &obj);
 }
 
diff --git a/test/ruby/test_object.rb b/test/ruby/test_object.rb
index 6a1a82546a..1e952533a6 100644
--- a/test/ruby/test_object.rb
+++ b/test/ruby/test_object.rb
@@ -23,7 +23,9 @@ class TestObject < Test::Unit::TestCase
     object = Object.new
     assert_same(self, object.yield_self {self}, feature)
     assert_same(object, object.yield_self {|x| break x}, feature)
-    assert_instance_of(Enumerator, object.yield_self)
+    enum = object.yield_self
+    assert_instance_of(Enumerator, enum)
+    assert_equal(1, enum.size)
   end
 
   def test_dup