From 93f859662e9f6f4036c20e8e6a9c3ac81a7f8090 Mon Sep 17 00:00:00 2001
From: Antonio Terceiro <terceiro@colivre.coop.br>
Date: Fri, 5 Feb 2010 14:27:17 -0300
Subject: [PATCH] Coloring test output

---
 test/colors.rb      | 57 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 test/test_helper.rb |  2 ++
 2 files changed, 59 insertions(+), 0 deletions(-)
 create mode 100644 test/colors.rb

diff --git a/test/colors.rb b/test/colors.rb
new file mode 100644
index 0000000..35d0de1
--- /dev/null
+++ b/test/colors.rb
@@ -0,0 +1,57 @@
+require 'test/unit'
+require 'test/unit/ui/console/testrunner'
+
+module Color
+  COLORS = { :clear => 0, :red => 31, :green => 32, :yellow => 33 }
+  def self.method_missing(color_name, *args)
+    colname = color_name.to_s
+    ansi_color = (colname =~ /^light/ ?
+                  light_color(colname.gsub(/light/, '')) :
+                  color(colname))
+    ansi_color + args.first + color(:clear)
+  end
+  def self.color(color)
+    "\e[#{COLORS[color.to_sym]}m"
+  end
+
+  def self.light_color(color)
+  "\e[1;#{COLORS[color.to_sym]}m"
+  end
+end
+
+class Test::Unit::UI::Console::TestRunner
+  def output_single(something, level=NORMAL)
+    return unless (output?(level))
+    something = case something
+                when '.' then Color.green('.')
+                when 'F' then Color.red("F")
+                when 'E' then Color.yellow("E")
+                else something
+                end
+    @io.write(something)
+    @io.flush
+  end
+end
+
+class Test::Unit::TestResult
+  alias :old_to_s :to_s
+  def to_s
+    if old_to_s =~ /\d+ tests, \d+ assertions, (\d+) failures, (\d+) errors/
+      Color.send($1.to_i != 0 || $2.to_i != 0 ? :red : :green, $&)
+    end
+  end
+end
+
+class Test::Unit::Failure
+  alias :old_long_display :long_display
+  def long_display
+    old_long_display.sub('Failure', Color.red('Failure'))
+  end
+end
+
+class Test::Unit::Error
+  alias :old_long_display :long_display
+  def long_display
+    old_long_display.sub('Error', Color.yellow('Error'))
+  end
+end
diff --git a/test/test_helper.rb b/test/test_helper.rb
index 6e64237..988eda3 100644
--- a/test/test_helper.rb
+++ b/test/test_helper.rb
@@ -208,4 +208,6 @@ class ActionController::IntegrationTest
 
 end
 
+require File.dirname(__FILE__) + '/colors'
+
 Profile
--
libgit2 0.21.2