Permalink
Browse files

Only check for color once

  • Loading branch information...
1 parent f4b7ddf commit 0e221391b0c0f3658bcb270e976c315d92b28157 @rf- rf- committed Sep 9, 2012
Showing with 3 additions and 7 deletions.
  1. +3 −7 test/helper.rb
View
@@ -11,14 +11,10 @@
# Colorize output (based on greeneggs (c) 2009 Michael Fleet)
# TODO: Make own gem (assigned to rking)
module Bacon
- COLORS = {'F' => 31, 'E' => 35, 'M' => 33, '.' => 32}
+ COLORS = {'F' => 31, 'E' => 35, 'M' => 33, '.' => 32}
+ USE_COLOR = Pry::Helpers::BaseHelpers.use_ansi_codes?
module TestUnitOutput
- def handle_specification(name)
- @use_color = Pry::Helpers::BaseHelpers.use_ansi_codes?
- yield
- end
-
def handle_requirement(description)
error = yield
@@ -44,7 +40,7 @@ def handle_summary
end
def colorize_string(text, color = nil)
- if @use_color
+ if USE_COLOR
"\e[#{ COLORS[color || text] }m#{ text }\e[0m"
else
text

0 comments on commit 0e22139

Please sign in to comment.