Parse methods in extend inline node

2 files changed, 34 insertions, 2 deletions

diff --git a/src/test/java/org/shedlang/compiler/ast/untyped/UntypedExtendInlineNodeMatcher.java b/src/test/java/org/shedlang/compiler/ast/untyped/UntypedExtendInlineNodeMatcher.java
index d8dc26a1..1c82f937 100644
--- a/src/test/java/org/shedlang/compiler/ast/untyped/UntypedExtendInlineNodeMatcher.java
+++ b/src/test/java/org/shedlang/compiler/ast/untyped/UntypedExtendInlineNodeMatcher.java
@@ -41,4 +41,14 @@ public class UntypedExtendInlineNodeMatcher extends MatcherBuilder<
             extendedType
         ));
     }
+
+    public UntypedExtendInlineNodeMatcher withMethods(
+        Matcher<Iterable<? extends UntypedFunctionDefinitionNode>> methods
+    ) {
+        return with(has(
+            "methods",
+            x -> x.methods(),
+            methods
+        ));
+    }
 }
diff --git a/src/test/java/org/shedlang/compiler/parser/ParserExtendInlineTests.java b/src/test/java/org/shedlang/compiler/parser/ParserExtendInlineTests.java
index 75030e4b..6652a46e 100644
--- a/src/test/java/org/shedlang/compiler/parser/ParserExtendInlineTests.java
+++ b/src/test/java/org/shedlang/compiler/parser/ParserExtendInlineTests.java
@@ -2,10 +2,10 @@ package org.shedlang.compiler.parser;
 
 import org.junit.jupiter.api.Test;
 
-import static org.shedlang.compiler.ast.untyped.UntypedNodeMatchers.isUntypedExtendInlineNode;
-import static org.shedlang.compiler.ast.untyped.UntypedNodeMatchers.isUntypedTypeLevelReferenceNode;
+import static org.shedlang.compiler.ast.untyped.UntypedNodeMatchers.*;
 import static org.shedlang.compiler.parser.ParserTesting.parseString;
 import static org.zwobble.precisely.AssertThat.assertThat;
+import static org.zwobble.precisely.Matchers.isSequence;
 
 public class ParserExtendInlineTests {
     @Test
@@ -20,6 +20,28 @@ public class ParserExtendInlineTests {
             node,
             isUntypedExtendInlineNode()
                 .withExtendedType(isUntypedTypeLevelReferenceNode("Count"))
+                .withMethods(isSequence())
+        );
+    }
+
+    @Test
+    public void canParseExtendInlineWithSingleMethod() {
+        var source = """
+            extend Count {
+                fun message(this) {
+                }
+            }
+        """;
+
+        var node = parseString(source, Parser::parseNamespaceStatement);
+
+        assertThat(
+            node,
+            isUntypedExtendInlineNode()
+                .withExtendedType(isUntypedTypeLevelReferenceNode("Count"))
+                .withMethods(isSequence(
+                    isUntypedFunctionDefinitionNode().withName("message")
+                ))
         );
     }
 }