Fix #277: CollectionProxy should include Enumerable explicitly

To precise the types of enumerable methods of CollectionProxy object, this adds include statement to the generated CollectionProxy types. refs: #277
pocke · Nov 10, 2024 · 9fe353c · 9fe353c
1 parent 5f5c8f8
commit 9fe353c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/lib/rbs_rails/active_record.rb b/lib/rbs_rails/active_record.rb
@@ -85,6 +85,7 @@ class #{relation_class_name(abs: false)} < ::ActiveRecord::Relation
           class ActiveRecord_Associations_CollectionProxy < ::ActiveRecord::Associations::CollectionProxy
             include #{generated_relation_methods_name}
             include ::_ActiveRecord_Relation[#{klass_name}, #{pk_type}]
+            include ::Enumerable[#{klass_name}]
           end
         RBS
       end

diff --git a/test/expectations/user.rbs b/test/expectations/user.rbs
@@ -252,5 +252,6 @@ class User < ::ApplicationRecord
   class ActiveRecord_Associations_CollectionProxy < ::ActiveRecord::Associations::CollectionProxy
     include ::User::GeneratedRelationMethods
     include ::_ActiveRecord_Relation[::User, ::Integer]
+    include ::Enumerable[::User]
   end
 end