From dba85b17910c8778413c0552f62912bcb6368780 Mon Sep 17 00:00:00 2001
From: tasuku-lab280 <56028684+tasuku-lab280@users.noreply.github.com>
Date: Thu, 21 Nov 2024 16:44:16 +0900
Subject: [PATCH] add ActiveRecord::Calculations module to ActiveRecord::Base

---
 gems/activerecord/6.0/_test/activerecord-generated.rb | 1 +
 gems/activerecord/6.0/activerecord.rbs                | 1 +
 2 files changed, 2 insertions(+)

diff --git a/gems/activerecord/6.0/_test/activerecord-generated.rb b/gems/activerecord/6.0/_test/activerecord-generated.rb
index df7eaa0e..42dfd7e7 100644
--- a/gems/activerecord/6.0/_test/activerecord-generated.rb
+++ b/gems/activerecord/6.0/_test/activerecord-generated.rb
@@ -30,6 +30,7 @@ class Article < ActiveRecord::Base
 User.preload(:address, friends: [:address, { followers: :users }]) # steep:ignore FallbackAny
 User.in_order_of(:id, [1, 5, 3])
 User.offset(5).limit(10)
+User.count(:id)
 
 t = User.arel_table
 User.limit(10).select(:id, "name", t[:age].as("years"), t[:email])
diff --git a/gems/activerecord/6.0/activerecord.rbs b/gems/activerecord/6.0/activerecord.rbs
index 043ab19e..66fe0351 100644
--- a/gems/activerecord/6.0/activerecord.rbs
+++ b/gems/activerecord/6.0/activerecord.rbs
@@ -17,6 +17,7 @@ module ActiveRecord
     extend ::ActiveRecord::AttributeMethods::Write::ClassMethods
     extend ::ActiveRecord::Attributes::ClassMethods
     extend ::ActiveRecord::AutosaveAssociation::ClassMethods
+    extend ::ActiveRecord::Calculations
     extend ::ActiveRecord::Core::ClassMethods
     extend ::ActiveRecord::CounterCache::ClassMethods
     extend ::ActiveRecord::Inheritance::ClassMethods