fix and improve ability coverage to account for tokens

[rails.git] / config / preinitializer.rb

diff --git a/config/preinitializer.rb b/config/preinitializer.rb
index 377e6bad4a6a13ac33095926085eccefccc925fb..07f10572286923c6b2b9acc074c83e84a6bc3d83 100644 (file)
--- a/config/preinitializer.rb
+++ b/config/preinitializer.rb
@@ -1,16 +1,17 @@
-require 'yaml'
+require "yaml"
+
+env = if defined?(Rake.application) && Rake.application.top_level_tasks.grep(/^(default$|test(:|$))/).any?
+        "test"
+      else
+        ENV["RAILS_ENV"] || "development"
+      end

 
 

-env = ENV['RAILS_ENV'] || 'development'

 config = YAML.load_file(File.expand_path(env == "test" ? "../example.application.yml" : "../application.yml", __FILE__))
 
 config = YAML.load_file(File.expand_path(env == "test" ? "../example.application.yml" : "../application.yml", __FILE__))
 

-ENV.each do |key,value|
-  if key.match(/^OSM_(.*)$/)
-    Object.const_set(Regexp.last_match(1).upcase, value)
-  end
+ENV.each do |key, value|
+  Object.const_set(Regexp.last_match(1).upcase, value) if key =~ /^OSM_(.*)$/

 end
 
 end
 

-config[env].each do |key,value|
-  unless Object.const_defined?(key.upcase)
-    Object.const_set(key.upcase, value)
-  end
+config[env].each do |key, value|
+  Object.const_set(key.upcase, value) unless Object.const_defined?(key.upcase)

 end
 end