From 1f73c6fcd01e6c3b3c0e7ca0b30fb46de56f0b59 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sun, 2 Mar 2025 19:52:06 +0000 Subject: Add native type type params to environment --- .../TypeCheckerNativeTypeDeclarationTests.java | 59 ++++++++++++++++++++++ 1 file changed, 59 insertions(+) (limited to 'src/test/java/org') diff --git a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerNativeTypeDeclarationTests.java b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerNativeTypeDeclarationTests.java index 562b7ab4..024773c8 100644 --- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerNativeTypeDeclarationTests.java +++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerNativeTypeDeclarationTests.java @@ -281,4 +281,63 @@ public class TypeCheckerNativeTypeDeclarationTests { )) ); } + + @Test + public void whenNativeTypeIsGenericThenTypeParamsCanBeUsedInNativeMethodDeclarations() { + var untyped = UntypedNativeTypeDeclarationNode.builder() + .withName("ExampleType") + .withTypeLevelParams(List.of(Untyped.typeParam("T"))) + .addMethod( + UntypedNativeFunctionDeclarationNode.builder() + .withName("a") + .addPositionalParam(Untyped.typeLevelReference("This")) + .addPositionalParam(Untyped.typeLevelReference("T")) + .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.constructedType( + Types.typeConstructor( + List.of( + TypeParam.forNativeType(Types.nativeType(NAMESPACE_NAME, "ExampleType"), "T") + ), + Types.nativeType(NAMESPACE_NAME, "ExampleType") + ), + List.of(TypeParam.forNativeType(Types.nativeType(NAMESPACE_NAME, "ExampleType"), "T")) + )) + .addPositionalParam(TypeParam.forNativeType(Types.nativeType(NAMESPACE_NAME, "ExampleType"), "T")) + .build() + ) + )) + ); + var nativeTypeInfo = typed.context().toTypeInfos() + .nativeTypeInfo(Types.nativeType(NAMESPACE_NAME, "ExampleType")) + .orElseThrow(); + assertThat( + nativeTypeInfo, + has("methods", x -> x.methods(), isSequence( + equalTo(new MethodInfo( + "a", + FunctionType.builder() + .addPositionalParam(TypeParam.forNativeType(Types.nativeType(NAMESPACE_NAME, "ExampleType"), "T")) + .withReturnType(Types.UNIT) + .build() + )) + )) + ); + } } -- cgit v1.2.3