sorbet-rails 0.6.3 → 0.7.1

data/spec/test_data/v5.1/expected_wand.rbi

@@ -3,8 +3,6 @@
 
 # typed: strong
 module Wand::EnumInstanceMethods
-  extend T::Sig
-
   sig { returns(T::Boolean) }
   def phoenix_feather?; end
 
@@ -36,8 +34,6 @@ module Wand::ActiveRelation_WhereNot
 end
 
 module Wand::GeneratedAttributeMethods
-  extend T::Sig
-
   sig { returns(T::Boolean) }
   def broken; end
 
@@ -167,11 +163,18 @@ class Wand::CoreType < T::Enum
 end
 
 module Wand::GeneratedAssociationMethods
-  extend T::Sig
-
   sig { returns(::Wizard) }
   def wizard; end
 
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::Wizard).void)).returns(::Wizard) }
+  def build_wizard(*args, &block); end
+
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::Wizard).void)).returns(::Wizard) }
+  def create_wizard(*args, &block); end
+
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::Wizard).void)).returns(::Wizard) }
+  def create_wizard!(*args, &block); end
+
   sig { params(value: ::Wizard).void }
   def wizard=(value); end
 end
@@ -198,8 +201,7 @@ class Wand < ApplicationRecord
   include Wand::GeneratedAttributeMethods
   include Wand::GeneratedAssociationMethods
   extend Wand::CustomFinderMethods
-  extend T::Sig
-  extend T::Generic
+  extend Wand::QueryMethodsReturningRelation
   RelationType = T.type_alias { T.any(Wand::ActiveRecord_Relation, Wand::ActiveRecord_Associations_CollectionProxy, Wand::ActiveRecord_AssociationRelation) }
 
   sig { returns(T::Hash[T.any(String, Symbol), Integer]) }
@@ -217,122 +219,85 @@ class Wand < ApplicationRecord
   sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
   def self.unicorn_tail_hair(*args); end
 
