compiler_test.rb

require 'test_helper'

module Noscript
  class CompilerTest < MiniTest::Unit::TestCase
    def test_compile_integer_literal
      assert_equal 1, compile("1")
    end

    def test_compile_unary_minus
      assert_equal -1, compile("-1")
    end

    def test_compile_string_literal
      assert_equal '1', compile("'1'")
    end

    def test_compile_string_literal_with_escaped_chars
      assert_equal "\n", compile("'\n'")
    end

    def test_compile_string_interpolation
      assert_equal '1', compile("'%s' % [1]")
    end

    def test_compile_string_interpolation_with_variables
      assert_equal '1', compile("foo = 1; '%s' % foo")
    end

    def test_compile_assignment
      assert_equal 1, compile("foo = 1; foo")
    end

    def test_compile_constant_assignment
      assert_equal 1, compile("Foo = 1; Foo")
    end

    def test_compile_constant_lookup
      assert_equal Rubinius, compile("Rubinius")
    end

    def test_compile_constant_lookup_with_namespace
      assert_equal Noscript::AST::Identifier, compile("Noscript.AST.Identifier")
    end

    def test_compile_array_literal
      assert_equal 1, compile("[1,2].at(0)")
    end

    def test_compile_array_literal_with_variables
      assert_equal [3], compile("bar = 3; [bar]")
    end

    def test_compile_tuple_literal
      assert_equal({'a' => true, 'b' => false}, compile("{a: true, b: false}"))
    end

    def test_compile_tuple_literal_referring_to_a_variable
      assert_equal({'a' => 34, 'b' => false}, compile("foo = 34; {a: foo, b: false}"))
    end

    def test_compile_tuple_literal_inside_a_function
      assert_equal({'a' => 34, 'b' => false}, compile("foo = -> bar; {a: bar, b: false}; end; foo(34)"))
    end

    def test_compile_tuple_literal_as_an_expression
      assert_equal(['a', 'b'], compile("{a: true, b: false}.keys()"))
    end

    def test_compile_function_without_arguments
      assert_equal 3, compile("foo = ->; 3; end; foo()")
    end

    def test_compile_function_with_unused_arguments
      assert_equal 3, compile("foo = -> a; 3; end; foo(5)")
    end

    def test_compile_function_with_used_arguments
      assert_equal 10, compile("foo = -> a; a + a; end; foo(5)")
    end

    def test_compile_function_with_call_inside
      assert_equal 1, compile("foo = ->; '3'.length(); '3'.length(); end; foo()")
    end

    def test_compile_function_call_twice
      assert_equal 3, compile("foo = ->; 3; end; '3'.length(); foo()")
    end

    def test_compile_function_call_twice_again
      assert_equal 1, compile("'3'.length(); '3'.length()")
    end

    # TODO: Fix this
    # def test_compile_function_scope
    #   assert_equal 3, compile(<<-CODE)
    #     hey = -> foo
    #       ho = -> bar
    #         foo + bar
    #       end
    #       ho(2)
    #     end
    #     hey(1)
    #   CODE
    # end

    def test_compile_multiple_expressions
      result = compile("
      if true != false
        '3'
        '3'
        '3'
        '3'
        1
      else
        '3'
        '3'
        @errors.push('Expected to be truthy.')
      end")
      assert_equal 1, result
    end

    def test_compile_function_with_multiple_arguments
      assert_equal 7, compile("foo = -> a, b, c; b + c; end; foo(5, 4, 3)")
    end

    def test_compile_function_as_an_object_slot
      assert_equal 4, compile("foo = Object.clone(); foo.bar = ->; 4; end; foo.bar()")
    end

    def test_compile_true
      assert_equal true, compile("true")
    end

    def test_compile_false
      assert_equal false, compile("false")
    end

    def test_compile_nil
      assert_equal nil, compile("nil")
    end

    def test_compile_call
      assert_equal "1", compile("1.inspect()")
    end

    def test_compile_local_assign
      assert_equal 3, compile("a = 3")
    end

    def test_compile_object_clone
      obj = compile("Object.clone()")
      assert_kind_of Object, obj
      assert_equal Runtime::Object, obj.__noscript_prototype__
    end

    def test_compile_object_clone_with_properties
      obj = compile("Object.clone({a: 1})")
      assert_equal 1, obj.__noscript_get__(:a)
    end

    def test_compile_slot_get
      assert_equal 1, compile("obj = Object.clone({a: 1}); obj.a")
    end

    def test_compile_slot_assign
      obj = compile("foo = Object.clone(); foo.a = 3; foo")
      assert_equal 3, obj.__noscript_get__(:a)
    end

    def test_compile_slot_assign_on_primitives
      obj = compile("foo = 1; foo.a = 3; foo")
      assert_equal 3, obj.__noscript_get__(:a)
    end

    def test_compile_if
      assert_equal 1, compile("if true; 1; end")
    end

    def test_compile_if_else
      assert_equal 2, compile("if false; 1; else; 2; end")
    end

    def test_compile_if_else_with_multiple_statements
      assert_equal 7, compile("if false; 1 * 7; 2 + 33; 3; else; 2 + 4; 3 + 9; 4 + 3; end")
    end

    def test_compile_while
      assert_equal 10, compile("a = 3; while a < 10; a = a + 1; end; a")
    end

    # Warning: this permanently pollutes self for the rest of the tests.
    def test_self_equals_deref
      assert_equal 3, compile("self.foo = 3; @foo")
    end

    # Warning: this permanently pollutes self for the rest of the tests.
    def test_self_equals_deref_on_asignment
      assert_equal 7, compile("@foo = 7; self.foo")
    end

    def test_ruby_call
      identifier = compile("Noscript.AST.Identifier.ruby('new', 1, 'foo')")
      assert_equal 1, identifier.line
      assert_equal 'foo', identifier.name
    end
  end
end