diff --git a/lib/hydra/master.rb b/lib/hydra/master.rb index dc1d850..baa088f 100644 --- a/lib/hydra/master.rb +++ b/lib/hydra/master.rb @@ -48,6 +48,7 @@ module Hydra #:nodoc: @verbose = opts.fetch('verbose') { false } @autosort = opts.fetch('autosort') { true } @sync = opts.fetch('sync') { nil } + @environment = opts.fetch('environment') { 'test' } if @autosort sort_files_from_report @@ -139,7 +140,7 @@ module Hydra #:nodoc: runners = worker.fetch('runners') { raise "You must specify the number of runners" } command = worker.fetch('command') { - "ruby -e \"require 'rubygems'; require 'hydra'; Hydra::Worker.new(:io => Hydra::Stdio.new, :runners => #{runners}, :verbose => #{@verbose});\"" + "RAILS_ENV=#{@environment} ruby -e \"require 'rubygems'; require 'hydra'; Hydra::Worker.new(:io => Hydra::Stdio.new, :runners => #{runners}, :verbose => #{@verbose});\"" } trace "Booting SSH worker" diff --git a/lib/hydra/tasks.rb b/lib/hydra/tasks.rb index 47e4b09..eb50f75 100644 --- a/lib/hydra/tasks.rb +++ b/lib/hydra/tasks.rb @@ -188,7 +188,7 @@ module Hydra #:nodoc: end @listeners.each{|l| l.join} $stdout.write "\n==== Hydra Running #{@name} COMPLETE ====\n\n" - $stdout.write @results.values.join('\n') + $stdout.write @results.values.join("\n") end end