diff options
Diffstat (limited to 'spec/ruby/optional/capi/ext/enumerator_spec.c')
| -rw-r--r-- | spec/ruby/optional/capi/ext/enumerator_spec.c | 19 |
1 files changed, 12 insertions, 7 deletions
diff --git a/spec/ruby/optional/capi/ext/enumerator_spec.c b/spec/ruby/optional/capi/ext/enumerator_spec.c index 6b08feab52..917621c003 100644 --- a/spec/ruby/optional/capi/ext/enumerator_spec.c +++ b/spec/ruby/optional/capi/ext/enumerator_spec.c @@ -5,21 +5,26 @@ extern "C" { #endif -#ifdef HAVE_RB_ENUMERATORIZE VALUE enumerator_spec_rb_enumeratorize(int argc, VALUE *argv, VALUE self) { VALUE obj, meth, args; rb_scan_args(argc, argv, "2*", &obj, &meth, &args); return rb_enumeratorize(obj, meth, (int)RARRAY_LEN(args), RARRAY_PTR(args)); } -#endif -void Init_enumerator_spec(void) { - VALUE cls; - cls = rb_define_class("CApiEnumeratorSpecs", rb_cObject); +VALUE enumerator_spec_size_fn(VALUE obj, VALUE args, VALUE anEnum) { + return INT2NUM(7); +} -#ifdef HAVE_RB_ENUMERATORIZE +VALUE enumerator_spec_rb_enumeratorize_with_size(int argc, VALUE *argv, VALUE self) { + VALUE obj, meth, args; + rb_scan_args(argc, argv, "2*", &obj, &meth, &args); + return rb_enumeratorize_with_size(obj, meth, (int)RARRAY_LEN(args), RARRAY_PTR(args), enumerator_spec_size_fn); +} + +void Init_enumerator_spec(void) { + VALUE cls = rb_define_class("CApiEnumeratorSpecs", rb_cObject); rb_define_method(cls, "rb_enumeratorize", enumerator_spec_rb_enumeratorize, -1); -#endif + rb_define_method(cls, "rb_enumeratorize_with_size", enumerator_spec_rb_enumeratorize_with_size, -1); } #ifdef __cplusplus |