Fixed Criteria#count to delegate to cursor (instead of result array).

CodeMonkeySteve · Jul 1, 2011 · f4a55fe · f4a55fe
1 parent d2a7568
commit f4a55fe
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lib/mongo_odm/criteria.rb b/lib/mongo_odm/criteria.rb
@@ -4,7 +4,8 @@
 module MongoODM
 
   class Criteria
-    delegate :to_xml, :to_yaml, :to_json, :include?, :first, :length, :count, :collect, :map, :each, :all?, :include?, :to => :to_a
+    delegate :to_xml, :to_yaml, :to_json, :include?, :first, :length, :collect, :map, :each, :all?, :include?, :to => :to_a
+    delegate :count, :to => :to_cursor
 
     def initialize(klass, selector = {}, opts = {})
       @_klass    = klass
@@ -65,4 +66,4 @@ def method_missing(method_name, *args, &block)
 
   end
 
-end
+end