-  sig { returns(Wand::ActiveRecord_Relation) }
-  def self.all; end
-
-  sig { params(block: T.nilable(T.proc.void)).returns(Wand::ActiveRecord_Relation) }
-  def self.unscoped(&block); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.select(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.order(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.reorder(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.group(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.limit(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.offset(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.joins(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.left_joins(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.left_outer_joins(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.where(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.rewhere(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.preload(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.eager_load(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.includes(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.from(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.lock(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.readonly(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.or(*args); end
+  sig { returns(T.nilable(Wand::CoreType)) }
+  def typed_core_type; end
 
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.having(*args); end
+  sig { params(value: T.nilable(Wand::CoreType)).void }
+  def typed_core_type=(value); end
 
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.create_with(*args); end
+  sig { returns(T::Array[Wand]) }
+  def self.mythicals; end
+end
 
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.distinct(*args); end
+class Wand::ActiveRecord_Relation < ActiveRecord::Relation
+  include Wand::ActiveRelation_WhereNot
+  include Wand::CustomFinderMethods
+  include Wand::QueryMethodsReturningRelation
+  Elem = type_member(fixed: Wand)
 
   sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.references(*args); end
+  def basilisk_horn(*args); end
 
   sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.none(*args); end
+  def dragon_heartstring(*args); end
 
   sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.unscope(*args); end
+  def phoenix_feather(*args); end
 
   sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.merge(*args); end
+  def unicorn_tail_hair(*args); end
+end
 
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
-  def self.except(*args); end
+class Wand::ActiveRecord_AssociationRelation < ActiveRecord::AssociationRelation
+  include Wand::ActiveRelation_WhereNot
+  include Wand::CustomFinderMethods
+  include Wand::QueryMethodsReturningAssociationRelation
+  Elem = type_member(fixed: Wand)
 
-  sig { params(args: T.untyped, block: T.nilable(T.proc.void)).returns(Wand::ActiveRecord_Relation) }
-  def self.extending(*args, &block); end
+  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
+  def basilisk_horn(*args); end
 
-  sig { returns(T.nilable(Wand::CoreType)) }
-  def typed_core_type; end
+  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
+  def dragon_heartstring(*args); end
 
-  sig { params(value: T.nilable(Wand::CoreType)).void }
-  def typed_core_type=(value); end
+  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
+  def phoenix_feather(*args); end
 
-  sig { returns(T::Array[Wand]) }
-  def self.mythicals; end
+  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
+  def unicorn_tail_hair(*args); end
 end
 
-class Wand::ActiveRecord_Relation < ActiveRecord::Relation
-  include Wand::ActiveRelation_WhereNot
+class Wand::ActiveRecord_Associations_CollectionProxy < ActiveRecord::Associations::CollectionProxy
   include Wand::CustomFinderMethods
-  extend T::Sig
-  extend T::Generic
+  include Wand::QueryMethodsReturningAssociationRelation
   Elem = type_member(fixed: Wand)
 
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
+  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
   def basilisk_horn(*args); end
 
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
+  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
   def dragon_heartstring(*args); end
 
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
+  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
   def phoenix_feather(*args); end
 
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_Relation) }
+  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
   def unicorn_tail_hair(*args); end
 
+  sig { params(records: T.any(Wand, T::Array[Wand])).returns(T.self_type) }
+  def <<(*records); end
+
+  sig { params(records: T.any(Wand, T::Array[Wand])).returns(T.self_type) }
+  def append(*records); end
+
+  sig { params(records: T.any(Wand, T::Array[Wand])).returns(T.self_type) }
+  def push(*records); end
+
+  sig { params(records: T.any(Wand, T::Array[Wand])).returns(T.self_type) }
+  def concat(*records); end
+end
+
+module Wand::QueryMethodsReturningRelation
   sig { returns(Wand::ActiveRecord_Relation) }
   def all; end
 
@@ -419,137 +384,25 @@ class Wand::ActiveRecord_Relation < ActiveRecord::Relation
 
   sig { params(args: T.untyped, block: T.nilable(T.proc.void)).returns(Wand::ActiveRecord_Relation) }
   def extending(*args, &block); end
-end
-
-class Wand::ActiveRecord_AssociationRelation < ActiveRecord::AssociationRelation
-  include Wand::ActiveRelation_WhereNot
-  include Wand::CustomFinderMethods
-  extend T::Sig
-  extend T::Generic
-  Elem = type_member(fixed: Wand)
 
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def basilisk_horn(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def dragon_heartstring(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def phoenix_feather(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def unicorn_tail_hair(*args); end
-
-  sig { returns(Wand::ActiveRecord_AssociationRelation) }
-  def all; end
-
-  sig { params(block: T.nilable(T.proc.void)).returns(Wand::ActiveRecord_AssociationRelation) }
-  def unscoped(&block); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def select(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def order(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def reorder(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def group(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def limit(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def offset(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def joins(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def left_joins(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def left_outer_joins(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def where(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def rewhere(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def preload(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def eager_load(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def includes(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def from(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def lock(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def readonly(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def or(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def having(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def create_with(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def distinct(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def references(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def none(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def unscope(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def merge(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def except(*args); end
-
-  sig { params(args: T.untyped, block: T.nilable(T.proc.void)).returns(Wand::ActiveRecord_AssociationRelation) }
-  def extending(*args, &block); end
+  sig do
+    params(
+      of: T.nilable(Integer),
+      start: T.nilable(Integer),
+      finish: T.nilable(Integer),
+      load: T.nilable(T::Boolean),
+      error_on_ignore: T.nilable(T::Boolean),
+      block: T.nilable(T.proc.params(e: Wand::ActiveRecord_Relation).void)
+    ).returns(ActiveRecord::Batches::BatchEnumerator)
+  end
+  def in_batches(of: 1000, start: nil, finish: nil, load: false, error_on_ignore: nil, &block); end
 end
 
-class Wand::ActiveRecord_Associations_CollectionProxy < ActiveRecord::Associations::CollectionProxy
-  include Wand::CustomFinderMethods
-  extend T::Sig
-  extend T::Generic
-  Elem = type_member(fixed: Wand)
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def basilisk_horn(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def dragon_heartstring(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def phoenix_feather(*args); end
-
-  sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
-  def unicorn_tail_hair(*args); end
-
+module Wand::QueryMethodsReturningAssociationRelation
   sig { returns(Wand::ActiveRecord_AssociationRelation) }
   def all; end
 
-  sig { params(block: T.nilable(T.proc.void)).returns(Wand::ActiveRecord_AssociationRelation) }
+  sig { params(block: T.nilable(T.proc.void)).returns(Wand::ActiveRecord_Relation) }
   def unscoped(&block); end
 
   sig { params(args: T.untyped).returns(Wand::ActiveRecord_AssociationRelation) }
@@ -633,15 +486,15 @@ class Wand::ActiveRecord_Associations_CollectionProxy < ActiveRecord::Associatio
   sig { params(args: T.untyped, block: T.nilable(T.proc.void)).returns(Wand::ActiveRecord_AssociationRelation) }
   def extending(*args, &block); end
 
-  sig { params(records: T.any(Wand, T::Array[Wand])).returns(T.self_type) }
-  def <<(*records); end
-
-  sig { params(records: T.any(Wand, T::Array[Wand])).returns(T.self_type) }
-  def append(*records); end
-
-  sig { params(records: T.any(Wand, T::Array[Wand])).returns(T.self_type) }
-  def push(*records); end
-
-  sig { params(records: T.any(Wand, T::Array[Wand])).returns(T.self_type) }
-  def concat(*records); end
+  sig do
+    params(
+      of: T.nilable(Integer),
+      start: T.nilable(Integer),
+      finish: T.nilable(Integer),
+      load: T.nilable(T::Boolean),
+      error_on_ignore: T.nilable(T::Boolean),
+      block: T.nilable(T.proc.params(e: Wand::ActiveRecord_AssociationRelation).void)
+    ).returns(ActiveRecord::Batches::BatchEnumerator)
+  end
+  def in_batches(of: 1000, start: nil, finish: nil, load: false, error_on_ignore: nil, &block); end
 end

data/spec/test_data/v5.1/expected_wizard.rbi

@@ -3,8 +3,6 @@
 
 # typed: strong
 module Wizard::EnumInstanceMethods
-  extend T::Sig
-
   sig { returns(T::Boolean) }
   def broom_nimbus?; end
 
@@ -114,8 +112,6 @@ module Wizard::ActiveRelation_WhereNot
 end
 
 module Wizard::GeneratedAttributeMethods
-  extend T::Sig
-
   sig { returns(T::Boolean) }
   def broom?; end
 
@@ -293,11 +289,18 @@ class Wizard::QuidditchPosition < T::Enum
 end
 
 module Wizard::GeneratedAssociationMethods
-  extend T::Sig
-
   sig { returns(T.nilable(::School)) }
   def school; end
 
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::School).void)).returns(::School) }
+  def build_school(*args, &block); end
+
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::School).void)).returns(::School) }
+  def create_school(*args, &block); end
+
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::School).void)).returns(::School) }
+  def create_school!(*args, &block); end
+
   sig { params(value: T.nilable(::School)).void }
   def school=(value); end
 
@@ -322,6 +325,15 @@ module Wizard::GeneratedAssociationMethods
   sig { returns(T.nilable(::Wand)) }
   def wand; end
 
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::Wand).void)).returns(::Wand) }
+  def build_wand(*args, &block); end
+
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::Wand).void)).returns(::Wand) }
+  def create_wand(*args, &block); end
+
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::Wand).void)).returns(::Wand) }
+  def create_wand!(*args, &block); end
+
   sig { params(value: T.nilable(::Wand)).void }
   def wand=(value); end
 end
@@ -348,8 +360,7 @@ class Wizard < ApplicationRecord
   include Wizard::GeneratedAttributeMethods
   include Wizard::GeneratedAssociationMethods
   extend Wizard::CustomFinderMethods
-  extend T::Sig
-  extend T::Generic
+  extend Wizard::QueryMethodsReturningRelation
   RelationType = T.type_alias { T.any(Wizard::ActiveRecord_Relation, Wizard::ActiveRecord_Associations_CollectionProxy, Wizard::ActiveRecord_AssociationRelation) }
 
   sig { returns(T::Hash[T.any(String, Symbol), String]) }
@@ -424,93 +435,6 @@ class Wizard < ApplicationRecord
   sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def self.recent(*args); end
 
-  sig { returns(Wizard::ActiveRecord_Relation) }
-  def self.all; end
-
-  sig { params(block: T.nilable(T.proc.void)).returns(Wizard::ActiveRecord_Relation) }
-  def self.unscoped(&block); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.select(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.order(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.reorder(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.group(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.limit(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.offset(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.joins(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.left_joins(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.left_outer_joins(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.where(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.rewhere(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.preload(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.eager_load(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.includes(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.from(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.lock(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.readonly(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.or(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.having(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.create_with(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.distinct(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.references(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.none(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.unscope(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.merge(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def self.except(*args); end
-
-  sig { params(args: T.untyped, block: T.nilable(T.proc.void)).returns(Wizard::ActiveRecord_Relation) }
-  def self.extending(*args, &block); end
-
   sig { returns(T.nilable(Wizard::Broom)) }
   def typed_broom; end
 
@@ -551,8 +475,7 @@ end
 class Wizard::ActiveRecord_Relation < ActiveRecord::Relation
   include Wizard::ActiveRelation_WhereNot
   include Wizard::CustomFinderMethods
-  extend T::Sig
-  extend T::Generic
+  include Wizard::QueryMethodsReturningRelation
   Elem = type_member(fixed: Wizard)
 
   sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
@@ -608,100 +531,72 @@ class Wizard::ActiveRecord_Relation < ActiveRecord::Relation
 
   sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def recent(*args); end
+end
 
-  sig { returns(Wizard::ActiveRecord_Relation) }
-  def all; end
-
-  sig { params(block: T.nilable(T.proc.void)).returns(Wizard::ActiveRecord_Relation) }
-  def unscoped(&block); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def select(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def order(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def reorder(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def group(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def limit(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def offset(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def joins(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def left_joins(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def left_outer_joins(*args); end
+class Wizard::ActiveRecord_AssociationRelation < ActiveRecord::AssociationRelation
+  include Wizard::ActiveRelation_WhereNot
+  include Wizard::CustomFinderMethods
+  include Wizard::QueryMethodsReturningAssociationRelation
+  Elem = type_member(fixed: Wizard)
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def where(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def Gryffindor(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def rewhere(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def Hagrid(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def preload(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def Hufflepuff(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def eager_load(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def Ravenclaw(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def includes(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def Slytherin(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def from(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def black_hair(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def lock(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def blonde_hair(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def readonly(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def broom_firebolt(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def or(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def broom_nimbus(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def having(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def brown_hair(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def create_with(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def color_blue_eyes(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def distinct(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def color_brown_eyes(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def references(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def color_green_eyes(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def none(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def quidditch_beater(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def unscope(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def quidditch_chaser(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def merge(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def quidditch_keeper(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def except(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def quidditch_seeker(*args); end
 
-  sig { params(args: T.untyped, block: T.nilable(T.proc.void)).returns(Wizard::ActiveRecord_Relation) }
-  def extending(*args, &block); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def recent(*args); end
 end
 
-class Wizard::ActiveRecord_AssociationRelation < ActiveRecord::AssociationRelation
-  include Wizard::ActiveRelation_WhereNot
+class Wizard::ActiveRecord_Associations_CollectionProxy < ActiveRecord::Associations::CollectionProxy
   include Wizard::CustomFinderMethods
-  extend T::Sig
-  extend T::Generic
+  include Wizard::QueryMethodsReturningAssociationRelation
   Elem = type_member(fixed: Wizard)
 
   sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
@@ -758,158 +653,125 @@ class Wizard::ActiveRecord_AssociationRelation < ActiveRecord::AssociationRelati
   sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
   def recent(*args); end
 
-  sig { returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(records: T.any(Wizard, T::Array[Wizard])).returns(T.self_type) }
+  def <<(*records); end
+
+  sig { params(records: T.any(Wizard, T::Array[Wizard])).returns(T.self_type) }
+  def append(*records); end
+
+  sig { params(records: T.any(Wizard, T::Array[Wizard])).returns(T.self_type) }
+  def push(*records); end
+
+  sig { params(records: T.any(Wizard, T::Array[Wizard])).returns(T.self_type) }
+  def concat(*records); end
+end
+
+module Wizard::QueryMethodsReturningRelation
+  sig { returns(Wizard::ActiveRecord_Relation) }
   def all; end
 
-  sig { params(block: T.nilable(T.proc.void)).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(block: T.nilable(T.proc.void)).returns(Wizard::ActiveRecord_Relation) }
   def unscoped(&block); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def select(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def order(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def reorder(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def group(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def limit(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def offset(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def joins(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def left_joins(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def left_outer_joins(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def where(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def rewhere(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def preload(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def eager_load(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def includes(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def from(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def lock(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def readonly(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def or(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def having(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def create_with(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def distinct(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def references(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def none(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def unscope(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def merge(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
   def except(*args); end
 
-  sig { params(args: T.untyped, block: T.nilable(T.proc.void)).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(args: T.untyped, block: T.nilable(T.proc.void)).returns(Wizard::ActiveRecord_Relation) }
   def extending(*args, &block); end
-end
-
-class Wizard::ActiveRecord_Associations_CollectionProxy < ActiveRecord::Associations::CollectionProxy
-  include Wizard::CustomFinderMethods
-  extend T::Sig
-  extend T::Generic
-  Elem = type_member(fixed: Wizard)
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
-  def Gryffindor(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
-  def Hagrid(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
-  def Hufflepuff(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
-  def Ravenclaw(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
-  def Slytherin(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
-  def black_hair(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
-  def blonde_hair(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
-  def broom_firebolt(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
-  def broom_nimbus(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
-  def brown_hair(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
-  def color_blue_eyes(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
-  def color_brown_eyes(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
-  def color_green_eyes(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
-  def quidditch_beater(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
-  def quidditch_chaser(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
-  def quidditch_keeper(*args); end
 
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
-  def quidditch_seeker(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
-  def recent(*args); end
+  sig do
+    params(
+      of: T.nilable(Integer),
+      start: T.nilable(Integer),
+      finish: T.nilable(Integer),
+      load: T.nilable(T::Boolean),
+      error_on_ignore: T.nilable(T::Boolean),
+      block: T.nilable(T.proc.params(e: Wizard::ActiveRecord_Relation).void)
+    ).returns(ActiveRecord::Batches::BatchEnumerator)
+  end
+  def in_batches(of: 1000, start: nil, finish: nil, load: false, error_on_ignore: nil, &block); end
+end
 
+module Wizard::QueryMethodsReturningAssociationRelation
   sig { returns(Wizard::ActiveRecord_AssociationRelation) }
   def all; end
 
-  sig { params(block: T.nilable(T.proc.void)).returns(Wizard::ActiveRecord_AssociationRelation) }
+  sig { params(block: T.nilable(T.proc.void)).returns(Wizard::ActiveRecord_Relation) }
   def unscoped(&block); end
 
   sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
@@ -993,15 +855,15 @@ class Wizard::ActiveRecord_Associations_CollectionProxy < ActiveRecord::Associat
   sig { params(args: T.untyped, block: T.nilable(T.proc.void)).returns(Wizard::ActiveRecord_AssociationRelation) }
   def extending(*args, &block); end
 
-  sig { params(records: T.any(Wizard, T::Array[Wizard])).returns(T.self_type) }
-  def <<(*records); end
-
-  sig { params(records: T.any(Wizard, T::Array[Wizard])).returns(T.self_type) }
-  def append(*records); end
-
-  sig { params(records: T.any(Wizard, T::Array[Wizard])).returns(T.self_type) }
-  def push(*records); end
-
-  sig { params(records: T.any(Wizard, T::Array[Wizard])).returns(T.self_type) }
-  def concat(*records); end
+  sig do
+    params(
+      of: T.nilable(Integer),
+      start: T.nilable(Integer),
+      finish: T.nilable(Integer),
+      load: T.nilable(T::Boolean),
+      error_on_ignore: T.nilable(T::Boolean),
+      block: T.nilable(T.proc.params(e: Wizard::ActiveRecord_AssociationRelation).void)
+    ).returns(ActiveRecord::Batches::BatchEnumerator)
+  end
+  def in_batches(of: 1000, start: nil, finish: nil, load: false, error_on_ignore: nil, &block); end
 end