require_relative "helper"

# Please read note in test_parser

require_relative 'helper'

class TestAst < MiniTest::Test

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

  def check
    syntax    = @parser.parse(@string_input)
    tree      = @transform.apply(syntax)
    # puts tree.inspect
    assert_equal @transform_output , tree
  end
  
  def test_expression_else
    @string_input    = <<HERE
4
5
else
HERE
    @transform_output = {:expressions=>[ Parser::IntegerExpression.new(4), Parser::IntegerExpression.new(5)]}
    
    @parser = @parser.expressions_else
    check
  end

  def test_expression_end
    @string_input    = <<HERE
5
name
call(4,6)
end
HERE
    @transform_output = {:expressions=> [Parser::IntegerExpression.new(5), Parser::NameExpression.new("name"),
      Parser::FuncallExpression.new("call", [Parser::IntegerExpression.new(4), Parser::IntegerExpression.new(6) ]) ] }
    @parser = @parser.expressions_end
    check
  end

  def test_conditional
    @string_input = <<HERE
if (0) 
  42
else
  667
end
HERE
    @transform_output = Parser::ConditionalExpression.new(  Parser::IntegerExpression.new(0),
                                            [Parser::IntegerExpression.new(42)],
                                            [Parser::IntegerExpression.new(667)])
    @parser = @parser.conditional
    check
  end
  
  def test_function_definition
    @string_input    = <<HERE
def foo(x) 
  5
end
HERE
    @transform_output = Parser::FunctionExpression.new('foo', 
                  [Parser::NameExpression.new('x')], 
                  [Parser::IntegerExpression.new(5)])
    @parser = @parser.function_definition
    check
  end
  def test_function_assignment
    @string_input    = <<HERE
def foo(x) 
 abba = 5 
end
HERE
    @transform_output = Parser::FunctionExpression.new( "foo", [Parser::NameExpression.new("x")],
                       [Parser::AssignmentExpression.new( "abba", Parser::IntegerExpression.new(5) ) ])
    check
  end
end