require 'minitest_helper'
class TypePair
  def to_s
    "#{last_arg_type} => #{rtn_type}"
  end
end
class TestRubype < MiniTest::Unit::TestCase
  def setup
    @string  = 'str'
    @numeric = 1
    @symbol  = :test
    @array   = [1, 2, 3]
    @hash    = { test: :hash }
  end

  def test_correct_type_by_class
    assert_correct_type({ [Numeric] => Numeric }, [@numeric], @numeric)
    assert_correct_type({ [Numeric] => Array   }, [@numeric], @array  )
    assert_correct_type({ [Numeric] => String  }, [@numeric], @string )
    assert_correct_type({ [Numeric] => Hash    }, [@numeric], @hash   )
    assert_correct_type({ [Numeric] => Symbol  }, [@numeric], @symbol )
    assert_correct_type({ [Numeric] => Boolean }, [@numeric], true    )
    assert_correct_type({ [Numeric] => Boolean }, [@numeric], false   )

    assert_correct_type({ [Boolean, Numeric] => Numeric }, [true, @numeric], @numeric)
    assert_correct_type({ [Boolean, Array  ] => Array   }, [true, @array  ], @array)
    assert_correct_type({ [Boolean, String ] => String  }, [true, @string ], @string)
    assert_correct_type({ [Boolean, Hash   ] => Hash    }, [true, @hash   ], @hash)
    assert_correct_type({ [Boolean, Symbol ] => Symbol  }, [true, @symbol ], @symbol)
  end

  def test_correct_type_by_sym
    assert_correct_type({ [Numeric] => :to_i }, [@numeric], @numeric)
    assert_correct_type({ [Numeric] => :to_i }, [@numeric], @string)

    assert_correct_type({ [Numeric] => :to_s }, [@numeric], @numeric)
    assert_correct_type({ [Numeric] => :to_s }, [@numeric], @string)
    assert_correct_type({ [Numeric] => :to_s }, [@numeric], @symbol)
    assert_correct_type({ [Numeric] => :to_s }, [@numeric], @array)
    assert_correct_type({ [Numeric] => :to_s }, [@numeric], @hash)
  end

  def test_wrong_return_type
    assert_wrong_rtn({ [Numeric] => Numeric }, [@numeric], @array)
    assert_wrong_rtn({ [Numeric] => Numeric }, [@numeric], @string)
    assert_wrong_rtn({ [Numeric] => Numeric }, [@numeric], @hash)
    assert_wrong_rtn({ [Numeric] => Numeric }, [@numeric], @symbol)
    assert_wrong_rtn({ [Numeric] => Numeric }, [@numeric], true)

    assert_wrong_rtn({ [Numeric, Numeric] => Numeric }, [@numeric, @numeric], @array)
    assert_wrong_rtn({ [Numeric, Numeric] => Numeric }, [@numeric, @numeric], @string)
    assert_wrong_rtn({ [Numeric, Numeric] => Numeric }, [@numeric, @numeric], @hash)
    assert_wrong_rtn({ [Numeric, Numeric] => Numeric }, [@numeric, @numeric], @symbol)
    assert_wrong_rtn({ [Numeric, Numeric] => Numeric }, [@numeric, @numeric], true)

    assert_wrong_rtn({ [Numeric] => :to_i }, [@numeric], @symbol)
    assert_wrong_rtn({ [Numeric] => :to_i }, [@numeric], @array)
    assert_wrong_rtn({ [Numeric] => :to_i }, [@numeric], @hash)
  end

  def test_wrong_args_type
    assert_wrong_arg({ [Numeric] => Numeric }, [@array ], @numeric)
    assert_wrong_arg({ [Numeric] => Numeric }, [@string], @numeric)
    assert_wrong_arg({ [Numeric] => Numeric }, [@hash  ], @numeric)
    assert_wrong_arg({ [Numeric] => Numeric }, [@symbol], @numeric)
    assert_wrong_arg({ [Numeric] => Numeric }, [true   ], @numeric)

    assert_wrong_arg({ [Numeric, Numeric] => Numeric }, [@numeric, @array ], @numeric)
    assert_wrong_arg({ [Numeric, Numeric] => Numeric }, [@numeric, @string], @numeric)
    assert_wrong_arg({ [Numeric, Numeric] => Numeric }, [@numeric, @hash  ], @numeric)
    assert_wrong_arg({ [Numeric, Numeric] => Numeric }, [@numeric, @symbol], @numeric)
    assert_wrong_arg({ [Numeric, Numeric] => Numeric }, [@numeric, true   ], @numeric)

    assert_wrong_arg({ [Numeric] => :to_i }, [@array ], @numeric)
    assert_wrong_arg({ [Numeric] => :to_i }, [@hash  ], @numeric)
    assert_wrong_arg({ [Numeric] => :to_i }, [@symbol], @numeric)
    assert_wrong_arg({ [Numeric] => :to_i }, [true   ], @numeric)
  end

  def test_any
    assert_correct_type({ [Any] => Any }, [@array ], @numeric)
    assert_correct_type({ [Any] => Any }, [@string], @numeric)
    assert_correct_type({ [Any] => Any }, [@hash  ], @numeric)
    assert_correct_type({ [Any] => Any }, [@symbol], @numeric)

    assert_correct_type({ [Any, Any] => Any }, [@numeric, @array ], @numeric)
    assert_correct_type({ [Any, Any] => Any }, [@numeric, @string], @numeric)
    assert_correct_type({ [Any, Any] => Any }, [@numeric, @hash  ], @numeric)
    assert_correct_type({ [Any, Any] => Any }, [@numeric, @symbol], @numeric)
  end

  def test_type_info
    klass = Class.new.class_eval <<-RUBY_CODE
      def test_mth
      end
      typesig :test_mth, [Numeric, Numeric] => String
    RUBY_CODE

    assert_equal klass.new.method(:test_mth).type_info, { [Numeric, Numeric] => String }
  end

  private
    def assert_equal_to_s(str, val)
      assert_equal str, val.to_s
    end

    def assert_correct_type(type_list, args, val)
      assert_equal val, define_test_method(type_list, args, val).call(*args)
    end

    def assert_wrong_arg(type_list, args, val)
      assert_raises(Rubype::ArgumentTypeError) { define_test_method(type_list, args, val).call(*args) }
    end

    def assert_wrong_rtn(type_list, args, val)
      assert_raises(Rubype::ReturnTypeError) { define_test_method(type_list, args, val).call(*args) }
    end

    def define_test_method(type_list, args, val)
      klass = Class.new.class_eval <<-RUBY_CODE
        def call(#{arg_literal(args.count)})
          #{obj_literal(val)}
        end
        typesig :call, #{obj_literal(type_list)}
      RUBY_CODE

      klass.new
    end

    def obj_literal(obj)
      "ObjectSpace._id2ref(#{obj.__id__})"
    end

    def arg_literal(count)
      ('a'..'z').to_a[0..count-1].join(',')
    end
end