sorbet-rails 0.6.2 → 0.7.0

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
 
@@ -167,10 +163,10 @@ module Wizard::GeneratedAttributeMethods
   sig { returns(T::Boolean) }
   def id?; end
 
-  sig { returns(T.nilable(String)) }
+  sig { returns(String) }
   def name; end
 
-  sig { params(value: T.nilable(T.any(String, Symbol))).void }
+  sig { params(value: T.any(String, Symbol)).void }
   def name=(value); end
 
   sig { returns(T::Boolean) }
@@ -236,64 +232,63 @@ end
 
 class Wizard::Broom < T::Enum
   enums do
-    Nimbus = new('nimbus')
-    Firebolt = new('firebolt')
+    Nimbus = new(%q{nimbus})
+    Firebolt = new(%q{firebolt})
   end
 
 end
 
 class Wizard::EyeColor < T::Enum
   enums do
-    Brown = new('brown')
-    Green = new('green')
-    Blue = new('blue')
+    Brown = new(%q{brown})
+    Green = new(%q{green})
+    Blue = new(%q{blue})
   end
 
 end
 
 class Wizard::HairColor < T::Enum
   enums do
-    Brown = new('brown')
-    Black = new('black')
-    Blonde = new('blonde')
+    Brown = new(%q{brown})
+    Black = new(%q{black})
+    Blonde = new(%q{blonde})
   end
 
 end
 
 class Wizard::House < T::Enum
   enums do
-    Gryffindor = new('Gryffindor')
-    Hufflepuff = new('Hufflepuff')
-    Ravenclaw = new('Ravenclaw')
-    Slytherin = new('Slytherin')
+    Gryffindor = new(%q{Gryffindor})
+    Hufflepuff = new(%q{Hufflepuff})
+    Ravenclaw = new(%q{Ravenclaw})
+    Slytherin = new(%q{Slytherin})
   end
 
 end
 
 class Wizard::ProfessorEnum < T::Enum
   enums do
-    SeverusSnape = new('Severus Snape')
-    MinervaMcGonagall = new('Minerva McGonagall')
-    PomonaSprout = new('Pomona Sprout')
-    FiliusFlitwick = new('Filius Flitwick')
-    Hagrid = new('Hagrid')
+    SeverusSnape = new(%q{Severus Snape})
+    MinervaMcGonagall = new(%q{Minerva McGonagall})
+    PomonaSprout = new(%q{Pomona Sprout})
+    FiliusFlitwick = new(%q{Filius Flitwick})
+    Hagrid = new(%q{Hagrid})
+    AlastorMadEyeMoody = new(%q{Alastor 'Mad-Eye' Moody})
   end
 
 end
 
 class Wizard::QuidditchPosition < T::Enum
   enums do
-    Keeper = new('keeper')
-    Seeker = new('seeker')
-    Beater = new('beater')
-    Chaser = new('chaser')
+    Keeper = new(%q{keeper})
+    Seeker = new(%q{seeker})
+    Beater = new(%q{beater})
+    Chaser = new(%q{chaser})
   end
 
 end
 
 module Wizard::GeneratedAssociationMethods
-  extend T::Sig
-
   sig { returns(T.nilable(::School)) }
   def school; end
 
@@ -303,9 +298,21 @@ module Wizard::GeneratedAssociationMethods
   sig { returns(::SpellBook::ActiveRecord_Associations_CollectionProxy) }
   def spell_books; end
 
+  sig { returns(T::Array[Integer]) }
+  def spell_book_ids; end
+
   sig { params(value: T::Enumerable[::SpellBook]).void }
   def spell_books=(value); end
 
+  sig { returns(::Subject::ActiveRecord_Associations_CollectionProxy) }
+  def subjects; end
+
+  sig { returns(T::Array[Integer]) }
+  def subject_ids; end
+
+  sig { params(value: T::Enumerable[::Subject]).void }
+  def subjects=(value); end
+
   sig { returns(T.nilable(::Wand)) }
   def wand; end
 
@@ -335,8 +342,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]) }
@@ -411,93 +417,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
 
@@ -533,32 +452,12 @@ class Wizard < ApplicationRecord
 
   sig { params(value: T.nilable(Wizard::QuidditchPosition)).void }
   def typed_quidditch_position=(value); end
