lib/multiverse/railtie.rb in multiverse-0.1.1 vs lib/multiverse/railtie.rb in multiverse-0.1.2

- old
+ new

@@ -13,10 +13,11 @@ rake_tasks do namespace :db do task :load_config do ActiveRecord::Tasks::DatabaseTasks.migrations_paths = [Multiverse.migrate_path] ActiveRecord::Tasks::DatabaseTasks.db_dir = [Multiverse.db_dir] + Rails.application.paths["db/seeds.rb"] = ["#{Multiverse.db_dir}/seeds.rb"] end namespace :test do task load_schema: %w(db:test:purge) do begin @@ -42,11 +43,18 @@ namespace :multiverse do task :load_config do ActiveRecord::Base.establish_connection(Multiverse.record_class.connection_config) end + + task :override_config do + ActiveRecord::Tasks::DatabaseTasks.current_config = ActiveRecord::Base.configurations[Multiverse.env(ActiveRecord::Tasks::DatabaseTasks.env)] + end end Rake::Task["db:migrate:status"].enhance ["multiverse:load_config"] + Rake::Task["db:structure:dump"].enhance ["multiverse:load_config", "multiverse:override_config"] + Rake::Task["db:schema:cache:dump"].enhance ["multiverse:load_config"] + Rake::Task["db:version"].enhance ["multiverse:load_config"] end end end