1berry/ruby
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Make Range#reverse_each raise TypeError if endless

Browse Source
This commit is contained in:
Kouhei Yanagita 2022-01-27 14:37:48 +09:00 committed by Yusuke Endoh
parent 0164da68c1
commit 1a16b6ffc2
2 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
range.c
View File

@ -1133,6 +1133,11 @@ range_reverse_each(VALUE range)
VALUE end = RANGE_END(range); VALUE end = RANGE_END(range);
int excl = EXCL(range); int excl = EXCL(range);
if (NIL_P(end)) {
rb_raise(rb_eTypeError, "can't iterate from %s",
rb_obj_classname(end));
}
if (FIXNUM_P(beg) && FIXNUM_P(end)) { if (FIXNUM_P(beg) && FIXNUM_P(end)) {
if (excl) { if (excl) {
if (end == LONG2FIX(FIXNUM_MIN)) return range; if (end == LONG2FIX(FIXNUM_MIN)) return range;

8
test/ruby/test_range.rb
View File

@ -582,6 +582,14 @@ class TestRange < Test::Unit::TestCase
assert_equal([fmin-2, fmin-3], a) assert_equal([fmin-2, fmin-3], a)
end end
def test_reverse_each_for_endless_range
assert_raise(TypeError) { (1..).reverse_each {} }
enum = nil
assert_nothing_raised { enum = (1..).reverse_each }
assert_raise(TypeError) { enum.each {} }
end
def test_reverse_each_for_single_point_range def test_reverse_each_for_single_point_range
fmin = RbConfig::LIMITS['FIXNUM_MIN'] fmin = RbConfig::LIMITS['FIXNUM_MIN']
fmax = RbConfig::LIMITS['FIXNUM_MAX'] fmax = RbConfig::LIMITS['FIXNUM_MAX']