class MiniTest::Unit

def self.after_tests
  at_exit { at_exit { yield } }
end

def self.autorun
  at_exit {
    next if $! # don't run if there was an exception
    exit_code = MiniTest::Unit.new.run ARGV
    exit false if exit_code && exit_code != 0
  } unless @@installed_at_exit
  @@installed_at_exit = true
end

def self.out
  warn "::out deprecated, use ::output instead." if $VERBOSE
  output
end

def self.output
  @@out
end

def self.output= stream
  @@out = stream
end

def self.plugins
  @@plugins ||= (["run_tests"] +
                 public_instance_methods(false).
                 grep(/^run_/).map { |s| s.to_s }).uniq
end

def _run_anything type
  suites = TestCase.send "#{type}_suites"
  return if suites.empty?
  start = Time.now
  puts
  puts "# Running #{type}s:"
  puts
  @test_count, @assertion_count = 0, 0
  sync = output.respond_to? :"sync=" # stupid emacs
  old_sync, output.sync = output.sync, true if sync
  results = _run_suites suites, type
  @test_count      = results.inject(0) { |sum, (tc, ac)| sum + tc }
  @assertion_count = results.inject(0) { |sum, (tc, ac)| sum + ac }
  output.sync = old_sync if sync
  t = Time.now - start
  puts
  puts
  puts "Finished #{type}s in %.6fs, %.4f tests/s, %.4f assertions/s." %
    [t, test_count / t, assertion_count / t]
  report.each_with_index do |msg, i|
    puts "\n%3d) %s" % [i + 1, msg]
  end
  puts
  status
end

def _run_suite suite, type
  header = "#{type}_suite_header"
  puts send(header, suite) if respond_to? header
  filter = options[:filter] || '/./'
  filter = Regexp.new $1 if filter =~ /\/(.*)\//
  assertions = suite.send("#{type}_methods").grep(filter).map { |method|
    inst = suite.new method
    inst._assertions = 0
    start_time = Time.now
    result = inst.run self
    time = Time.now - start_time
    print "#{suite}##{method} = %.2f s = " % time if @verbose
    print result
    puts if @verbose
    inst._assertions
  }
  return assertions.size, assertions.inject(0) { |sum, n| sum + n }
end

def _run_suites suites, type
  suites.map { |suite| _run_suite suite, type }
end

def benchmark_suite_header suite
  "\n#{suite}\t#{suite.bench_range.join("\t")}"
end

def initialize # :nodoc:
  @report = []
  @errors = @failures = @skips = 0
  @verbose = false
end

def location e # :nodoc:
  last_before_assertion = ""
  e.backtrace.reverse_each do |s|
    break if s =~ /in .(assert|refute|flunk|pass|fail|raise|must|wont)/
    last_before_assertion = s
  end
  last_before_assertion.sub(/:in .*$/, '')
end

def output
  self.class.output
end

def print *a # :nodoc:
  output.print(*a)
end

def process_args args = []
  options = {}
  orig_args = args.dup
  OptionParser.new do |opts|
    opts.banner  = 'minitest options:'
    opts.version = MiniTest::Unit::VERSION
    opts.on '-h', '--help', 'Display this help.' do
      puts opts
      exit
    end
    opts.on '-s', '--seed SEED', Integer, "Sets random seed" do |m|
      options[:seed] = m.to_i
    end
    opts.on '-v', '--verbose', "Verbose. Show progress processing files." do
      options[:verbose] = true
    end
    opts.on '-n', '--name PATTERN', "Filter test names on pattern." do |a|
      options[:filter] = a
    end
    opts.parse! args
    orig_args -= args
  end
  unless options[:seed] then
    srand
    options[:seed] = srand % 0xFFFF
    orig_args << "--seed" << options[:seed].to_s
  end
  srand options[:seed]
  self.verbose = options[:verbose]
  @help = orig_args.map { |s| s =~ /[\s|&<>$()]/ ? s.inspect : s }.join " "
  options
end

def puke klass, meth, e
  e = case e
      when MiniTest::Skip then
        @skips += 1
        "Skipped:\n#{meth}(#{klass}) [#{location e}]:\n#{e.message}\n"
      when MiniTest::Assertion then
        @failures += 1
        "Failure:\n#{meth}(#{klass}) [#{location e}]:\n#{e.message}\n"
      else
        @errors += 1
        bt = MiniTest::filter_backtrace(e.backtrace).join "\n    "
        "Error:\n#{meth}(#{klass}):\n#{e.class}: #{e.message}\n    #{bt}\n"
      end
  @report << e
  e[0, 1]
end

def puts *a  # :nodoc:
  output.puts(*a)
end

def run args = []
  self.options = process_args args
  puts "Run options: #{help}"
  self.class.plugins.each do |plugin|
    send plugin
    break unless report.empty?
  end
  return failures + errors if @test_count > 0 # or return nil...
rescue Interrupt
  abort 'Interrupted'
end

def run_benchmarks
  _run_anything :benchmark
end

def run_tests
  _run_anything :test
end

def status io = self.output
  format = "%d tests, %d assertions, %d failures, %d errors, %d skips"
  io.puts format % [test_count, assertion_count, failures, errors, skips]
end