Reset the counter for two consecutive runs

This commit is contained in:
Satoshi Tagomori 2024-07-31 10:15:29 +09:00 committed by Koichi Sasada
parent 7aea269b89
commit 19ec803179
Notes: git 2024-07-31 02:00:10 +00:00
2 changed files with 9 additions and 0 deletions

View File

@ -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);
}

View File

@ -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,