ruby-on-rails – 在Rails中重新生成夹具测试文件

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

这是我的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

用法,说我想为users表生成fixture

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

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

HTH