# frozen_string_literal: true

require "test_helper"

module Byebug
  #
  # Tests +finish+ functionality when it needs to stop after method return
  # events.
  #
  class FinishAfterReturnTest < TestCase
    def program
      strip_line_numbers <<-RUBY
         1:  module Byebug
         2:    #
         3:    # Toy class to test the +finish+ command
         4:    #
         5:    class #{example_class}
         6:      def a
         7:        b
         8:      end
         9:
        10:      def b
        11:        c
        12:        2
        13:      end
        14:
        15:      def c
        16:        d
        17:        3
        18:      end
        19:
        20:      def d
        21:        5
        22:      end
        23:    end
        24:
        25:    byebug
        26:
        27:    #{example_class}.new.a
        28:  end
      RUBY
    end

    def test_finish_stops_after_current_single_line_frame_is_finished
      enter "break 21", "cont", "finish"

      debug_code(program) { assert_equal 17, frame.line }
    end

    def test_finish_stops_after_current_multiline_frame_is_finished
      enter "break 16", "cont", "finish"

      debug_code(program) { assert_equal 12, frame.line }
    end

    def test_finish_1_stops_after_current_frame_is_finished
      enter "break 21", "cont", "finish 1"

      debug_code(program) { assert_equal 17, frame.line }
    end

    def test_finish_works_for_frame_numbers_higher_than_one
      enter "break 21", "cont", "finish 2"

      debug_code(program) { assert_equal 12, frame.line }
    end

    def test_finish_behaves_consistently_even_if_current_frame_has_been_changed
      enter "break 21", "cont", "up", "finish"

      debug_code(program) { assert_equal 12, frame.line }
    end

    def test_finish_shows_an_error_if_incorrect_frame_number_specified
      enter "break 21", "cont", "finish foo"
      debug_code(program)

      check_error_includes '"finish" argument "foo" needs to be a number'
    end

    def test_finish_stays_at_the_same_line_if_incorrect_frame_number_specified
      enter "break 21", "cont", "finish foo"

      debug_code(program) { assert_equal 21, frame.line }
    end

    def test_finish_does_not_stop_in_byebug_internal_frames
      enter "break 21", "cont", "finish 4"

      debug_code(program) { assert_program_finished }
    end
  end

  #
  # Tests +finish+ functionality when it needs to stop before method return
  # events.
  #
  class FinishBeforeReturnTest < TestCase
    def program
      strip_line_numbers <<-RUBY
         1:  module Byebug
         2:    #
         3:    # Toy class to test the +finish+ command
         4:    #
         5:    class #{example_class}
         6:      def a
         7:        b
         8:      end
         9:
        10:      def b
        11:        (1..5).map do |i|
        12:          i**2
        13:        end
        14:      end
        15:    end
        16:
        17:    byebug
        18:
        19:    #{example_class}.new.a
        20:  end
      RUBY
    end

    def test_finish_0_stops_right_before_frame_returns__simple_case
      enter "b 7", "cont", "finish 0"

      debug_code(program) { assert_equal 8, frame.line }
    end

    def test_finish_0_shows_information_about_the_return_value
      enter "b 7", "cont", "finish 0"
      debug_code(program)

      check_output_includes "Return value is: [1, 4, 9, 16, 25]"
    end

    def test_finish_0_stops_right_before_frame_returns__convoluted_case
      enter "b 11", "cont", "finish 0"

      debug_code(program) { assert_equal 14, frame.line }
    end
  end

  #
  # Tests +finish+ functionality inside autoloaded files.
  #
  class FinishInsideAutoloadedFilesTest < TestCase
    def program
      strip_line_numbers <<-RUBY
         1:  autoload :ByebugBar, "./byebug_bar"
         2:
         3:  def main
         4:    b = ByebugBar
         5:    b
         6:  end
         7:
         8:  main
         9:
        10:  sleep 0
      RUBY
    end

    def teardown
      super

      force_remove_const(Object, :ByebugBar)
    end

    def test_finish_inside_autoloaded_files
      with_new_file("byebug_bar.rb", "byebug\nmodule ByebugBar; end") do
        enter "finish"

        debug_code(program) { assert_equal 5, frame.line }
      end
    end
  end
end
