lib/active_mocker/mock/relation.rb in active_mocker-2.5.1 vs lib/active_mocker/mock/relation.rb in active_mocker-2.5.2

@@ -1,9 +1,10 @@
 # frozen_string_literal: true
 module ActiveMocker
   class Relation < Collection
     include Queries
+    include MockableMethod
 
     def initialize(collection = [])
       super
       @from_limit = false
     end
@@ -16,16 +17,16 @@
 
     def from_limit?
       @from_limit
     end
 
+    def name
+      self.class.name
+    end
+
     private
 
     def set_from_limit
       @from_limit = true
-    end
-
-    def name
-      self.class.name
     end
   end
 end