Dont use alias_method to make it easier to override methods in use

lib/quo/collection_backed_query.rb

@@ -31,9 +31,6 @@ def self.wrap(data = nil, props: {}, &block)
     def count
       @total_count || underlying_query.size
     end
-    # @rbs override
-    alias_method :total_count, :count
-    alias_method :size, :count
 
     # @rbs override
     def page_count
@@ -73,7 +70,9 @@ def group(*options)
     end
 
     # @rbs override
-    alias_method :includes, :preload
+    def includes(*options)
+      preload(*options)
+    end
 
     # The default implementation of `query` calls `collection` and preloads the includes, however you can also
     # override this method to return an ActiveRecord::Relation or any other query-like object as usual in a Query object.

lib/quo/query.rb

@@ -229,8 +229,15 @@ def count #: Integer
       count_query(underlying_query)
     end
 
-    alias_method :total_count, :count
-    alias_method :size, :count
+    # Total number of items without paging applied.
+    def total_count #: Integer
+      count
+    end
+
+    # Alias for count
+    def size #: Integer
+      count
+    end
 
     # Gets the actual count of elements in the page of results (assuming paging is being used, otherwise the count of
     # all results)

test/quo/collection_backed_query_test.rb

@@ -80,15 +80,17 @@ def collection
   end
 
   test "#includes" do
-    q = Quo::CollectionBackedQuery.wrap([1, 2, 3]).new
-    included = q.includes(:itself)
-    assert_equal [1, 2, 3], included.to_a
+    author = Author.create!(name: "John")
+    q = Quo::CollectionBackedQuery.wrap([author]).new
+    q = q.includes(:posts)
+    assert q.first.posts.loaded?
   end
 
   test "#preload" do
-    q = Quo::CollectionBackedQuery.wrap([1, 2, 3]).new
-    preloaded = q.preload(:itself)
-    assert_equal [1, 2, 3], preloaded.to_a
+    author = Author.create!(name: "John")
+    q = Quo::CollectionBackedQuery.wrap([author]).new
+    q = q.preload(:posts)
+    assert q.first.posts.loaded?
   end
 
   test "#first" do