sorbet-rails 0.6.5 → 0.7.3

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

@@ -111,6 +111,44 @@ module Wizard::ActiveRelation_WhereNot
   def not(opts, *rest); end
 end
 
+module Wizard::GeneratedSerializedAttributeMethods
+  sig { returns(T.nilable(T.any(T::Array[T.untyped], T::Boolean, Float, T::Hash[T.untyped, T.untyped], Integer, String))) }
+  def newt_subjects; end
+
+  sig { params(value: T.nilable(T.any(T::Array[T.untyped], T::Boolean, Float, T::Hash[T.untyped, T.untyped], Integer, String))).void }
+  def newt_subjects=(value); end
+
+  sig { returns(T::Boolean) }
+  def newt_subjects?; end
+
+  sig { returns(T.nilable(T::Hash[T.untyped, T.untyped])) }
+  def owl_results; end
+
+  sig { params(value: T.nilable(T::Hash[T.untyped, T.untyped])).void }
+  def owl_results=(value); end
+
+  sig { returns(T::Boolean) }
+  def owl_results?; end
+
+  sig { returns(T.nilable(T.any(T::Array[T.untyped], T::Boolean, Float, T::Hash[T.untyped, T.untyped], Integer, String))) }
+  def patronus_characteristics; end
+
+  sig { params(value: T.nilable(T.any(T::Array[T.untyped], T::Boolean, Float, T::Hash[T.untyped, T.untyped], Integer, String))).void }
+  def patronus_characteristics=(value); end
+
+  sig { returns(T::Boolean) }
+  def patronus_characteristics?; end
+
+  sig { returns(T.nilable(T::Array[T.untyped])) }
+  def pets; end
+
+  sig { params(value: T.nilable(T::Array[T.untyped])).void }
+  def pets=(value); end
+
+  sig { returns(T::Boolean) }
+  def pets?; end
+end
+
 module Wizard::GeneratedAttributeMethods
   sig { returns(T::Boolean) }
   def broom?; end
@@ -230,69 +268,11 @@ module Wizard::GeneratedAttributeMethods
   def updated_at?; end
 end
 
-class Wizard::Broom < T::Enum
-  enums do
-    Nimbus = new(%q{nimbus})
-    Firebolt = new(%q{firebolt})
-  end
-
-end
-
-class Wizard::EyeColor < T::Enum
-  enums do
-    Brown = new(%q{brown})
-    Green = new(%q{green})
-    Blue = new(%q{blue})
-  end
-
-end
-
-class Wizard::HairColor < T::Enum
-  enums do
-    Brown = new(%q{brown})
-    Black = new(%q{black})
-    Blonde = new(%q{blonde})
-  end
-
-end
-
-class Wizard::House < T::Enum
-  enums do
-    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(%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(%q{keeper})
-    Seeker = new(%q{seeker})
-    Beater = new(%q{beater})
-    Chaser = new(%q{chaser})
-  end
-
-end
-
 module Wizard::GeneratedAssociationMethods
   sig { returns(::ActiveStorage::Attachment::ActiveRecord_Associations_CollectionProxy) }
   def hats_attachments; end
 
-  sig { returns(T::Array[Integer]) }
+  sig { returns(T::Array[T.untyped]) }
   def hats_attachment_ids; end
 
   sig { params(value: T::Enumerable[::ActiveStorage::Attachment]).void }
@@ -301,7 +281,7 @@ module Wizard::GeneratedAssociationMethods
   sig { returns(::ActiveStorage::Blob::ActiveRecord_Associations_CollectionProxy) }
   def hats_blobs; end
 
-  sig { returns(T::Array[Integer]) }
+  sig { returns(T::Array[T.untyped]) }
   def hats_blob_ids; end
 
   sig { params(value: T::Enumerable[::ActiveStorage::Blob]).void }
@@ -310,21 +290,57 @@ module Wizard::GeneratedAssociationMethods
   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
 
+  sig { returns(T.nilable(::School)) }
+  def reload_school; end
+
   sig { returns(T.nilable(::ActiveStorage::Attachment)) }
   def school_photo_attachment; end
 
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::ActiveStorage::Attachment).void)).returns(::ActiveStorage::Attachment) }
+  def build_school_photo_attachment(*args, &block); end
+
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::ActiveStorage::Attachment).void)).returns(::ActiveStorage::Attachment) }
+  def create_school_photo_attachment(*args, &block); end
+
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::ActiveStorage::Attachment).void)).returns(::ActiveStorage::Attachment) }
+  def create_school_photo_attachment!(*args, &block); end
+
   sig { params(value: T.nilable(::ActiveStorage::Attachment)).void }
   def school_photo_attachment=(value); end
 
