Как принудительно включить RAILS_ENV в задаче rake?

Для этой конкретной задачи вам нужно только изменить соединение с БД, поэтому, как указал Адам, вы можете сделать это:

namespace :db do 
  namespace :test do 
    task :reset do 
      ActiveRecord::Base.establish_connection('test')
      Rake::Task['db:drop'].invoke
      Rake::Task['db:create'].invoke
      Rake::Task['db:migrate'].invoke
      ActiveRecord::Base.establish_connection(ENV['RAILS_ENV'])  #Make sure you don't have side-effects!
    end
  end
end

Если ваша задача более сложная и вам нужны другие аспекты ENV, безопаснее всего создать новый процесс rake:

namespace :db do 
  namespace :test do 
    task :reset do 
      system("rake db:drop RAILS_ENV=test")
      system("rake db:create RAILS_ENV=test")
      system("rake db:migrate RAILS_ENV=test")
    end
  end
end

or

namespace :db do 
  namespace :test do 
    task :reset do 
      if (ENV['RAILS_ENV'] == "test")
        Rake::Task['db:drop'].invoke
        Rake::Task['db:create'].invoke
        Rake::Task['db:migrate'].invoke
      else
        system("rake db:test:reset RAILS_ENV=test")
      end
    end
  end
end

В Rails 3 вам придется использовать

Rails.env = "test"
Rake::Task["db:drop"].invoke

вместо

RAILS_ENV = "test"
Rake::Task["db:drop"].invoke 

Другой вариант — проверить env и отказаться от продолжения:

unless Rails.env.development?
  puts "This task can only be run in development environment"
  exit
end

или спросите, действительно ли они хотят продолжить:

unless Rails.env.development?
  puts "You are using #{Rails.env} environment, are you sure? y/n"
  continue = STDIN.gets.chomp
  exit unless continue == 'y'
end

Самым чистым и простым решением было бы переопределить RAILS_ENV (не ENV['RAILS_ENV'])

namespace :db do
  namespace :test do  
    task :reset do 
      RAILS_ENV = "test" 
      Rake::Task['db:drop'].invoke
      Rake::Task['db:create'].invoke
      Rake::Task['db:migrate'].invoke
    end
  end
end

Во время процесса загрузки приложения Rails RAILS_ENV инициализируется следующим образом.

RAILS_ENV = (ENV['RAILS_ENV'] || 'development').dup unless defined?(RAILS_ENV)

Остальной код Rails использует RAILS_ENV напрямую.

Однако, как указал Майкл в комментарии к своему ответу, переключение RAILS_ENV на лету может быть рискованным. Другим подходом было бы переключение соединения с базой данных, это решение фактически используется db:test задачами по умолчанию.

ActiveRecord::Base.establish_connection(:test)

Лучший способ, конечно, указать среду из командной строки при запуске задачи rake, но если по какой-то причине это не то, что вы хотите сделать, вы можете сделать это:

ENV["RAILS_ENV"] = 'test'
RAILS_ENV.replace('test') if defined?(RAILS_ENV)

load "#{RAILS_ROOT}/config/environment.rb"

И это должно помочь.

В database_tasks.rb есть какой-то странный код:

  def each_current_configuration(environment)
    environments = [environment]
    environments << 'test' if environment == 'development'

    configurations = ActiveRecord::Base.configurations.values_at(*environments)
    configurations.compact.each do |configuration|
      yield configuration unless configuration['database'].blank?
    end
  end

Он всегда добавляет test, если env равно development. Я решил, что хочу выполнить пользовательскую задачу db:rebuild для одновременного выполнения development и test, запустив сначала development, а затем test. Кроме того, перед запуском задач я вызываю свой метод set_env, который гарантирует установку ActiveRecord::Tasks::DatabaseTasks.env, без этого соединения с базой данных, похоже, не обрабатываются дискретно для сред, как ожидалось. Я пробовал все другие виды отключения и т. Д., Но это сработало без дополнительного кода.

def set_env(env)
  Rails.env = env.to_s
  ENV['RAILS_ENV'] = env.to_s
  ActiveRecord::Tasks::DatabaseTasks.env = env.to_s
end

Вот суть моего полного файла db.rake с одновременной мультисредой db:rebuild и db:truncate