mirror of
https://github.com/LaCasemate/fab-manager.git
synced 2024-11-29 10:24:20 +01:00
34 lines
1.3 KiB
Ruby
34 lines
1.3 KiB
Ruby
# frozen_string_literal: false
|
|
|
|
namespace :db do
|
|
# Usage example:
|
|
# RAILS_ENV=test rails db:to_fixtures[chained_elements]
|
|
desc 'Convert development DB to Rails test fixtures'
|
|
task :to_fixtures, [:table] => :environment do |_task, args|
|
|
tables_to_skip = %w[ar_internal_metadata delayed_jobs schema_info schema_migrations].freeze
|
|
|
|
begin
|
|
ActiveRecord::Base.establish_connection
|
|
ActiveRecord::Base.connection.tables.each do |table_name|
|
|
next if tables_to_skip.include?(table_name)
|
|
next if args.table && args.table != table_name
|
|
|
|
counter = '000'
|
|
file_path = Rails.root.join("test/fixtures/#{table_name}.yml")
|
|
File.open(file_path, File::WRONLY | File::CREAT) do |file|
|
|
rows = ActiveRecord::Base.connection.select_all("SELECT * FROM #{table_name}")
|
|
data = rows.each_with_object({}) do |record, hash|
|
|
suffix = record['id'].presence || counter.succ!
|
|
# FIXME, this is broken with jsonb columns: it records a String but a Hash must be saved
|
|
hash["#{table_name.singularize}#{suffix}"] = record
|
|
end
|
|
puts "Writing table '#{table_name}' to '#{file_path}'"
|
|
file.write(data.to_yaml)
|
|
end
|
|
end
|
|
ensure
|
|
ActiveRecord::Base.connection&.close
|
|
end
|
|
end
|
|
end
|