# frozen_string_literal: true

require 'minitest/autorun'
require 'minitest/pride'

require_relative '../lib/mutual_recursion'

class InventoryTest < Minitest::Test
  include MutualRecursion

  def test_terminal_value
    tail = terminal_value(42)
    assert_equal(42, tail.invoke)
  end

  def test_single_recursion
    tail = tail_call { terminal_value(42) }
    assert_equal(42, tail.invoke)
  end

  def test_several_recursions
    tail = tail_call { tail_call { tail_call { terminal_value(42) } } }
    assert_equal(42, tail.invoke)
  end

  def test_mutual_tail_recursion
    tail = one(50_000)
    assert_equal(50_001, tail.invoke)
  end

  def test_direct_tail_recursion
    tail = direct(50_000)
    assert_equal(50_001, tail.invoke)
  end

  def test_recursive_function_can_return_lambda
    tail = lambda_returning
    assert_kind_of(Proc, tail.invoke)
  end

  def test_recursive_function_can_return_proc
    tail = proc_returning(20)
    assert_kind_of(Proc, tail.invoke)
  end

  def test_non_tail_call_detected
    tail = bad_return
    assert_raises(MissingTailCallError) { tail.invoke }
  end

  def test_module_functions
    tail = MutualRecursion.tail_call { MutualRecursion.terminal_value(42) }
    assert_equal(42, tail.invoke)
  end
end

def one(x, y = 0)
  return terminal_value(y) if x.negative?

  tail_call { two(x, y + 1) }
end

def two(x, y)
  tail_call { one(x - 1, y) }
end

def direct(x, y = 0)
  return terminal_value(y) if x.negative?

  tail_call { direct(x - 1, y + 1) }
end

def lambda_returning
  terminal_value(-> { 24 })
end

def proc_returning(x, y = 0)
  return terminal_value(proc { "|#{y}|" }) if x.negative?

  tail_call { proc_returning(x - 1, y + 1) }
end

def bad_return
  tail_call do
    Class.new do
      attr_reader :value, :block

      def initialize
        @value = 99
        @block = proc { 25 }
      end
    end.new
  end
end