class Test::Unit::UI::Console::TestRunner

def finished_with_rspec(elapsed_time)
  @ran_test ||= false
  if @ran_test
    finished_without_rspec(elapsed_time)
  end
end

def setup_mediator_with_rspec
  orig_io = @io
  @io = StringIO.new
  setup_mediator_without_rspec
ensure
  @rspec_io = @io
  @io = orig_io
end

def started_with_rspec(result)
  @result = result
  @need_to_output_started = true
end

def test_finished_with_rspec(name)
  test_finished_without_rspec(name)
  @ran_test = true
end

def test_started_with_rspec(name)
  if @need_to_output_started
    if @rspec_io
      @rspec_io.rewind
      output(@rspec_io.read)
    end
    output("Started")
    @need_to_output_started = false
  end
  test_started_without_rspec(name)
end