Allow seeding a specific tenant with db:dev_seed

Until now, running `db:dev_seed` created development data for the
default tenant but it was impossible to create this data for other
tenants.

Now the tenant can be provided as a parameter.

Note that, in order to be able to execute this task twice while running
the tests, we need to use `load` instead of `require_relative`, since
`require_relative` doesn't load the file again if it's already loaded.

Also note that having two optional parameters in a rake task results in
a cumbersome syntax to execute it. To avoid this, we're simply removing
the `print_log` parameter, which was used mainly for the test
environment. Now we use a different logic to get the same result.

From now on it won't be possible to pass the option to avoid the log in
the development environment. I don't know a developer who's ever used
this option, though, and it can always be done using `> /dev/null`.
This commit is contained in:
Javi Martín
2022-10-08 22:10:43 +02:00
parent 7189a2368e
commit a98c363d4d
3 changed files with 44 additions and 32 deletions

View File

@@ -1,11 +1,15 @@
unless Rails.env.test? unless Rails.env.test?
Tenant.destroy_all Tenant.destroy_all if Tenant.default?
ActiveRecord::Tasks::DatabaseTasks.truncate_all ActiveRecord::Tasks::DatabaseTasks.truncate_all
end end
@logger = Logger.new(STDOUT) @logger = Logger.new(STDOUT)
@logger.formatter = proc do |_severity, _datetime, _progname, msg| @logger.formatter = proc do |_severity, _datetime, _progname, msg|
msg unless @avoid_log msg unless Rails.env.test?
end
def load_dev_seeds(dev_seeds_file)
load Rails.root.join("db", "dev_seeds", "#{dev_seeds_file}.rb")
end end
def section(section_title) def section(section_title)
@@ -32,28 +36,30 @@ def random_locales_attributes(**attribute_names_with_values)
end end
end end
require_relative "dev_seeds/settings" log "Creating dev seeds for tenant #{Tenant.current_schema}" unless Tenant.default?
require_relative "dev_seeds/geozones"
require_relative "dev_seeds/users" load_dev_seeds "settings"
require_relative "dev_seeds/tags_categories" load_dev_seeds "geozones"
require_relative "dev_seeds/debates" load_dev_seeds "users"
require_relative "dev_seeds/proposals" load_dev_seeds "tags_categories"
require_relative "dev_seeds/budgets" load_dev_seeds "debates"
require_relative "dev_seeds/comments" load_dev_seeds "proposals"
require_relative "dev_seeds/votes" load_dev_seeds "budgets"
require_relative "dev_seeds/flags" load_dev_seeds "comments"
require_relative "dev_seeds/hiddings" load_dev_seeds "votes"
require_relative "dev_seeds/banners" load_dev_seeds "flags"
require_relative "dev_seeds/polls" load_dev_seeds "hiddings"
require_relative "dev_seeds/communities" load_dev_seeds "banners"
require_relative "dev_seeds/legislation_processes" load_dev_seeds "polls"
require_relative "dev_seeds/newsletters" load_dev_seeds "communities"
require_relative "dev_seeds/notifications" load_dev_seeds "legislation_processes"
require_relative "dev_seeds/widgets" load_dev_seeds "newsletters"
require_relative "dev_seeds/admin_notifications" load_dev_seeds "notifications"
require_relative "dev_seeds/legislation_proposals" load_dev_seeds "widgets"
require_relative "dev_seeds/milestones" load_dev_seeds "admin_notifications"
require_relative "dev_seeds/pages" load_dev_seeds "legislation_proposals"
require_relative "dev_seeds/sdg" load_dev_seeds "milestones"
load_dev_seeds "pages"
load_dev_seeds "sdg"
log "All dev seeds created successfuly 👍" log "All dev seeds created successfuly 👍"

View File

@@ -1,9 +1,8 @@
namespace :db do namespace :db do
desc "Resets the database and loads it from db/dev_seeds.rb" desc "Resets the database and loads it from db/dev_seeds.rb"
task :dev_seed, [:print_log] => [:environment] do |t, args| task :dev_seed, [:tenant] => [:environment] do |_, args|
@avoid_log = args[:print_log] == "avoid_log"
I18n.enforce_available_locales = false I18n.enforce_available_locales = false
load(Rails.root.join("db", "dev_seeds.rb")) Tenant.switch(args[:tenant]) { load(Rails.root.join("db", "dev_seeds.rb")) }
end end
desc "Calculates the TSV column for all comments and proposal notifications" desc "Calculates the TSV column for all comments and proposal notifications"

View File

@@ -1,11 +1,18 @@
require "rails_helper" require "rails_helper"
describe "rake db:dev_seed" do describe "rake db:dev_seed" do
let :run_rake_task do before { Rake::Task["db:dev_seed"].reenable }
Rake.application.invoke_task("db:dev_seed[avoid_log]")
end
it "seeds the database without errors" do it "seeds the database without errors" do
expect { run_rake_task }.not_to raise_error expect { Rake.application.invoke_task("db:dev_seed") }.not_to raise_error
end
it "can seed a tenant" do
create(:tenant, schema: "democracy")
Rake.application.invoke_task("db:dev_seed[democracy]")
expect(Debate.count).to eq 0
Tenant.switch("democracy") { expect(Debate.count).not_to eq 0 }
end end
end end