namespace :db do
namespace :fixtures do
desc 'Create YAML test fixtures from data in an existing database.
Defaults to development database. Set RAILS_ENV to override.'
task :dump => :environment do
sql = "SELECT * FROM %s"
sql << " WHERE " + ENV['WHERE'] if ENV['WHERE']
ActiveRecord::Base.connection.add_limit_offset!(sql, { :limit => ENV['LIMIT'].to_i, :offset => ENV['OFFSET'].to_i }) if ENV['LIMIT'] || ENV['OFFSET']
skip_tables = ["schema_info"]
ActiveRecord::Base.establish_connection(RAILS_ENV)
tables = ENV['TABLES'].split(',')
tables ||= (ActiveRecord::Base.connection.tables - skip_tables)
tables.each do |table_name|
i = "000"
File.open("#{RAILS_ROOT}/test/fixtures/#{table_name}.yml", 'w') do |file|
data = ActiveRecord::Base.connection.select_all(sql % table_name)
file.write data.inject({}) { |hash, record|
hash["#{table_name}_#{i.succ!}"] = record
hash
}.to_yaml
end
end
end
end
end