使用RSpec测试CSV.generate

时间:2011-11-04 19:48:24

标签: ruby-on-rails ruby csv rspec

我在rails 3.1项目中有以下帮助器 - 我只是想知道是否有办法测试CSV.generate调用。我想说我有一些想法怎么做,但事实是我甚至不知道从哪里开始。任何想法都赞赏。

require 'csv'

module Admin::PurchasesHelper
  def csv_purchase_list
    columns = ['course', 'amount', 'first_name', 'last_name', 'contact_phone', 'contact_mobile', 'created_at']
    CSV.generate(:col_sep => ";", :row_sep => "\r\n", :headers => true, :write_headers => true, :return_headers => true) do |csv|
      csv << ["Course", "Amount", "First Name", "Last Name", "Phone", "Mobile", "Created At"]
      Purchase.all.each do |p|
        csv << columns.collect{ |name| p.send(name)}
      end
    end
  end
end

1 个答案:

答案 0 :(得分:10)

我会写的:

# helper class to get access to csv_purchase_list
class Foo
  include Admin::PurchasesHelper
end

it "should return Purchases as csv" do
  # create some objects - you have to know its values
  Factory(:purchase)
  Factory(:purchase)

  # you have to prepare file.csv with content from objects created above
  expected_csv = File.read('file.csv')
  generated_csv = Foo.new.csv_purchase_list

  # sometimes it is better to parse generated_csv (ie. when you testing other formats like json or xml
  generated_csv.should == expected_csv
end