From 0aeb59d65c6a87f5805193f6f7ac62b71225c8f0 Mon Sep 17 00:00:00 2001
From: John Bintz <john@coswellproductions.com>
Date: Wed, 27 Feb 2013 17:24:31 -0500
Subject: [PATCH] slightly better handling of template exceptions

---
 lib/capybara/rails-log-inspection.rb | 27 +++++++++++++++++++++++----
 1 file changed, 23 insertions(+), 4 deletions(-)

diff --git a/lib/capybara/rails-log-inspection.rb b/lib/capybara/rails-log-inspection.rb
index 92a0b94..f2c3df2 100644
--- a/lib/capybara/rails-log-inspection.rb
+++ b/lib/capybara/rails-log-inspection.rb
@@ -8,9 +8,11 @@ module Capybara
       end
 
       def logger(level = nil)
-        logger = Logger.new(logger_target)
-        logger.level = level || default_log_level
-        logger
+        return @logger if @logger
+
+        @logger = Logger.new(logger_target)
+        @logger.level = level || default_log_level
+        @logger
       end
 
       attr_writer :backtrace_clean_patterns, :default_log_level, :rack_log_level, :filter_request_starts
@@ -42,6 +44,11 @@ module Capybara
       def add_backtrace(exception)
         clean_backtrace(exception).each { |line| logger_target << "  #{line.strip}\n" }
       end
+
+      def output_line(line, target = $stderr)
+        target.print(Term::ANSIColor.red, line, Term::ANSIColor.reset)
+        target.flush
+      end
     end
 
     def reset_logs
@@ -55,7 +62,7 @@ module Capybara
       lines = Capybara::RailsLogInspection.logger_target.read.lines
 
       Capybara::RailsLogInspection.clean_rack_output(lines).each do |line|
-        target.print(Term::ANSIColor.red, line, Term::ANSIColor.reset)
+        output_line(line, target)
       end
 
       reset_logs
@@ -63,8 +70,20 @@ module Capybara
   end
 end
 
+
 Rails.logger = Capybara::RailsLogInspection.logger
 
+class LogInspectionSubscriber < ActiveSupport::LogSubscriber
+  def process_action(event)
+    if event.payload[:exception]
+      Capybara::RailsLogInspection.output_line "Processing #{event.payload[:method]} #{event.payload[:controller]}##{event.payload[:action]} as #{event.payload[:format]}\n"
+      Capybara::RailsLogInspection.output_line "  #{event.payload[:exception].first}: #{event.payload[:exception].last}\n"
+    end
+  end
+end
+
+LogInspectionSubscriber.attach_to :action_controller
+
 Capybara.server do |app, port|
   require 'rack/handler/webrick'