Type check native type methods

1 files changed, 38 insertions, 0 deletions

diff --git a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerNativeTypeDeclarationTests.java b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerNativeTypeDeclarationTests.java
index c9dca725..938b6075 100644
--- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerNativeTypeDeclarationTests.java
+++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerNativeTypeDeclarationTests.java
@@ -2,14 +2,19 @@ package org.shedlang.compiler.typechecker;
 
 import org.junit.jupiter.api.Test;
 import org.shedlang.compiler.ast.untyped.Untyped;
+import org.shedlang.compiler.ast.untyped.UntypedNativeFunctionDeclarationNode;
 import org.shedlang.compiler.ast.untyped.UntypedNativeTypeDeclarationNode;
 import org.shedlang.compiler.typechecker.errors.CannotDefineNativeTypeOutsideOfNamespaceError;
+import org.shedlang.compiler.types.FunctionType;
 import org.shedlang.compiler.types.NamespaceName;
 import org.shedlang.compiler.types.Types;
+import org.shedlang.compiler.util.P;
 
 import java.util.List;
+import java.util.Map;
 
 import static org.junit.jupiter.api.Assertions.assertThrows;
+import static org.shedlang.compiler.ast.typed.TypedNodeMatchers.isTypedNativeFunctionDeclarationNode;
 import static org.shedlang.compiler.ast.typed.TypedNodeMatchers.isTypedNativeTypeDeclarationNode;
 import static org.shedlang.compiler.typechecker.TypeCheckerTesting.typeCheckNamespaceStatement;
 import static org.shedlang.compiler.types.TypeMatchers.*;
@@ -119,4 +124,37 @@ public class TypeCheckerNativeTypeDeclarationTests {
             ))
         );
     }
+
+    @Test
+    public void canTypeCheckMethodWithOnlyThisParam() {
+        var untyped = UntypedNativeTypeDeclarationNode.builder()
+            .withName("ExampleType")
+            .addMethod(
+                UntypedNativeFunctionDeclarationNode.builder()
+                    .addPositionalParam(Untyped.typeLevelReference("This"))
+                    .withName("a")
+                    .withReturnType(Untyped.typeLevelReference("Unit"))
+                    .build()
+            )
+            .build();
+        var context = TypeCheckerContext.EMPTY
+            .enterGlobal(P.map(
+                Map.entry("Unit", Types.metaType(Types.UNIT))
+            ));
+
+        var typed = typeCheckNamespaceStatement(untyped, context);
+
+        assertThat(
+            typed.value(),
+            isTypedNativeTypeDeclarationNode()
+                .withMethods(isSequence(
+                    isTypedNativeFunctionDeclarationNode()
+                        .withSignature(
+                            FunctionType.builder()
+                                .addPositionalParam(Types.nativeType(NAMESPACE_NAME, "ExampleType"))
+                                .build()
+                        )
+                ))
+        );
+    }
 }