fake_arel 0.9.4 → 0.9.5

data/fake_arel.gemspec

@@ -6,7 +6,7 @@ require 'bundler/version'
  
 Gem::Specification.new do |s|
   s.name        = "fake_arel"
-  s.version     = "0.9.4"
+  s.version     = "0.9.5"
   s.platform    = Gem::Platform::RUBY
   s.author     = "Grant Ammons"
   s.email       = ["grant@pipelinedealsco.com"]

data/lib/fake_arel/rails_3_finders.rb

@@ -58,6 +58,16 @@ module Rails3Finders
         scoped.where(where.join(" OR "))
       end
       named_scope :or, __or_fn
+
+      def self.find_each(options = {:batch_size => 1000}, &block)
+        count = self.scoped.count
+        return self.scoped if count <= options[:batch_size]
+        offset = 0
+        while offset < count
+          self.scoped(:limit => options[:batch_size], :offset => offset).each { |entry| yield entry }
+          offset += options[:batch_size]
+        end
+      end
     end
   end
 end

data/lib/fake_arel.rb

@@ -9,7 +9,7 @@ require 'fake_arel/rails_3_finders'
 require 'fake_arel/calculations'
 
 module FakeArel
-  VERSION = '0.9.4'
+  VERSION = '0.9.5'
   ActiveRecord::Base.send :include, Rails3Finders
   ActiveRecord::Base.send :include, WithScopeReplacement
 end

data/spec/fake_arel_spec.rb

@@ -187,6 +187,20 @@ describe "Fake Arel" do
     Reply.all.select {|r| r.id == 1 }.first.id.should == 1
     lambda { Reply.select(:id).first.content }.should raise_error(ActiveRecord::MissingAttributeError)
   end
+
+  it "should respond to batch_find" do
+    10.times {|i| Reply.create(:content => "This is reply number #{i}") }
+
+    count = 0
+    entries = []
+    Reply.where("content LIKE('This is reply number%')").find_each(:batch_size => 5) do |reply|
+      count += 1
+      reply.class.should == Reply
+      entries << reply
+    end
+    count.should == 10
+    entries.should == Reply.where("content LIKE('This is reply number%')")
+  end
 end
 
 describe "NameScope bug" do

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: fake_arel
 version: !ruby/object:Gem::Version 
-  hash: 51
+  hash: 49
   prerelease: false
   segments: 
   - 0
   - 9
-  - 4
-  version: 0.9.4
+  - 5
+  version: 0.9.5
 platform: ruby
 authors: 
 - Grant Ammons