¿Cómo fuerzo a RAILS_ENV en una tarea de rake?

Question

tengo esta pequeña tarea rake:

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

Ahora, cuando yo haga, se ignorará la RAILS_ENV Traté de codificar. Cómo hago que esta tarea funcione como se espera

Answer 1

Para esta tarea en particular, sólo es necesario cambiar la conexión de base de datos, así como Adán señaló, se puede hacer esto:

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 

Si su tarea es más complicada, y que necesita otros aspectos de la ENV , que son más seguros genere un nuevo proceso de desprendimiento:

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 

o

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 

Answer 2

La mejor manera, por supuesto, es especificar el entorno desde la línea de comando cuando ejecuta la tarea de rake, pero si por algún motivo no es lo que quiere hacer, puede hacer esto:

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

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

Y eso debería hacer el truco.

Answer 3

el más limpio y solución más simple sería redefinir RAILS_ENV (no 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 

Durante el proceso de arranque de un RAILS_ENV aplicación Rails se inicializa como sigue

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

El resto del código Rails utiliza RAILS_ENV directamente.

Sin embargo, como Michael ha señalado en un comentario a su respuesta, cambiar RAILS_ENV sobre la marcha puede ser riesgoso. Otro enfoque sería cambiar la conexión de base de datos, esta solución es, de hecho, el valor por defecto utilizado por db:test tareas

ActiveRecord::Base.establish_connection(:test) 

Answer 4

En Rails 3, que tendrá que utilizar

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

en lugar de

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

Answer 5

Otra opción es verificar el env y rechazar continuar:

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

o pregunte si realmente quieren seguir:

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 

Answer 6

Hay algo de código extraño en 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 

Siempre añade test si env es development. Resolví el caso de querer hacer una tarea personalizada db:rebuild para development y test simultáneos ejecutando primero development y test en segundo lugar.Además, antes de ejecutar las tareas, llamo a mi método set_env que se asegura de establecer ActiveRecord::Tasks::DatabaseTasks.env, sin esto, las conexiones de la base de datos no parecen manejarse discretamente para entornos como se esperaba. Probé todos los otros tipos de desconexión, etc., pero esto funcionó sin más código.

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

---

Here is a gist of my full db.rake file with simultaneous multi-environment db:rebuild and db:truncate