add test

Author: aceinetx

zenlang/tests/vm_this.rs zenlang/tests/vm_self.rszenlang/tests/vm_this.rs renamed to zenlang/tests/vm_self.rs

@@ -1,6 +1,4 @@
 extern crate alloc;
-use alloc::rc::Rc;
-use core::cell::RefCell;
 
 use zenlang::compiler::*;
 use zenlang::parser::*;
@@ -33,34 +31,6 @@ fn expect_to_return(code: String, value: Value) {
     assert!(vm.ret.equal(&value, &vm));
 }
 
-fn expect_to_return_obj(code: String, object: Object) {
-    let mut tokenizer = Tokenizer::new(code);
-    let mut parser = Parser::new(&mut tokenizer);
-    let mut compiler = Compiler::new(&mut parser);
-    if let Err(e) = compiler.compile() {
-        assert_eq!(e, "");
-    }
-    let mut vm = VM::new();
-    let module = compiler.get_module();
-    println!("{:?}", module.opcodes);
-
-    let _ = vm.load_module(&zenlang::stdlib::compile_stdlib_module());
-    let _ = vm.load_module(module);
-
-    if let Err(e) = vm.set_entry_function("main") {
-        assert_eq!(e, "");
-    }
-
-    vm.run_until_halt();
-
-    println!("vm.ret: {}", vm.ret);
-    assert_eq!(vm.error, "");
-    assert!(
-        vm.ret
-            .equal(&Value::Object(Rc::new(RefCell::new(object))), &vm)
-    );
-}
-
 #[test]
 fn vm_test_self_1() {
     expect_to_return(
@@ -115,3 +85,34 @@ fn main {
         Value::Number(2.0),
     );
 }
+
+#[test]
+fn vm_test_self_3() {
+    expect_to_return(
+        r#"
+fn f2 {
+    let self.hi = 3;
+}
+
+fn f {
+    let self.hi = 1;
+    let obj = {
+        "test" = f2
+    };
+    obj.test();
+    let self.hi = obj.hi - 1;
+}
+
+fn main {
+    let obj = {
+        "obj" = {
+            "test" = f,
+        }
+    };
+    obj.obj.test();
+    return obj.obj.hi;
+} "#
+        .into(),
+        Value::Number(2.0),
+    );
+}