env.rb in xcpretty-0.0.4

- old
+ new

@@ -1,25 +1,25 @@
 $:.unshift File.expand_path('../../..', __FILE__)
 
 require "tempfile"
-require "spec/fixtures/constants.rb"
+require "spec/fixtures/constants"
+require "spec/support/matchers/colors"
+require "lib/xcpretty/ansi"
 
+include XCPretty::ANSI
+
+TEST_RUN_START_MATCHER  = /Test Suite .+ started/
 TEST_SUITE_COMPLETION_MATCHER = /Executed \d+ tests, with \d+ failures \(\d+ unexpected\) in \d+\.\d+ \(\d+\.\d+\) seconds/
-TEST_SUITE_BEGINNING_MATCHER  = /Test Suite '.+' started/
+TEST_SUITE_START_MATCHER = /[\w]*(Spec|Tests)$/
 TEST_PATH_MATCHER = %r{[\w/\-\s]+:\d+}
-PASSING_TEST_NAME_MATCHER = %r{\w+\s\w+\s\(\d+\.\d+\sseconds\)}
-FAILING_TEST_NAME_MATCHER = %r{\w+\s\w+, expected:}
+PASSING_TEST_NAME_MATCHER = %r{\w+\s\(\d+\.\d+\sseconds\)}
+FAILING_TEST_NAME_MATCHER = %r{\w+, expected:}
 
-INPUT_FILE  = "xcpretty_input"
-RED_START   = "\e[31m"
-COLOR_END   = "\e[0m"
-GREEN_START = "\e[32;1m"
-
 def run_xcpretty flags
   add_run_input SAMPLE_OCUNIT_SUITE_COMPLETION
   add_run_input SAMPLE_EXECUTED_TESTS
-  input_file = Tempfile.new(INPUT_FILE)
+  input_file = Tempfile.new("xcpretty_input")
   File.open(input_file.path, 'w') do |file|
     file.print run_input
   end
   @output = %x(cat '#{input_file.path}' | bin/xcpretty #{flags})
   input_file.unlink
@@ -35,9 +35,13 @@
 
 def run_output
   @output ||= ""
 end
 
+Before do
+ self.colorize = true
+end
+
 After do
   @input  = ""
   @output = ""
-end
\ No newline at end of file
+end