module Test::Unit

def at_exit(&hook)
  @@at_exit_hooks << hook
end

def at_start(&hook)
  @@at_start_hooks << hook
end

def run=(have_run)
  AutoRunner.need_auto_run = (not have_run)
end

def run?
  not AutoRunner.need_auto_run?
end

def run_at_exit_hooks
  @@at_exit_hooks.each do |hook|
    hook.call
  end
end

def run_at_start_hooks
  @@at_start_hooks.each do |hook|
    hook.call
  end
end