env.rb in xcpretty-0.0.6

- old
+ new

@@ -2,10 +2,13 @@
 
 require "tempfile"
 require "spec/fixtures/constants"
 require "spec/support/matchers/colors"
 require "lib/xcpretty/ansi"
+require "rexml/document"
+require "lib/xcpretty/printer"
+require "lib/xcpretty/reporters/junit"
 
 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/
@@ -35,13 +38,18 @@
 
 def run_output
   @output ||= ""
 end
 
+def junit_report
+  REXML::Document.new(File.open(XCPretty::JUnit::FILEPATH, 'r').read)
+end
+
 Before do
  self.colorize = true
 end
 
 After do
   @input  = ""
   @output = ""
+  FileUtils.rm_rf(XCPretty::JUnit::FILEPATH)
 end