Type check int literals

3 files changed, 45 insertions, 0 deletions

diff --git a/src/test/java/org/shedlang/compiler/ast/typed/TypedNodeMatchers.java b/src/test/java/org/shedlang/compiler/ast/typed/TypedNodeMatchers.java
index 9c2164b5..37f8e7f4 100644
--- a/src/test/java/org/shedlang/compiler/ast/typed/TypedNodeMatchers.java
+++ b/src/test/java/org/shedlang/compiler/ast/typed/TypedNodeMatchers.java
@@ -3,12 +3,25 @@ package org.shedlang.compiler.ast.typed;
 import org.shedlang.compiler.types.Type;
 import org.zwobble.precisely.Matcher;
 
+import java.math.BigInteger;
+
 import static org.zwobble.precisely.Matchers.*;
 
 public class TypedNodeMatchers {
     private TypedNodeMatchers() {
     }
 
+    public static Matcher<TypedExpressionNode> isTypedIntLiteralNode(int value) {
+        return isTypedIntLiteralNode(BigInteger.valueOf(value));
+    }
+
+    public static Matcher<TypedExpressionNode> isTypedIntLiteralNode(BigInteger value) {
+        return instanceOf(
+            TypedIntLiteralNode.class,
+            has("value", x -> x.value(), equalTo(value))
+        );
+    }
+
     public static Matcher<TypedExpressionNode> isTypedReferenceNode(String name, Type type) {
         return instanceOf(
             TypedReferenceNode.class,
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 32a01ec4..7751bc05 100644
--- a/src/test/java/org/shedlang/compiler/ast/untyped/Untyped.java
+++ b/src/test/java/org/shedlang/compiler/ast/untyped/Untyped.java
@@ -2,10 +2,20 @@ package org.shedlang.compiler.ast.untyped;
 
 import org.shedlang.compiler.sources.Sources;
 
+import java.math.BigInteger;
+
 public class Untyped {
     private Untyped() {
     }
 
+    public static UntypedIntLiteralNode intLiteral(int value) {
+        return new UntypedIntLiteralNode(BigInteger.valueOf(value), Sources.NULL);
+    }
+
+    public static UntypedIntLiteralNode intLiteral(BigInteger value) {
+        return new UntypedIntLiteralNode(value, Sources.NULL);
+    }
+
     public static UntypedReferenceNode reference(String name) {
         return new UntypedReferenceNode(name, Sources.NULL);
     }
diff --git a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerIntLiteralTests.java b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerIntLiteralTests.java
new file mode 100644
index 00000000..945c39f2
--- /dev/null
+++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerIntLiteralTests.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.isTypedIntLiteralNode;
+import static org.shedlang.compiler.ast.typed.TypedNodeMatchers.isTypedStringLiteralNode;
+import static org.zwobble.precisely.AssertThat.assertThat;
+import static org.zwobble.precisely.Matchers.equalTo;
+
+public class TypeCheckerIntLiteralTests {
+    @Test
+    public void intLiteralIsTypeCheckedAsInt() {
+        var untyped = Untyped.intLiteral(42);
+
+        var typed = TypeChecker.typeCheckExpression(untyped, TypeCheckerContext.EMPTY);
+
+        assertThat(typed, isTypedIntLiteralNode(42));
+        assertThat(typed.type(), equalTo(Types.INT));
+    }
+}