Check number of type-level arguments

1 files changed, 31 insertions, 0 deletions

diff --git a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerCallTests.java b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerCallTests.java
index 21a3e718..3027304e 100644
--- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerCallTests.java
+++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerCallTests.java
@@ -632,4 +632,35 @@ public class TypeCheckerCallTests {
                 .withHasEffects(true)
         );
     }
+
+    @Test
+    public void whenReceiverIsCalledWithWrongNumberOfTypeLevelArgsThenErrorIsThrown() {
+        var untyped = UntypedCallNode.builder()
+            .withReceiver(Untyped.reference("identity"))
+            .withTypeLevelArgs(List.of(
+                Untyped.typeLevelReference("Int"),
+                Untyped.typeLevelReference("Int")
+            ))
+            .addPositionalArg(Untyped.intLiteral(42))
+            .build();
+        var context = TypeCheckerContext.EMPTY
+            .enterGlobal(P.map(
+                Map.entry(
+                    "identity",
+                    FunctionType.builder()
+                        .withTypeLevelParams(List.of(
+                            TypeParam.forFunction(NamespaceName.of("Example"), "identity", "T")
+                        ))
+                        .addPositionalParam(TypeParam.forFunction(NamespaceName.of("Example"), "identity", "T"))
+                        .withReturnType(TypeParam.forFunction(NamespaceName.of("Example"), "identity", "T"))
+                        .build()
+                ),
+                Map.entry("Int", Types.metaType(Types.INT))
+            ));
+
+        var error = assertThrows(WrongNumberOfTypeLevelArgsError.class, () -> typeCheckExpression(untyped, context));
+
+        assertThat(error.expected(), equalTo(1));
+        assertThat(error.actual(), equalTo(2));
+    }
 }