Question

我不知道从哪里开始测试这个rake任务。我需要存根吗？如果是这样，如何使用它们？任何帮助，将不胜感激。谢谢！

desc "Import CSV file"
  task :import => [:environment] do
    data = "db/data.csv" 
    headers = CSV.open(data, 'r') { |csv| csv.first }
    cs = headers[2..-1].map { |c| Model1.where(name: c).first_or_create }
    ls = Model2.find_ls

    csv_contents = CSV.read(photos)
    csv_contents.shift

    csv_contents.each do |row|
      p = Model2.where(id: row[0], f_name: row[1]).first_or_create
      p_d = FastImage.size(p.file.url(:small))
      p.update_attributes(dimensions: p_d)

      row[2..-1].each_with_index do |ls, i|
        unless ls.nil?
          ls.split(',').each { |l|
          cl = Model3.where(name: l.strip, model_1_id: cs[i].id).first_or_create
          Model4.where(p_id: p.id, model_3_id: cl.id).first_or_create
        }
      end
    end
  end
end

Answer 1

以下是我的表现：

1）快乐路径测试

这样的耙子任务很难测试。将rake任务的主体提取到一个类中：

whatever.rake

desc "Import CSV file"
  task :import => [:environment] do
    CSVImporter.new.import "db/data.csv"
  end
end

LIB / csv_importer.rb

class CsvImporter
  def import(data)
    headers = CSV.open(data, 'r') { |csv| csv.first }
    cs = headers[2..-1].map { |c| Model1.where(name: c).first_or_create }
    ls = Model2.find_ls

    csv_contents = CSV.read(photos)
    csv_contents.shift

    csv_contents.each do |row|
      p = Model2.where(id: row[0], f_name: row[1]).first_or_create
      p_d = FastImage.size(p.file.url(:small))
      p.update_attributes(dimensions: p_d)

      row[2..-1].each_with_index do |ls, i|
        unless ls.nil?
          ls.split(',').each { |l|
          cl = Model3.where(name: l.strip, model_1_id: cs[i].id).first_or_create
          Model4.where(p_id: p.id, model_3_id: cl.id).first_or_create
        }
      end
    end
  end
end

现在编写一个在测试文件上调用CSVImporter.new.import的测试很容易（这就是import将文件作为参数而非硬编码的原因）并且预计结果。如果在测试环境中导入db/data.csv是合理的，您可以根据需要在测试中执行此操作。你可能只需要这样一个测试。不需要存根。

2）边缘和错误案例

这里有很多逻辑，为了简单和速度，你想要在不创建实际模型对象的情况下进行测试。也就是说，是的，你想要存根。 Model2.find_ls和FastImage.size已经很容易存根了。让我们提取一个方法，使其他模型调用易于存根：

class CsvImporter
  def import(data)
    headers = CSV.open(data, 'r') { |csv| csv.first }
    cs = headers[2..-1].map { |c| Model1.first_or_create_with(name: c) }
    ls = Model2.find_ls

    csv_contents = CSV.read(photos)
    csv_contents.shift

    csv_contents.each do |row|
      p = Model2.first_or_create_with(id: row[0], f_name: row[1])
      p_d = FastImage.size(p.file.url(:small))
      p.update_attributes(dimensions: p_d)

      row[2..-1].each_with_index do |ls, i|
        unless ls.nil?
          ls.split(',').each { |l|
          cl = Model3.first_or_create_with(name: l.strip, model_1_id: cs[i].id)
          Model4.first_or_create_with(p_id: p.id, model_3_id: cl.id)
        }
      end
    end
  end
end

应用程序/模型/关切/ active_record_extensions.rb

module ActiveRecordExtensions
  def first_or_create_with(attributes)
    where(attributes).first_or_create
  end
end

并将模块包含在所有需要它的模型中。

现在很容易存根所有模型方法，因此您可以编写模拟您喜欢的任何数据库情况的测试。

如何测试此CSV导入rake任务？

1 个答案: