summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--ext/-test-/ensure_and_callcc/ensure_and_callcc.c8
-rw-r--r--test/-ext-/test_ensure_and_callcc.rb1
2 files changed, 9 insertions, 0 deletions
diff --git a/ext/-test-/ensure_and_callcc/ensure_and_callcc.c b/ext/-test-/ensure_and_callcc/ensure_and_callcc.c
index aef83769a6..1a92de69c3 100644
--- a/ext/-test-/ensure_and_callcc/ensure_and_callcc.c
+++ b/ext/-test-/ensure_and_callcc/ensure_and_callcc.c
@@ -40,11 +40,19 @@ ensure_called(VALUE self)
return rb_iv_get(rb_mEnsureAndCallcc, "@ensure_called");
}
+static VALUE
+reset(VALUE self)
+{
+ rb_iv_set(rb_mEnsureAndCallcc, "@ensure_called", INT2FIX(0));
+ return Qnil;
+}
+
void
Init_ensure_and_callcc(void)
{
rb_mEnsureAndCallcc = rb_define_module("EnsureAndCallcc");
rb_iv_set(rb_mEnsureAndCallcc, "@ensure_called", INT2FIX(0));
+ rb_define_singleton_method(rb_mEnsureAndCallcc, "reset", reset, 0);
rb_define_singleton_method(rb_mEnsureAndCallcc, "ensure_called", ensure_called, 0);
rb_define_singleton_method(rb_mEnsureAndCallcc, "require_with_ensure", require_with_ensure, 1);
}
diff --git a/test/-ext-/test_ensure_and_callcc.rb b/test/-ext-/test_ensure_and_callcc.rb
index 695ca3b646..9303a094ea 100644
--- a/test/-ext-/test_ensure_and_callcc.rb
+++ b/test/-ext-/test_ensure_and_callcc.rb
@@ -23,6 +23,7 @@ class TestEnsureAndCallcc < Test::Unit::TestCase
def test_bug20655_extension_using_rb_ensure
need_continuation
require '-test-/ensure_and_callcc'
+ EnsureAndCallcc.reset
assert_equal(0, EnsureAndCallcc.ensure_called)
EnsureAndCallcc.require_with_ensure(File.join(__dir__, 'required'))
assert_equal(1, EnsureAndCallcc.ensure_called,