Add WASM code gen for namespace field stores

1 files changed, 41 insertions, 0 deletions

diff --git a/src/test/java/org/shedlang/compiler/backends/wasm/WasmCodeGenNamespaceFieldStoreTests.java b/src/test/java/org/shedlang/compiler/backends/wasm/WasmCodeGenNamespaceFieldStoreTests.java
new file mode 100644
index 00000000..c35f3e1a
--- /dev/null
+++ b/src/test/java/org/shedlang/compiler/backends/wasm/WasmCodeGenNamespaceFieldStoreTests.java
@@ -0,0 +1,41 @@
+package org.shedlang.compiler.backends.wasm;
+
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.shedlang.compiler.backends.wasm.structure.ast.modules.WasmAlignment;
+import org.shedlang.compiler.stackir.instructions.Instructions;
+import org.shedlang.compiler.testing.snapshots.SnapshotResolver;
+import org.shedlang.compiler.testing.snapshots.Snapshotter;
+import org.shedlang.compiler.typechecker.TypeCheckerContext;
+import org.shedlang.compiler.types.NamespaceName;
+import org.shedlang.compiler.types.Types;
+import org.shedlang.compiler.util.P;
+
+import java.io.IOException;
+import java.util.List;
+import java.util.Map;
+
+import static org.shedlang.compiler.backends.wasm.WasmCodeGenTesting.objdumpContext;
+
+@ExtendWith({SnapshotResolver.class})
+public class WasmCodeGenNamespaceFieldStoreTests {
+    @Test
+    public void namespaceFieldStorePopsFromStackAndUpdatesFieldOnNamespace(Snapshotter snapshotter) throws IOException, InterruptedException {
+        var namespaceName = new NamespaceName(List.of("Example", "Project"));
+        var instruction = Instructions.namespaceFieldStore(namespaceName, "y");
+        var typeInfo = TypeCheckerContext.EMPTY
+            .addNamespaceFields(namespaceName, P.orderedMap(
+                Map.entry("x", Types.INT),
+                Map.entry("y", Types.INT),
+                Map.entry("z", Types.INT)
+            ))
+            .toTypeInfo();
+        var context = WasmCodeGenContext.fromTypeInfo(typeInfo)
+            .addData("shed__namespace__Example_Project", new byte[12], WasmAlignment.ALIGN_32_BIT)
+            .context();
+
+        var result = WasmCodeGen.generateInstruction(instruction, context);
+
+        snapshotter.assertSnapshot(objdumpContext(result));
+    }
+}