fix: minor correction in Deref bytecode

jac3km4 · May 25, 2024 · f12f7ab · f12f7ab
1 parent f0926a1
commit f12f7ab
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 1 deletion.
diff --git a/compiler/src/assembler.rs b/compiler/src/assembler.rs
@@ -553,7 +553,8 @@ impl<'a> Assembler<'a> {
                 self.emit(Instr::AsRef(get_arg_type(0)?));
             }
             IntrinsicOp::Deref => {
-                self.emit(Instr::Deref(get_arg_type(0)?));
+                let type_idx = scope.get_type_index(return_type, pool).with_span(span)?;
+                self.emit(Instr::Deref(type_idx));
             }
             IntrinsicOp::RefToWeakRef => {
                 self.emit(Instr::RefToWeakRef);

diff --git a/compiler/tests/bytecode.rs b/compiler/tests/bytecode.rs
@@ -566,6 +566,34 @@ fn compile_string_literals() {
     TestContext::compiled(vec![sources]).unwrap().run("Testing", check);
 }
 
+#[test]
+fn compile_deref_asref() {
+    let sources = r#"
+        func Testing(str: String) {
+            let a = ToString(str);
+            let b = AsRef(str);
+            let c = Deref(b);
+        }
+        "#;
+
+    let check = check_code![
+        pat!(Assign),
+        mem!(Local(a)),
+        mem!(ToString(str_type)),
+        mem!(Param(str)),
+        pat!(Assign),
+        mem!(Local(b)),
+        mem!(AsRef(str_type)),
+        mem!(Param(str)),
+        pat!(Assign),
+        mem!(Local(c)),
+        mem!(Deref(str_type)),
+        mem!(Local(b)),
+        pat!(Nop)
+    ];
+    TestContext::compiled(vec![sources]).unwrap().run("Testing", check);
+}
+
 #[test]
 fn compile_is_defined() {
     let sources = "