diff options
Diffstat (limited to 'spec/ruby/optional/capi/ext/enumerator_spec.c')
| -rw-r--r-- | spec/ruby/optional/capi/ext/enumerator_spec.c | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/spec/ruby/optional/capi/ext/enumerator_spec.c b/spec/ruby/optional/capi/ext/enumerator_spec.c index 6b08feab52..d9e34afd96 100644 --- a/spec/ruby/optional/capi/ext/enumerator_spec.c +++ b/spec/ruby/optional/capi/ext/enumerator_spec.c @@ -13,6 +13,18 @@ VALUE enumerator_spec_rb_enumeratorize(int argc, VALUE *argv, VALUE self) { } #endif +#ifdef HAVE_RB_ENUMERATORIZE_WITH_SIZE +VALUE enumerator_spec_size_fn(VALUE obj, VALUE args, VALUE anEnum) { + return INT2NUM(7); +} + +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); +} +#endif + void Init_enumerator_spec(void) { VALUE cls; cls = rb_define_class("CApiEnumeratorSpecs", rb_cObject); @@ -20,6 +32,9 @@ void Init_enumerator_spec(void) { #ifdef HAVE_RB_ENUMERATORIZE rb_define_method(cls, "rb_enumeratorize", enumerator_spec_rb_enumeratorize, -1); #endif +#ifdef HAVE_RB_ENUMERATORIZE_WITH_SIZE + rb_define_method(cls, "rb_enumeratorize_with_size", enumerator_spec_rb_enumeratorize_with_size, -1); +#endif } #ifdef __cplusplus |