From f39ec01be157d0a93202dfbb1638ccf899e2e392 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sun, 19 Nov 2023 12:16:39 +0000 Subject: Generate Patio code for defining an effect --- .../patio/PatioCodeGenDefineEffectTests.java | 47 ++++++++++++++++++++++ 1 file changed, 47 insertions(+) create mode 100644 src/test/java/org/shedlang/compiler/patio/PatioCodeGenDefineEffectTests.java (limited to 'src/test/java') diff --git a/src/test/java/org/shedlang/compiler/patio/PatioCodeGenDefineEffectTests.java b/src/test/java/org/shedlang/compiler/patio/PatioCodeGenDefineEffectTests.java new file mode 100644 index 00000000..a6135d0a --- /dev/null +++ b/src/test/java/org/shedlang/compiler/patio/PatioCodeGenDefineEffectTests.java @@ -0,0 +1,47 @@ +package org.shedlang.compiler.patio; + +import org.junit.jupiter.api.Test; +import org.junit.jupiter.api.extension.ExtendWith; +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.TypeInfos; +import org.shedlang.compiler.types.*; + +import java.io.IOException; + +import static org.shedlang.compiler.patio.PatioTesting.serialize; + +@ExtendWith({SnapshotResolver.class}) +public class PatioCodeGenDefineEffectTests { + @Test + public void canGenerateCodeForDefineEffect(Snapshotter snapshotter) throws IOException { + var effect = Effects.userDefined(NamespaceName.of("Example", "Project"), "State"); + var instruction = Instructions.defineEffect(effect); + var typeInfos = TypeInfos.EMPTY + .addUserDefinedEffectInfo( + effect, + UserDefinedEffectInfo.builder() + .addMethod( + "read", + FunctionType.builder() + .addNamedParam("fence", Types.BOOL) + .withReturnType(Types.INT) + .build() + ) + .addMethod( + "write", + FunctionType.builder() + .addPositionalParam(Types.INT) + .withReturnType(Types.UNIT) + .build() + ) + .build() + ); + var context = PatioCodeGenContext.forTypeInfos(typeInfos); + + var result = PatioCodeGen.generateInstruction(instruction, context); + + snapshotter.assertSnapshot(serialize(result)); + } +} -- cgit v1.2.3