Fix how to show host name in error

mizzy · Aug 17, 2014 · 2aca137 · 2aca137
1 parent 52252c5
commit 2aca137
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/serverspec.rb b/lib/serverspec.rb
@@ -13,9 +13,10 @@
 module RSpec::Core::Notifications
   class FailedExampleNotification < ExampleNotification
     def failure_lines
+      host = ENV['TARGET_HOST'] || Specinfra.configuration.host
       @failure_lines ||=
         begin
-          lines = ["On host `#{Specinfra.configuration.host}`"]
+          lines = ["On host `#{host}`"]
           lines << "Failure/Error: #{read_failed_line.strip}"
           lines << "#{exception_class_name}:" unless exception_class_name =~ /RSpec/
           exception.message.to_s.split("\n").each do |line|