lib/active_mocker/mock/relation.rb in active_mocker-1.5.2 vs lib/active_mocker/mock/relation.rb in active_mocker-1.6

@@ -2,8 +2,29 @@
 module Mock
   class Relation < Collection
 
     include Queries
 
+    def initialize(collection=[])
+      super
+      @from_limit = false
+    end
+
+    def inspect
+      entries = to_a.take(11).map!(&:inspect)
+      entries[10] = '...' if entries.size == 11
+      "#<#{self.class.name} [#{entries.join(', ')}]>"
+    end
+
+    def from_limit?
+      @from_limit
+    end
+
+    private
+
+    def set_from_limit
+      @from_limit = true
+    end
+
   end
 end
 end
\ No newline at end of file