sorbet-rails 0.6.3 → 0.7.1

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

@@ -8,8 +8,6 @@ module Wizard::HABTM_Subjects::ActiveRelation_WhereNot
 end
 
 module Wizard::HABTM_Subjects::GeneratedAttributeMethods
-  extend T::Sig
-
   sig { returns(T.nilable(Integer)) }
   def subject_id; end
 
@@ -30,17 +28,33 @@ module Wizard::HABTM_Subjects::GeneratedAttributeMethods
 end
 
 module Wizard::HABTM_Subjects::GeneratedAssociationMethods
-  extend T::Sig
-
   sig { returns(T.nilable(::Wizard)) }
   def left_side; end
 
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::Wizard).void)).returns(::Wizard) }
+  def build_left_side(*args, &block); end
+
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::Wizard).void)).returns(::Wizard) }
+  def create_left_side(*args, &block); end
+
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::Wizard).void)).returns(::Wizard) }
+  def create_left_side!(*args, &block); end
+
   sig { params(value: T.nilable(::Wizard)).void }
   def left_side=(value); end
 
   sig { returns(T.nilable(::Subject)) }
   def subject; end
 
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::Subject).void)).returns(::Subject) }
+  def build_subject(*args, &block); end
+
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::Subject).void)).returns(::Subject) }
+  def create_subject(*args, &block); end
+
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::Subject).void)).returns(::Subject) }
+  def create_subject!(*args, &block); end
+
   sig { params(value: T.nilable(::Subject)).void }
   def subject=(value); end
 end
@@ -66,105 +80,11 @@ class Wizard::HABTM_Subjects < ActiveRecord::Base
   include Wizard::HABTM_Subjects::GeneratedAttributeMethods
   include Wizard::HABTM_Subjects::GeneratedAssociationMethods
   extend Wizard::HABTM_Subjects::CustomFinderMethods
-  extend T::Sig
-  extend T::Generic
+  extend Wizard::HABTM_Subjects::QueryMethodsReturningRelation
   RelationType = T.type_alias { T.any(Wizard::HABTM_Subjects::ActiveRecord_Relation, Wizard::HABTM_Subjects::ActiveRecord_Associations_CollectionProxy, Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-
-  sig { returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.all; end
-
-  sig { params(block: T.nilable(T.proc.void)).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.unscoped(&block); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.select(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.order(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.reorder(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.group(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.limit(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.offset(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.joins(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.left_joins(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.left_outer_joins(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.where(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.rewhere(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.preload(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.eager_load(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.includes(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.from(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.lock(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.readonly(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.or(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.having(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.create_with(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.distinct(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.references(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.none(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.unscope(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.merge(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.except(*args); end
-
-  sig { params(args: T.untyped, block: T.nilable(T.proc.void)).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
-  def self.extending(*args, &block); end
 end
 
-class Wizard::HABTM_Subjects::ActiveRecord_Relation < ActiveRecord::Relation
-  include Wizard::HABTM_Subjects::ActiveRelation_WhereNot
-  include Wizard::HABTM_Subjects::CustomFinderMethods
-  extend T::Sig
-  extend T::Generic
-  Elem = type_member(fixed: Wizard::HABTM_Subjects)
-
+module Wizard::HABTM_Subjects::QueryMethodsReturningRelation
   sig { returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
   def all; end
 
@@ -251,19 +171,25 @@ class Wizard::HABTM_Subjects::ActiveRecord_Relation < ActiveRecord::Relation
 
   sig { params(args: T.untyped, block: T.nilable(T.proc.void)).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
   def extending(*args, &block); end
-end
 
-class Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation < ActiveRecord::AssociationRelation
-  include Wizard::HABTM_Subjects::ActiveRelation_WhereNot
-  include Wizard::HABTM_Subjects::CustomFinderMethods
-  extend T::Sig
-  extend T::Generic
-  Elem = type_member(fixed: Wizard::HABTM_Subjects)
+  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::HABTM_Subjects::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::HABTM_Subjects::QueryMethodsReturningAssociationRelation
   sig { returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
   def all; end
 
-  sig { params(block: T.nilable(T.proc.void)).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
+  sig { params(block: T.nilable(T.proc.void)).returns(Wizard::HABTM_Subjects::ActiveRecord_Relation) }
   def unscoped(&block); end
 
   sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
@@ -346,100 +272,38 @@ class Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation < ActiveRecord::A
 
   sig { params(args: T.untyped, block: T.nilable(T.proc.void)).returns(Wizard::HABTM_Subjects::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: Wizard::HABTM_Subjects::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
 
-class Wizard::HABTM_Subjects::ActiveRecord_Associations_CollectionProxy < ActiveRecord::Associations::CollectionProxy
+class Wizard::HABTM_Subjects::ActiveRecord_Relation < ActiveRecord::Relation
+  include Wizard::HABTM_Subjects::ActiveRelation_WhereNot
   include Wizard::HABTM_Subjects::CustomFinderMethods
-  extend T::Sig
-  extend T::Generic
+  include Wizard::HABTM_Subjects::QueryMethodsReturningRelation
   Elem = type_member(fixed: Wizard::HABTM_Subjects)
+end
 
-  sig { returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def all; end
-
-  sig { params(block: T.nilable(T.proc.void)).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def unscoped(&block); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def select(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def order(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def reorder(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def group(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def limit(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def offset(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def joins(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def left_joins(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def left_outer_joins(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def where(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def rewhere(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def preload(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def eager_load(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def includes(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def from(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def lock(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def readonly(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def or(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def having(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def create_with(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def distinct(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def references(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def none(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def unscope(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def merge(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def except(*args); end
+class Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation < ActiveRecord::AssociationRelation
+  include Wizard::HABTM_Subjects::ActiveRelation_WhereNot
+  include Wizard::HABTM_Subjects::CustomFinderMethods
+  include Wizard::HABTM_Subjects::QueryMethodsReturningAssociationRelation
+  Elem = type_member(fixed: Wizard::HABTM_Subjects)
+end
 
-  sig { params(args: T.untyped, block: T.nilable(T.proc.void)).returns(Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation) }
-  def extending(*args, &block); end
+class Wizard::HABTM_Subjects::ActiveRecord_Associations_CollectionProxy < ActiveRecord::Associations::CollectionProxy
+  include Wizard::HABTM_Subjects::CustomFinderMethods
+  include Wizard::HABTM_Subjects::QueryMethodsReturningAssociationRelation
+  Elem = type_member(fixed: Wizard::HABTM_Subjects)
 
   sig { params(records: T.any(Wizard::HABTM_Subjects, T::Array[Wizard::HABTM_Subjects])).returns(T.self_type) }
   def <<(*records); end

data/spec/test_data/v5.1/expected_wizard_wo_spellbook.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(T.untyped)) }
   def school; end
 
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: T.untyped).void)).returns(T.untyped) }
+  def build_school(*args, &block); end
+
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: T.untyped).void)).returns(T.untyped) }
+  def create_school(*args, &block); end
+
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: T.untyped).void)).returns(T.untyped) }
+  def create_school!(*args, &block); end
+
   sig { params(value: T.nilable(T.untyped)).void }
   def school=(value); end
 
@@ -316,6 +319,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
@@ -342,8 +354,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]) }
@@ -418,93 +429,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
 
@@ -545,8 +469,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) }
@@ -602,100 +525,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) }
@@ -752,158 +647,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) }
@@ -987,15 +849,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