From 68470bef862f14501ff0d8a3a6ac78ea38ab383b Mon Sep 17 00:00:00 2001 From: "Yukihiro \"Matz\" Matsumoto" Date: Fri, 11 Aug 2017 09:21:09 +0900 Subject: Check method existence in `to_enum'; ref #3773 --- mrbgems/mruby-enum-lazy/mrblib/lazy.rb | 3 +++ 1 file changed, 3 insertions(+) (limited to 'mrbgems/mruby-enum-lazy') diff --git a/mrbgems/mruby-enum-lazy/mrblib/lazy.rb b/mrbgems/mruby-enum-lazy/mrblib/lazy.rb index 8a7ac7fdc..c98681edf 100644 --- a/mrbgems/mruby-enum-lazy/mrblib/lazy.rb +++ b/mrbgems/mruby-enum-lazy/mrblib/lazy.rb @@ -43,6 +43,9 @@ class Enumerator end def to_enum(meth=:each, *args, &block) + unless self.respond_to?(meth) + raise NoMethodError, "undefined method #{meth}" + end lz = Lazy.new(self, &block) lz.obj = self lz.meth = meth -- cgit v1.2.3