raise TypeError if meth does not respond to :for_define_method

rubinius · Feb 5, 2016 · 510919d · 510919d
1 parent 3f9756b
commit 510919d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/core/module.rb b/core/module.rb
@@ -442,7 +442,11 @@ def define_method(name, meth = undefined, &prc)
 
     name = Rubinius::Type.coerce_to_symbol name
 
-    code, scope = meth.for_define_method(name, self.class)
+    if meth.respond_to?(:for_define_method)
+      code, scope = meth.for_define_method(name, self.class)
+    else
+      raise TypeError, "wrong argument type #{meth.class} (expected Proc/Method)"
+    end
 
     Rubinius.add_method name, code, self, scope, 0, :public