koi-reference-compiler 0.0.1

data/test/compiler/unit/node_extensions/functions/function_definition_test.rb

@@ -0,0 +1,37 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', '..', '..', 'test_helper.rb'))
+
+class FunctionDefinitionTest < Test::Unit::TestCase
+
+  include KoiReferenceCompiler
+  
+  test "should compile FunctionDefinition" do
+    tree = FunctionDefinition.new("function( arg )\n arg = 1\n end", 0, [
+      Identifier.new("arg", 10),
+      Block.new("arg = 1\n", 17,[
+        Statement.new("arg = 1", 17, [
+          Assignment.new("arg = 1", 17, [
+            Identifier.new("arg", 17),
+            AssignmentOperator.new("=", 21),
+            Expression.new("1", 23, [
+              IntegerLiteral.new("1", 23)
+            ])
+          ])
+        ])
+      ])
+    ])
+    bytecode = tree.compile
+    assert_equal [
+      PUSH_FUNCTION, 0,
+      
+      SET_LOCAL, :arg,
+      
+      PUSH_INT, 1,
+      SET_LOCAL, :arg,
+      
+      RETURN,
+      END_FUNCTION, 0,
+      END_FUNCTION, 0
+    ], bytecode
+  end
+  
+end

data/test/compiler/unit/node_extensions/hash_literals/hash_accessor_list_test.rb

@@ -0,0 +1,83 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', '..', '..', 'test_helper.rb'))
+
+class HashAccessorListTest < Test::Unit::TestCase
+
+  include KoiReferenceCompiler
+  
+  test "should compile lvalue HashAccessorList to SET_KEY" do
+    tree = HashAssignment.new("test[1] = 1", 0, [
+      Identifier.new("test", 0),
+      HashAccessorList.new("[1]", 4, [
+        HashAccessor.new("[1]", 4, [
+          Expression.new("1", 5, [
+            IntegerLiteral.new("1", 5)
+          ])
+        ])
+      ]),
+      AssignmentOperator.new("=", 8),
+      Expression.new("1", 10, [
+        IntegerLiteral.new("1", 10)
+      ])
+    ])
+    bytecode = tree.compile
+    assert_equal [
+      GET_LOCAL, :test,
+      PUSH_INT, 1,
+      PUSH_INT, 1, 
+      SET_KEY,
+      POP
+    ], bytecode
+  end
+  
+  test "should compile lvalue HashAccessorList to GET_KEY, SET_KEY" do
+    tree = HashAssignment.new("test[1][1] = 1", 0, [
+      Identifier.new("test", 0),
+      HashAccessorList.new("[1][1]", 4, [
+        HashAccessor.new("[1]", 4, [
+          Expression.new("1", 5, [
+            IntegerLiteral.new("1", 5)
+          ])
+        ]),
+        HashAccessor.new("[1]", 7, [
+          Expression.new("1", 8, [
+            IntegerLiteral.new("1", 8)
+          ])
+        ])
+      ]),
+      AssignmentOperator.new("=", 11),
+      Expression.new("1", 13, [
+        IntegerLiteral.new("1", 13)
+      ])
+    ])
+    bytecode = tree.compile
+    assert_equal [
+      GET_LOCAL, :test,
+      PUSH_INT, 1,
+      GET_KEY,
+      PUSH_INT, 1,
+      PUSH_INT, 1, 
+      SET_KEY,
+      POP
+    ], bytecode
+  end
+  
+  test "should compile rvalue HashAccess to GET_KEY" do
+    tree = HashAccess.new("test[1]", 0, [
+      Identifier.new("test", 0),
+      HashAccessorList.new("[1]", 4, [
+        HashAccessor.new("[1]", 4, [
+          Expression.new("1", 5, [
+            IntegerLiteral.new("1", 5)
+          ])
+        ])
+      ])
+    ])
+    bytecode = tree.compile
+    assert_equal [
+      GET_LOCAL, :test,
+      PUSH_INT, 1,
+      GET_KEY
+    ], bytecode
+  end
+  
+end

data/test/compiler/unit/node_extensions/hash_literals/hash_accessor_test.rb

@@ -0,0 +1,33 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', '..', '..', 'test_helper.rb'))
+
+class HashAccessorTest < Test::Unit::TestCase
+
+  include KoiReferenceCompiler
+  
+  test "should compile HashAccessor to GET_KEY" do
+    tree =  HashAccessor.new("[1]", 0, [
+      Expression.new("1", 1, [
+        IntegerLiteral.new("1", 1)
+      ])
+    ])
+    bytecode = tree.compile_get
+    assert_equal [
+      PUSH_INT, 1,
+      GET_KEY
+    ], bytecode
+  end
+  
+  test "should compile HashAccessor to SET_KEY" do
+    tree =  HashAccessor.new("[1]", 0, [
+      Expression.new("1", 1, [
+        IntegerLiteral.new("1", 1)
+      ])
+    ])
+    bytecode = tree.compile_set
+    assert_equal [
+      PUSH_INT, 1,
+      SET_KEY
+    ], bytecode
+  end
+  
+end