+  sig { returns(T.nilable(::ActiveStorage::Attachment)) }
+  def reload_school_photo_attachment; end
+
   sig { returns(T.nilable(::ActiveStorage::Blob)) }
   def school_photo_blob; end
 
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::ActiveStorage::Blob).void)).returns(::ActiveStorage::Blob) }
+  def build_school_photo_blob(*args, &block); end
+
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::ActiveStorage::Blob).void)).returns(::ActiveStorage::Blob) }
+  def create_school_photo_blob(*args, &block); end
+
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::ActiveStorage::Blob).void)).returns(::ActiveStorage::Blob) }
+  def create_school_photo_blob!(*args, &block); end
+
   sig { params(value: T.nilable(::ActiveStorage::Blob)).void }
   def school_photo_blob=(value); end
 
+  sig { returns(T.nilable(::ActiveStorage::Blob)) }
+  def reload_school_photo_blob; end
+
   sig { returns(::SpellBook::ActiveRecord_Associations_CollectionProxy) }
   def spell_books; end
 
@@ -346,8 +362,20 @@ 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
+
+  sig { returns(T.nilable(::Wand)) }
+  def reload_wand; end
 end
 
 module Wizard::CustomFinderMethods
@@ -369,6 +397,7 @@ end
 
 class Wizard < ApplicationRecord
   include Wizard::EnumInstanceMethods
+  include Wizard::GeneratedSerializedAttributeMethods
   include Wizard::GeneratedAttributeMethods
   include Wizard::GeneratedAssociationMethods
   extend Wizard::CustomFinderMethods
@@ -488,6 +517,64 @@ class Wizard < ApplicationRecord
 
   sig { params(value: T.nilable(Wizard::QuidditchPosition)).void }
   def typed_quidditch_position=(value); end
+
+  class Broom < T::Enum
+    enums do
+      Nimbus = new(%q{nimbus})
+      Firebolt = new(%q{firebolt})
+    end
+
+  end
+
+  class EyeColor < T::Enum
+    enums do
+      Brown = new(%q{brown})
+      Green = new(%q{green})
+      Blue = new(%q{blue})
+    end
+
+  end
+
+  class HairColor < T::Enum
+    enums do
+      Brown = new(%q{brown})
+      Black = new(%q{black})
+      Blonde = new(%q{blonde})
+    end
+
+  end
+
+  class House < T::Enum
+    enums do
+      Gryffindor = new(%q{Gryffindor})
+      Hufflepuff = new(%q{Hufflepuff})
+      Ravenclaw = new(%q{Ravenclaw})
+      Slytherin = new(%q{Slytherin})
+    end
+
+  end
+
+  class ProfessorEnum < T::Enum
+    enums do
+      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 QuidditchPosition < T::Enum
+    enums do
+      Keeper = new(%q{keeper})
+      Seeker = new(%q{seeker})
+      Beater = new(%q{beater})
+      Chaser = new(%q{chaser})
+    end
+
+  end
 end
 
 class Wizard::ActiveRecord_Relation < ActiveRecord::Relation
@@ -798,7 +885,7 @@ module Wizard::QueryMethodsReturningRelation
       load: T.nilable(T::Boolean),
       error_on_ignore: T.nilable(T::Boolean),
       block: T.nilable(T.proc.params(e: Wizard::ActiveRecord_Relation).void)
-    ).returns(T::Enumerable[Wizard::ActiveRecord_Relation])
+    ).returns(ActiveRecord::Batches::BatchEnumerator)
   end
   def in_batches(of: 1000, start: nil, finish: nil, load: false, error_on_ignore: nil, &block); end
 end
@@ -899,7 +986,7 @@ module Wizard::QueryMethodsReturningAssociationRelation
       load: T.nilable(T::Boolean),
       error_on_ignore: T.nilable(T::Boolean),
       block: T.nilable(T.proc.params(e: Wizard::ActiveRecord_AssociationRelation).void)
-    ).returns(T::Enumerable[Wizard::ActiveRecord_AssociationRelation])
+    ).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.2/expected_wizard/habtm_subjects.rbi

@@ -31,14 +31,38 @@ module Wizard::HABTM_Subjects::GeneratedAssociationMethods
   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(::Wizard)) }
+  def reload_left_side; 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
+
+  sig { returns(T.nilable(::Subject)) }
+  def reload_subject; end
 end
 
 module Wizard::HABTM_Subjects::CustomFinderMethods
@@ -162,7 +186,7 @@ module Wizard::HABTM_Subjects::QueryMethodsReturningRelation
       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(T::Enumerable[Wizard::HABTM_Subjects::ActiveRecord_Relation])
