Convert kwarg arg via to_hash if not a hash. procs should not strip o…

…ff kwargs arg during prepareArguments
jruby · Apr 8, 2015 · 4cdc6e9 · 4cdc6e9
1 parent 6196101
commit 4cdc6e9
Showing 3 changed files with 19 additions and 5 deletions.
diff --git a/core/src/main/java/org/jruby/RubyProc.java b/core/src/main/java/org/jruby/RubyProc.java
@@ -36,6 +36,7 @@
 
 import org.jruby.anno.JRubyClass;
 import org.jruby.anno.JRubyMethod;
+import org.jruby.ir.interpreter.Interpreter;
 import org.jruby.lexer.yacc.ISourcePosition;
 import org.jruby.lexer.yacc.SimpleSourcePosition;
 import org.jruby.parser.StaticScope;
@@ -44,7 +45,9 @@
 import org.jruby.runtime.Block;
 import org.jruby.runtime.BlockBody;
 import org.jruby.runtime.ClassIndex;
+import org.jruby.runtime.CompiledIRBlockBody;
 import org.jruby.runtime.Helpers;
+import org.jruby.runtime.IRBlockBody;
 import org.jruby.runtime.InterpretedIRBlockBody;
 import org.jruby.runtime.MethodBlock;
 import org.jruby.runtime.ObjectAllocator;
@@ -266,9 +269,13 @@ public static IRubyObject[] prepareArgs(ThreadContext context, Block.Type type,
             actual = args.length;
         }
 
-        // fixed arity > 0 with mismatch needs a new args array
-        if (isFixed && required > 0 && required != actual) {
+        // FIXME: This code is horrible so I named it poorly.  We add 1 if we have kwargs so we do not accidentally
+        // chomp it off of args array.  Much of this logic needs to be encapsulated as part of our transition from
+        // Arity to Signature.
+        int fudge = blockBody instanceof IRBlockBody && ((IRBlockBody) blockBody).getSignature().kwargs() ? 1 : 0;
 
+        // fixed arity > 0 with mismatch needs a new args array
+        if (isFixed && required > 0 && required+fudge != actual) {
             IRubyObject[] newArgs = Arrays.copyOf(args, required);
 
             // pad with nil

diff --git a/core/src/main/java/org/jruby/ir/runtime/IRRuntimeHelpers.java b/core/src/main/java/org/jruby/ir/runtime/IRRuntimeHelpers.java
@@ -544,7 +544,15 @@ public static RubyHash extractKwargsHash(Object[] args, int requiredArgsCount, b
 
         Object lastArg = args[args.length - 1];
 
-        return !(lastArg instanceof RubyHash) ? null : (RubyHash) lastArg;
+        if (lastArg instanceof RubyHash) return (RubyHash) lastArg;
+
+        if (((IRubyObject) lastArg).respondsTo("to_hash")) {
+            lastArg = ((IRubyObject) lastArg).callMethod(((IRubyObject) lastArg).getRuntime().getCurrentContext(), "to_hash");
+
+            if (lastArg instanceof RubyHash) return (RubyHash) lastArg;
+        }
+
+        return null;
     }
 
     public static void checkForExtraUnwantedKeywordArgs(final ThreadContext context, RubyHash keywordArgs) {

diff --git a/test/mri/excludes/TestKeywordArguments.rb b/test/mri/excludes/TestKeywordArguments.rb
@@ -1,5 +1,4 @@
-exclude :test_block_required_keyword, "needs investigation"
-exclude :test_implicit_hash_conversion, "needs investigation"
+exclude :test_implicit_hash_conversion, "JIT-only"
 exclude :test_lambda, "JIT-only"
 exclude :test_p6, "JIT-only"
 exclude :test_required_keyword_with_reserved, "needs investigation"