module Byebug
  class RestartExample
    def concat_args(a, b, c)
      a.to_s + b.to_s + c.to_s
    end
  end

  class RestartTestCase < TestCase
    def setup
      @example = -> do
        byebug
        a = ARGV[0]
        b = ARGV[1]
        c = ARGV[2]
        RestartExample.new.concat_args(a, b, c)
      end

      super
    end

    def must_restart(cmd = nil)
      expectation = RestartCommand.any_instance.expects(:exec)
      expectation = expectation.with(cmd) if cmd
      expectation
    end

    def test_restarts_with_manual_arguments
      force_set_const(Byebug, 'BYEBUG_SCRIPT', 'byebug_script')
      cmd = "#{BYEBUG_SCRIPT} #{Byebug.debugged_program} 1 2 3"
      must_restart(cmd)

      enter 'restart 1 2 3'
      debug_proc(@example)
      check_output_includes "Re exec'ing:\n\t#{cmd}"
    end

    def test_restart_with_a_default_script_if_nil_script_specified
      force_set_const(Byebug, 'BYEBUG_SCRIPT', 'byebug_script')
      Byebug.debugged_program = nil
      must_restart

      enter 'restart'
      debug_proc(@example)
      check_output_includes(/Re exec'ing:\s*#{BYEBUG_SCRIPT} #{$PROGRAM_NAME}/)
    end

    def test_does_not_restart_when_script_specified_does_not_exist
      Byebug.debugged_program = 'blabla'
      must_restart.never
      enter 'restart'
      debug_proc(@example)
      check_error_includes 'Ruby program blabla doesn\'t exist'
    end

    def test_still_restarts_when_byebug_attached_to_running_program
      force_unset_const(Byebug, 'BYEBUG_SCRIPT')
      must_restart
      enter 'restart'
      debug_proc(@example)
      check_output_includes 'Byebug was not called from the outset...'
      check_output_includes \
        "Ruby program #{Byebug.debugged_program} not executable... " \
        "We'll wrap it in a ruby call"
    end
  end
end