Use size for Integer ranges unless it comes back nil

headius · headius · commit 79ee9204d678 · 2024-03-15T15:12:20.000-05:00
Without this we would hang on (1..2**100).count, added to
test_range.rb:test_count after 3.1.
diff --git a/core/src/main/java/org/jruby/RubyRange.java b/core/src/main/java/org/jruby/RubyRange.java
@@ -1167,6 +1167,13 @@ public IRubyObject call(ThreadContext ctx, IRubyObject larg, Block blk) {
     public IRubyObject count(ThreadContext context, Block block) {
         if (isBeginless || isEndless) return RubyFloat.newFloat(context.runtime, RubyFloat.INFINITY);
 
+        if (begin instanceof RubyInteger) {
+            IRubyObject size = size(context);
+            if (!size.isNil()) {
+                return size;
+            }
+        }
+
         return RubyEnumerable.count(context, this, block);
     }
 
