[Truffle] Add BasicObject#singleton_method_undefined hook for undef_m…

…ethod
jruby · Sep 5, 2016 · c87a85d · c87a85d
1 parent edb207a
commit c87a85d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/spec/truffle/tags/core/basicobject/singleton_method_undefined_tags.txt b/spec/truffle/tags/core/basicobject/singleton_method_undefined_tags.txt
diff --git a/truffle/src/main/java/org/jruby/truffle/core/module/ModuleNodes.java b/truffle/src/main/java/org/jruby/truffle/core/module/ModuleNodes.java
@@ -1820,7 +1820,12 @@ private void undefMethod(VirtualFrame frame, DynamicObject module, String name)
             raiseIfFrozenNode.execute(frame);
 
             Layouts.MODULE.getFields(module).undefMethod(getContext(), this, name);
-            methodUndefinedNode.call(frame, module, "method_undefined", getSymbol(name));
+            if (Layouts.CLASS.isClass(module) && Layouts.CLASS.getIsSingleton(module)) {
+                final DynamicObject receiver = Layouts.CLASS.getAttached(module);
+                methodUndefinedNode.call(frame, receiver, "singleton_method_undefined", getSymbol(name));
+            } else {
+                methodUndefinedNode.call(frame, module, "method_undefined", getSymbol(name));
+            }
         }
 
     }