+    ).returns(ActiveRecord::Batches::BatchEnumerator)
   end
   def in_batches(of: 1000, start: nil, finish: nil, load: false, error_on_ignore: nil, &block); end
 end
@@ -263,7 +287,7 @@ module Wizard::HABTM_Subjects::QueryMethodsReturningAssociationRelation
       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(T::Enumerable[Wizard::HABTM_Subjects::ActiveRecord_AssociationRelation])
+    ).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.2/expected_wizard_wo_spellbook.rbi

@@ -111,6 +111,44 @@ module Wizard::ActiveRelation_WhereNot
   def not(opts, *rest); end
 end
 
+module Wizard::GeneratedSerializedAttributeMethods
+  sig { returns(T.nilable(T.any(T::Array[T.untyped], T::Boolean, Float, T::Hash[T.untyped, T.untyped], Integer, String))) }
+  def newt_subjects; end
+
+  sig { params(value: T.nilable(T.any(T::Array[T.untyped], T::Boolean, Float, T::Hash[T.untyped, T.untyped], Integer, String))).void }
+  def newt_subjects=(value); end
+
+  sig { returns(T::Boolean) }
+  def newt_subjects?; end
+
+  sig { returns(T.nilable(T::Hash[T.untyped, T.untyped])) }
+  def owl_results; end
+
+  sig { params(value: T.nilable(T::Hash[T.untyped, T.untyped])).void }
+  def owl_results=(value); end
+
+  sig { returns(T::Boolean) }
+  def owl_results?; end
+
+  sig { returns(T.nilable(T.any(T::Array[T.untyped], T::Boolean, Float, T::Hash[T.untyped, T.untyped], Integer, String))) }
+  def patronus_characteristics; end
+
+  sig { params(value: T.nilable(T.any(T::Array[T.untyped], T::Boolean, Float, T::Hash[T.untyped, T.untyped], Integer, String))).void }
+  def patronus_characteristics=(value); end
+
+  sig { returns(T::Boolean) }
+  def patronus_characteristics?; end
+
+  sig { returns(T.nilable(T::Array[T.untyped])) }
+  def pets; end
+
+  sig { params(value: T.nilable(T::Array[T.untyped])).void }
+  def pets=(value); end
+
+  sig { returns(T::Boolean) }
+  def pets?; end
+end
+
 module Wizard::GeneratedAttributeMethods
   sig { returns(T::Boolean) }
   def broom?; end
@@ -230,69 +268,11 @@ module Wizard::GeneratedAttributeMethods
   def updated_at?; end
 end
 
-class Wizard::Broom < T::Enum
-  enums do
-    Nimbus = new(%q{nimbus})
-    Firebolt = new(%q{firebolt})
-  end
-
-end
-
-class Wizard::EyeColor < T::Enum
-  enums do
-    Brown = new(%q{brown})
-    Green = new(%q{green})
-    Blue = new(%q{blue})
-  end
-
-end
-
-class Wizard::HairColor < T::Enum
-  enums do
-    Brown = new(%q{brown})
-    Black = new(%q{black})
-    Blonde = new(%q{blonde})
-  end
-
-end
-
-class Wizard::House < T::Enum
-  enums do
-    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(%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(%q{keeper})
-    Seeker = new(%q{seeker})
-    Beater = new(%q{beater})
-    Chaser = new(%q{chaser})
-  end
-
-end
-
 module Wizard::GeneratedAssociationMethods
   sig { returns(::ActiveStorage::Attachment::ActiveRecord_Associations_CollectionProxy) }
   def hats_attachments; end
 
-  sig { returns(T::Array[Integer]) }
+  sig { returns(T::Array[T.untyped]) }
   def hats_attachment_ids; end
 
   sig { params(value: T::Enumerable[::ActiveStorage::Attachment]).void }
@@ -301,7 +281,7 @@ module Wizard::GeneratedAssociationMethods
   sig { returns(::ActiveStorage::Blob::ActiveRecord_Associations_CollectionProxy) }
   def hats_blobs; end
 
-  sig { returns(T::Array[Integer]) }
+  sig { returns(T::Array[T.untyped]) }
   def hats_blob_ids; end
 
   sig { params(value: T::Enumerable[::ActiveStorage::Blob]).void }
@@ -310,21 +290,57 @@ module Wizard::GeneratedAssociationMethods
   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
 
+  sig { returns(T.nilable(T.untyped)) }
+  def reload_school; end
+
   sig { returns(T.nilable(::ActiveStorage::Attachment)) }
   def school_photo_attachment; end
 
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::ActiveStorage::Attachment).void)).returns(::ActiveStorage::Attachment) }
+  def build_school_photo_attachment(*args, &block); end
+
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::ActiveStorage::Attachment).void)).returns(::ActiveStorage::Attachment) }
+  def create_school_photo_attachment(*args, &block); end
+
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::ActiveStorage::Attachment).void)).returns(::ActiveStorage::Attachment) }
+  def create_school_photo_attachment!(*args, &block); end
+
   sig { params(value: T.nilable(::ActiveStorage::Attachment)).void }
   def school_photo_attachment=(value); end
 
