在Rails中重新生成夹具测试文件

时间:2012-03-28 21:12:24

标签: ruby-on-rails testing yaml fixtures

如何重新生成所有YML fixture文件?我不小心删了它们。

1 个答案:

答案 0 :(得分:7)

@布赖恩,

我正在使用以下脚本从给定的sql生成灯具

这是我的lib / task目录下的rake任务

namespace :fixture_generator do
  desc "generate fixtures for a given sql query from the current development database"

  task :fixture_generator, [:sql, :file_name] => :environment do |t, args|
    args.with_defaults(:sql => nil, :file_name => nil)
    i = "000"
    p "creating fixture - #{args.file_name}"
    File.open("#{Rails.root}/test/fixtures/#{args.file_name}.yml", 'a+') do |file|
      data = ActiveRecord::Base.connection.select_all(args.sql)
      file.write data.inject({}) { |hash, record|
        number = i.succ!
        hash["#{args.file_name}_#{number}"] = record
        hash
      }.to_yaml
    end

  end
end

用法,说我想为用户表生成夹具

rake fixture_generator:fixture_generator["select * from users","users"]

此外,如果您使用相同的fixture文件名运行另一个查询,它将附加到现有的

HTH