require_relative 'helper'

include Vm

class TransformTest <  MiniTest::Test

  def setup
    @transform = Parser::Transform.new
  end

  def check_equals
    is = @transform.apply @input
    assert_equal @expected , is
  end
  def test_number
    @input    = {:integer => '42'}
    @expected = Vm::IntegerExpression.new(42)
    check_equals
    assert_equal 42 , @expected.value
  end

  def test_name
    @input    = {:name => 'foo'}
    @expected = Vm::NameExpression.new('foo')
    check_equals
  end

  def test_argument_list
    @input    = {:argument_list => [{:argument => {:integer => '42'}},
                          {:argument => {:name   => 'foo'}}]}
    @expected = [Vm::IntegerExpression.new(42),
                Vm::NameExpression.new('foo')]
    check_equals
  end

  def test_single_argument
    @input = {:function_call => {:name => 'foo'},
             :argument_list    => {:argument => {:integer => '42'} } }
    @expected = Vm::FuncallExpression.new 'foo', [Vm::IntegerExpression.new(42)]

    check_equals
  end

  def test_multi_argument
    @input = {:function_call => {:name => 'baz'},
             :argument_list    => [{:argument => {:integer => '42'}},
                          {:argument => {:name => 'foo'}}]}
    @expected = Vm::FuncallExpression.new 'baz', [Vm::IntegerExpression.new(42),
                                          Vm::NameExpression.new('foo')]

    check_equals
  end

  def test_conditional
    @input = { :conditional => { :integer => "0"}, 
                  :if_true => {  :expressions => [ { :integer => "42" } ] } , 
                  :if_false => { :expressions => [ { :integer => "667" } ] } }
    @expected = Vm::ConditionalExpression.new(  Vm::IntegerExpression.new(0),
                                                [Vm::IntegerExpression.new(42)],
                                                [Vm::IntegerExpression.new(667)])
    check_equals
  end

  def test_parmeter
    @input = {:parmeter => { :name => "foo"}} 
    @expected = Vm::NameExpression.new('foo')
    check_equals
  end
  def test_parmeter_list
    @input = {:parmeter_list => [{:parmeter => { :name => "foo"}}]}
    @expected = [Vm::NameExpression.new('foo')]
    check_equals
  end
  
  def test_function_definition
    @input = {:function_definition   => {:name => 'foo'},
                :parmeter_list => {:parmeter => {:name => 'x'}},
                :expressions   => [{:integer => '5'}]}
    @expected = Vm::FunctionExpression.new('foo', 
                [Vm::NameExpression.new('x')], 
                [Vm::IntegerExpression.new(5)])
    check_equals
  end
end