data/test/compiler/unit/node_extensions/hash_literals/hash_literal_test.rb

@@ -0,0 +1,29 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', '..', '..', 'test_helper.rb'))
+
+class HashLiteralTest < Test::Unit::TestCase
+
+  include KoiReferenceCompiler
+  
+  test "should compile HashLiteral" do
+    tree = HashLiteral.new("{1 => 2}", 0, [
+      KeyValueList.new("1 => 2", 1, [
+        KeyValue.new("1 => 2", 1, [
+          Expression.new("1", 1, [
+            IntegerLiteral.new("1", 1)
+          ]),
+          Expression.new("2", 6, [
+            IntegerLiteral.new("2", 6)
+          ])
+        ])
+      ])
+    ])
+    bytecode = tree.compile
+    assert_equal [
+      PUSH_HASH,
+      PUSH_INT, 1,
+      PUSH_INT, 2, 
+      SET_KEY
+    ], bytecode
+  end
+  
+end

data/test/compiler/unit/node_extensions/hash_literals/key_value_list_test.rb

@@ -0,0 +1,37 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', '..', '..', 'test_helper.rb'))
+
+class KeyValueListTest < Test::Unit::TestCase
+
+  include KoiReferenceCompiler
+  
+  test "should compile KeyValueList" do
+    tree = KeyValueList.new("1 => 2, 2 => 3", 0, [
+      KeyValue.new("1 => 2", 0, [
+        Expression.new("1", 0, [
+          IntegerLiteral.new("1", 0)
+        ]),
+        Expression.new("2", 5, [
+          IntegerLiteral.new("2", 5)
+        ])
+      ]),
+      KeyValue.new("2 => 3", 8, [
+        Expression.new("2", 8, [
+          IntegerLiteral.new("2", 8)
+        ]),
+        Expression.new("3", 13, [
+          IntegerLiteral.new("3", 13)
+        ])
+      ])
+    ])
+    bytecode = tree.compile
+    assert_equal [
+      PUSH_INT, 1,
+      PUSH_INT, 2, 
+      SET_KEY,
+      PUSH_INT, 2,
+      PUSH_INT, 3, 
+      SET_KEY
+    ], bytecode
+  end
+  
+end

data/test/compiler/unit/node_extensions/hash_literals/key_value_test.rb

@@ -0,0 +1,24 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', '..', '..', 'test_helper.rb'))
+
+class KeyValueTest < Test::Unit::TestCase
+
+  include KoiReferenceCompiler
+  
+  test "should compile KeyValue" do
+    tree = KeyValue.new("1 => 2", 0...6, [
+      Expression.new("1", 0...1, [
+        IntegerLiteral.new("1", 0...1)
+      ]),
+      Expression.new("2", 5...6, [
+        IntegerLiteral.new("2", 5...6)
+      ])
+    ])
+    bytecode = tree.compile
+    assert_equal [
+      PUSH_INT, 1,
+      PUSH_INT, 2, 
+      SET_KEY
+    ], bytecode
+  end
+  
+end

data/test/compiler/unit/node_extensions/identifiers/identifier_test.rb

