Parse type-level params for function type

1 files changed, 32 insertions, 0 deletions

diff --git a/src/test/java/org/shedlang/compiler/parser/ParserFunctionTypeTests.java b/src/test/java/org/shedlang/compiler/parser/ParserFunctionTypeTests.java
index 209220ad..319b0252 100644
--- a/src/test/java/org/shedlang/compiler/parser/ParserFunctionTypeTests.java
+++ b/src/test/java/org/shedlang/compiler/parser/ParserFunctionTypeTests.java
@@ -164,4 +164,36 @@ public class ParserFunctionTypeTests {
                 .withEffect(isOptionalOf(isUntypedTypeLevelReferenceNode("Io")))
         );
     }
+
+    @Test
+    public void canParseSingleTypeLevelParam() {
+        var source = "fn [A]() -> Unit";
+
+        var node = parseString(source, Parser::parseTopLevelTypeLevelExpression);
+
+        assertThat(
+            node,
+            isUntypedFunctionTypeNode()
+                .withTypeLevelParams(isOptionalOf(isSequence(
+                    isUntypedTypeParamNode("A")
+                )))
+        );
+    }
+
+    @Test
+    public void canParseManyTypeLevelParams() {
+        var source = "fn [A, B, C]() -> Unit";
+
+        var node = parseString(source, Parser::parseTopLevelTypeLevelExpression);
+
+        assertThat(
+            node,
+            isUntypedFunctionTypeNode()
+                .withTypeLevelParams(isOptionalOf(isSequence(
+                    isUntypedTypeParamNode("A"),
+                    isUntypedTypeParamNode("B"),
+                    isUntypedTypeParamNode("C")
+                )))
+        );
+    }
 }