From 1368ae647bc14ad97095c059ce69d8906179440c Mon Sep 17 00:00:00 2001
From: Jun Hiroe <Jun.Hiroe@gmail.com>
Date: Sun, 6 Apr 2014 01:41:18 +0900
Subject: Add args in Enumerator#inspect

---
 mrbgems/mruby-enumerator/mrblib/enumerator.rb | 10 +++++++++-
 mrbgems/mruby-enumerator/test/enumerator.rb   |  4 ++++
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/mrbgems/mruby-enumerator/mrblib/enumerator.rb b/mrbgems/mruby-enumerator/mrblib/enumerator.rb
index e1c7d7ef5..7915a8472 100644
--- a/mrbgems/mruby-enumerator/mrblib/enumerator.rb
+++ b/mrbgems/mruby-enumerator/mrblib/enumerator.rb
@@ -216,7 +216,15 @@ class Enumerator
 
   def inspect
     return "#<#{self.class}: uninitialized>" unless @obj
-    "#<#{self.class}: #{@obj}:#{@meth}>"
+
+    args = ""
+    if @args && @args.size > 0
+      args = "("
+      @args.each {|arg| args << "#{arg}, " }
+      args = args[0, args.size-2]
+      args << ")"
+    end
+    "#<#{self.class}: #{@obj}:#{@meth}#{args}>"
   end
 
   ##
diff --git a/mrbgems/mruby-enumerator/test/enumerator.rb b/mrbgems/mruby-enumerator/test/enumerator.rb
index 5954af4e0..b22ed7754 100644
--- a/mrbgems/mruby-enumerator/test/enumerator.rb
+++ b/mrbgems/mruby-enumerator/test/enumerator.rb
@@ -90,6 +90,10 @@ end
 assert 'Enumerator#inspect' do
   e = (0..10).each
   assert_equal("#<Enumerator: 0..10:each>", e.inspect)
+  e = Enumerator.new("FooObject", :foo, 1)
+  assert_equal("#<Enumerator: FooObject:foo(1)>", e.inspect)
+  e = Enumerator.new("FooObject", :foo, 1, 2, 3)
+  assert_equal("#<Enumerator: FooObject:foo(1, 2, 3)>", e.inspect)
 end
 
 assert 'Enumerator#each' do
-- 
cgit v1.2.3