Add type checking for unit literal

2 files changed, 26 insertions, 0 deletions

diff --git a/src/test/java/org/shedlang/compiler/ast/untyped/Untyped.java b/src/test/java/org/shedlang/compiler/ast/untyped/Untyped.java
index 031dd362..07f3021d 100644
--- a/src/test/java/org/shedlang/compiler/ast/untyped/Untyped.java
+++ b/src/test/java/org/shedlang/compiler/ast/untyped/Untyped.java
@@ -212,6 +212,10 @@ public class Untyped {
         return new UntypedTypeLevelReferenceNode(name, Sources.NULL);
     }
 
+    public static UntypedUnitLiteralNode unitLiteral() {
+        return new UntypedUnitLiteralNode(Sources.NULL);
+    }
+
     public static UntypedAssignNode val(
         String targetName,
         UntypedExpressionNode expression
diff --git a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerUnitLiteralTests.java b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerUnitLiteralTests.java
new file mode 100644
index 00000000..07f57003
--- /dev/null
+++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerUnitLiteralTests.java
@@ -0,0 +1,22 @@
+package org.shedlang.compiler.typechecker;
+
+import org.junit.jupiter.api.Test;
+import org.shedlang.compiler.ast.untyped.Untyped;
+import org.shedlang.compiler.types.Types;
+
+import static org.shedlang.compiler.ast.typed.TypedNodeMatchers.isTypedUnitNode;
+import static org.shedlang.compiler.typechecker.TypeCheckerTesting.typeCheckExpression;
+import static org.zwobble.precisely.AssertThat.assertThat;
+import static org.zwobble.precisely.Matchers.equalTo;
+
+public class TypeCheckerUnitLiteralTests {
+    @Test
+    public void unitLiteralIsTypeCheckedAsUnit() {
+        var untyped = Untyped.unitLiteral();
+
+        var typed = typeCheckExpression(untyped, TypeCheckerContext.EMPTY);
+
+        assertThat(typed.value(), isTypedUnitNode());
+        assertThat(typed.value().type(), equalTo(Types.UNIT));
+    }
+}