Support mutual recursion

2 files changed, 9 insertions, 12 deletions

diff --git a/src/test/java/org/shedlang/compiler/backends/wasm/WasmCodeGenDefineFunctionTests.java b/src/test/java/org/shedlang/compiler/backends/wasm/WasmCodeGenDefineFunctionTests.java
index d32f716a..a996dbcc 100644
--- a/src/test/java/org/shedlang/compiler/backends/wasm/WasmCodeGenDefineFunctionTests.java
+++ b/src/test/java/org/shedlang/compiler/backends/wasm/WasmCodeGenDefineFunctionTests.java
@@ -50,9 +50,9 @@ public class WasmCodeGenDefineFunctionTests {
             Sources.NULL
         );
         var context = WasmCodeGenContext.EMPTY
-            .addShedLocal("b").context()
-            .addShedLocal("c").context()
-            .addShedLocal("a").context();
+            .addShedArg("b")
+            .addShedArg("c")
+            .addShedArg("a");
 
         var result = WasmCodeGen.generateInstruction(instruction, context);
 
@@ -77,9 +77,9 @@ public class WasmCodeGenDefineFunctionTests {
             Sources.NULL
         );
         var context = WasmCodeGenContext.EMPTY
-            .addShedLocal("b").context()
-            .addShedLocal("c").context()
-            .addShedLocal("a").context();
+            .addShedArg("b")
+            .addShedArg("c")
+            .addShedArg("a");
 
         var result = WasmCodeGen.generateInstruction(instruction, context);
 
diff --git a/src/test/java/org/shedlang/compiler/backends/wasm/WasmCodeGenLocalLoadTests.java b/src/test/java/org/shedlang/compiler/backends/wasm/WasmCodeGenLocalLoadTests.java
index 62aefa1d..737f97cb 100644
--- a/src/test/java/org/shedlang/compiler/backends/wasm/WasmCodeGenLocalLoadTests.java
+++ b/src/test/java/org/shedlang/compiler/backends/wasm/WasmCodeGenLocalLoadTests.java
@@ -16,12 +16,9 @@ public class WasmCodeGenLocalLoadTests {
     public void localLoadCompilesToLocalGet(Snapshotter snapshotter) throws IOException, InterruptedException {
         var instruction = Instructions.localLoad("x");
         var context = WasmCodeGenContext.EMPTY
-            .addShedLocal("y")
-            .context()
-            .addShedLocal("x")
-            .context()
-            .addShedLocal("z")
-            .context();
+            .addShedArg("y")
+            .addShedArg("x")
+            .addShedArg("z");
 
         var result = WasmCodeGen.generateInstruction(instruction, context);