1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00

* vm_insnhelper.c: set keyword hash on Proc/block calls.

[ruby-core:51172] [Bug #7630]

* test/ruby/test_keyword.rb: add tests for above.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@38657 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
ktsj 2012-12-30 02:06:28 +00:00
parent b0767cf8c0
commit 371c07f2a2
3 changed files with 161 additions and 23 deletions

View file

@ -1,3 +1,10 @@
Sun Dec 30 10:51:29 2012 Kazuki Tsujimoto <kazuki@callcc.net>
* vm_insnhelper.c: set keyword hash on Proc/block calls.
[ruby-core:51172] [Bug #7630]
* test/ruby/test_keyword.rb: add tests for above.
Sat Dec 29 21:57:11 2012 Keiju Ishitsuka <keiju@ishitsuka.com>
* lib/irb/completion.rb: treat rightly comletion for symbol on irb

View file

@ -135,4 +135,123 @@ class TestKeywordArguments < Test::Unit::TestCase
assert_equal(["foo", 111111], f[num: 111111])
assert_equal(["bar", 111111], f[str: "bar", num: 111111])
end
def p1
Proc.new do |str: "foo", num: 424242|
[str, num]
end
end
def test_p1
assert_equal(["foo", 424242], p1[])
assert_equal(["bar", 424242], p1[str: "bar"])
assert_equal(["foo", 111111], p1[num: 111111])
assert_equal(["bar", 111111], p1[str: "bar", num: 111111])
assert_raise(ArgumentError) { p1[str: "bar", check: true] }
assert_equal(["foo", 424242], p1["string"] )
end
def p2
Proc.new do |x, str: "foo", num: 424242|
[x, str, num]
end
end
def test_p2
assert_equal([nil, "foo", 424242], p2[])
assert_equal([:xyz, "foo", 424242], p2[:xyz])
end
def p3
Proc.new do |str: "foo", num: 424242, **h|
[str, num, h]
end
end
def test_p3
assert_equal(["foo", 424242, {}], p3[])
assert_equal(["bar", 424242, {}], p3[str: "bar"])
assert_equal(["foo", 111111, {}], p3[num: 111111])
assert_equal(["bar", 111111, {}], p3[str: "bar", num: 111111])
assert_equal(["bar", 424242, {:check=>true}], p3[str: "bar", check: true])
assert_equal(["foo", 424242, {}], p3["string"])
end
def p4
Proc.new do |str: "foo", num: 424242, **h, &blk|
[str, num, h, blk]
end
end
def test_p4
assert_equal(["foo", 424242, {}, nil], p4[])
assert_equal(["bar", 424242, {}, nil], p4[str: "bar"])
assert_equal(["foo", 111111, {}, nil], p4[num: 111111])
assert_equal(["bar", 111111, {}, nil], p4[str: "bar", num: 111111])
assert_equal(["bar", 424242, {:check=>true}, nil], p4[str: "bar", check: true])
a = p4.call {|x| x + 42 }
assert_equal(["foo", 424242, {}], a[0, 3])
assert_equal(43, a.last.call(1))
end
def p5
Proc.new do |*r, str: "foo", num: 424242, **h|
[r, str, num, h]
end
end
def test_p5
assert_equal([[], "foo", 424242, {}], p5[])
assert_equal([[], "bar", 424242, {}], p5[str: "bar"])
assert_equal([[], "foo", 111111, {}], p5[num: 111111])
assert_equal([[], "bar", 111111, {}], p5[str: "bar", num: 111111])
assert_equal([[1], "foo", 424242, {}], p5[1])
assert_equal([[1, 2], "foo", 424242, {}], p5[1, 2])
assert_equal([[1, 2, 3], "foo", 424242, {}], p5[1, 2, 3])
assert_equal([[1], "bar", 424242, {}], p5[1, str: "bar"])
assert_equal([[1, 2], "bar", 424242, {}], p5[1, 2, str: "bar"])
assert_equal([[1, 2, 3], "bar", 424242, {}], p5[1, 2, 3, str: "bar"])
end
def p6
Proc.new do |o1, o2=42, *args, p, k: :key, **kw, &b|
[o1, o2, args, p, k, kw, b]
end
end
def test_p6
assert_equal([nil, 42, [], nil, :key, {}, nil], p6[])
assert_equal([1, 42, [], 2, :key, {}, nil], p6[1, 2])
assert_equal([1, 2, [], 3, :key, {}, nil], p6[1, 2, 3])
assert_equal([1, 2, [3], 4, :key, {}, nil], p6[1, 2, 3, 4])
assert_equal([1, 2, [3, 4], 5, :key, {str: "bar"}, nil], p6[1, 2, 3, 4, 5, str: "bar"])
end
def test_proc_parameters
assert_equal([[:key, :str], [:key, :num]], p1.parameters);
assert_equal([[:opt, :x], [:key, :str], [:key, :num]], p2.parameters);
assert_equal([[:key, :str], [:key, :num], [:keyrest, :h]], p3.parameters);
assert_equal([[:key, :str], [:key, :num], [:keyrest, :h], [:block, :blk]], p4.parameters);
assert_equal([[:rest, :r], [:key, :str], [:key, :num], [:keyrest, :h]], p5.parameters);
assert_equal([[:opt, :o1], [:opt, :o2], [:rest, :args], [:opt, :p], [:key, :k],
[:keyrest, :kw], [:block, :b]], p6.parameters)
end
def m1(*args)
yield *args
end
def test_block
blk = Proc.new {|str: "foo", num: 424242| [str, num] }
assert_equal(["foo", 424242], m1(&blk))
assert_equal(["bar", 424242], m1(str: "bar", &blk))
assert_equal(["foo", 111111], m1(num: 111111, &blk))
assert_equal(["bar", 111111], m1(str: "bar", num: 111111, &blk))
end
end

View file

@ -1064,6 +1064,34 @@ vm_caller_setup_args(const rb_thread_t *th, rb_control_frame_t *cfp, rb_call_inf
}
}
static inline int
vm_callee_setup_keyword_arg(const rb_iseq_t *iseq, int argc, VALUE *orig_argv)
{
VALUE keyword_hash = Qnil;
int i, j;
if (argc > 0) keyword_hash = rb_check_hash_type(orig_argv[argc-1]);
if (!NIL_P(keyword_hash)) {
argc--;
keyword_hash = rb_hash_dup(keyword_hash);
if (iseq->arg_keyword_check) {
for (i = j = 0; i < iseq->arg_keywords; i++) {
if (st_lookup(RHASH_TBL(keyword_hash), ID2SYM(iseq->arg_keyword_table[i]), 0)) j++;
}
if (RHASH_TBL(keyword_hash)->num_entries > (unsigned int) j) {
unknown_keyword_error(iseq, keyword_hash);
}
}
}
else {
keyword_hash = rb_hash_new();
}
orig_argv[iseq->arg_keyword] = keyword_hash;
return argc;
}
static inline int
vm_callee_setup_arg_complex(rb_thread_t *th, rb_call_info_t *ci, const rb_iseq_t *iseq, VALUE *orig_argv)
{
@ -1074,29 +1102,13 @@ vm_callee_setup_arg_complex(rb_thread_t *th, rb_call_info_t *ci, const rb_iseq_t
const int orig_argc = ci->argc;
int argc = orig_argc;
VALUE *argv = orig_argv;
VALUE keyword_hash = Qnil;
rb_num_t opt_pc = 0;
th->mark_stack_len = argc + iseq->arg_size;
/* keyword argument */
if (iseq->arg_keyword != -1) {
int i, j;
if (argc > 0) keyword_hash = rb_check_hash_type(argv[argc-1]);
if (!NIL_P(keyword_hash)) {
argc--;
keyword_hash = rb_hash_dup(keyword_hash);
if (iseq->arg_keyword_check) {
for (i = j = 0; i < iseq->arg_keywords; i++) {
if (st_lookup(RHASH_TBL(keyword_hash), ID2SYM(iseq->arg_keyword_table[i]), 0)) j++;
}
if (RHASH_TBL(keyword_hash)->num_entries > (unsigned int) j) {
unknown_keyword_error(iseq, keyword_hash);
}
}
}
else {
keyword_hash = rb_hash_new();
}
argc = vm_callee_setup_keyword_arg(iseq, argc, orig_argv);
}
/* mandatory */
@ -1142,11 +1154,6 @@ vm_callee_setup_arg_complex(rb_thread_t *th, rb_call_info_t *ci, const rb_iseq_t
argc = 0;
}
/* keyword argument */
if (iseq->arg_keyword != -1) {
orig_argv[iseq->arg_keyword] = keyword_hash;
}
/* block arguments */
if (iseq->arg_block != -1) {
VALUE blockval = Qnil;
@ -2085,6 +2092,11 @@ vm_yield_setup_block_args(rb_thread_t *th, const rb_iseq_t * iseq,
th->mark_stack_len = argc;
/* keyword argument */
if (iseq->arg_keyword != -1) {
argc = vm_callee_setup_keyword_arg(iseq, argc, argv);
}
/*
* yield [1, 2]
* => {|a|} => a = [1, 2]