@@ -0,0 +1,103 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', '..', '..', 'test_helper.rb'))
+
+class IdentifierTest < Test::Unit::TestCase
+
+  include KoiReferenceCompiler
+  
+  test "should compile identifier with Assignment as parent to SET_LOCAL" do
+    tree = Assignment.new("test = 2", 0, [
+      Identifier.new('test', 0)
+    ])
+    bytecode = tree.elements.first.compile
+    assert_equal [SET_LOCAL, :test], bytecode
+  end
+  
+  test "should compile identifier with FunctionDefinition as parent to SET_LOCAL" do
+    tree = FunctionDefinition.new("test = 2", 0, [
+      Identifier.new('test', 0)
+    ])
+    bytecode = tree.elements.first.compile
+    assert_equal [SET_LOCAL, :test], bytecode
+  end
+  
+  test "should compile identifier with Expression as parent to GET_LOCAL" do
+    tree = Expression.new("test = 2", 0, [
+      Identifier.new('test', 0)
+    ])
+    bytecode = tree.elements.first.compile
+    assert_equal [GET_LOCAL, :test], bytecode
+  end
+  
+  test "should compile identifier with MultitiveExpression as parent to GET_LOCAL" do
+    tree = MultitiveExpression.new("test = 2", 0, [
+      Identifier.new('test', 0)
+    ])
+    bytecode = tree.elements.first.compile
+    assert_equal [GET_LOCAL, :test], bytecode
+  end
+  
+  test "should compile identifier with AdditiveExpression as parent to GET_LOCAL" do
+    tree = AdditiveExpression.new("test = 2", 0, [
+      Identifier.new('test', 0)
+    ])
+    bytecode = tree.elements.first.compile
+    assert_equal [GET_LOCAL, :test], bytecode
+  end
+  
+  test "should compile identifier with ComparativeExpression as parent to GET_LOCAL" do
+    tree = ComparativeExpression.new("test = 2", 0, [
+      Identifier.new('test', 0)
+    ])
+    bytecode = tree.elements.first.compile
+    assert_equal [GET_LOCAL, :test], bytecode
+  end
+  
+  test "should compile identifier with Assignment as parent to SET_GLOBAL" do
+    tree = Assignment.new("$test = 2", 0, [
+      Identifier.new('$test', 0)
+    ])
+    bytecode = tree.elements.first.compile
+    assert_equal [SET_GLOBAL, :$test], bytecode
+  end
+  
+  test "should compile identifier with FunctionDefinition as parent to SET_GLOBAL" do
+    tree = FunctionDefinition.new("$test = 2", 0, [
+      Identifier.new('$test', 0)
+    ])
+    bytecode = tree.elements.first.compile
+    assert_equal [SET_GLOBAL, :$test], bytecode
+  end
+  
+  test "should compile identifier with Expression as parent to GET_GLOBAL" do
+    tree = Expression.new("$test = 2", 0, [
+      Identifier.new('$test', 0)
+    ])
+    bytecode = tree.elements.first.compile
+    assert_equal [GET_GLOBAL, :$test], bytecode
+  end
+  
+  test "should compile identifier with MultitiveExpression as parent to GET_GLOBAL" do
+    tree = MultitiveExpression.new("$test = 2", 0, [
+      Identifier.new('$test', 0)
+    ])
+    bytecode = tree.elements.first.compile
+    assert_equal [GET_GLOBAL, :$test], bytecode
+  end
+  
+  test "should compile identifier with AdditiveExpression as parent to GET_GLOBAL" do
+    tree = AdditiveExpression.new("$test = 2", 0, [
+      Identifier.new('$test', 0)
+    ])
+    bytecode = tree.elements.first.compile
+    assert_equal [GET_GLOBAL, :$test], bytecode
+  end
+  
+  test "should compile identifier with ComarativeExpression as parent to GET_GLOBAL" do
+    tree = ComparativeExpression.new("$test = 2", 0, [
+      Identifier.new('$test', 0)
+    ])
+    bytecode = tree.elements.first.compile
+    assert_equal [GET_GLOBAL, :$test], bytecode
+  end
+  
+end

data/test/compiler/unit/node_extensions/literals/false_literal_test.rb

@@ -0,0 +1,13 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', '..', '..', 'test_helper.rb'))
+
+class FalseLiteralTest < Test::Unit::TestCase
+
+  include KoiReferenceCompiler
+  
+  test "should compile FalseLiteral" do
+    tree = FalseLiteral.new("false", 0)
+    bytecode = tree.compile
+    assert_equal [PUSH_BOOL, false], bytecode
+  end
+  
+end

data/test/compiler/unit/node_extensions/literals/float_literal_test.rb

@@ -0,0 +1,13 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', '..', '..', 'test_helper.rb'))
+
+class IntegerLiteralTest < Test::Unit::TestCase
+
+  include KoiReferenceCompiler
+  
+  test "should compile FloatLiteral" do
+    tree = FloatLiteral.new("99.00019", 0)
+    bytecode = tree.compile
+    assert_equal [PUSH_FLOAT, 99.00019], bytecode
+  end
+  
+end

data/test/compiler/unit/node_extensions/literals/integer_literal_test.rb

@@ -0,0 +1,13 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', '..', '..', 'test_helper.rb'))
+
+class IntegerLiteralTest < Test::Unit::TestCase
+
+  include KoiReferenceCompiler
+  
+  test "should compile IntegerLiteral" do
+    tree = IntegerLiteral.new("99", 0)
+    bytecode = tree.compile
+    assert_equal [PUSH_INT, 99], bytecode
+  end
+  
+end

data/test/compiler/unit/node_extensions/literals/nil_literal_test.rb

@@ -0,0 +1,13 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', '..', '..', 'test_helper.rb'))
+
+class NilLiteralTest < Test::Unit::TestCase
+
+  include KoiReferenceCompiler
+  
+  test "should compile NilLiteral" do
+    tree = NilLiteral.new("nil", 0)
+    bytecode = tree.compile
+    assert_equal [PUSH_NIL], bytecode
+  end
+  
+end

data/test/compiler/unit/node_extensions/literals/string_literal_test.rb

@@ -0,0 +1,13 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', '..', '..', 'test_helper.rb'))
+
+class StringLiteralTest < Test::Unit::TestCase
+
+  include KoiReferenceCompiler
+  
+  test "should compile StringLiteral" do
+    tree = StringLiteral.new('"test"', 0)
+    bytecode = tree.compile
+    assert_equal [PUSH_STRING, 'test'], bytecode
+  end
+  
+end

data/test/compiler/unit/node_extensions/literals/true_literal_test.rb

@@ -0,0 +1,13 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', '..', '..', 'test_helper.rb'))
+
+class TrueLiteralTest < Test::Unit::TestCase
+
+  include KoiReferenceCompiler
+  
+  test "should compile TrueLiteral" do
+    tree = TrueLiteral.new("true", 0)
+    bytecode = tree.compile
+    assert_equal [PUSH_BOOL, true], bytecode
+  end
+  
+end