CsvBuilder uses find_each

lib/csv_builder.rb

@@ -29,7 +29,7 @@ class CsvBuilder
     CSV.generate do |csv|
       csv << headers
 
-      @collection.each do |object|
+      @collection.find_each do |object|
         csv << row(object)
       end
     end

spec/lib/csv_builder_spec.rb

@@ -2,9 +2,14 @@ require 'spec_helper'
 
 describe CsvBuilder, lib: true do
   let(:object) { double(question: :answer) }
-  let(:subject) { CsvBuilder.new([object], 'Q & A' => :question, 'Reversed' => -> (o) { o.question.to_s.reverse }) }
+  let(:fake_relation) { [object] }
+  let(:subject) { CsvBuilder.new(fake_relation, 'Q & A' => :question, 'Reversed' => -> (o) { o.question.to_s.reverse }) }
   let(:csv_data) { subject.render }
 
+  before do
+    allow(fake_relation).to receive(:find_each).and_yield(object)
+  end
+
   it 'generates a csv' do
     expect(csv_data.scan(/(,|\n)/).join).to include ",\n,"
   end