-
-  sig { params(conditions: T.untyped).returns(T::Boolean) }
-  def self.exists?(conditions = nil); end
-
-  sig { params(args: T.untyped).returns(T::Boolean) }
-  def self.any?(*args); end
-
-  sig { params(args: T.untyped).returns(T::Boolean) }
-  def self.many?(*args); end
-
-  sig { params(args: T.untyped).returns(T::Boolean) }
-  def self.none?(*args); end
-
-  sig { params(args: T.untyped).returns(T::Boolean) }
-  def self.one?(*args); end
-
-  sig { params(args: T.untyped).returns(T::Boolean) }
-  def self.empty?(*args); end
 end
 
 class Wizard::ActiveRecord_Relation < ActiveRecord::Relation
   include Wizard::ActiveRelation_WhereNot
   include Wizard::CustomFinderMethods
-  include Enumerable
-  extend T::Sig
-  extend T::Generic
+  include Wizard::QueryMethodsReturningRelation
   Elem = type_member(fixed: Wizard)
 
   sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
@@ -614,119 +513,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
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def where(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def rewhere(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def preload(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def eager_load(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def includes(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_Relation) }
-  def from(*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 lock(*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 readonly(*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 or(*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 having(*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 create_with(*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 distinct(*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 references(*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 none(*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 unscope(*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 merge(*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 except(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def color_blue_eyes(*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 color_brown_eyes(*args); end
 
-  sig { params(conditions: T.untyped).returns(T::Boolean) }
-  def exists?(conditions = nil); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def color_green_eyes(*args); end
 
-  sig { params(args: T.untyped).returns(T::Boolean) }
-  def any?(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def quidditch_beater(*args); end
 
-  sig { params(args: T.untyped).returns(T::Boolean) }
-  def many?(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def quidditch_chaser(*args); end
 
-  sig { params(args: T.untyped).returns(T::Boolean) }
-  def none?(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def quidditch_keeper(*args); end
 
-  sig { params(args: T.untyped).returns(T::Boolean) }
-  def one?(*args); end
+  sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
+  def quidditch_seeker(*args); end
 
-  sig { params(args: T.untyped).returns(T::Boolean) }
-  def empty?(*args); 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
-  include Enumerable
-  extend T::Sig
-  extend T::Generic
+  include Wizard::QueryMethodsReturningAssociationRelation
   Elem = type_member(fixed: Wizard)
 
   sig { params(args: T.untyped).returns(Wizard::ActiveRecord_AssociationRelation) }
@@ -783,248 +635,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
 
-  sig { params(args: T.untyped).returns(Wizard) }
-  def find(*args); end
-
-  sig { params(args: T.untyped).returns(T.nilable(Wizard)) }
-  def find_by(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard) }
-  def find_by!(*args); end
-
-  sig { params(attributes: T.untyped, block: T.nilable(T.proc.params(object: Wizard).void)).returns(Wizard) }
-  def find_or_initialize_by(attributes, &block); end
-
-  sig { params(attributes: T.untyped, block: T.nilable(T.proc.params(object: Wizard).void)).returns(Wizard) }
-  def find_or_create_by(attributes, &block); end
-
-  sig { params(attributes: T.untyped, block: T.nilable(T.proc.params(object: Wizard).void)).returns(Wizard) }
-  def find_or_create_by!(attributes, &block); end
-
-  sig { returns(T.nilable(Wizard)) }
-  def first; end
-
-  sig { returns(Wizard) }
-  def first!; end
-
-  sig { returns(T.nilable(Wizard)) }
-  def second; end
-
-  sig { returns(Wizard) }
-  def second!; end
-
-  sig { returns(T.nilable(Wizard)) }
-  def third; end
-
-  sig { returns(Wizard) }
-  def third!; end
-
-  sig { returns(T.nilable(Wizard)) }
-  def third_to_last; end
-
-  sig { returns(Wizard) }
-  def third_to_last!; end
-
-  sig { returns(T.nilable(Wizard)) }
-  def second_to_last; end
-
-  sig { returns(Wizard) }
-  def second_to_last!; end
-
-  sig { returns(T.nilable(Wizard)) }
-  def last; end
-
-  sig { returns(Wizard) }
-  def last!; end
-
-  sig { params(conditions: T.untyped).returns(T::Boolean) }
-  def exists?(conditions = nil); end
-
-  sig { params(args: T.untyped).returns(T::Boolean) }
-  def any?(*args); end
-
-  sig { params(args: T.untyped).returns(T::Boolean) }
-  def many?(*args); end
-
-  sig { params(args: T.untyped).returns(T::Boolean) }
-  def none?(*args); end
-
-  sig { params(args: T.untyped).returns(T::Boolean) }
-  def one?(*args); end
-
-  sig { params(args: T.untyped).returns(T::Boolean) }
-  def empty?(*args); end
-
-  sig { override.params(block: T.proc.params(e: Wizard).void).returns(T::Array[Wizard]) }
-  def each(&block); end
-
-  sig { params(level: T.nilable(Integer)).returns(T::Array[Wizard]) }
-  def flatten(level); end
-
-  sig { returns(T::Array[Wizard]) }
-  def to_a; end
-
   sig do
-    type_parameters(:U).params(
-        blk: T.proc.params(arg0: Elem).returns(T.type_parameter(:U)),
-    )
-    .returns(T::Array[T.type_parameter(:U)])
+    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 map(&blk); end
+  def in_batches(of: 1000, start: nil, finish: nil, load: false, error_on_ignore: nil, &block); end
 end
 
-class Wizard::ActiveRecord_Associations_CollectionProxy < ActiveRecord::Associations::CollectionProxy
-  include Wizard::CustomFinderMethods
-  include Enumerable
-  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
-
+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) }
@@ -1108,104 +837,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(args: T.untyped).returns(Wizard) }
-  def find(*args); end
-
-  sig { params(args: T.untyped).returns(T.nilable(Wizard)) }
-  def find_by(*args); end
-
-  sig { params(args: T.untyped).returns(Wizard) }
-  def find_by!(*args); end
-
-  sig { params(attributes: T.untyped, block: T.nilable(T.proc.params(object: Wizard).void)).returns(Wizard) }
-  def find_or_initialize_by(attributes, &block); end
-
-  sig { params(attributes: T.untyped, block: T.nilable(T.proc.params(object: Wizard).void)).returns(Wizard) }
-  def find_or_create_by(attributes, &block); end
-
-  sig { params(attributes: T.untyped, block: T.nilable(T.proc.params(object: Wizard).void)).returns(Wizard) }
-  def find_or_create_by!(attributes, &block); end
-
-  sig { returns(T.nilable(Wizard)) }
-  def first; end
-
-  sig { returns(Wizard) }
-  def first!; end
-
-  sig { returns(T.nilable(Wizard)) }
-  def second; end
-
-  sig { returns(Wizard) }
-  def second!; end
-
-  sig { returns(T.nilable(Wizard)) }
-  def third; end
-
-  sig { returns(Wizard) }
-  def third!; end
-
-  sig { returns(T.nilable(Wizard)) }
-  def third_to_last; end
-
-  sig { returns(Wizard) }
-  def third_to_last!; end
-
-  sig { returns(T.nilable(Wizard)) }
-  def second_to_last; end
-
-  sig { returns(Wizard) }
-  def second_to_last!; end
-
-  sig { returns(T.nilable(Wizard)) }
-  def last; end
-
-  sig { returns(Wizard) }
-  def last!; end
-
-  sig { params(conditions: T.untyped).returns(T::Boolean) }
-  def exists?(conditions = nil); end
-
-  sig { params(args: T.untyped).returns(T::Boolean) }
-  def any?(*args); end
-
-  sig { params(args: T.untyped).returns(T::Boolean) }
-  def many?(*args); end
-
-  sig { params(args: T.untyped).returns(T::Boolean) }
-  def none?(*args); end
-
-  sig { params(args: T.untyped).returns(T::Boolean) }
-  def one?(*args); end
-
-  sig { params(args: T.untyped).returns(T::Boolean) }
-  def empty?(*args); end
-
-  sig { override.params(block: T.proc.params(e: Wizard).void).returns(T::Array[Wizard]) }
-  def each(&block); end
-
-  sig { params(level: T.nilable(Integer)).returns(T::Array[Wizard]) }
-  def flatten(level); end
-
-  sig { returns(T::Array[Wizard]) }
-  def to_a; end
-
   sig do
-    type_parameters(:U).params(
-        blk: T.proc.params(arg0: Elem).returns(T.type_parameter(:U)),
-    )
-    .returns(T::Array[T.type_parameter(:U)])
+    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 map(&blk); 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
+  def in_batches(of: 1000, start: nil, finish: nil, load: false, error_on_ignore: nil, &block); end
 end