+  sig { returns(T.nilable(::ActiveStorage::Attachment)) }
+  def reload_school_photo_attachment; end
+
   sig { returns(T.nilable(::ActiveStorage::Blob)) }
   def school_photo_blob; end
 
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::ActiveStorage::Blob).void)).returns(::ActiveStorage::Blob) }
+  def build_school_photo_blob(*args, &block); end
+
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::ActiveStorage::Blob).void)).returns(::ActiveStorage::Blob) }
+  def create_school_photo_blob(*args, &block); end
+
+  sig { params(args: T.untyped, block: T.nilable(T.proc.params(object: ::ActiveStorage::Blob).void)).returns(::ActiveStorage::Blob) }
+  def create_school_photo_blob!(*args, &block); end
+
   sig { params(value: T.nilable(::ActiveStorage::Blob)).void }
   def school_photo_blob=(value); end
 
+  sig { returns(T.nilable(::ActiveStorage::Blob)) }
+  def reload_school_photo_blob; end
+
   sig { returns(ActiveRecord::Associations::CollectionProxy) }
   def spell_books; end
 
@@ -340,8 +356,20 @@ 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
+
+  sig { returns(T.nilable(::Wand)) }
+  def reload_wand; end
 end
 
 module Wizard::CustomFinderMethods
@@ -363,6 +391,7 @@ end
 
 class Wizard < ApplicationRecord
   include Wizard::EnumInstanceMethods
+  include Wizard::GeneratedSerializedAttributeMethods
   include Wizard::GeneratedAttributeMethods
   include Wizard::GeneratedAssociationMethods
   extend Wizard::CustomFinderMethods
@@ -482,6 +511,64 @@ class Wizard < ApplicationRecord
 
   sig { params(value: T.nilable(Wizard::QuidditchPosition)).void }
   def typed_quidditch_position=(value); end
+
+  class Broom < T::Enum
+    enums do
+      Nimbus = new(%q{nimbus})
+      Firebolt = new(%q{firebolt})
+    end
+
+  end
+
+  class EyeColor < T::Enum
+    enums do
+      Brown = new(%q{brown})
+      Green = new(%q{green})
+      Blue = new(%q{blue})
+    end
+
+  end
+
+  class HairColor < T::Enum
+    enums do
+      Brown = new(%q{brown})
+      Black = new(%q{black})
+      Blonde = new(%q{blonde})
+    end
+
+  end
+
+  class House < T::Enum
+    enums do
+      Gryffindor = new(%q{Gryffindor})
+      Hufflepuff = new(%q{Hufflepuff})
+      Ravenclaw = new(%q{Ravenclaw})
+      Slytherin = new(%q{Slytherin})
+    end
+
+  end
+
+  class ProfessorEnum < T::Enum
+    enums do
+      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 QuidditchPosition < T::Enum
+    enums do
+      Keeper = new(%q{keeper})
+      Seeker = new(%q{seeker})
+      Beater = new(%q{beater})
+      Chaser = new(%q{chaser})
+    end
+
+  end
 end
 
 class Wizard::ActiveRecord_Relation < ActiveRecord::Relation
@@ -792,7 +879,7 @@ module Wizard::QueryMethodsReturningRelation
       load: T.nilable(T::Boolean),
       error_on_ignore: T.nilable(T::Boolean),
       block: T.nilable(T.proc.params(e: Wizard::ActiveRecord_Relation).void)
-    ).returns(T::Enumerable[Wizard::ActiveRecord_Relation])
+    ).returns(ActiveRecord::Batches::BatchEnumerator)
   end
   def in_batches(of: 1000, start: nil, finish: nil, load: false, error_on_ignore: nil, &block); end
 end
@@ -893,7 +980,7 @@ module Wizard::QueryMethodsReturningAssociationRelation
       load: T.nilable(T::Boolean),
       error_on_ignore: T.nilable(T::Boolean),
       block: T.nilable(T.proc.params(e: Wizard::ActiveRecord_AssociationRelation).void)
-    ).returns(T::Enumerable[Wizard::ActiveRecord_AssociationRelation])
+    ).returns(ActiveRecord::Batches::BatchEnumerator)
   end
   def in_batches(of: 1000, start: nil, finish: nil, load: false, error_on_ignore: nil, &block); end
 end