Commit 93f859662e9f6f4036c20e8e6a9c3ac81a7f8090
1 parent
10bed1b8
Exists in
master
and in
22 other branches
Coloring test output
Taken from a comment by Edvard Majakari (http://majakari.net/) at http://errtheblog.com/posts/34-a-ruby-rainbow Changed the coloring to use the dark versions of the colores instead of the light ones
Showing
2 changed files
with
59 additions
and
0 deletions
Show diff stats
| @@ -0,0 +1,57 @@ | @@ -0,0 +1,57 @@ | ||
| 1 | +require 'test/unit' | ||
| 2 | +require 'test/unit/ui/console/testrunner' | ||
| 3 | + | ||
| 4 | +module Color | ||
| 5 | + COLORS = { :clear => 0, :red => 31, :green => 32, :yellow => 33 } | ||
| 6 | + def self.method_missing(color_name, *args) | ||
| 7 | + colname = color_name.to_s | ||
| 8 | + ansi_color = (colname =~ /^light/ ? | ||
| 9 | + light_color(colname.gsub(/light/, '')) : | ||
| 10 | + color(colname)) | ||
| 11 | + ansi_color + args.first + color(:clear) | ||
| 12 | + end | ||
| 13 | + def self.color(color) | ||
| 14 | + "\e[#{COLORS[color.to_sym]}m" | ||
| 15 | + end | ||
| 16 | + | ||
| 17 | + def self.light_color(color) | ||
| 18 | + "\e[1;#{COLORS[color.to_sym]}m" | ||
| 19 | + end | ||
| 20 | +end | ||
| 21 | + | ||
| 22 | +class Test::Unit::UI::Console::TestRunner | ||
| 23 | + def output_single(something, level=NORMAL) | ||
| 24 | + return unless (output?(level)) | ||
| 25 | + something = case something | ||
| 26 | + when '.' then Color.green('.') | ||
| 27 | + when 'F' then Color.red("F") | ||
| 28 | + when 'E' then Color.yellow("E") | ||
| 29 | + else something | ||
| 30 | + end | ||
| 31 | + @io.write(something) | ||
| 32 | + @io.flush | ||
| 33 | + end | ||
| 34 | +end | ||
| 35 | + | ||
| 36 | +class Test::Unit::TestResult | ||
| 37 | + alias :old_to_s :to_s | ||
| 38 | + def to_s | ||
| 39 | + if old_to_s =~ /\d+ tests, \d+ assertions, (\d+) failures, (\d+) errors/ | ||
| 40 | + Color.send($1.to_i != 0 || $2.to_i != 0 ? :red : :green, $&) | ||
| 41 | + end | ||
| 42 | + end | ||
| 43 | +end | ||
| 44 | + | ||
| 45 | +class Test::Unit::Failure | ||
| 46 | + alias :old_long_display :long_display | ||
| 47 | + def long_display | ||
| 48 | + old_long_display.sub('Failure', Color.red('Failure')) | ||
| 49 | + end | ||
| 50 | +end | ||
| 51 | + | ||
| 52 | +class Test::Unit::Error | ||
| 53 | + alias :old_long_display :long_display | ||
| 54 | + def long_display | ||
| 55 | + old_long_display.sub('Error', Color.yellow('Error')) | ||
| 56 | + end | ||
| 57 | +end |